NCManageDatabase.swift 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854
  1. //
  2. // NCManageDatabase.swift
  3. // Nextcloud
  4. //
  5. // Created by Marino Faggiana on 06/05/17.
  6. // Copyright © 2017 Marino Faggiana. All rights reserved.
  7. //
  8. // Author Marino Faggiana <marino.faggiana@nextcloud.com>
  9. // Author Henrik Storch <henrik.storch@nextcloud.com>
  10. //
  11. // This program is free software: you can redistribute it and/or modify
  12. // it under the terms of the GNU General Public License as published by
  13. // the Free Software Foundation, either version 3 of the License, or
  14. // (at your option) any later version.
  15. //
  16. // This program is distributed in the hope that it will be useful,
  17. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  18. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  19. // GNU General Public License for more details.
  20. //
  21. // You should have received a copy of the GNU General Public License
  22. // along with this program. If not, see <http://www.gnu.org/licenses/>.
  23. //
  24. import UIKit
  25. import RealmSwift
  26. import NextcloudKit
  27. import CoreMedia
  28. import Photos
  29. class NCManageDatabase: NSObject {
  30. @objc static let shared: NCManageDatabase = {
  31. let instance = NCManageDatabase()
  32. return instance
  33. }()
  34. override init() {
  35. let dirGroup = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: NCBrandOptions.shared.capabilitiesGroups)
  36. let databaseFileUrlPath = dirGroup?.appendingPathComponent(NCGlobal.shared.appDatabaseNextcloud + "/" + databaseName)
  37. let bundleUrl: URL = Bundle.main.bundleURL
  38. let bundlePathExtension: String = bundleUrl.pathExtension
  39. let isAppex: Bool = bundlePathExtension == "appex"
  40. if let databaseFilePath = databaseFileUrlPath?.path {
  41. if FileManager.default.fileExists(atPath: databaseFilePath) {
  42. NextcloudKit.shared.nkCommonInstance.writeLog("DATABASE FOUND in " + databaseFilePath)
  43. } else {
  44. NextcloudKit.shared.nkCommonInstance.writeLog("DATABASE NOT FOUND in " + databaseFilePath)
  45. }
  46. }
  47. // Disable file protection for directory DB
  48. // https://docs.mongodb.com/realm/sdk/ios/examples/configure-and-open-a-realm/#std-label-ios-open-a-local-realm
  49. if let folderPathURL = dirGroup?.appendingPathComponent(NCGlobal.shared.appDatabaseNextcloud) {
  50. let folderPath = folderPathURL.path
  51. do {
  52. try FileManager.default.setAttributes([FileAttributeKey.protectionKey: FileProtectionType.completeUntilFirstUserAuthentication], ofItemAtPath: folderPath)
  53. } catch {
  54. print("Dangerous error")
  55. }
  56. }
  57. if isAppex {
  58. // App Extension config
  59. let config = Realm.Configuration(
  60. fileURL: dirGroup?.appendingPathComponent(NCGlobal.shared.appDatabaseNextcloud + "/" + databaseName),
  61. schemaVersion: databaseSchemaVersion,
  62. objectTypes: [tableMetadata.self, tableLocalFile.self, tableDirectory.self, tableTag.self, tableAccount.self, tableCapabilities.self, tablePhotoLibrary.self, tableE2eEncryption.self, tableE2eEncryptionLock.self, tableE2eMetadata.self, tableShare.self, tableChunk.self, tableAvatar.self, tableDashboardWidget.self, tableDashboardWidgetButton.self, NCDBLayoutForView.self]
  63. )
  64. Realm.Configuration.defaultConfiguration = config
  65. } else {
  66. // App config
  67. let configCompact = Realm.Configuration(
  68. fileURL: databaseFileUrlPath,
  69. schemaVersion: databaseSchemaVersion,
  70. migrationBlock: { migration, oldSchemaVersion in
  71. if oldSchemaVersion < 255 {
  72. migration.deleteData(forType: tableActivity.className())
  73. migration.deleteData(forType: tableActivityLatestId.className())
  74. migration.deleteData(forType: tableActivityPreview.className())
  75. migration.deleteData(forType: tableActivitySubjectRich.className())
  76. migration.deleteData(forType: tableDirectory.className())
  77. migration.deleteData(forType: tableMetadata.className())
  78. }
  79. if oldSchemaVersion < 292 {
  80. migration.deleteData(forType: tableVideo.className())
  81. }
  82. }, shouldCompactOnLaunch: { totalBytes, usedBytes in
  83. // totalBytes refers to the size of the file on disk in bytes (data + free space)
  84. // usedBytes refers to the number of bytes used by data in the file
  85. // Compact if the file is over 100MB in size and less than 50% 'used'
  86. let oneHundredMB = 100 * 1024 * 1024
  87. return (totalBytes > oneHundredMB) && (Double(usedBytes) / Double(totalBytes)) < 0.5
  88. }
  89. )
  90. do {
  91. _ = try Realm(configuration: configCompact)
  92. } catch let error {
  93. if let databaseFileUrlPath = databaseFileUrlPath {
  94. do {
  95. #if !EXTENSION
  96. let nkError = NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: error.localizedDescription)
  97. NCContentPresenter.shared.showError(error: nkError, priority: .max)
  98. #endif
  99. NextcloudKit.shared.nkCommonInstance.writeLog("DATABASE ERROR: \(error.localizedDescription)")
  100. try FileManager.default.removeItem(at: databaseFileUrlPath)
  101. } catch {}
  102. }
  103. }
  104. let config = Realm.Configuration(
  105. fileURL: dirGroup?.appendingPathComponent(NCGlobal.shared.appDatabaseNextcloud + "/" + databaseName),
  106. schemaVersion: databaseSchemaVersion
  107. )
  108. Realm.Configuration.defaultConfiguration = config
  109. }
  110. // Verify Database, if corrupt remove it
  111. do {
  112. _ = try Realm()
  113. } catch let error {
  114. if let databaseFileUrlPath = databaseFileUrlPath {
  115. do {
  116. #if !EXTENSION
  117. let nkError = NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: error.localizedDescription)
  118. NCContentPresenter.shared.showError(error: nkError, priority: .max)
  119. #endif
  120. NextcloudKit.shared.nkCommonInstance.writeLog("DATABASE ERROR: \(error.localizedDescription)")
  121. try FileManager.default.removeItem(at: databaseFileUrlPath)
  122. } catch { }
  123. }
  124. }
  125. // Open Real
  126. do {
  127. _ = try Realm()
  128. } catch let error as NSError {
  129. NextcloudKit.shared.nkCommonInstance.writeLog("Could not open database: \(error)")
  130. }
  131. }
  132. // MARK: -
  133. // MARK: Utility Database
  134. @objc func clearTable(_ table: Object.Type, account: String? = nil) {
  135. do {
  136. let realm = try Realm()
  137. try realm.write {
  138. var results: Results<Object>
  139. if let account = account {
  140. results = realm.objects(table).filter("account == %@", account)
  141. } else {
  142. results = realm.objects(table)
  143. }
  144. realm.delete(results)
  145. }
  146. } catch let error {
  147. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  148. }
  149. }
  150. @objc func clearDatabase(account: String?, removeAccount: Bool) {
  151. self.clearTable(tableActivity.self, account: account)
  152. self.clearTable(tableActivityLatestId.self, account: account)
  153. self.clearTable(tableActivityPreview.self, account: account)
  154. self.clearTable(tableActivitySubjectRich.self, account: account)
  155. self.clearTable(tableAvatar.self)
  156. self.clearTable(tableCapabilities.self, account: account)
  157. self.clearTable(tableChunk.self, account: account)
  158. self.clearTable(tableComments.self, account: account)
  159. self.clearTable(tableDashboardWidget.self, account: account)
  160. self.clearTable(tableDashboardWidgetButton.self, account: account)
  161. self.clearTable(tableDirectEditingCreators.self, account: account)
  162. self.clearTable(tableDirectEditingEditors.self, account: account)
  163. self.clearTable(tableDirectory.self, account: account)
  164. self.clearTable(tableE2eEncryption.self, account: account)
  165. self.clearTable(tableE2eEncryptionLock.self, account: account)
  166. self.clearTable(tableE2eMetadata.self, account: account)
  167. self.clearTable(tableExternalSites.self, account: account)
  168. self.clearTable(tableGPS.self, account: nil)
  169. self.clearTable(TableGroupfolders.self, account: account)
  170. self.clearTable(TableGroupfoldersGroups.self, account: account)
  171. self.clearTable(NCDBLayoutForView.self, account: account)
  172. self.clearTable(tableLocalFile.self, account: account)
  173. self.clearTable(tableMetadata.self, account: account)
  174. self.clearTable(tablePhotoLibrary.self, account: account)
  175. self.clearTable(tableShare.self, account: account)
  176. self.clearTable(tableTag.self, account: account)
  177. self.clearTable(tableTip.self)
  178. self.clearTable(tableTrash.self, account: account)
  179. self.clearTable(tableUserStatus.self, account: account)
  180. self.clearTable(tableVideo.self, account: account)
  181. if removeAccount {
  182. self.clearTable(tableAccount.self, account: account)
  183. }
  184. }
  185. @objc func removeDB() {
  186. let realmURL = Realm.Configuration.defaultConfiguration.fileURL!
  187. let realmURLs = [
  188. realmURL,
  189. realmURL.appendingPathExtension("lock"),
  190. realmURL.appendingPathExtension("note"),
  191. realmURL.appendingPathExtension("management")
  192. ]
  193. for URL in realmURLs {
  194. do {
  195. try FileManager.default.removeItem(at: URL)
  196. } catch let error {
  197. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  198. }
  199. }
  200. }
  201. func getThreadConfined(_ object: Object) -> Any {
  202. return ThreadSafeReference(to: object)
  203. }
  204. func putThreadConfined(_ tableRef: ThreadSafeReference<Object>) -> Object? {
  205. do {
  206. let realm = try Realm()
  207. return realm.resolve(tableRef)
  208. } catch let error as NSError {
  209. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  210. }
  211. return nil
  212. }
  213. func isTableInvalidated(_ object: Object) -> Bool {
  214. return object.isInvalidated
  215. }
  216. // MARK: -
  217. // MARK: Table Chunk
  218. func getChunkFolder(account: String, ocId: String) -> String {
  219. do {
  220. let realm = try Realm()
  221. realm.refresh()
  222. guard let result = realm.objects(tableChunk.self).filter("account == %@ AND ocId == %@", account, ocId).first else { return NSUUID().uuidString }
  223. return result.chunkFolder
  224. } catch let error as NSError {
  225. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  226. }
  227. return NSUUID().uuidString
  228. }
  229. func getChunks(account: String, ocId: String) -> [String] {
  230. var filesNames: [String] = []
  231. do {
  232. let realm = try Realm()
  233. realm.refresh()
  234. let results = realm.objects(tableChunk.self).filter("account == %@ AND ocId == %@", account, ocId).sorted(byKeyPath: "fileName", ascending: true)
  235. for result in results {
  236. filesNames.append(result.fileName)
  237. }
  238. return filesNames
  239. } catch let error as NSError {
  240. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  241. }
  242. return filesNames
  243. }
  244. func addChunks(account: String, ocId: String, chunkFolder: String, fileNames: [String]) {
  245. var size: Int64 = 0
  246. do {
  247. let realm = try Realm()
  248. try realm.write {
  249. for fileName in fileNames {
  250. let object = tableChunk()
  251. size += NCUtilityFileSystem.shared.getFileSize(filePath: CCUtility.getDirectoryProviderStorageOcId(ocId, fileNameView: fileName)!)
  252. object.account = account
  253. object.chunkFolder = chunkFolder
  254. object.fileName = fileName
  255. object.index = ocId + fileName
  256. object.ocId = ocId
  257. object.size = size
  258. realm.add(object, update: .all)
  259. }
  260. }
  261. } catch let error {
  262. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  263. }
  264. }
  265. func getChunk(account: String, fileName: String) -> tableChunk? {
  266. do {
  267. let realm = try Realm()
  268. realm.refresh()
  269. guard let result = realm.objects(tableChunk.self).filter("account == %@ AND fileName == %@", account, fileName).first else { return nil }
  270. return tableChunk.init(value: result)
  271. } catch let error as NSError {
  272. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  273. }
  274. return nil
  275. }
  276. func deleteChunk(account: String, ocId: String, fileName: String) {
  277. do {
  278. let realm = try Realm()
  279. try realm.write {
  280. let result = realm.objects(tableChunk.self).filter(NSPredicate(format: "account == %@ AND ocId == %@ AND fileName == %@", account, ocId, fileName))
  281. realm.delete(result)
  282. }
  283. } catch let error {
  284. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  285. }
  286. }
  287. func deleteChunks(account: String, ocId: String) {
  288. do {
  289. let realm = try Realm()
  290. try realm.write {
  291. let result = realm.objects(tableChunk.self).filter(NSPredicate(format: "account == %@ AND ocId == %@", account, ocId))
  292. realm.delete(result)
  293. }
  294. } catch let error {
  295. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  296. }
  297. }
  298. // MARK: -
  299. // MARK: Table Direct Editing
  300. func addDirectEditing(account: String, editors: [NKEditorDetailsEditors], creators: [NKEditorDetailsCreators]) {
  301. do {
  302. let realm = try Realm()
  303. try realm.write {
  304. let resultsCreators = realm.objects(tableDirectEditingCreators.self).filter("account == %@", account)
  305. realm.delete(resultsCreators)
  306. let resultsEditors = realm.objects(tableDirectEditingEditors.self).filter("account == %@", account)
  307. realm.delete(resultsEditors)
  308. for creator in creators {
  309. let addObject = tableDirectEditingCreators()
  310. addObject.account = account
  311. addObject.editor = creator.editor
  312. addObject.ext = creator.ext
  313. addObject.identifier = creator.identifier
  314. addObject.mimetype = creator.mimetype
  315. addObject.name = creator.name
  316. addObject.templates = creator.templates
  317. realm.add(addObject)
  318. }
  319. for editor in editors {
  320. let addObject = tableDirectEditingEditors()
  321. addObject.account = account
  322. for mimeType in editor.mimetypes {
  323. addObject.mimetypes.append(mimeType)
  324. }
  325. addObject.name = editor.name
  326. if editor.name.lowercased() == NCGlobal.shared.editorOnlyoffice {
  327. addObject.editor = NCGlobal.shared.editorOnlyoffice
  328. } else {
  329. addObject.editor = NCGlobal.shared.editorText
  330. }
  331. for mimeType in editor.optionalMimetypes {
  332. addObject.optionalMimetypes.append(mimeType)
  333. }
  334. addObject.secure = editor.secure
  335. realm.add(addObject)
  336. }
  337. }
  338. } catch let error {
  339. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  340. }
  341. }
  342. func getDirectEditingCreators(account: String) -> [tableDirectEditingCreators]? {
  343. do {
  344. let realm = try Realm()
  345. realm.refresh()
  346. let results = realm.objects(tableDirectEditingCreators.self).filter("account == %@", account)
  347. if results.isEmpty {
  348. return nil
  349. } else {
  350. return Array(results.map { tableDirectEditingCreators.init(value: $0) })
  351. }
  352. } catch let error as NSError {
  353. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  354. }
  355. return nil
  356. }
  357. func getDirectEditingCreators(predicate: NSPredicate) -> [tableDirectEditingCreators]? {
  358. do {
  359. let realm = try Realm()
  360. realm.refresh()
  361. let results = realm.objects(tableDirectEditingCreators.self).filter(predicate)
  362. if results.isEmpty {
  363. return nil
  364. } else {
  365. return Array(results.map { tableDirectEditingCreators.init(value: $0) })
  366. }
  367. } catch let error as NSError {
  368. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  369. }
  370. return nil
  371. }
  372. func getDirectEditingEditors(account: String) -> [tableDirectEditingEditors]? {
  373. do {
  374. let realm = try Realm()
  375. realm.refresh()
  376. let results = realm.objects(tableDirectEditingEditors.self).filter("account == %@", account)
  377. if results.isEmpty {
  378. return nil
  379. } else {
  380. return Array(results.map { tableDirectEditingEditors.init(value: $0) })
  381. }
  382. } catch let error as NSError {
  383. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  384. }
  385. return nil
  386. }
  387. // MARK: -
  388. // MARK: Table External Sites
  389. func addExternalSites(_ externalSite: NKExternalSite, account: String) {
  390. do {
  391. let realm = try Realm()
  392. try realm.write {
  393. let addObject = tableExternalSites()
  394. addObject.account = account
  395. addObject.idExternalSite = externalSite.idExternalSite
  396. addObject.icon = externalSite.icon
  397. addObject.lang = externalSite.lang
  398. addObject.name = externalSite.name
  399. addObject.url = externalSite.url
  400. addObject.type = externalSite.type
  401. realm.add(addObject)
  402. }
  403. } catch let error {
  404. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  405. }
  406. }
  407. func deleteExternalSites(account: String) {
  408. do {
  409. let realm = try Realm()
  410. try realm.write {
  411. let results = realm.objects(tableExternalSites.self).filter("account == %@", account)
  412. realm.delete(results)
  413. }
  414. } catch let error {
  415. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  416. }
  417. }
  418. func getAllExternalSites(account: String) -> [tableExternalSites]? {
  419. do {
  420. let realm = try Realm()
  421. realm.refresh()
  422. let results = realm.objects(tableExternalSites.self).filter("account == %@", account).sorted(byKeyPath: "idExternalSite", ascending: true)
  423. if results.isEmpty {
  424. return nil
  425. } else {
  426. return Array(results.map { tableExternalSites.init(value: $0) })
  427. }
  428. } catch let error as NSError {
  429. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  430. }
  431. return nil
  432. }
  433. // MARK: -
  434. // MARK: Table GPS
  435. @objc func addGeocoderLocation(_ location: String, placemarkAdministrativeArea: String, placemarkCountry: String, placemarkLocality: String, placemarkPostalCode: String, placemarkThoroughfare: String, latitude: String, longitude: String) {
  436. do {
  437. let realm = try Realm()
  438. guard realm.objects(tableGPS.self).filter("latitude == %@ AND longitude == %@", latitude, longitude).first == nil else { return }
  439. try realm.write {
  440. let addObject = tableGPS()
  441. addObject.latitude = latitude
  442. addObject.location = location
  443. addObject.longitude = longitude
  444. addObject.placemarkAdministrativeArea = placemarkAdministrativeArea
  445. addObject.placemarkCountry = placemarkCountry
  446. addObject.placemarkLocality = placemarkLocality
  447. addObject.placemarkPostalCode = placemarkPostalCode
  448. addObject.placemarkThoroughfare = placemarkThoroughfare
  449. realm.add(addObject)
  450. }
  451. } catch let error as NSError {
  452. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  453. }
  454. }
  455. @objc func getLocationFromGeoLatitude(_ latitude: String, longitude: String) -> String? {
  456. do {
  457. let realm = try Realm()
  458. realm.refresh()
  459. let result = realm.objects(tableGPS.self).filter("latitude == %@ AND longitude == %@", latitude, longitude).first
  460. return result?.location
  461. } catch let error as NSError {
  462. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  463. }
  464. return nil
  465. }
  466. // MARK: -
  467. // MARK: Table Photo Library
  468. @discardableResult
  469. func addPhotoLibrary(_ assets: [PHAsset], account: String) -> Bool {
  470. do {
  471. let realm = try Realm()
  472. try realm.write {
  473. for asset in assets {
  474. var creationDateString = ""
  475. let addObject = tablePhotoLibrary()
  476. addObject.account = account
  477. addObject.assetLocalIdentifier = asset.localIdentifier
  478. addObject.mediaType = asset.mediaType.rawValue
  479. if let creationDate = asset.creationDate {
  480. addObject.creationDate = creationDate as NSDate
  481. creationDateString = String(describing: creationDate)
  482. }
  483. if let modificationDate = asset.modificationDate {
  484. addObject.modificationDate = modificationDate as NSDate
  485. }
  486. addObject.idAsset = account + asset.localIdentifier + creationDateString
  487. realm.add(addObject, update: .all)
  488. }
  489. }
  490. } catch let error {
  491. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  492. return false
  493. }
  494. return true
  495. }
  496. func getPhotoLibraryIdAsset(image: Bool, video: Bool, account: String) -> [String]? {
  497. var predicate = NSPredicate()
  498. if image && video {
  499. predicate = NSPredicate(format: "account == %@ AND (mediaType == %d OR mediaType == %d)", account, PHAssetMediaType.image.rawValue, PHAssetMediaType.video.rawValue)
  500. } else if image {
  501. predicate = NSPredicate(format: "account == %@ AND mediaType == %d", account, PHAssetMediaType.image.rawValue)
  502. } else if video {
  503. predicate = NSPredicate(format: "account == %@ AND mediaType == %d", account, PHAssetMediaType.video.rawValue)
  504. }
  505. do {
  506. let realm = try Realm()
  507. realm.refresh()
  508. let results = realm.objects(tablePhotoLibrary.self).filter(predicate)
  509. let idsAsset = results.map { $0.idAsset }
  510. return Array(idsAsset)
  511. } catch let error as NSError {
  512. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  513. }
  514. return nil
  515. }
  516. // MARK: -
  517. // MARK: Table Tag
  518. func addTag(_ ocId: String, tagIOS: Data?, account: String) {
  519. do {
  520. let realm = try Realm()
  521. try realm.write {
  522. let addObject = tableTag()
  523. addObject.account = account
  524. addObject.ocId = ocId
  525. addObject.tagIOS = tagIOS
  526. realm.add(addObject, update: .all)
  527. }
  528. } catch let error {
  529. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  530. }
  531. }
  532. func deleteTag(_ ocId: String) {
  533. do {
  534. let realm = try Realm()
  535. try realm.write {
  536. let results = realm.objects(tableTag.self).filter("ocId == %@", ocId)
  537. realm.delete(results)
  538. }
  539. } catch let error {
  540. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  541. }
  542. }
  543. func getTags(predicate: NSPredicate) -> [tableTag] {
  544. do {
  545. let realm = try Realm()
  546. realm.refresh()
  547. let results = realm.objects(tableTag.self).filter(predicate)
  548. return Array(results.map { tableTag.init(value: $0) })
  549. } catch let error as NSError {
  550. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  551. }
  552. return []
  553. }
  554. func getTag(predicate: NSPredicate) -> tableTag? {
  555. do {
  556. let realm = try Realm()
  557. realm.refresh()
  558. guard let result = realm.objects(tableTag.self).filter(predicate).first else { return nil }
  559. return tableTag.init(value: result)
  560. } catch let error as NSError {
  561. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  562. }
  563. return nil
  564. }
  565. // MARK: -
  566. // MARK: Table Tip
  567. func tipExists(_ tipName: String) -> Bool {
  568. do {
  569. let realm = try Realm()
  570. guard (realm.objects(tableTip.self).where {
  571. $0.tipName == tipName
  572. }.first) == nil else {
  573. return true
  574. }
  575. } catch let error as NSError {
  576. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  577. }
  578. return false
  579. }
  580. func addTip(_ tipName: String) {
  581. do {
  582. let realm = try Realm()
  583. try realm.write {
  584. let addObject = tableTip()
  585. addObject.tipName = tipName
  586. realm.add(addObject, update: .all)
  587. }
  588. } catch let error {
  589. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  590. }
  591. }
  592. // MARK: -
  593. // MARK: Table Trash
  594. func addTrash(account: String, items: [NKTrash]) {
  595. do {
  596. let realm = try Realm()
  597. try realm.write {
  598. for trash in items {
  599. let object = tableTrash()
  600. object.account = account
  601. object.contentType = trash.contentType
  602. object.date = trash.date
  603. object.directory = trash.directory
  604. object.fileId = trash.fileId
  605. object.fileName = trash.fileName
  606. object.filePath = trash.filePath
  607. object.hasPreview = trash.hasPreview
  608. object.iconName = trash.iconName
  609. object.size = trash.size
  610. object.trashbinDeletionTime = trash.trashbinDeletionTime
  611. object.trashbinFileName = trash.trashbinFileName
  612. object.trashbinOriginalLocation = trash.trashbinOriginalLocation
  613. object.classFile = trash.classFile
  614. realm.add(object, update: .all)
  615. }
  616. }
  617. } catch let error {
  618. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  619. }
  620. }
  621. func deleteTrash(filePath: String?, account: String) {
  622. var predicate = NSPredicate()
  623. do {
  624. let realm = try Realm()
  625. try realm.write {
  626. if filePath == nil {
  627. predicate = NSPredicate(format: "account == %@", account)
  628. } else {
  629. predicate = NSPredicate(format: "account == %@ AND filePath == %@", account, filePath!)
  630. }
  631. let result = realm.objects(tableTrash.self).filter(predicate)
  632. realm.delete(result)
  633. }
  634. } catch let error {
  635. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  636. }
  637. }
  638. func deleteTrash(fileId: String?, account: String) {
  639. var predicate = NSPredicate()
  640. do {
  641. let realm = try Realm()
  642. try realm.write {
  643. if fileId == nil {
  644. predicate = NSPredicate(format: "account == %@", account)
  645. } else {
  646. predicate = NSPredicate(format: "account == %@ AND fileId == %@", account, fileId!)
  647. }
  648. let result = realm.objects(tableTrash.self).filter(predicate)
  649. realm.delete(result)
  650. }
  651. } catch let error {
  652. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  653. }
  654. }
  655. func getTrash(filePath: String, sort: String?, ascending: Bool?, account: String) -> [tableTrash]? {
  656. let sort = sort ?? "date"
  657. let ascending = ascending ?? false
  658. do {
  659. let realm = try Realm()
  660. realm.refresh()
  661. let results = realm.objects(tableTrash.self).filter("account == %@ AND filePath == %@", account, filePath).sorted(byKeyPath: sort, ascending: ascending)
  662. return Array(results.map { tableTrash.init(value: $0) })
  663. } catch let error as NSError {
  664. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  665. }
  666. return nil
  667. }
  668. func getTrashItem(fileId: String, account: String) -> tableTrash? {
  669. do {
  670. let realm = try Realm()
  671. realm.refresh()
  672. guard let result = realm.objects(tableTrash.self).filter("account == %@ AND fileId == %@", account, fileId).first else { return nil }
  673. return tableTrash.init(value: result)
  674. } catch let error as NSError {
  675. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  676. }
  677. return nil
  678. }
  679. // MARK: -
  680. // MARK: Table UserStatus
  681. func addUserStatus(_ userStatuses: [NKUserStatus], account: String, predefined: Bool) {
  682. do {
  683. let realm = try Realm()
  684. try realm.write {
  685. let results = realm.objects(tableUserStatus.self).filter("account == %@ AND predefined == %@", account, predefined)
  686. realm.delete(results)
  687. for userStatus in userStatuses {
  688. let object = tableUserStatus()
  689. object.account = account
  690. object.clearAt = userStatus.clearAt
  691. object.clearAtTime = userStatus.clearAtTime
  692. object.clearAtType = userStatus.clearAtType
  693. object.icon = userStatus.icon
  694. object.id = userStatus.id
  695. object.message = userStatus.message
  696. object.predefined = userStatus.predefined
  697. object.status = userStatus.status
  698. object.userId = userStatus.userId
  699. realm.add(object)
  700. }
  701. }
  702. } catch let error {
  703. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  704. }
  705. }
  706. }