NCManageDatabase.swift 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978
  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. _ = try! Realm()
  127. }
  128. // MARK: -
  129. // MARK: Utility Database
  130. @objc func clearTable(_ table: Object.Type, account: String? = nil) {
  131. let realm = try! Realm()
  132. do {
  133. try realm.write {
  134. var results: Results<Object>
  135. if let account = account {
  136. results = realm.objects(table).filter("account == %@", account)
  137. } else {
  138. results = realm.objects(table)
  139. }
  140. realm.delete(results)
  141. }
  142. } catch let error {
  143. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  144. }
  145. }
  146. @objc func clearDatabase(account: String?, removeAccount: Bool) {
  147. self.clearTable(tableActivity.self, account: account)
  148. self.clearTable(tableActivityLatestId.self, account: account)
  149. self.clearTable(tableActivityPreview.self, account: account)
  150. self.clearTable(tableActivitySubjectRich.self, account: account)
  151. self.clearTable(tableAvatar.self)
  152. self.clearTable(tableCapabilities.self, account: account)
  153. self.clearTable(tableChunk.self, account: account)
  154. self.clearTable(tableComments.self, account: account)
  155. self.clearTable(tableDashboardWidget.self, account: account)
  156. self.clearTable(tableDashboardWidgetButton.self, account: account)
  157. self.clearTable(tableDirectEditingCreators.self, account: account)
  158. self.clearTable(tableDirectEditingEditors.self, account: account)
  159. self.clearTable(tableDirectory.self, account: account)
  160. self.clearTable(tableE2eEncryption.self, account: account)
  161. self.clearTable(tableE2eEncryptionLock.self, account: account)
  162. self.clearTable(tableE2eMetadata.self, account: account)
  163. self.clearTable(tableExternalSites.self, account: account)
  164. self.clearTable(tableGPS.self, account: nil)
  165. self.clearTable(TableGroupfolders.self, account: account)
  166. self.clearTable(TableGroupfoldersGroups.self, account: account)
  167. self.clearTable(NCDBLayoutForView.self, account: account)
  168. self.clearTable(tableLocalFile.self, account: account)
  169. self.clearTable(tableMetadata.self, account: account)
  170. self.clearTable(tablePhotoLibrary.self, account: account)
  171. self.clearTable(tableShare.self, account: account)
  172. self.clearTable(tableTag.self, account: account)
  173. self.clearTable(tableTip.self)
  174. self.clearTable(tableTrash.self, account: account)
  175. self.clearTable(tableUserStatus.self, account: account)
  176. self.clearTable(tableVideo.self, account: account)
  177. if removeAccount {
  178. self.clearTable(tableAccount.self, account: account)
  179. }
  180. }
  181. @objc func removeDB() {
  182. let realmURL = Realm.Configuration.defaultConfiguration.fileURL!
  183. let realmURLs = [
  184. realmURL,
  185. realmURL.appendingPathExtension("lock"),
  186. realmURL.appendingPathExtension("note"),
  187. realmURL.appendingPathExtension("management")
  188. ]
  189. for URL in realmURLs {
  190. do {
  191. try FileManager.default.removeItem(at: URL)
  192. } catch let error {
  193. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  194. }
  195. }
  196. }
  197. @objc func getThreadConfined(_ object: Object) -> Any {
  198. // id tradeReference = [[NCManageDatabase shared] getThreadConfined:metadata];
  199. return ThreadSafeReference(to: object)
  200. }
  201. @objc func putThreadConfined(_ tableRef: Any) -> Object? {
  202. // tableMetadata *metadataThread = (tableMetadata *)[[NCManageDatabase shared] putThreadConfined:tradeReference];
  203. let realm = try! Realm()
  204. return realm.resolve(tableRef as! ThreadSafeReference<Object>)
  205. }
  206. @objc func isTableInvalidated(_ object: Object) -> Bool {
  207. return object.isInvalidated
  208. }
  209. // MARK: -
  210. // MARK: Table Chunk
  211. func getChunkFolder(account: String, ocId: String) -> String {
  212. let realm = try! Realm()
  213. if let result = realm.objects(tableChunk.self).filter("account == %@ AND ocId == %@", account, ocId).first {
  214. return result.chunkFolder
  215. }
  216. return NSUUID().uuidString
  217. }
  218. func getChunks(account: String, ocId: String) -> [String] {
  219. let realm = try! Realm()
  220. var filesNames: [String] = []
  221. let results = realm.objects(tableChunk.self).filter("account == %@ AND ocId == %@", account, ocId).sorted(byKeyPath: "fileName", ascending: true)
  222. for result in results {
  223. filesNames.append(result.fileName)
  224. }
  225. return filesNames
  226. }
  227. func addChunks(account: String, ocId: String, chunkFolder: String, fileNames: [String]) {
  228. let realm = try! Realm()
  229. var size: Int64 = 0
  230. do {
  231. try realm.write {
  232. for fileName in fileNames {
  233. let object = tableChunk()
  234. size += NCUtilityFileSystem.shared.getFileSize(filePath: CCUtility.getDirectoryProviderStorageOcId(ocId, fileNameView: fileName)!)
  235. object.account = account
  236. object.chunkFolder = chunkFolder
  237. object.fileName = fileName
  238. object.index = ocId + fileName
  239. object.ocId = ocId
  240. object.size = size
  241. realm.add(object, update: .all)
  242. }
  243. }
  244. } catch let error {
  245. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  246. }
  247. }
  248. func getChunk(account: String, fileName: String) -> tableChunk? {
  249. let realm = try! Realm()
  250. if let result = realm.objects(tableChunk.self).filter("account == %@ AND fileName == %@", account, fileName).first {
  251. return tableChunk.init(value: result)
  252. } else {
  253. return nil
  254. }
  255. }
  256. func deleteChunk(account: String, ocId: String, fileName: String) {
  257. let realm = try! Realm()
  258. do {
  259. try realm.write {
  260. let result = realm.objects(tableChunk.self).filter(NSPredicate(format: "account == %@ AND ocId == %@ AND fileName == %@", account, ocId, fileName))
  261. realm.delete(result)
  262. }
  263. } catch let error {
  264. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  265. }
  266. }
  267. func deleteChunks(account: String, ocId: String) {
  268. let realm = try! Realm()
  269. do {
  270. try realm.write {
  271. let result = realm.objects(tableChunk.self).filter(NSPredicate(format: "account == %@ AND ocId == %@", account, ocId))
  272. realm.delete(result)
  273. }
  274. } catch let error {
  275. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  276. }
  277. }
  278. // MARK: -
  279. // MARK: Table Direct Editing
  280. func addDirectEditing(account: String, editors: [NKEditorDetailsEditors], creators: [NKEditorDetailsCreators]) {
  281. let realm = try! Realm()
  282. do {
  283. try realm.write {
  284. let resultsCreators = realm.objects(tableDirectEditingCreators.self).filter("account == %@", account)
  285. realm.delete(resultsCreators)
  286. let resultsEditors = realm.objects(tableDirectEditingEditors.self).filter("account == %@", account)
  287. realm.delete(resultsEditors)
  288. for creator in creators {
  289. let addObject = tableDirectEditingCreators()
  290. addObject.account = account
  291. addObject.editor = creator.editor
  292. addObject.ext = creator.ext
  293. addObject.identifier = creator.identifier
  294. addObject.mimetype = creator.mimetype
  295. addObject.name = creator.name
  296. addObject.templates = creator.templates
  297. realm.add(addObject)
  298. }
  299. for editor in editors {
  300. let addObject = tableDirectEditingEditors()
  301. addObject.account = account
  302. for mimeType in editor.mimetypes {
  303. addObject.mimetypes.append(mimeType)
  304. }
  305. addObject.name = editor.name
  306. if editor.name.lowercased() == NCGlobal.shared.editorOnlyoffice {
  307. addObject.editor = NCGlobal.shared.editorOnlyoffice
  308. } else {
  309. addObject.editor = NCGlobal.shared.editorText
  310. }
  311. for mimeType in editor.optionalMimetypes {
  312. addObject.optionalMimetypes.append(mimeType)
  313. }
  314. addObject.secure = editor.secure
  315. realm.add(addObject)
  316. }
  317. }
  318. } catch let error {
  319. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  320. }
  321. }
  322. func getDirectEditingCreators(account: String) -> [tableDirectEditingCreators]? {
  323. let realm = try! Realm()
  324. let results = realm.objects(tableDirectEditingCreators.self).filter("account == %@", account)
  325. if results.count > 0 {
  326. return Array(results.map { tableDirectEditingCreators.init(value: $0) })
  327. } else {
  328. return nil
  329. }
  330. }
  331. func getDirectEditingCreators(predicate: NSPredicate) -> [tableDirectEditingCreators]? {
  332. let realm = try! Realm()
  333. let results = realm.objects(tableDirectEditingCreators.self).filter(predicate)
  334. if results.count > 0 {
  335. return Array(results.map { tableDirectEditingCreators.init(value: $0) })
  336. } else {
  337. return nil
  338. }
  339. }
  340. func getDirectEditingEditors(account: String) -> [tableDirectEditingEditors]? {
  341. let realm = try! Realm()
  342. let results = realm.objects(tableDirectEditingEditors.self).filter("account == %@", account)
  343. if results.count > 0 {
  344. return Array(results.map { tableDirectEditingEditors.init(value: $0) })
  345. } else {
  346. return nil
  347. }
  348. }
  349. // MARK: -
  350. // MARK: Table External Sites
  351. func addExternalSites(_ externalSite: NKExternalSite, account: String) {
  352. let realm = try! Realm()
  353. do {
  354. try realm.write {
  355. let addObject = tableExternalSites()
  356. addObject.account = account
  357. addObject.idExternalSite = externalSite.idExternalSite
  358. addObject.icon = externalSite.icon
  359. addObject.lang = externalSite.lang
  360. addObject.name = externalSite.name
  361. addObject.url = externalSite.url
  362. addObject.type = externalSite.type
  363. realm.add(addObject)
  364. }
  365. } catch let error {
  366. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  367. }
  368. }
  369. func deleteExternalSites(account: String) {
  370. let realm = try! Realm()
  371. do {
  372. try realm.write {
  373. let results = realm.objects(tableExternalSites.self).filter("account == %@", account)
  374. realm.delete(results)
  375. }
  376. } catch let error {
  377. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  378. }
  379. }
  380. func getAllExternalSites(account: String) -> [tableExternalSites]? {
  381. let realm = try! Realm()
  382. let results = realm.objects(tableExternalSites.self).filter("account == %@", account).sorted(byKeyPath: "idExternalSite", ascending: true)
  383. if results.count > 0 {
  384. return Array(results.map { tableExternalSites.init(value: $0) })
  385. } else {
  386. return nil
  387. }
  388. }
  389. // MARK: -
  390. // MARK: Table GPS
  391. @objc func addGeocoderLocation(_ location: String, placemarkAdministrativeArea: String, placemarkCountry: String, placemarkLocality: String, placemarkPostalCode: String, placemarkThoroughfare: String, latitude: String, longitude: String) {
  392. let realm = try! Realm()
  393. realm.beginWrite()
  394. // Verify if exists
  395. guard realm.objects(tableGPS.self).filter("latitude == %@ AND longitude == %@", latitude, longitude).first == nil else {
  396. realm.cancelWrite()
  397. return
  398. }
  399. // Add new GPS
  400. let addObject = tableGPS()
  401. addObject.latitude = latitude
  402. addObject.location = location
  403. addObject.longitude = longitude
  404. addObject.placemarkAdministrativeArea = placemarkAdministrativeArea
  405. addObject.placemarkCountry = placemarkCountry
  406. addObject.placemarkLocality = placemarkLocality
  407. addObject.placemarkPostalCode = placemarkPostalCode
  408. addObject.placemarkThoroughfare = placemarkThoroughfare
  409. realm.add(addObject)
  410. do {
  411. try realm.commitWrite()
  412. } catch let error {
  413. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  414. }
  415. }
  416. @objc func getLocationFromGeoLatitude(_ latitude: String, longitude: String) -> String? {
  417. let realm = try! Realm()
  418. let result = realm.objects(tableGPS.self).filter("latitude == %@ AND longitude == %@", latitude, longitude).first
  419. return result?.location
  420. }
  421. // MARK: -
  422. // MARK: Table LocalFile
  423. func addLocalFile(metadata: tableMetadata) {
  424. let realm = try! Realm()
  425. do {
  426. try realm.write {
  427. let addObject = getTableLocalFile(predicate: NSPredicate(format: "ocId == %@", metadata.ocId)) ?? tableLocalFile()
  428. addObject.account = metadata.account
  429. addObject.etag = metadata.etag
  430. addObject.exifDate = NSDate()
  431. addObject.exifLatitude = "-1"
  432. addObject.exifLongitude = "-1"
  433. addObject.ocId = metadata.ocId
  434. addObject.fileName = metadata.fileName
  435. realm.add(addObject, update: .all)
  436. }
  437. } catch let error {
  438. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  439. }
  440. }
  441. func addLocalFile(account: String, etag: String, ocId: String, fileName: String) {
  442. let realm = try! Realm()
  443. do {
  444. try realm.write {
  445. let addObject = tableLocalFile()
  446. addObject.account = account
  447. addObject.etag = etag
  448. addObject.exifDate = NSDate()
  449. addObject.exifLatitude = "-1"
  450. addObject.exifLongitude = "-1"
  451. addObject.ocId = ocId
  452. addObject.fileName = fileName
  453. realm.add(addObject, update: .all)
  454. }
  455. } catch let error {
  456. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  457. }
  458. }
  459. func deleteLocalFile(predicate: NSPredicate) {
  460. let realm = try! Realm()
  461. do {
  462. try realm.write {
  463. let results = realm.objects(tableLocalFile.self).filter(predicate)
  464. realm.delete(results)
  465. }
  466. } catch let error {
  467. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  468. }
  469. }
  470. func setLocalFile(ocId: String, fileName: String?, etag: String?) {
  471. let realm = try! Realm()
  472. do {
  473. try realm.write {
  474. let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first
  475. if let fileName = fileName {
  476. result?.fileName = fileName
  477. }
  478. if let etag = etag {
  479. result?.etag = etag
  480. }
  481. }
  482. } catch let error {
  483. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  484. }
  485. }
  486. @objc func setLocalFile(ocId: String, exifDate: NSDate?, exifLatitude: String, exifLongitude: String, exifLensModel: String?) {
  487. let realm = try! Realm()
  488. do {
  489. try realm.write {
  490. if let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first {
  491. result.exifDate = exifDate
  492. result.exifLatitude = exifLatitude
  493. result.exifLongitude = exifLongitude
  494. if exifLensModel?.count ?? 0 > 0 {
  495. result.exifLensModel = exifLensModel
  496. }
  497. }
  498. }
  499. } catch let error {
  500. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  501. }
  502. }
  503. func getTableLocalFile(account: String) -> [tableLocalFile] {
  504. let realm = try! Realm()
  505. let results = realm.objects(tableLocalFile.self).filter("account == %@", account)
  506. return Array(results.map { tableLocalFile.init(value: $0) })
  507. }
  508. func getTableLocalFile(predicate: NSPredicate) -> tableLocalFile? {
  509. let realm = try! Realm()
  510. guard let result = realm.objects(tableLocalFile.self).filter(predicate).first else {
  511. return nil
  512. }
  513. return tableLocalFile.init(value: result)
  514. }
  515. func getTableLocalFiles(predicate: NSPredicate, sorted: String, ascending: Bool) -> [tableLocalFile] {
  516. let realm = try! Realm()
  517. let results = realm.objects(tableLocalFile.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  518. return Array(results.map { tableLocalFile.init(value: $0) })
  519. }
  520. func setLocalFile(ocId: String, offline: Bool) {
  521. let realm = try! Realm()
  522. do {
  523. try realm.write {
  524. let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first
  525. result?.offline = offline
  526. }
  527. } catch let error {
  528. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  529. }
  530. }
  531. // MARK: -
  532. // MARK: Table Photo Library
  533. @discardableResult
  534. func addPhotoLibrary(_ assets: [PHAsset], account: String) -> Bool {
  535. let realm = try! Realm()
  536. do {
  537. try realm.write {
  538. for asset in assets {
  539. var creationDateString = ""
  540. let addObject = tablePhotoLibrary()
  541. addObject.account = account
  542. addObject.assetLocalIdentifier = asset.localIdentifier
  543. addObject.mediaType = asset.mediaType.rawValue
  544. if let creationDate = asset.creationDate {
  545. addObject.creationDate = creationDate as NSDate
  546. creationDateString = String(describing: creationDate)
  547. }
  548. if let modificationDate = asset.modificationDate {
  549. addObject.modificationDate = modificationDate as NSDate
  550. }
  551. addObject.idAsset = account + asset.localIdentifier + creationDateString
  552. realm.add(addObject, update: .all)
  553. }
  554. }
  555. } catch let error {
  556. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  557. return false
  558. }
  559. return true
  560. }
  561. func getPhotoLibraryIdAsset(image: Bool, video: Bool, account: String) -> [String]? {
  562. let realm = try! Realm()
  563. var predicate = NSPredicate()
  564. if image && video {
  565. predicate = NSPredicate(format: "account == %@ AND (mediaType == %d OR mediaType == %d)", account, PHAssetMediaType.image.rawValue, PHAssetMediaType.video.rawValue)
  566. } else if image {
  567. predicate = NSPredicate(format: "account == %@ AND mediaType == %d", account, PHAssetMediaType.image.rawValue)
  568. } else if video {
  569. predicate = NSPredicate(format: "account == %@ AND mediaType == %d", account, PHAssetMediaType.video.rawValue)
  570. }
  571. let results = realm.objects(tablePhotoLibrary.self).filter(predicate)
  572. let idsAsset = results.map { $0.idAsset }
  573. return Array(idsAsset)
  574. }
  575. // MARK: -
  576. // MARK: Table Tag
  577. func addTag(_ ocId: String, tagIOS: Data?, account: String) {
  578. let realm = try! Realm()
  579. do {
  580. try realm.write {
  581. // Add new
  582. let addObject = tableTag()
  583. addObject.account = account
  584. addObject.ocId = ocId
  585. addObject.tagIOS = tagIOS
  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. func deleteTag(_ ocId: String) {
  593. let realm = try! Realm()
  594. realm.beginWrite()
  595. let result = realm.objects(tableTag.self).filter("ocId == %@", ocId)
  596. realm.delete(result)
  597. do {
  598. try realm.commitWrite()
  599. } catch let error {
  600. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  601. }
  602. }
  603. func getTags(predicate: NSPredicate) -> [tableTag] {
  604. let realm = try! Realm()
  605. let results = realm.objects(tableTag.self).filter(predicate)
  606. return Array(results.map { tableTag.init(value: $0) })
  607. }
  608. func getTag(predicate: NSPredicate) -> tableTag? {
  609. let realm = try! Realm()
  610. guard let result = realm.objects(tableTag.self).filter(predicate).first else {
  611. return nil
  612. }
  613. return tableTag.init(value: result)
  614. }
  615. // MARK: -
  616. // MARK: Table Tip
  617. func tipExists(_ tipName: String) -> Bool {
  618. let realm = try! Realm()
  619. guard (realm.objects(tableTip.self).where {
  620. $0.tipName == tipName
  621. }.first) == nil else {
  622. return true
  623. }
  624. return false
  625. }
  626. func addTip(_ tipName: String) {
  627. let realm = try! Realm()
  628. do {
  629. try realm.write {
  630. let addObject = tableTip()
  631. addObject.tipName = tipName
  632. realm.add(addObject, update: .all)
  633. }
  634. } catch let error {
  635. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  636. }
  637. }
  638. // MARK: -
  639. // MARK: Table Trash
  640. func addTrash(account: String, items: [NKTrash]) {
  641. let realm = try! Realm()
  642. do {
  643. try realm.write {
  644. for trash in items {
  645. let object = tableTrash()
  646. object.account = account
  647. object.contentType = trash.contentType
  648. object.date = trash.date
  649. object.directory = trash.directory
  650. object.fileId = trash.fileId
  651. object.fileName = trash.fileName
  652. object.filePath = trash.filePath
  653. object.hasPreview = trash.hasPreview
  654. object.iconName = trash.iconName
  655. object.size = trash.size
  656. object.trashbinDeletionTime = trash.trashbinDeletionTime
  657. object.trashbinFileName = trash.trashbinFileName
  658. object.trashbinOriginalLocation = trash.trashbinOriginalLocation
  659. object.classFile = trash.classFile
  660. realm.add(object, update: .all)
  661. }
  662. }
  663. } catch let error {
  664. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  665. }
  666. }
  667. func deleteTrash(filePath: String?, account: String) {
  668. let realm = try! Realm()
  669. var predicate = NSPredicate()
  670. do {
  671. try realm.write {
  672. if filePath == nil {
  673. predicate = NSPredicate(format: "account == %@", account)
  674. } else {
  675. predicate = NSPredicate(format: "account == %@ AND filePath == %@", account, filePath!)
  676. }
  677. let result = realm.objects(tableTrash.self).filter(predicate)
  678. realm.delete(result)
  679. }
  680. } catch let error {
  681. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  682. }
  683. }
  684. func deleteTrash(fileId: String?, account: String) {
  685. let realm = try! Realm()
  686. var predicate = NSPredicate()
  687. do {
  688. try realm.write {
  689. if fileId == nil {
  690. predicate = NSPredicate(format: "account == %@", account)
  691. } else {
  692. predicate = NSPredicate(format: "account == %@ AND fileId == %@", account, fileId!)
  693. }
  694. let result = realm.objects(tableTrash.self).filter(predicate)
  695. realm.delete(result)
  696. }
  697. } catch let error {
  698. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  699. }
  700. }
  701. func getTrash(filePath: String, sort: String?, ascending: Bool?, account: String) -> [tableTrash]? {
  702. let realm = try! Realm()
  703. let sort = sort ?? "date"
  704. let ascending = ascending ?? false
  705. let results = realm.objects(tableTrash.self).filter("account == %@ AND filePath == %@", account, filePath).sorted(byKeyPath: sort, ascending: ascending)
  706. return Array(results.map { tableTrash.init(value: $0) })
  707. }
  708. func getTrashItem(fileId: String, account: String) -> tableTrash? {
  709. let realm = try! Realm()
  710. guard let result = realm.objects(tableTrash.self).filter("account == %@ AND fileId == %@", account, fileId).first else {
  711. return nil
  712. }
  713. return tableTrash.init(value: result)
  714. }
  715. // MARK: -
  716. // MARK: Table UserStatus
  717. func addUserStatus(_ userStatuses: [NKUserStatus], account: String, predefined: Bool) {
  718. let realm = try! Realm()
  719. do {
  720. try realm.write {
  721. let results = realm.objects(tableUserStatus.self).filter("account == %@ AND predefined == %@", account, predefined)
  722. realm.delete(results)
  723. for userStatus in userStatuses {
  724. let object = tableUserStatus()
  725. object.account = account
  726. object.clearAt = userStatus.clearAt
  727. object.clearAtTime = userStatus.clearAtTime
  728. object.clearAtType = userStatus.clearAtType
  729. object.icon = userStatus.icon
  730. object.id = userStatus.id
  731. object.message = userStatus.message
  732. object.predefined = userStatus.predefined
  733. object.status = userStatus.status
  734. object.userId = userStatus.userId
  735. realm.add(object)
  736. }
  737. }
  738. } catch let error {
  739. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  740. }
  741. }
  742. }