NCManageDatabase.swift 36 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090
  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 SwiftyJSON
  28. import CoreMedia
  29. import Photos
  30. class NCManageDatabase: NSObject {
  31. @objc static let shared: NCManageDatabase = {
  32. let instance = NCManageDatabase()
  33. return instance
  34. }()
  35. override init() {
  36. let dirGroup = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: NCBrandOptions.shared.capabilitiesGroups)
  37. let databaseFileUrlPath = dirGroup?.appendingPathComponent(NCGlobal.shared.appDatabaseNextcloud + "/" + databaseName)
  38. let bundleUrl: URL = Bundle.main.bundleURL
  39. let bundlePathExtension: String = bundleUrl.pathExtension
  40. let isAppex: Bool = bundlePathExtension == "appex"
  41. if let databaseFilePath = databaseFileUrlPath?.path {
  42. if FileManager.default.fileExists(atPath: databaseFilePath) {
  43. NextcloudKit.shared.nkCommonInstance.writeLog("DATABASE FOUND in " + databaseFilePath)
  44. } else {
  45. NextcloudKit.shared.nkCommonInstance.writeLog("DATABASE NOT FOUND in " + databaseFilePath)
  46. }
  47. }
  48. // Disable file protection for directory DB
  49. // https://docs.mongodb.com/realm/sdk/ios/examples/configure-and-open-a-realm/#std-label-ios-open-a-local-realm
  50. if let folderPathURL = dirGroup?.appendingPathComponent(NCGlobal.shared.appDatabaseNextcloud) {
  51. let folderPath = folderPathURL.path
  52. do {
  53. try FileManager.default.setAttributes([FileAttributeKey.protectionKey: FileProtectionType.completeUntilFirstUserAuthentication], ofItemAtPath: folderPath)
  54. } catch {
  55. print("Dangerous error")
  56. }
  57. }
  58. if isAppex {
  59. // App Extension config
  60. let config = Realm.Configuration(
  61. fileURL: dirGroup?.appendingPathComponent(NCGlobal.shared.appDatabaseNextcloud + "/" + databaseName),
  62. schemaVersion: databaseSchemaVersion,
  63. objectTypes: [tableMetadata.self, tableLocalFile.self, tableDirectory.self, tableTag.self, tableAccount.self, tableCapabilities.self, tablePhotoLibrary.self, tableE2eEncryption.self, tableE2eEncryptionLock.self, tableShare.self, tableChunk.self, tableAvatar.self, tableDashboardWidget.self, tableDashboardWidgetButton.self, NCDBLayoutForView.self]
  64. )
  65. Realm.Configuration.defaultConfiguration = config
  66. } else {
  67. // App config
  68. let configCompact = Realm.Configuration(
  69. fileURL: databaseFileUrlPath,
  70. schemaVersion: databaseSchemaVersion,
  71. migrationBlock: { migration, oldSchemaVersion in
  72. if oldSchemaVersion < 255 {
  73. migration.deleteData(forType: tableActivity.className())
  74. migration.deleteData(forType: tableActivityLatestId.className())
  75. migration.deleteData(forType: tableActivityPreview.className())
  76. migration.deleteData(forType: tableActivitySubjectRich.className())
  77. migration.deleteData(forType: tableDirectory.className())
  78. migration.deleteData(forType: tableMetadata.className())
  79. }
  80. if oldSchemaVersion < 292 {
  81. migration.deleteData(forType: tableVideo.className())
  82. }
  83. }, shouldCompactOnLaunch: { totalBytes, usedBytes in
  84. // totalBytes refers to the size of the file on disk in bytes (data + free space)
  85. // usedBytes refers to the number of bytes used by data in the file
  86. // Compact if the file is over 100MB in size and less than 50% 'used'
  87. let oneHundredMB = 100 * 1024 * 1024
  88. return (totalBytes > oneHundredMB) && (Double(usedBytes) / Double(totalBytes)) < 0.5
  89. }
  90. )
  91. do {
  92. _ = try Realm(configuration: configCompact)
  93. } catch let error {
  94. if let databaseFileUrlPath = databaseFileUrlPath {
  95. do {
  96. #if !EXTENSION
  97. let nkError = NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: error.localizedDescription)
  98. NCContentPresenter.shared.showError(error: nkError, priority: .max)
  99. #endif
  100. NextcloudKit.shared.nkCommonInstance.writeLog("DATABASE ERROR: \(error.localizedDescription)")
  101. try FileManager.default.removeItem(at: databaseFileUrlPath)
  102. } catch {}
  103. }
  104. }
  105. let config = Realm.Configuration(
  106. fileURL: dirGroup?.appendingPathComponent(NCGlobal.shared.appDatabaseNextcloud + "/" + databaseName),
  107. schemaVersion: databaseSchemaVersion
  108. )
  109. Realm.Configuration.defaultConfiguration = config
  110. }
  111. // Verify Database, if corrupt remove it
  112. do {
  113. _ = try Realm()
  114. } catch let error {
  115. if let databaseFileUrlPath = databaseFileUrlPath {
  116. do {
  117. #if !EXTENSION
  118. let nkError = NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: error.localizedDescription)
  119. NCContentPresenter.shared.showError(error: nkError, priority: .max)
  120. #endif
  121. NextcloudKit.shared.nkCommonInstance.writeLog("DATABASE ERROR: \(error.localizedDescription)")
  122. try FileManager.default.removeItem(at: databaseFileUrlPath)
  123. } catch { }
  124. }
  125. }
  126. // Open Real
  127. _ = try! Realm()
  128. }
  129. // MARK: -
  130. // MARK: Utility Database
  131. @objc func clearTable(_ table: Object.Type, account: String? = nil) {
  132. let realm = try! Realm()
  133. do {
  134. try realm.write {
  135. var results: Results<Object>
  136. if let account = account {
  137. results = realm.objects(table).filter("account == %@", account)
  138. } else {
  139. results = realm.objects(table)
  140. }
  141. realm.delete(results)
  142. }
  143. } catch let error {
  144. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  145. }
  146. }
  147. @objc func clearDatabase(account: String?, removeAccount: Bool) {
  148. self.clearTable(tableActivity.self, account: account)
  149. self.clearTable(tableActivityLatestId.self, account: account)
  150. self.clearTable(tableActivityPreview.self, account: account)
  151. self.clearTable(tableActivitySubjectRich.self, account: account)
  152. self.clearTable(tableAvatar.self)
  153. self.clearTable(tableCapabilities.self, account: account)
  154. self.clearTable(tableChunk.self, account: account)
  155. self.clearTable(tableComments.self, account: account)
  156. self.clearTable(tableDashboardWidget.self, account: account)
  157. self.clearTable(tableDashboardWidgetButton.self, account: account)
  158. self.clearTable(tableDirectEditingCreators.self, account: account)
  159. self.clearTable(tableDirectEditingEditors.self, account: account)
  160. self.clearTable(tableDirectory.self, account: account)
  161. self.clearTable(tableE2eEncryption.self, account: account)
  162. self.clearTable(tableE2eEncryptionLock.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 Avatar
  211. // MARK: -
  212. // MARK: Table Capabilities
  213. func addCapabilitiesJSon(_ data: Data, account: String) {
  214. let realm = try! Realm()
  215. do {
  216. try realm.write {
  217. let addObject = tableCapabilities()
  218. addObject.account = account
  219. addObject.jsondata = data
  220. realm.add(addObject, update: .all)
  221. }
  222. } catch let error {
  223. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  224. }
  225. }
  226. func getCapabilities(account: String) -> String? {
  227. let realm = try! Realm()
  228. guard let result = realm.objects(tableCapabilities.self).filter("account == %@", account).first else {
  229. return nil
  230. }
  231. guard let jsondata = result.jsondata else {
  232. return nil
  233. }
  234. let json = JSON(jsondata)
  235. return json.rawString()?.replacingOccurrences(of: "\\/", with: "/")
  236. }
  237. @objc func getCapabilitiesServerString(account: String, elements: [String]) -> String? {
  238. let realm = try! Realm()
  239. guard let result = realm.objects(tableCapabilities.self).filter("account == %@", account).first else {
  240. return nil
  241. }
  242. guard let jsondata = result.jsondata else {
  243. return nil
  244. }
  245. let json = JSON(jsondata)
  246. return json[elements].string
  247. }
  248. func getCapabilitiesServerInt(account: String, elements: [String]) -> Int {
  249. let realm = try! Realm()
  250. guard let result = realm.objects(tableCapabilities.self).filter("account == %@", account).first,
  251. let jsondata = result.jsondata else {
  252. return 0
  253. }
  254. let json = JSON(jsondata)
  255. return json[elements].intValue
  256. }
  257. @objc func getCapabilitiesServerBool(account: String, elements: [String], exists: Bool) -> Bool {
  258. let realm = try! Realm()
  259. guard let result = realm.objects(tableCapabilities.self).filter("account == %@", account).first else {
  260. return false
  261. }
  262. guard let jsondata = result.jsondata else {
  263. return false
  264. }
  265. let json = JSON(jsondata)
  266. if exists {
  267. return json[elements].exists()
  268. } else {
  269. return json[elements].boolValue
  270. }
  271. }
  272. func getCapabilitiesServerArray(account: String, elements: [String]) -> [String]? {
  273. let realm = try! Realm()
  274. var resultArray: [String] = []
  275. guard let result = realm.objects(tableCapabilities.self).filter("account == %@", account).first else {
  276. return nil
  277. }
  278. guard let jsondata = result.jsondata else {
  279. return nil
  280. }
  281. let json = JSON(jsondata)
  282. if let results = json[elements].array {
  283. for result in results {
  284. resultArray.append(result.string ?? "")
  285. }
  286. return resultArray
  287. }
  288. return nil
  289. }
  290. // MARK: -
  291. // MARK: Table Chunk
  292. func getChunkFolder(account: String, ocId: String) -> String {
  293. let realm = try! Realm()
  294. if let result = realm.objects(tableChunk.self).filter("account == %@ AND ocId == %@", account, ocId).first {
  295. return result.chunkFolder
  296. }
  297. return NSUUID().uuidString
  298. }
  299. func getChunks(account: String, ocId: String) -> [String] {
  300. let realm = try! Realm()
  301. var filesNames: [String] = []
  302. let results = realm.objects(tableChunk.self).filter("account == %@ AND ocId == %@", account, ocId).sorted(byKeyPath: "fileName", ascending: true)
  303. for result in results {
  304. filesNames.append(result.fileName)
  305. }
  306. return filesNames
  307. }
  308. func addChunks(account: String, ocId: String, chunkFolder: String, fileNames: [String]) {
  309. let realm = try! Realm()
  310. var size: Int64 = 0
  311. do {
  312. try realm.write {
  313. for fileName in fileNames {
  314. let object = tableChunk()
  315. size += NCUtilityFileSystem.shared.getFileSize(filePath: CCUtility.getDirectoryProviderStorageOcId(ocId, fileNameView: fileName)!)
  316. object.account = account
  317. object.chunkFolder = chunkFolder
  318. object.fileName = fileName
  319. object.index = ocId + fileName
  320. object.ocId = ocId
  321. object.size = size
  322. realm.add(object, update: .all)
  323. }
  324. }
  325. } catch let error {
  326. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  327. }
  328. }
  329. func getChunk(account: String, fileName: String) -> tableChunk? {
  330. let realm = try! Realm()
  331. if let result = realm.objects(tableChunk.self).filter("account == %@ AND fileName == %@", account, fileName).first {
  332. return tableChunk.init(value: result)
  333. } else {
  334. return nil
  335. }
  336. }
  337. func deleteChunk(account: String, ocId: String, fileName: String) {
  338. let realm = try! Realm()
  339. do {
  340. try realm.write {
  341. let result = realm.objects(tableChunk.self).filter(NSPredicate(format: "account == %@ AND ocId == %@ AND fileName == %@", account, ocId, fileName))
  342. realm.delete(result)
  343. }
  344. } catch let error {
  345. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  346. }
  347. }
  348. func deleteChunks(account: String, ocId: String) {
  349. let realm = try! Realm()
  350. do {
  351. try realm.write {
  352. let result = realm.objects(tableChunk.self).filter(NSPredicate(format: "account == %@ AND ocId == %@", account, ocId))
  353. realm.delete(result)
  354. }
  355. } catch let error {
  356. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  357. }
  358. }
  359. // MARK: -
  360. // MARK: Table Direct Editing
  361. func addDirectEditing(account: String, editors: [NKEditorDetailsEditors], creators: [NKEditorDetailsCreators]) {
  362. let realm = try! Realm()
  363. do {
  364. try realm.write {
  365. let resultsCreators = realm.objects(tableDirectEditingCreators.self).filter("account == %@", account)
  366. realm.delete(resultsCreators)
  367. let resultsEditors = realm.objects(tableDirectEditingEditors.self).filter("account == %@", account)
  368. realm.delete(resultsEditors)
  369. for creator in creators {
  370. let addObject = tableDirectEditingCreators()
  371. addObject.account = account
  372. addObject.editor = creator.editor
  373. addObject.ext = creator.ext
  374. addObject.identifier = creator.identifier
  375. addObject.mimetype = creator.mimetype
  376. addObject.name = creator.name
  377. addObject.templates = creator.templates
  378. realm.add(addObject)
  379. }
  380. for editor in editors {
  381. let addObject = tableDirectEditingEditors()
  382. addObject.account = account
  383. for mimeType in editor.mimetypes {
  384. addObject.mimetypes.append(mimeType)
  385. }
  386. addObject.name = editor.name
  387. if editor.name.lowercased() == NCGlobal.shared.editorOnlyoffice {
  388. addObject.editor = NCGlobal.shared.editorOnlyoffice
  389. } else {
  390. addObject.editor = NCGlobal.shared.editorText
  391. }
  392. for mimeType in editor.optionalMimetypes {
  393. addObject.optionalMimetypes.append(mimeType)
  394. }
  395. addObject.secure = editor.secure
  396. realm.add(addObject)
  397. }
  398. }
  399. } catch let error {
  400. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  401. }
  402. }
  403. func getDirectEditingCreators(account: String) -> [tableDirectEditingCreators]? {
  404. let realm = try! Realm()
  405. let results = realm.objects(tableDirectEditingCreators.self).filter("account == %@", account)
  406. if results.count > 0 {
  407. return Array(results.map { tableDirectEditingCreators.init(value: $0) })
  408. } else {
  409. return nil
  410. }
  411. }
  412. func getDirectEditingCreators(predicate: NSPredicate) -> [tableDirectEditingCreators]? {
  413. let realm = try! Realm()
  414. let results = realm.objects(tableDirectEditingCreators.self).filter(predicate)
  415. if results.count > 0 {
  416. return Array(results.map { tableDirectEditingCreators.init(value: $0) })
  417. } else {
  418. return nil
  419. }
  420. }
  421. func getDirectEditingEditors(account: String) -> [tableDirectEditingEditors]? {
  422. let realm = try! Realm()
  423. let results = realm.objects(tableDirectEditingEditors.self).filter("account == %@", account)
  424. if results.count > 0 {
  425. return Array(results.map { tableDirectEditingEditors.init(value: $0) })
  426. } else {
  427. return nil
  428. }
  429. }
  430. // MARK: -
  431. // MARK: Table External Sites
  432. func addExternalSites(_ externalSite: NKExternalSite, account: String) {
  433. let realm = try! Realm()
  434. do {
  435. try realm.write {
  436. let addObject = tableExternalSites()
  437. addObject.account = account
  438. addObject.idExternalSite = externalSite.idExternalSite
  439. addObject.icon = externalSite.icon
  440. addObject.lang = externalSite.lang
  441. addObject.name = externalSite.name
  442. addObject.url = externalSite.url
  443. addObject.type = externalSite.type
  444. realm.add(addObject)
  445. }
  446. } catch let error {
  447. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  448. }
  449. }
  450. func deleteExternalSites(account: String) {
  451. let realm = try! Realm()
  452. do {
  453. try realm.write {
  454. let results = realm.objects(tableExternalSites.self).filter("account == %@", account)
  455. realm.delete(results)
  456. }
  457. } catch let error {
  458. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  459. }
  460. }
  461. func getAllExternalSites(account: String) -> [tableExternalSites]? {
  462. let realm = try! Realm()
  463. let results = realm.objects(tableExternalSites.self).filter("account == %@", account).sorted(byKeyPath: "idExternalSite", ascending: true)
  464. if results.count > 0 {
  465. return Array(results.map { tableExternalSites.init(value: $0) })
  466. } else {
  467. return nil
  468. }
  469. }
  470. // MARK: -
  471. // MARK: Table GPS
  472. @objc func addGeocoderLocation(_ location: String, placemarkAdministrativeArea: String, placemarkCountry: String, placemarkLocality: String, placemarkPostalCode: String, placemarkThoroughfare: String, latitude: String, longitude: String) {
  473. let realm = try! Realm()
  474. realm.beginWrite()
  475. // Verify if exists
  476. guard realm.objects(tableGPS.self).filter("latitude == %@ AND longitude == %@", latitude, longitude).first == nil else {
  477. realm.cancelWrite()
  478. return
  479. }
  480. // Add new GPS
  481. let addObject = tableGPS()
  482. addObject.latitude = latitude
  483. addObject.location = location
  484. addObject.longitude = longitude
  485. addObject.placemarkAdministrativeArea = placemarkAdministrativeArea
  486. addObject.placemarkCountry = placemarkCountry
  487. addObject.placemarkLocality = placemarkLocality
  488. addObject.placemarkPostalCode = placemarkPostalCode
  489. addObject.placemarkThoroughfare = placemarkThoroughfare
  490. realm.add(addObject)
  491. do {
  492. try realm.commitWrite()
  493. } catch let error {
  494. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  495. }
  496. }
  497. @objc func getLocationFromGeoLatitude(_ latitude: String, longitude: String) -> String? {
  498. let realm = try! Realm()
  499. let result = realm.objects(tableGPS.self).filter("latitude == %@ AND longitude == %@", latitude, longitude).first
  500. return result?.location
  501. }
  502. // MARK: -
  503. // MARK: Table LocalFile
  504. func addLocalFile(metadata: tableMetadata) {
  505. let realm = try! Realm()
  506. do {
  507. try realm.write {
  508. let addObject = getTableLocalFile(predicate: NSPredicate(format: "ocId == %@", metadata.ocId)) ?? tableLocalFile()
  509. addObject.account = metadata.account
  510. addObject.etag = metadata.etag
  511. addObject.exifDate = NSDate()
  512. addObject.exifLatitude = "-1"
  513. addObject.exifLongitude = "-1"
  514. addObject.ocId = metadata.ocId
  515. addObject.fileName = metadata.fileName
  516. realm.add(addObject, update: .all)
  517. }
  518. } catch let error {
  519. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  520. }
  521. }
  522. func addLocalFile(account: String, etag: String, ocId: String, fileName: String) {
  523. let realm = try! Realm()
  524. do {
  525. try realm.write {
  526. let addObject = tableLocalFile()
  527. addObject.account = account
  528. addObject.etag = etag
  529. addObject.exifDate = NSDate()
  530. addObject.exifLatitude = "-1"
  531. addObject.exifLongitude = "-1"
  532. addObject.ocId = ocId
  533. addObject.fileName = fileName
  534. realm.add(addObject, update: .all)
  535. }
  536. } catch let error {
  537. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  538. }
  539. }
  540. func deleteLocalFile(predicate: NSPredicate) {
  541. let realm = try! Realm()
  542. do {
  543. try realm.write {
  544. let results = realm.objects(tableLocalFile.self).filter(predicate)
  545. realm.delete(results)
  546. }
  547. } catch let error {
  548. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  549. }
  550. }
  551. func setLocalFile(ocId: String, fileName: String?, etag: String?) {
  552. let realm = try! Realm()
  553. do {
  554. try realm.write {
  555. let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first
  556. if let fileName = fileName {
  557. result?.fileName = fileName
  558. }
  559. if let etag = etag {
  560. result?.etag = etag
  561. }
  562. }
  563. } catch let error {
  564. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  565. }
  566. }
  567. @objc func setLocalFile(ocId: String, exifDate: NSDate?, exifLatitude: String, exifLongitude: String, exifLensModel: String?) {
  568. let realm = try! Realm()
  569. do {
  570. try realm.write {
  571. if let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first {
  572. result.exifDate = exifDate
  573. result.exifLatitude = exifLatitude
  574. result.exifLongitude = exifLongitude
  575. if exifLensModel?.count ?? 0 > 0 {
  576. result.exifLensModel = exifLensModel
  577. }
  578. }
  579. }
  580. } catch let error {
  581. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  582. }
  583. }
  584. func getTableLocalFile(account: String) -> [tableLocalFile] {
  585. let realm = try! Realm()
  586. let results = realm.objects(tableLocalFile.self).filter("account == %@", account)
  587. return Array(results.map { tableLocalFile.init(value: $0) })
  588. }
  589. func getTableLocalFile(predicate: NSPredicate) -> tableLocalFile? {
  590. let realm = try! Realm()
  591. guard let result = realm.objects(tableLocalFile.self).filter(predicate).first else {
  592. return nil
  593. }
  594. return tableLocalFile.init(value: result)
  595. }
  596. func getTableLocalFiles(predicate: NSPredicate, sorted: String, ascending: Bool) -> [tableLocalFile] {
  597. let realm = try! Realm()
  598. let results = realm.objects(tableLocalFile.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  599. return Array(results.map { tableLocalFile.init(value: $0) })
  600. }
  601. func setLocalFile(ocId: String, offline: Bool) {
  602. let realm = try! Realm()
  603. do {
  604. try realm.write {
  605. let result = realm.objects(tableLocalFile.self).filter("ocId == %@", ocId).first
  606. result?.offline = offline
  607. }
  608. } catch let error {
  609. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  610. }
  611. }
  612. // MARK: -
  613. // MARK: Table Photo Library
  614. @discardableResult
  615. func addPhotoLibrary(_ assets: [PHAsset], account: String) -> Bool {
  616. let realm = try! Realm()
  617. do {
  618. try realm.write {
  619. for asset in assets {
  620. var creationDateString = ""
  621. let addObject = tablePhotoLibrary()
  622. addObject.account = account
  623. addObject.assetLocalIdentifier = asset.localIdentifier
  624. addObject.mediaType = asset.mediaType.rawValue
  625. if let creationDate = asset.creationDate {
  626. addObject.creationDate = creationDate as NSDate
  627. creationDateString = String(describing: creationDate)
  628. }
  629. if let modificationDate = asset.modificationDate {
  630. addObject.modificationDate = modificationDate as NSDate
  631. }
  632. addObject.idAsset = account + asset.localIdentifier + creationDateString
  633. realm.add(addObject, update: .all)
  634. }
  635. }
  636. } catch let error {
  637. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  638. return false
  639. }
  640. return true
  641. }
  642. func getPhotoLibraryIdAsset(image: Bool, video: Bool, account: String) -> [String]? {
  643. let realm = try! Realm()
  644. var predicate = NSPredicate()
  645. if image && video {
  646. predicate = NSPredicate(format: "account == %@ AND (mediaType == %d OR mediaType == %d)", account, PHAssetMediaType.image.rawValue, PHAssetMediaType.video.rawValue)
  647. } else if image {
  648. predicate = NSPredicate(format: "account == %@ AND mediaType == %d", account, PHAssetMediaType.image.rawValue)
  649. } else if video {
  650. predicate = NSPredicate(format: "account == %@ AND mediaType == %d", account, PHAssetMediaType.video.rawValue)
  651. }
  652. let results = realm.objects(tablePhotoLibrary.self).filter(predicate)
  653. let idsAsset = results.map { $0.idAsset }
  654. return Array(idsAsset)
  655. }
  656. // MARK: -
  657. // MARK: Table Tag
  658. func addTag(_ ocId: String, tagIOS: Data?, account: String) {
  659. let realm = try! Realm()
  660. do {
  661. try realm.write {
  662. // Add new
  663. let addObject = tableTag()
  664. addObject.account = account
  665. addObject.ocId = ocId
  666. addObject.tagIOS = tagIOS
  667. realm.add(addObject, update: .all)
  668. }
  669. } catch let error {
  670. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  671. }
  672. }
  673. func deleteTag(_ ocId: String) {
  674. let realm = try! Realm()
  675. realm.beginWrite()
  676. let result = realm.objects(tableTag.self).filter("ocId == %@", ocId)
  677. realm.delete(result)
  678. do {
  679. try realm.commitWrite()
  680. } catch let error {
  681. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  682. }
  683. }
  684. func getTags(predicate: NSPredicate) -> [tableTag] {
  685. let realm = try! Realm()
  686. let results = realm.objects(tableTag.self).filter(predicate)
  687. return Array(results.map { tableTag.init(value: $0) })
  688. }
  689. func getTag(predicate: NSPredicate) -> tableTag? {
  690. let realm = try! Realm()
  691. guard let result = realm.objects(tableTag.self).filter(predicate).first else {
  692. return nil
  693. }
  694. return tableTag.init(value: result)
  695. }
  696. // MARK: -
  697. // MARK: Table Tip
  698. func tipExists(_ tipName: String) -> Bool {
  699. let realm = try! Realm()
  700. guard (realm.objects(tableTip.self).where {
  701. $0.tipName == tipName
  702. }.first) == nil else {
  703. return true
  704. }
  705. return false
  706. }
  707. func addTip(_ tipName: String) {
  708. let realm = try! Realm()
  709. do {
  710. try realm.write {
  711. let addObject = tableTip()
  712. addObject.tipName = tipName
  713. realm.add(addObject, update: .all)
  714. }
  715. } catch let error {
  716. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  717. }
  718. }
  719. // MARK: -
  720. // MARK: Table Trash
  721. func addTrash(account: String, items: [NKTrash]) {
  722. let realm = try! Realm()
  723. do {
  724. try realm.write {
  725. for trash in items {
  726. let object = tableTrash()
  727. object.account = account
  728. object.contentType = trash.contentType
  729. object.date = trash.date
  730. object.directory = trash.directory
  731. object.fileId = trash.fileId
  732. object.fileName = trash.fileName
  733. object.filePath = trash.filePath
  734. object.hasPreview = trash.hasPreview
  735. object.iconName = trash.iconName
  736. object.size = trash.size
  737. object.trashbinDeletionTime = trash.trashbinDeletionTime
  738. object.trashbinFileName = trash.trashbinFileName
  739. object.trashbinOriginalLocation = trash.trashbinOriginalLocation
  740. object.classFile = trash.classFile
  741. realm.add(object, update: .all)
  742. }
  743. }
  744. } catch let error {
  745. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  746. }
  747. }
  748. func deleteTrash(filePath: String?, account: String) {
  749. let realm = try! Realm()
  750. var predicate = NSPredicate()
  751. do {
  752. try realm.write {
  753. if filePath == nil {
  754. predicate = NSPredicate(format: "account == %@", account)
  755. } else {
  756. predicate = NSPredicate(format: "account == %@ AND filePath == %@", account, filePath!)
  757. }
  758. let result = realm.objects(tableTrash.self).filter(predicate)
  759. realm.delete(result)
  760. }
  761. } catch let error {
  762. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  763. }
  764. }
  765. func deleteTrash(fileId: String?, account: String) {
  766. let realm = try! Realm()
  767. var predicate = NSPredicate()
  768. do {
  769. try realm.write {
  770. if fileId == nil {
  771. predicate = NSPredicate(format: "account == %@", account)
  772. } else {
  773. predicate = NSPredicate(format: "account == %@ AND fileId == %@", account, fileId!)
  774. }
  775. let result = realm.objects(tableTrash.self).filter(predicate)
  776. realm.delete(result)
  777. }
  778. } catch let error {
  779. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  780. }
  781. }
  782. func getTrash(filePath: String, sort: String?, ascending: Bool?, account: String) -> [tableTrash]? {
  783. let realm = try! Realm()
  784. let sort = sort ?? "date"
  785. let ascending = ascending ?? false
  786. let results = realm.objects(tableTrash.self).filter("account == %@ AND filePath == %@", account, filePath).sorted(byKeyPath: sort, ascending: ascending)
  787. return Array(results.map { tableTrash.init(value: $0) })
  788. }
  789. func getTrashItem(fileId: String, account: String) -> tableTrash? {
  790. let realm = try! Realm()
  791. guard let result = realm.objects(tableTrash.self).filter("account == %@ AND fileId == %@", account, fileId).first else {
  792. return nil
  793. }
  794. return tableTrash.init(value: result)
  795. }
  796. // MARK: -
  797. // MARK: Table UserStatus
  798. func addUserStatus(_ userStatuses: [NKUserStatus], account: String, predefined: Bool) {
  799. let realm = try! Realm()
  800. do {
  801. try realm.write {
  802. let results = realm.objects(tableUserStatus.self).filter("account == %@ AND predefined == %@", account, predefined)
  803. realm.delete(results)
  804. for userStatus in userStatuses {
  805. let object = tableUserStatus()
  806. object.account = account
  807. object.clearAt = userStatus.clearAt
  808. object.clearAtTime = userStatus.clearAtTime
  809. object.clearAtType = userStatus.clearAtType
  810. object.icon = userStatus.icon
  811. object.id = userStatus.id
  812. object.message = userStatus.message
  813. object.predefined = userStatus.predefined
  814. object.status = userStatus.status
  815. object.userId = userStatus.userId
  816. realm.add(object)
  817. }
  818. }
  819. } catch let error {
  820. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  821. }
  822. }
  823. }