NCManageDatabase+Metadata.swift 42 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119
  1. //
  2. // NCManageDatabase+Metadata.swift
  3. // Nextcloud
  4. //
  5. // Created by Henrik Storch on 30.11.21.
  6. // Copyright © 2021 Marino Faggiana. All rights reserved.
  7. //
  8. // Author Marino Faggiana <marino.faggiana@nextcloud.com>
  9. //
  10. // This program is free software: you can redistribute it and/or modify
  11. // it under the terms of the GNU General Public License as published by
  12. // the Free Software Foundation, either version 3 of the License, or
  13. // (at your option) any later version.
  14. //
  15. // This program is distributed in the hope that it will be useful,
  16. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  18. // GNU General Public License for more details.
  19. //
  20. // You should have received a copy of the GNU General Public License
  21. // along with this program. If not, see <http://www.gnu.org/licenses/>.
  22. //
  23. import Foundation
  24. import RealmSwift
  25. import NextcloudKit
  26. class tableMetadata: Object, NCUserBaseUrl {
  27. override func isEqual(_ object: Any?) -> Bool {
  28. if let object = object as? tableMetadata {
  29. return self.fileId == object.fileId && self.account == object.account
  30. && self.path == object.path && self.fileName == object.fileName
  31. } else {
  32. return false
  33. }
  34. }
  35. @objc dynamic var account = ""
  36. @objc dynamic var assetLocalIdentifier = ""
  37. @objc dynamic var checksums = ""
  38. @objc dynamic var chunk: Int = 0
  39. @objc dynamic var classFile = ""
  40. @objc dynamic var commentsUnread: Bool = false
  41. @objc dynamic var contentType = ""
  42. @objc dynamic var creationDate = NSDate()
  43. @objc dynamic var dataFingerprint = ""
  44. @objc dynamic var date = NSDate()
  45. @objc dynamic var directory: Bool = false
  46. @objc dynamic var deleteAssetLocalIdentifier: Bool = false
  47. @objc dynamic var downloadURL = ""
  48. @objc dynamic var e2eEncrypted: Bool = false
  49. @objc dynamic var edited: Bool = false
  50. @objc dynamic var etag = ""
  51. @objc dynamic var etagResource = ""
  52. @objc dynamic var favorite: Bool = false
  53. @objc dynamic var fileId = ""
  54. @objc dynamic var fileName = ""
  55. @objc dynamic var fileNameView = ""
  56. @objc dynamic var hasPreview: Bool = false
  57. @objc dynamic var hidden: Bool = false
  58. @objc dynamic var iconName = ""
  59. @objc dynamic var iconUrl = ""
  60. @objc dynamic var isFlaggedAsLivePhotoByServer: Bool = false
  61. @objc dynamic var isExtractFile: Bool = false
  62. @objc dynamic var livePhotoFile = ""
  63. @objc dynamic var mountType = ""
  64. @objc dynamic var name = "" // for unifiedSearch is the provider.id
  65. @objc dynamic var note = ""
  66. @objc dynamic var ocId = ""
  67. @objc dynamic var ownerId = ""
  68. @objc dynamic var ownerDisplayName = ""
  69. @objc public var lock = false
  70. @objc public var lockOwner = ""
  71. @objc public var lockOwnerEditor = ""
  72. @objc public var lockOwnerType = 0
  73. @objc public var lockOwnerDisplayName = ""
  74. @objc public var lockTime: Date?
  75. @objc public var lockTimeOut: Date?
  76. @objc dynamic var path = ""
  77. @objc dynamic var permissions = ""
  78. @objc dynamic var quotaUsedBytes: Int64 = 0
  79. @objc dynamic var quotaAvailableBytes: Int64 = 0
  80. @objc dynamic var resourceType = ""
  81. @objc dynamic var richWorkspace: String?
  82. @objc dynamic var serverUrl = ""
  83. @objc dynamic var session = ""
  84. @objc dynamic var sessionError = ""
  85. @objc dynamic var sessionSelector = ""
  86. @objc dynamic var sessionTaskIdentifier: Int = 0
  87. @objc dynamic var sharePermissionsCollaborationServices: Int = 0
  88. let sharePermissionsCloudMesh = List<String>()
  89. let shareType = List<Int>()
  90. @objc dynamic var size: Int64 = 0
  91. @objc dynamic var status: Int = 0
  92. @objc dynamic var subline: String?
  93. let tags = List<String>()
  94. @objc dynamic var trashbinFileName = ""
  95. @objc dynamic var trashbinOriginalLocation = ""
  96. @objc dynamic var trashbinDeletionTime = NSDate()
  97. @objc dynamic var uploadDate = NSDate()
  98. @objc dynamic var url = ""
  99. @objc dynamic var urlBase = ""
  100. @objc dynamic var user = ""
  101. @objc dynamic var userId = ""
  102. @objc dynamic var latitude: Double = 0
  103. @objc dynamic var longitude: Double = 0
  104. @objc dynamic var height: Int = 0
  105. @objc dynamic var width: Int = 0
  106. @objc dynamic var errorCode: Int = 0
  107. @objc dynamic var errorCodeCounter: Int = 0
  108. @objc dynamic var errorCodeDate: Date?
  109. override static func primaryKey() -> String {
  110. return "ocId"
  111. }
  112. }
  113. extension tableMetadata {
  114. var fileExtension: String {
  115. (fileNameView as NSString).pathExtension
  116. }
  117. var fileNoExtension: String {
  118. (fileNameView as NSString).deletingPathExtension
  119. }
  120. var isRenameable: Bool {
  121. if lock {
  122. return false
  123. }
  124. if !isDirectoryE2EE && e2eEncrypted {
  125. return false
  126. }
  127. return true
  128. }
  129. var isPrintable: Bool {
  130. if isDocumentViewableOnly {
  131. return false
  132. }
  133. if ["application/pdf", "com.adobe.pdf"].contains(contentType) || contentType.hasPrefix("text/") || classFile == NKCommon.TypeClassFile.image.rawValue {
  134. return true
  135. }
  136. return false
  137. }
  138. var isSavebleInCameraRoll: Bool {
  139. return (classFile == NKCommon.TypeClassFile.image.rawValue && contentType != "image/svg+xml") || classFile == NKCommon.TypeClassFile.video.rawValue
  140. }
  141. var isDocumentViewableOnly: Bool {
  142. sharePermissionsCollaborationServices == NCGlobal.shared.permissionReadShare && classFile == NKCommon.TypeClassFile.document.rawValue
  143. }
  144. var isAudioOrVideo: Bool {
  145. return classFile == NKCommon.TypeClassFile.audio.rawValue || classFile == NKCommon.TypeClassFile.video.rawValue
  146. }
  147. var isVideo: Bool {
  148. return classFile == NKCommon.TypeClassFile.video.rawValue
  149. }
  150. var isAudio: Bool {
  151. return classFile == NKCommon.TypeClassFile.audio.rawValue
  152. }
  153. var isImage: Bool {
  154. return classFile == NKCommon.TypeClassFile.image.rawValue
  155. }
  156. var isSavebleAsImage: Bool {
  157. classFile == NKCommon.TypeClassFile.image.rawValue && contentType != "image/svg+xml"
  158. }
  159. var isCopyableInPasteboard: Bool {
  160. !isDocumentViewableOnly && !directory
  161. }
  162. var isCopyableMovable: Bool {
  163. !isDocumentViewableOnly && !isDirectoryE2EE && !e2eEncrypted
  164. }
  165. var isModifiableWithQuickLook: Bool {
  166. if directory || isDocumentViewableOnly || isDirectoryE2EE {
  167. return false
  168. }
  169. return contentType == "com.adobe.pdf" || contentType == "application/pdf" || classFile == NKCommon.TypeClassFile.image.rawValue
  170. }
  171. var isDeletable: Bool {
  172. if !isDirectoryE2EE && e2eEncrypted {
  173. return false
  174. }
  175. return true
  176. }
  177. var isSettableOnOffline: Bool {
  178. return session.isEmpty && !isDocumentViewableOnly
  179. }
  180. var canOpenIn: Bool {
  181. return session.isEmpty && !isDocumentViewableOnly && !directory && !NCBrandOptions.shared.disable_openin_file
  182. }
  183. var isDirectoySettableE2EE: Bool {
  184. return directory && size == 0 && !e2eEncrypted && NCKeychain().isEndToEndEnabled(account: account)
  185. }
  186. var isDirectoryUnsettableE2EE: Bool {
  187. return !isDirectoryE2EE && directory && size == 0 && e2eEncrypted && NCKeychain().isEndToEndEnabled(account: account)
  188. }
  189. var canOpenExternalEditor: Bool {
  190. if isDocumentViewableOnly {
  191. return false
  192. }
  193. let utility = NCUtility()
  194. let editors = utility.isDirectEditing(account: account, contentType: contentType)
  195. let isRichDocument = utility.isRichDocument(self)
  196. return classFile == NKCommon.TypeClassFile.document.rawValue && editors.contains(NCGlobal.shared.editorText) && ((editors.contains(NCGlobal.shared.editorOnlyoffice) || isRichDocument))
  197. }
  198. var isWaitingTransfer: Bool {
  199. status == NCGlobal.shared.metadataStatusWaitDownload || status == NCGlobal.shared.metadataStatusWaitUpload || status == NCGlobal.shared.metadataStatusUploadError
  200. }
  201. var isInTransfer: Bool {
  202. status == NCGlobal.shared.metadataStatusInDownload || status == NCGlobal.shared.metadataStatusDownloading || status == NCGlobal.shared.metadataStatusInUpload || status == NCGlobal.shared.metadataStatusUploading
  203. }
  204. var isTransferInForeground: Bool {
  205. (status > 0 && (chunk > 0 || e2eEncrypted))
  206. }
  207. var isDownload: Bool {
  208. status == NCGlobal.shared.metadataStatusInDownload || status == NCGlobal.shared.metadataStatusDownloading
  209. }
  210. var isUpload: Bool {
  211. status == NCGlobal.shared.metadataStatusInUpload || status == NCGlobal.shared.metadataStatusUploading
  212. }
  213. @objc var isDirectoryE2EE: Bool {
  214. NCUtilityFileSystem().isDirectoryE2EE(account: account, urlBase: urlBase, userId: userId, serverUrl: serverUrl)
  215. }
  216. var isDirectoryE2EETop: Bool {
  217. NCUtilityFileSystem().isDirectoryE2EETop(account: account, serverUrl: serverUrl)
  218. }
  219. var isLivePhoto: Bool {
  220. !livePhotoFile.isEmpty
  221. }
  222. var isNotFlaggedAsLivePhotoByServer: Bool {
  223. !isFlaggedAsLivePhotoByServer
  224. }
  225. /// Returns false if the user is lokced out of the file. I.e. The file is locked but by somone else
  226. func canUnlock(as user: String) -> Bool {
  227. return !lock || (lockOwner == user && lockOwnerType == 0)
  228. }
  229. // Return if is sharable
  230. func isSharable() -> Bool {
  231. guard NCGlobal.shared.capabilityFileSharingApiEnabled else { return false }
  232. if isDirectoryE2EE || e2eEncrypted {
  233. guard directory, NCGlobal.shared.capabilityE2EEEnabled else { return false }
  234. return true
  235. }
  236. return true
  237. }
  238. }
  239. extension NCManageDatabase {
  240. func convertFileToMetadata(_ file: NKFile, isDirectoryE2EE: Bool) -> tableMetadata {
  241. let metadata = tableMetadata()
  242. metadata.account = file.account
  243. metadata.checksums = file.checksums
  244. metadata.commentsUnread = file.commentsUnread
  245. metadata.contentType = file.contentType
  246. if let date = file.creationDate {
  247. metadata.creationDate = date
  248. } else {
  249. metadata.creationDate = file.date
  250. }
  251. metadata.dataFingerprint = file.dataFingerprint
  252. metadata.date = file.date
  253. metadata.directory = file.directory
  254. metadata.downloadURL = file.downloadURL
  255. metadata.e2eEncrypted = file.e2eEncrypted
  256. metadata.etag = file.etag
  257. metadata.favorite = file.favorite
  258. metadata.fileId = file.fileId
  259. metadata.fileName = file.fileName
  260. metadata.fileNameView = file.fileName
  261. metadata.hasPreview = file.hasPreview
  262. metadata.hidden = file.hidden
  263. metadata.iconName = file.iconName
  264. metadata.mountType = file.mountType
  265. metadata.name = file.name
  266. metadata.note = file.note
  267. metadata.ocId = file.ocId
  268. metadata.ownerId = file.ownerId
  269. metadata.ownerDisplayName = file.ownerDisplayName
  270. metadata.lock = file.lock
  271. metadata.lockOwner = file.lockOwner
  272. metadata.lockOwnerEditor = file.lockOwnerEditor
  273. metadata.lockOwnerType = file.lockOwnerType
  274. metadata.lockOwnerDisplayName = file.lockOwnerDisplayName
  275. metadata.lockTime = file.lockTime
  276. metadata.lockTimeOut = file.lockTimeOut
  277. metadata.path = file.path
  278. metadata.permissions = file.permissions
  279. metadata.quotaUsedBytes = file.quotaUsedBytes
  280. metadata.quotaAvailableBytes = file.quotaAvailableBytes
  281. metadata.richWorkspace = file.richWorkspace
  282. metadata.resourceType = file.resourceType
  283. metadata.serverUrl = file.serverUrl
  284. metadata.sharePermissionsCollaborationServices = file.sharePermissionsCollaborationServices
  285. for element in file.sharePermissionsCloudMesh {
  286. metadata.sharePermissionsCloudMesh.append(element)
  287. }
  288. for element in file.shareType {
  289. metadata.shareType.append(element)
  290. }
  291. for element in file.tags {
  292. metadata.tags.append(element)
  293. }
  294. metadata.size = file.size
  295. metadata.classFile = file.classFile
  296. // iOS 12.0,* don't detect UTI text/markdown, text/x-markdown
  297. if (metadata.contentType == "text/markdown" || metadata.contentType == "text/x-markdown") && metadata.classFile == NKCommon.TypeClassFile.unknow.rawValue {
  298. metadata.classFile = NKCommon.TypeClassFile.document.rawValue
  299. }
  300. if let date = file.uploadDate {
  301. metadata.uploadDate = date
  302. } else {
  303. metadata.uploadDate = file.date
  304. }
  305. metadata.urlBase = file.urlBase
  306. metadata.user = file.user
  307. metadata.userId = file.userId
  308. metadata.latitude = file.latitude
  309. metadata.longitude = file.longitude
  310. metadata.height = file.height
  311. metadata.width = file.width
  312. metadata.livePhotoFile = file.livePhotoFile
  313. metadata.isFlaggedAsLivePhotoByServer = file.isFlaggedAsLivePhotoByServer
  314. // E2EE find the fileName for fileNameView
  315. if isDirectoryE2EE || file.e2eEncrypted {
  316. if let tableE2eEncryption = NCManageDatabase.shared.getE2eEncryption(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileNameIdentifier == %@", file.account, file.serverUrl, file.fileName)) {
  317. metadata.fileNameView = tableE2eEncryption.fileName
  318. let results = NextcloudKit.shared.nkCommonInstance.getInternalType(fileName: metadata.fileNameView, mimeType: file.contentType, directory: file.directory)
  319. metadata.contentType = results.mimeType
  320. metadata.iconName = results.iconName
  321. metadata.classFile = results.classFile
  322. }
  323. }
  324. return metadata
  325. }
  326. func convertFilesToMetadatas(_ files: [NKFile], useMetadataFolder: Bool, completion: @escaping (_ metadataFolder: tableMetadata, _ metadatasFolder: [tableMetadata], _ metadatas: [tableMetadata]) -> Void) {
  327. var counter: Int = 0
  328. var isDirectoryE2EE: Bool = false
  329. let listServerUrl = ThreadSafeDictionary<String, Bool>()
  330. var metadataFolder = tableMetadata()
  331. var metadataFolders: [tableMetadata] = []
  332. var metadatas: [tableMetadata] = []
  333. for file in files {
  334. if let key = listServerUrl[file.serverUrl] {
  335. isDirectoryE2EE = key
  336. } else {
  337. isDirectoryE2EE = NCUtilityFileSystem().isDirectoryE2EE(file: file)
  338. listServerUrl[file.serverUrl] = isDirectoryE2EE
  339. }
  340. let metadata = convertFileToMetadata(file, isDirectoryE2EE: isDirectoryE2EE)
  341. if counter == 0 && useMetadataFolder {
  342. metadataFolder = tableMetadata.init(value: metadata)
  343. } else {
  344. metadatas.append(metadata)
  345. if metadata.directory {
  346. metadataFolders.append(metadata)
  347. }
  348. }
  349. counter += 1
  350. }
  351. completion(metadataFolder, metadataFolders, metadatas)
  352. }
  353. func createMetadata(account: String, user: String, userId: String, fileName: String, fileNameView: String, ocId: String, serverUrl: String, urlBase: String, url: String, contentType: String, isUrl: Bool = false, name: String = NCGlobal.shared.appName, subline: String? = nil, iconName: String? = nil, iconUrl: String? = nil) -> tableMetadata {
  354. let metadata = tableMetadata()
  355. if isUrl {
  356. metadata.contentType = "text/uri-list"
  357. if let iconName = iconName {
  358. metadata.iconName = iconName
  359. } else {
  360. metadata.iconName = NKCommon.TypeClassFile.url.rawValue
  361. }
  362. metadata.classFile = NKCommon.TypeClassFile.url.rawValue
  363. } else {
  364. let (mimeType, classFile, iconName, _, _, _) = NextcloudKit.shared.nkCommonInstance.getInternalType(fileName: fileName, mimeType: contentType, directory: false)
  365. metadata.contentType = mimeType
  366. metadata.iconName = iconName
  367. metadata.classFile = classFile
  368. // iOS 12.0,* don't detect UTI text/markdown, text/x-markdown
  369. if classFile == NKCommon.TypeClassFile.unknow.rawValue && (mimeType == "text/x-markdown" || mimeType == "text/markdown") {
  370. metadata.iconName = NKCommon.TypeIconFile.txt.rawValue
  371. metadata.classFile = NKCommon.TypeClassFile.document.rawValue
  372. }
  373. }
  374. if let iconUrl = iconUrl {
  375. metadata.iconUrl = iconUrl
  376. }
  377. let fileName = fileName.trimmingCharacters(in: .whitespacesAndNewlines)
  378. metadata.account = account
  379. metadata.creationDate = Date() as NSDate
  380. metadata.date = Date() as NSDate
  381. metadata.hasPreview = true
  382. metadata.etag = ocId
  383. metadata.fileName = fileName
  384. metadata.fileNameView = fileName
  385. metadata.name = name
  386. metadata.ocId = ocId
  387. metadata.permissions = "RGDNVW"
  388. metadata.serverUrl = serverUrl
  389. metadata.subline = subline
  390. metadata.uploadDate = Date() as NSDate
  391. metadata.url = url
  392. metadata.urlBase = urlBase
  393. metadata.user = user
  394. metadata.userId = userId
  395. if !metadata.urlBase.isEmpty, metadata.serverUrl.hasPrefix(metadata.urlBase) {
  396. metadata.path = String(metadata.serverUrl.dropFirst(metadata.urlBase.count)) + "/"
  397. }
  398. return metadata
  399. }
  400. @discardableResult
  401. func addMetadata(_ metadata: tableMetadata) -> tableMetadata? {
  402. let result = tableMetadata.init(value: metadata)
  403. do {
  404. let realm = try Realm()
  405. try realm.write {
  406. realm.add(result, update: .all)
  407. }
  408. } catch let error {
  409. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  410. return nil
  411. }
  412. return tableMetadata.init(value: result)
  413. }
  414. func addMetadatas(_ metadatas: [tableMetadata]) {
  415. do {
  416. let realm = try Realm()
  417. try realm.write {
  418. realm.add(metadatas, update: .all)
  419. }
  420. } catch let error {
  421. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  422. }
  423. }
  424. func deleteMetadata(predicate: NSPredicate) {
  425. do {
  426. let realm = try Realm()
  427. try realm.write {
  428. let results = realm.objects(tableMetadata.self).filter(predicate)
  429. realm.delete(results)
  430. }
  431. } catch let error {
  432. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  433. }
  434. }
  435. func moveMetadata(ocId: String, serverUrlTo: String) {
  436. do {
  437. let realm = try Realm()
  438. try realm.write {
  439. if let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first {
  440. result.serverUrl = serverUrlTo
  441. }
  442. }
  443. } catch let error {
  444. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  445. }
  446. }
  447. func renameMetadata(fileNameTo: String, ocId: String) {
  448. do {
  449. let realm = try Realm()
  450. try realm.write {
  451. if let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first {
  452. let resultsType = NextcloudKit.shared.nkCommonInstance.getInternalType(fileName: fileNameTo, mimeType: "", directory: result.directory)
  453. result.fileName = fileNameTo
  454. result.fileNameView = fileNameTo
  455. result.iconName = resultsType.iconName
  456. result.contentType = resultsType.mimeType
  457. result.classFile = resultsType.classFile
  458. }
  459. }
  460. } catch let error {
  461. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  462. }
  463. }
  464. func setMetadataSession(ocId: String, newFileName: String? = nil, session: String?, sessionError: String?, sessionSelector: String?, sessionTaskIdentifier: Int?, status: Int?, etag: String? = nil, errorCode: Int?) {
  465. do {
  466. let realm = try Realm()
  467. try realm.write {
  468. if let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first {
  469. if let newFileName = newFileName {
  470. result.fileName = newFileName
  471. result.fileNameView = newFileName
  472. }
  473. if let session {
  474. result.session = session
  475. }
  476. if let sessionError {
  477. result.sessionError = sessionError
  478. }
  479. if let sessionSelector {
  480. result.sessionSelector = sessionSelector
  481. }
  482. if let sessionTaskIdentifier {
  483. result.sessionTaskIdentifier = sessionTaskIdentifier
  484. }
  485. if let status {
  486. result.status = status
  487. }
  488. if let etag {
  489. result.etag = etag
  490. }
  491. if let errorCode {
  492. result.errorCode = errorCode
  493. if errorCode == 0 {
  494. result.errorCodeCounter = 0
  495. } else {
  496. result.errorCodeCounter += 1
  497. result.errorCodeDate = Date()
  498. }
  499. }
  500. }
  501. }
  502. } catch let error {
  503. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  504. }
  505. }
  506. @discardableResult
  507. func setMetadataStatus(ocId: String, status: Int) -> tableMetadata? {
  508. var result: tableMetadata?
  509. do {
  510. let realm = try Realm()
  511. try realm.write {
  512. result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  513. result?.status = status
  514. }
  515. } catch let error {
  516. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  517. }
  518. if let result = result {
  519. return tableMetadata.init(value: result)
  520. } else {
  521. return nil
  522. }
  523. }
  524. func setMetadataEtagResource(ocId: String, etagResource: String?) {
  525. guard let etagResource = etagResource else { return }
  526. do {
  527. let realm = try Realm()
  528. try realm.write {
  529. let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  530. result?.etagResource = etagResource
  531. }
  532. } catch let error {
  533. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  534. }
  535. }
  536. func setMetadataFavorite(ocId: String, favorite: Bool) {
  537. do {
  538. let realm = try Realm()
  539. try realm.write {
  540. let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  541. result?.favorite = favorite
  542. }
  543. } catch let error {
  544. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  545. }
  546. }
  547. func setMetadataLivePhotoByServer(account: String, ocId: String, livePhotoFile: String) {
  548. do {
  549. let realm = try Realm()
  550. try realm.write {
  551. let result = realm.objects(tableMetadata.self).filter("account == %@ AND ocId == %@", account, ocId).first
  552. result?.isFlaggedAsLivePhotoByServer = true
  553. result?.livePhotoFile = livePhotoFile
  554. }
  555. } catch let error {
  556. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  557. }
  558. }
  559. func updateMetadatasFavorite(account: String, metadatas: [tableMetadata]) {
  560. do {
  561. let realm = try Realm()
  562. try realm.write {
  563. let results = realm.objects(tableMetadata.self).filter("account == %@ AND favorite == true", account)
  564. for result in results {
  565. result.favorite = false
  566. }
  567. for metadata in metadatas {
  568. realm.add(metadata, update: .all)
  569. }
  570. }
  571. } catch let error {
  572. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  573. }
  574. }
  575. func setMetadataEncrypted(ocId: String, encrypted: Bool) {
  576. do {
  577. let realm = try Realm()
  578. try realm.write {
  579. let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  580. result?.e2eEncrypted = encrypted
  581. }
  582. } catch let error {
  583. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  584. }
  585. }
  586. func setMetadataFileNameView(serverUrl: String, fileName: String, newFileNameView: String, account: String) {
  587. do {
  588. let realm = try Realm()
  589. try realm.write {
  590. let result = realm.objects(tableMetadata.self).filter("account == %@ AND serverUrl == %@ AND fileName == %@", account, serverUrl, fileName).first
  591. result?.fileNameView = newFileNameView
  592. }
  593. } catch let error {
  594. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  595. }
  596. }
  597. func getMetadata(predicate: NSPredicate) -> tableMetadata? {
  598. do {
  599. let realm = try Realm()
  600. realm.refresh()
  601. guard let result = realm.objects(tableMetadata.self).filter(predicate).first else { return nil }
  602. return tableMetadata.init(value: result)
  603. } catch let error as NSError {
  604. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  605. }
  606. return nil
  607. }
  608. func getMetadata(predicate: NSPredicate, sorted: String, ascending: Bool) -> tableMetadata? {
  609. do {
  610. let realm = try Realm()
  611. realm.refresh()
  612. guard let result = realm.objects(tableMetadata.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending).first else { return nil }
  613. return tableMetadata.init(value: result)
  614. } catch let error as NSError {
  615. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  616. }
  617. return nil
  618. }
  619. func getMetadatas(predicate: NSPredicate) -> [tableMetadata] {
  620. do {
  621. let realm = try Realm()
  622. realm.refresh()
  623. let results = realm.objects(tableMetadata.self).filter(predicate)
  624. return Array(results.map { tableMetadata.init(value: $0) })
  625. } catch let error as NSError {
  626. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  627. }
  628. return []
  629. }
  630. func getResultsMetadatas(predicate: NSPredicate, sorted: String? = nil, ascending: Bool = false) -> Results<tableMetadata>? {
  631. do {
  632. let realm = try Realm()
  633. if let sorted {
  634. return realm.objects(tableMetadata.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  635. } else {
  636. return realm.objects(tableMetadata.self).filter(predicate)
  637. }
  638. } catch let error as NSError {
  639. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  640. }
  641. return nil
  642. }
  643. func getResultMetadata(predicate: NSPredicate) -> tableMetadata? {
  644. do {
  645. let realm = try Realm()
  646. return realm.objects(tableMetadata.self).filter(predicate).first
  647. } catch let error as NSError {
  648. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  649. }
  650. return nil
  651. }
  652. func getAdvancedMetadatas(predicate: NSPredicate, page: Int = 0, limit: Int = 0, sorted: String, ascending: Bool) -> [tableMetadata] {
  653. var metadatas: [tableMetadata] = []
  654. do {
  655. let realm = try Realm()
  656. realm.refresh()
  657. let results = realm.objects(tableMetadata.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  658. if !results.isEmpty {
  659. if page == 0 || limit == 0 {
  660. return Array(results.map { tableMetadata.init(value: $0) })
  661. } else {
  662. let nFrom = (page - 1) * limit
  663. let nTo = nFrom + (limit - 1)
  664. for n in nFrom...nTo {
  665. if n == results.count {
  666. break
  667. }
  668. metadatas.append(tableMetadata.init(value: results[n]))
  669. }
  670. }
  671. }
  672. } catch let error as NSError {
  673. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  674. }
  675. return metadatas
  676. }
  677. func getMetadataAtIndex(predicate: NSPredicate, sorted: String, ascending: Bool, index: Int) -> tableMetadata? {
  678. do {
  679. let realm = try Realm()
  680. realm.refresh()
  681. let results = realm.objects(tableMetadata.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  682. if results.isEmpty {
  683. return nil
  684. } else {
  685. return tableMetadata.init(value: results[index])
  686. }
  687. } catch let error as NSError {
  688. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  689. }
  690. return nil
  691. }
  692. func getMetadataFromOcId(_ ocId: String?) -> tableMetadata? {
  693. guard let ocId else { return nil }
  694. do {
  695. let realm = try Realm()
  696. realm.refresh()
  697. guard let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first else { return nil }
  698. return tableMetadata.init(value: result)
  699. } catch let error as NSError {
  700. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  701. }
  702. return nil
  703. }
  704. func getTableMetadataFromOcId(_ ocId: String?) -> tableMetadata? {
  705. guard let ocId else { return nil }
  706. do {
  707. let realm = try Realm()
  708. realm.refresh()
  709. return realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  710. } catch let error as NSError {
  711. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  712. }
  713. return nil
  714. }
  715. func getMetadataFromFileId(_ fileId: String?) -> tableMetadata? {
  716. do {
  717. let realm = try Realm()
  718. realm.refresh()
  719. guard let fileId = fileId else { return nil }
  720. guard let result = realm.objects(tableMetadata.self).filter("fileId == %@", fileId).first else { return nil }
  721. return tableMetadata.init(value: result)
  722. } catch let error as NSError {
  723. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  724. }
  725. return nil
  726. }
  727. func getMetadataFolder(account: String, urlBase: String, userId: String, serverUrl: String) -> tableMetadata? {
  728. var serverUrl = serverUrl
  729. var fileName = ""
  730. let serverUrlHome = utilityFileSystem.getHomeServer(urlBase: urlBase, userId: userId)
  731. if serverUrlHome == serverUrl {
  732. fileName = "."
  733. serverUrl = ".."
  734. } else {
  735. fileName = (serverUrl as NSString).lastPathComponent
  736. if let path = utilityFileSystem.deleteLastPath(serverUrlPath: serverUrl) {
  737. serverUrl = path
  738. }
  739. }
  740. do {
  741. let realm = try Realm()
  742. realm.refresh()
  743. guard let result = realm.objects(tableMetadata.self).filter("account == %@ AND serverUrl == %@ AND fileName == %@", account, serverUrl, fileName).first else { return nil }
  744. return tableMetadata.init(value: result)
  745. } catch let error as NSError {
  746. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  747. }
  748. return nil
  749. }
  750. func getTableMetadatasDirectoryFavoriteIdentifierRank(account: String) -> [String: NSNumber] {
  751. var listIdentifierRank: [String: NSNumber] = [:]
  752. var counter = 10 as Int64
  753. do {
  754. let realm = try Realm()
  755. realm.refresh()
  756. let results = realm.objects(tableMetadata.self).filter("account == %@ AND directory == true AND favorite == true", account).sorted(byKeyPath: "fileNameView", ascending: true)
  757. for result in results {
  758. counter += 1
  759. listIdentifierRank[result.ocId] = NSNumber(value: Int64(counter))
  760. }
  761. } catch let error as NSError {
  762. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  763. }
  764. return listIdentifierRank
  765. }
  766. @objc func clearMetadatasUpload(account: String) {
  767. do {
  768. let realm = try Realm()
  769. try realm.write {
  770. let results = realm.objects(tableMetadata.self).filter("account == %@ AND (status == %d OR status == %@)", account, NCGlobal.shared.metadataStatusWaitUpload, NCGlobal.shared.metadataStatusUploadError)
  771. realm.delete(results)
  772. }
  773. } catch let error {
  774. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  775. }
  776. }
  777. func readMarkerMetadata(account: String, fileId: String) {
  778. do {
  779. let realm = try Realm()
  780. try realm.write {
  781. let results = realm.objects(tableMetadata.self).filter("account == %@ AND fileId == %@", account, fileId)
  782. for result in results {
  783. result.commentsUnread = false
  784. }
  785. }
  786. } catch let error {
  787. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  788. }
  789. }
  790. func getAssetLocalIdentifiersUploaded(account: String) -> [String] {
  791. var assetLocalIdentifiers: [String] = []
  792. do {
  793. let realm = try Realm()
  794. let results = realm.objects(tableMetadata.self).filter("account == %@ AND assetLocalIdentifier != '' AND deleteAssetLocalIdentifier == true", account)
  795. for result in results {
  796. assetLocalIdentifiers.append(result.assetLocalIdentifier)
  797. }
  798. } catch let error as NSError {
  799. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  800. }
  801. return assetLocalIdentifiers
  802. }
  803. func clearAssetLocalIdentifiers(_ assetLocalIdentifiers: [String], account: String) {
  804. do {
  805. let realm = try Realm()
  806. try realm.write {
  807. let results = realm.objects(tableMetadata.self).filter("account == %@ AND assetLocalIdentifier IN %@", account, assetLocalIdentifiers)
  808. for result in results {
  809. result.assetLocalIdentifier = ""
  810. result.deleteAssetLocalIdentifier = false
  811. }
  812. }
  813. } catch let error {
  814. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  815. }
  816. }
  817. func getMetadataLivePhoto(metadata: tableMetadata) -> tableMetadata? {
  818. guard metadata.isLivePhoto else { return nil }
  819. do {
  820. let realm = try Realm()
  821. guard let result = realm.objects(tableMetadata.self).filter(NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileId == %@", metadata.account, metadata.serverUrl, metadata.livePhotoFile)).first else { return nil }
  822. return tableMetadata.init(value: result)
  823. } catch let error as NSError {
  824. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  825. }
  826. return nil
  827. }
  828. func isMetadataShareOrMounted(metadata: tableMetadata, metadataFolder: tableMetadata?) -> Bool {
  829. var isShare = false
  830. var isMounted = false
  831. if metadataFolder != nil {
  832. isShare = metadata.permissions.contains(NCGlobal.shared.permissionShared) && !metadataFolder!.permissions.contains(NCGlobal.shared.permissionShared)
  833. isMounted = metadata.permissions.contains(NCGlobal.shared.permissionMounted) && !metadataFolder!.permissions.contains(NCGlobal.shared.permissionMounted)
  834. } else if let directory = NCManageDatabase.shared.getTableDirectory(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", metadata.account, metadata.serverUrl)) {
  835. isShare = metadata.permissions.contains(NCGlobal.shared.permissionShared) && !directory.permissions.contains(NCGlobal.shared.permissionShared)
  836. isMounted = metadata.permissions.contains(NCGlobal.shared.permissionMounted) && !directory.permissions.contains(NCGlobal.shared.permissionMounted)
  837. }
  838. if isShare || isMounted {
  839. return true
  840. } else {
  841. return false
  842. }
  843. }
  844. func getMetadataConflict(account: String, serverUrl: String, fileNameView: String) -> tableMetadata? {
  845. // verify exists conflict
  846. let fileNameExtension = (fileNameView as NSString).pathExtension.lowercased()
  847. let fileNameNoExtension = (fileNameView as NSString).deletingPathExtension
  848. var fileNameConflict = fileNameView
  849. if fileNameExtension == "heic", NCKeychain().formatCompatibility {
  850. fileNameConflict = fileNameNoExtension + ".jpg"
  851. }
  852. return getMetadata(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileNameView == %@", account, serverUrl, fileNameConflict))
  853. }
  854. func getNumMetadatasInUpload() -> Int {
  855. do {
  856. let realm = try Realm()
  857. realm.refresh()
  858. return realm.objects(tableMetadata.self).filter(NSPredicate(format: "status == %i || status == %i", NCGlobal.shared.metadataStatusInUpload, NCGlobal.shared.metadataStatusUploading)).count
  859. } catch let error as NSError {
  860. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  861. }
  862. return 0
  863. }
  864. func getMetadataFromDirectory(account: String, serverUrl: String) -> tableMetadata? {
  865. do {
  866. let realm = try Realm()
  867. realm.refresh()
  868. guard let directory = realm.objects(tableDirectory.self).filter("account == %@ AND serverUrl == %@", account, serverUrl).first else { return nil }
  869. guard let result = realm.objects(tableMetadata.self).filter("ocId == %@", directory.ocId).first else { return nil }
  870. return tableMetadata.init(value: result)
  871. } catch let error as NSError {
  872. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  873. }
  874. return nil
  875. }
  876. func getMetadatasFromGroupfolders(account: String, urlBase: String, userId: String) -> [tableMetadata] {
  877. var metadatas: [tableMetadata] = []
  878. let homeServerUrl = utilityFileSystem.getHomeServer(urlBase: urlBase, userId: userId)
  879. do {
  880. let realm = try Realm()
  881. realm.refresh()
  882. let groupfolders = realm.objects(TableGroupfolders.self).filter("account == %@", account)
  883. for groupfolder in groupfolders {
  884. let mountPoint = groupfolder.mountPoint.hasPrefix("/") ? groupfolder.mountPoint : "/" + groupfolder.mountPoint
  885. let serverUrlFileName = homeServerUrl + mountPoint
  886. if let directory = realm.objects(tableDirectory.self).filter("account == %@ AND serverUrl == %@", account, serverUrlFileName).first,
  887. let metadata = realm.objects(tableMetadata.self).filter("ocId == %@", directory.ocId).first {
  888. metadatas.append(tableMetadata(value: metadata))
  889. }
  890. }
  891. } catch let error as NSError {
  892. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  893. }
  894. return metadatas
  895. }
  896. func getMetadatasInError(account: String) -> Results<tableMetadata>? {
  897. do {
  898. let realm = try Realm()
  899. let results = realm.objects(tableMetadata.self).filter("account == %@ AND errorCodeCounter > 1", account)
  900. return results
  901. } catch let error as NSError {
  902. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  903. }
  904. return nil
  905. }
  906. func clearErrorCodeMetadatas(metadatas: Results<tableMetadata>?) {
  907. guard let metadatas else { return }
  908. do {
  909. let realm = try Realm()
  910. try realm.write {
  911. for metadata in metadatas {
  912. metadata.errorCode = 0
  913. metadata.errorCodeCounter = 0
  914. metadata.errorCodeDate = nil
  915. }
  916. }
  917. } catch let error as NSError {
  918. NextcloudKit.shared.nkCommonInstance.writeLog("Could not access database: \(error)")
  919. }
  920. }
  921. @discardableResult
  922. func updateMetadatas(_ metadatas: [tableMetadata], predicate: NSPredicate) -> (metadatasChangedCount: Int, metadatasChanged: Bool) {
  923. var metadatasChangedCount: Int = 0
  924. var metadatasChanged: Bool = false
  925. do {
  926. let realm = try Realm()
  927. try realm.write {
  928. let results = realm.objects(tableMetadata.self).filter(predicate)
  929. metadatasChangedCount = metadatas.count - results.count
  930. for metadata in metadatas {
  931. if let result = results.filter({ $0.ocId == metadata.ocId }).first,
  932. metadata.isEqual(result) { } else {
  933. metadatasChanged = true
  934. break
  935. }
  936. }
  937. if metadatasChangedCount != 0 || metadatasChanged {
  938. realm.delete(results)
  939. for metadata in metadatas {
  940. realm.add(tableMetadata(value: metadata), update: .all)
  941. }
  942. }
  943. }
  944. } catch let error {
  945. NextcloudKit.shared.nkCommonInstance.writeLog("Could not write to database: \(error)")
  946. }
  947. return (metadatasChangedCount, metadatasChanged)
  948. }
  949. }