NCManageDatabase.swift 36 KB

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