NCManageDatabase.swift 36 KB

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