NCManageDatabase.swift 37 KB

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