NCManageDatabase.swift 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983
  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. do {
  132. let realm = try Realm()
  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. func getThreadConfined(_ object: Object) -> Any {
  198. return ThreadSafeReference(to: object)
  199. }
  200. func putThreadConfined(_ tableRef: ThreadSafeReference<Object>) -> Object? {
  201. do {
  202. let realm = try Realm()
  203. return realm.resolve(tableRef)
  204. } catch let error as NSError {
  205. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  206. }
  207. return nil
  208. }
  209. func isTableInvalidated(_ object: Object) -> Bool {
  210. return object.isInvalidated
  211. }
  212. // MARK: -
  213. // MARK: Table Chunk
  214. func getChunkFolder(account: String, ocId: String) -> String {
  215. do {
  216. let realm = try Realm()
  217. guard let result = realm.objects(tableChunk.self).filter("account == %@ AND ocId == %@", account, ocId).first else { return NSUUID().uuidString }
  218. return result.chunkFolder
  219. } catch let error as NSError {
  220. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  221. }
  222. return NSUUID().uuidString
  223. }
  224. func getChunks(account: String, ocId: String) -> [String] {
  225. var filesNames: [String] = []
  226. do {
  227. let realm = try Realm()
  228. let results = realm.objects(tableChunk.self).filter("account == %@ AND ocId == %@", account, ocId).sorted(byKeyPath: "fileName", ascending: true)
  229. for result in results {
  230. filesNames.append(result.fileName)
  231. }
  232. return filesNames
  233. } catch let error as NSError {
  234. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  235. }
  236. return filesNames
  237. }
  238. func addChunks(account: String, ocId: String, chunkFolder: String, fileNames: [String]) {
  239. var size: Int64 = 0
  240. do {
  241. let realm = try Realm()
  242. try realm.write {
  243. for fileName in fileNames {
  244. let object = tableChunk()
  245. size += NCUtilityFileSystem.shared.getFileSize(filePath: CCUtility.getDirectoryProviderStorageOcId(ocId, fileNameView: fileName)!)
  246. object.account = account
  247. object.chunkFolder = chunkFolder
  248. object.fileName = fileName
  249. object.index = ocId + fileName
  250. object.ocId = ocId
  251. object.size = size
  252. realm.add(object, update: .all)
  253. }
  254. }
  255. } catch let error {
  256. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  257. }
  258. }
  259. func getChunk(account: String, fileName: String) -> tableChunk? {
  260. do {
  261. let realm = try Realm()
  262. guard let result = realm.objects(tableChunk.self).filter("account == %@ AND fileName == %@", account, fileName).first else { return nil }
  263. return tableChunk.init(value: result)
  264. } catch let error as NSError {
  265. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  266. }
  267. return nil
  268. }
  269. func deleteChunk(account: String, ocId: String, fileName: String) {
  270. do {
  271. let realm = try Realm()
  272. try realm.write {
  273. let result = realm.objects(tableChunk.self).filter(NSPredicate(format: "account == %@ AND ocId == %@ AND fileName == %@", account, ocId, fileName))
  274. realm.delete(result)
  275. }
  276. } catch let error {
  277. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  278. }
  279. }
  280. func deleteChunks(account: String, ocId: String) {
  281. do {
  282. let realm = try Realm()
  283. try realm.write {
  284. let result = realm.objects(tableChunk.self).filter(NSPredicate(format: "account == %@ AND ocId == %@", account, ocId))
  285. realm.delete(result)
  286. }
  287. } catch let error {
  288. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  289. }
  290. }
  291. // MARK: -
  292. // MARK: Table Direct Editing
  293. func addDirectEditing(account: String, editors: [NKEditorDetailsEditors], creators: [NKEditorDetailsCreators]) {
  294. do {
  295. let realm = try Realm()
  296. try realm.write {
  297. let resultsCreators = realm.objects(tableDirectEditingCreators.self).filter("account == %@", account)
  298. realm.delete(resultsCreators)
  299. let resultsEditors = realm.objects(tableDirectEditingEditors.self).filter("account == %@", account)
  300. realm.delete(resultsEditors)
  301. for creator in creators {
  302. let addObject = tableDirectEditingCreators()
  303. addObject.account = account
  304. addObject.editor = creator.editor
  305. addObject.ext = creator.ext
  306. addObject.identifier = creator.identifier
  307. addObject.mimetype = creator.mimetype
  308. addObject.name = creator.name
  309. addObject.templates = creator.templates
  310. realm.add(addObject)
  311. }
  312. for editor in editors {
  313. let addObject = tableDirectEditingEditors()
  314. addObject.account = account
  315. for mimeType in editor.mimetypes {
  316. addObject.mimetypes.append(mimeType)
  317. }
  318. addObject.name = editor.name
  319. if editor.name.lowercased() == NCGlobal.shared.editorOnlyoffice {
  320. addObject.editor = NCGlobal.shared.editorOnlyoffice
  321. } else {
  322. addObject.editor = NCGlobal.shared.editorText
  323. }
  324. for mimeType in editor.optionalMimetypes {
  325. addObject.optionalMimetypes.append(mimeType)
  326. }
  327. addObject.secure = editor.secure
  328. realm.add(addObject)
  329. }
  330. }
  331. } catch let error {
  332. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  333. }
  334. }
  335. func getDirectEditingCreators(account: String) -> [tableDirectEditingCreators]? {
  336. do {
  337. let realm = try Realm()
  338. let results = realm.objects(tableDirectEditingCreators.self).filter("account == %@", account)
  339. if results.isEmpty {
  340. return nil
  341. } else {
  342. return Array(results.map { tableDirectEditingCreators.init(value: $0) })
  343. }
  344. } catch let error as NSError {
  345. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  346. }
  347. return nil
  348. }
  349. func getDirectEditingCreators(predicate: NSPredicate) -> [tableDirectEditingCreators]? {
  350. do {
  351. let realm = try Realm()
  352. let results = realm.objects(tableDirectEditingCreators.self).filter(predicate)
  353. if results.isEmpty {
  354. return nil
  355. } else {
  356. return Array(results.map { tableDirectEditingCreators.init(value: $0) })
  357. }
  358. } catch let error as NSError {
  359. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  360. }
  361. return nil
  362. }
  363. func getDirectEditingEditors(account: String) -> [tableDirectEditingEditors]? {
  364. do {
  365. let realm = try Realm()
  366. let results = realm.objects(tableDirectEditingEditors.self).filter("account == %@", account)
  367. if results.isEmpty {
  368. return nil
  369. } else {
  370. return Array(results.map { tableDirectEditingEditors.init(value: $0) })
  371. }
  372. } catch let error as NSError {
  373. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  374. }
  375. return nil
  376. }
  377. // MARK: -
  378. // MARK: Table External Sites
  379. func addExternalSites(_ externalSite: NKExternalSite, account: String) {
  380. do {
  381. let realm = try Realm()
  382. try realm.write {
  383. let addObject = tableExternalSites()
  384. addObject.account = account
  385. addObject.idExternalSite = externalSite.idExternalSite
  386. addObject.icon = externalSite.icon
  387. addObject.lang = externalSite.lang
  388. addObject.name = externalSite.name
  389. addObject.url = externalSite.url
  390. addObject.type = externalSite.type
  391. realm.add(addObject)
  392. }
  393. } catch let error {
  394. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  395. }
  396. }
  397. func deleteExternalSites(account: String) {
  398. do {
  399. let realm = try Realm()
  400. try realm.write {
  401. let results = realm.objects(tableExternalSites.self).filter("account == %@", account)
  402. realm.delete(results)
  403. }
  404. } catch let error {
  405. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  406. }
  407. }
  408. func getAllExternalSites(account: String) -> [tableExternalSites]? {
  409. do {
  410. let realm = try Realm()
  411. let results = realm.objects(tableExternalSites.self).filter("account == %@", account).sorted(byKeyPath: "idExternalSite", ascending: true)
  412. if results.isEmpty {
  413. return nil
  414. } else {
  415. return Array(results.map { tableExternalSites.init(value: $0) })
  416. }
  417. } catch let error as NSError {
  418. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  419. }
  420. return nil
  421. }
  422. // MARK: -
  423. // MARK: Table GPS
  424. @objc func addGeocoderLocation(_ location: String, placemarkAdministrativeArea: String, placemarkCountry: String, placemarkLocality: String, placemarkPostalCode: String, placemarkThoroughfare: String, latitude: String, longitude: String) {
  425. do {
  426. let realm = try Realm()
  427. guard realm.objects(tableGPS.self).filter("latitude == %@ AND longitude == %@", latitude, longitude).first == nil else { return }
  428. try realm.write {
  429. let addObject = tableGPS()
  430. addObject.latitude = latitude
  431. addObject.location = location
  432. addObject.longitude = longitude
  433. addObject.placemarkAdministrativeArea = placemarkAdministrativeArea
  434. addObject.placemarkCountry = placemarkCountry
  435. addObject.placemarkLocality = placemarkLocality
  436. addObject.placemarkPostalCode = placemarkPostalCode
  437. addObject.placemarkThoroughfare = placemarkThoroughfare
  438. realm.add(addObject)
  439. }
  440. } catch let error as NSError {
  441. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  442. }
  443. }
  444. @objc func getLocationFromGeoLatitude(_ latitude: String, longitude: String) -> String? {
  445. do {
  446. let realm = try Realm()
  447. let result = realm.objects(tableGPS.self).filter("latitude == %@ AND longitude == %@", latitude, longitude).first
  448. return result?.location
  449. } catch let error as NSError {
  450. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  451. }
  452. return nil
  453. }
  454. // MARK: -
  455. // MARK: Table LocalFile
  456. func addLocalFile(metadata: tableMetadata) {
  457. do {
  458. let realm = try Realm()
  459. try realm.write {
  460. let addObject = getTableLocalFile(predicate: NSPredicate(format: "ocId == %@", metadata.ocId)) ?? tableLocalFile()
  461. addObject.account = metadata.account
  462. addObject.etag = metadata.etag
  463. addObject.exifDate = NSDate()
  464. addObject.exifLatitude = "-1"
  465. addObject.exifLongitude = "-1"
  466. addObject.ocId = metadata.ocId
  467. addObject.fileName = metadata.fileName
  468. realm.add(addObject, update: .all)
  469. }
  470. } catch let error {
  471. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  472. }
  473. }
  474. func addLocalFile(account: String, etag: String, ocId: String, fileName: String) {
  475. do {
  476. let realm = try Realm()
  477. try realm.write {
  478. let addObject = tableLocalFile()
  479. addObject.account = account
  480. addObject.etag = etag
  481. addObject.exifDate = NSDate()
  482. addObject.exifLatitude = "-1"
  483. addObject.exifLongitude = "-1"
  484. addObject.ocId = ocId
  485. addObject.fileName = fileName
  486. realm.add(addObject, update: .all)
  487. }
  488. } catch let error {
  489. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  490. }
  491. }
  492. func deleteLocalFile(predicate: NSPredicate) {
  493. do {
  494. let realm = try Realm()
  495. try realm.write {
  496. let results = realm.objects(tableLocalFile.self).filter(predicate)
  497. realm.delete(results)
  498. }
  499. } catch let error {
  500. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  501. }
  502. }
  503. func setLocalFile(ocId: String, fileName: String?, etag: String?) {
  504. do {
  505. let realm = try Realm()
  506. try realm.write {
  507. let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first
  508. if let fileName = fileName {
  509. result?.fileName = fileName
  510. }
  511. if let etag = etag {
  512. result?.etag = etag
  513. }
  514. }
  515. } catch let error {
  516. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  517. }
  518. }
  519. @objc func setLocalFile(ocId: String, exifDate: NSDate?, exifLatitude: String, exifLongitude: String, exifLensModel: String?) {
  520. do {
  521. let realm = try Realm()
  522. try realm.write {
  523. if let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first {
  524. result.exifDate = exifDate
  525. result.exifLatitude = exifLatitude
  526. result.exifLongitude = exifLongitude
  527. if exifLensModel?.count ?? 0 > 0 {
  528. result.exifLensModel = exifLensModel
  529. }
  530. }
  531. }
  532. } catch let error {
  533. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  534. }
  535. }
  536. func getTableLocalFile(account: String) -> [tableLocalFile] {
  537. do {
  538. let realm = try Realm()
  539. let results = realm.objects(tableLocalFile.self).filter("account == %@", account)
  540. return Array(results.map { tableLocalFile.init(value: $0) })
  541. } catch let error as NSError {
  542. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  543. }
  544. return []
  545. }
  546. func getTableLocalFile(predicate: NSPredicate) -> tableLocalFile? {
  547. do {
  548. let realm = try Realm()
  549. guard let result = realm.objects(tableLocalFile.self).filter(predicate).first else { return nil }
  550. return tableLocalFile.init(value: result)
  551. } catch let error as NSError {
  552. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  553. }
  554. return nil
  555. }
  556. func getTableLocalFiles(predicate: NSPredicate, sorted: String, ascending: Bool) -> [tableLocalFile] {
  557. do {
  558. let realm = try Realm()
  559. let results = realm.objects(tableLocalFile.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  560. return Array(results.map { tableLocalFile.init(value: $0) })
  561. } catch let error as NSError {
  562. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  563. }
  564. return []
  565. }
  566. func setLocalFile(ocId: String, offline: Bool) {
  567. do {
  568. let realm = try Realm()
  569. try realm.write {
  570. let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first
  571. result?.offline = offline
  572. }
  573. } catch let error {
  574. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  575. }
  576. }
  577. // MARK: -
  578. // MARK: Table Photo Library
  579. @discardableResult
  580. func addPhotoLibrary(_ assets: [PHAsset], account: String) -> Bool {
  581. do {
  582. let realm = try Realm()
  583. try realm.write {
  584. for asset in assets {
  585. var creationDateString = ""
  586. let addObject = tablePhotoLibrary()
  587. addObject.account = account
  588. addObject.assetLocalIdentifier = asset.localIdentifier
  589. addObject.mediaType = asset.mediaType.rawValue
  590. if let creationDate = asset.creationDate {
  591. addObject.creationDate = creationDate as NSDate
  592. creationDateString = String(describing: creationDate)
  593. }
  594. if let modificationDate = asset.modificationDate {
  595. addObject.modificationDate = modificationDate as NSDate
  596. }
  597. addObject.idAsset = account + asset.localIdentifier + creationDateString
  598. realm.add(addObject, update: .all)
  599. }
  600. }
  601. } catch let error {
  602. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  603. return false
  604. }
  605. return true
  606. }
  607. func getPhotoLibraryIdAsset(image: Bool, video: Bool, account: String) -> [String]? {
  608. var predicate = NSPredicate()
  609. if image && video {
  610. predicate = NSPredicate(format: "account == %@ AND (mediaType == %d OR mediaType == %d)", account, PHAssetMediaType.image.rawValue, PHAssetMediaType.video.rawValue)
  611. } else if image {
  612. predicate = NSPredicate(format: "account == %@ AND mediaType == %d", account, PHAssetMediaType.image.rawValue)
  613. } else if video {
  614. predicate = NSPredicate(format: "account == %@ AND mediaType == %d", account, PHAssetMediaType.video.rawValue)
  615. }
  616. do {
  617. let realm = try Realm()
  618. let results = realm.objects(tablePhotoLibrary.self).filter(predicate)
  619. let idsAsset = results.map { $0.idAsset }
  620. return Array(idsAsset)
  621. } catch let error as NSError {
  622. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  623. }
  624. return nil
  625. }
  626. // MARK: -
  627. // MARK: Table Tag
  628. func addTag(_ ocId: String, tagIOS: Data?, account: String) {
  629. do {
  630. let realm = try Realm()
  631. try realm.write {
  632. let addObject = tableTag()
  633. addObject.account = account
  634. addObject.ocId = ocId
  635. addObject.tagIOS = tagIOS
  636. realm.add(addObject, update: .all)
  637. }
  638. } catch let error {
  639. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  640. }
  641. }
  642. func deleteTag(_ ocId: String) {
  643. do {
  644. let realm = try Realm()
  645. try realm.write {
  646. let results = realm.objects(tableTag.self).filter("ocId == %@", ocId)
  647. realm.delete(results)
  648. }
  649. } catch let error {
  650. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  651. }
  652. }
  653. func getTags(predicate: NSPredicate) -> [tableTag] {
  654. do {
  655. let realm = try Realm()
  656. let results = realm.objects(tableTag.self).filter(predicate)
  657. return Array(results.map { tableTag.init(value: $0) })
  658. } catch let error as NSError {
  659. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  660. }
  661. return []
  662. }
  663. func getTag(predicate: NSPredicate) -> tableTag? {
  664. do {
  665. let realm = try Realm()
  666. guard let result = realm.objects(tableTag.self).filter(predicate).first else { return nil }
  667. return tableTag.init(value: result)
  668. } catch let error as NSError {
  669. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  670. }
  671. return nil
  672. }
  673. // MARK: -
  674. // MARK: Table Tip
  675. func tipExists(_ tipName: String) -> Bool {
  676. do {
  677. let realm = try Realm()
  678. guard (realm.objects(tableTip.self).where {
  679. $0.tipName == tipName
  680. }.first) == nil else {
  681. return true
  682. }
  683. } catch let error as NSError {
  684. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  685. }
  686. return false
  687. }
  688. func addTip(_ tipName: String) {
  689. do {
  690. let realm = try Realm()
  691. try realm.write {
  692. let addObject = tableTip()
  693. addObject.tipName = tipName
  694. realm.add(addObject, update: .all)
  695. }
  696. } catch let error {
  697. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  698. }
  699. }
  700. // MARK: -
  701. // MARK: Table Trash
  702. func addTrash(account: String, items: [NKTrash]) {
  703. do {
  704. let realm = try Realm()
  705. try realm.write {
  706. for trash in items {
  707. let object = tableTrash()
  708. object.account = account
  709. object.contentType = trash.contentType
  710. object.date = trash.date
  711. object.directory = trash.directory
  712. object.fileId = trash.fileId
  713. object.fileName = trash.fileName
  714. object.filePath = trash.filePath
  715. object.hasPreview = trash.hasPreview
  716. object.iconName = trash.iconName
  717. object.size = trash.size
  718. object.trashbinDeletionTime = trash.trashbinDeletionTime
  719. object.trashbinFileName = trash.trashbinFileName
  720. object.trashbinOriginalLocation = trash.trashbinOriginalLocation
  721. object.classFile = trash.classFile
  722. realm.add(object, update: .all)
  723. }
  724. }
  725. } catch let error {
  726. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  727. }
  728. }
  729. func deleteTrash(filePath: String?, account: String) {
  730. var predicate = NSPredicate()
  731. do {
  732. let realm = try Realm()
  733. try realm.write {
  734. if filePath == nil {
  735. predicate = NSPredicate(format: "account == %@", account)
  736. } else {
  737. predicate = NSPredicate(format: "account == %@ AND filePath == %@", account, filePath!)
  738. }
  739. let result = realm.objects(tableTrash.self).filter(predicate)
  740. realm.delete(result)
  741. }
  742. } catch let error {
  743. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  744. }
  745. }
  746. func deleteTrash(fileId: String?, account: String) {
  747. var predicate = NSPredicate()
  748. do {
  749. let realm = try Realm()
  750. try realm.write {
  751. if fileId == nil {
  752. predicate = NSPredicate(format: "account == %@", account)
  753. } else {
  754. predicate = NSPredicate(format: "account == %@ AND fileId == %@", account, fileId!)
  755. }
  756. let result = realm.objects(tableTrash.self).filter(predicate)
  757. realm.delete(result)
  758. }
  759. } catch let error {
  760. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  761. }
  762. }
  763. func getTrash(filePath: String, sort: String?, ascending: Bool?, account: String) -> [tableTrash]? {
  764. let sort = sort ?? "date"
  765. let ascending = ascending ?? false
  766. do {
  767. let realm = try Realm()
  768. let results = realm.objects(tableTrash.self).filter("account == %@ AND filePath == %@", account, filePath).sorted(byKeyPath: sort, ascending: ascending)
  769. return Array(results.map { tableTrash.init(value: $0) })
  770. } catch let error as NSError {
  771. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  772. }
  773. return nil
  774. }
  775. func getTrashItem(fileId: String, account: String) -> tableTrash? {
  776. do {
  777. let realm = try Realm()
  778. guard let result = realm.objects(tableTrash.self).filter("account == %@ AND fileId == %@", account, fileId).first else { return nil }
  779. return tableTrash.init(value: result)
  780. } catch let error as NSError {
  781. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  782. }
  783. return nil
  784. }
  785. // MARK: -
  786. // MARK: Table UserStatus
  787. func addUserStatus(_ userStatuses: [NKUserStatus], account: String, predefined: Bool) {
  788. do {
  789. let realm = try Realm()
  790. try realm.write {
  791. let results = realm.objects(tableUserStatus.self).filter("account == %@ AND predefined == %@", account, predefined)
  792. realm.delete(results)
  793. for userStatus in userStatuses {
  794. let object = tableUserStatus()
  795. object.account = account
  796. object.clearAt = userStatus.clearAt
  797. object.clearAtTime = userStatus.clearAtTime
  798. object.clearAtType = userStatus.clearAtType
  799. object.icon = userStatus.icon
  800. object.id = userStatus.id
  801. object.message = userStatus.message
  802. object.predefined = userStatus.predefined
  803. object.status = userStatus.status
  804. object.userId = userStatus.userId
  805. realm.add(object)
  806. }
  807. }
  808. } catch let error {
  809. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  810. }
  811. }
  812. }