NCManageDatabase+Metadata.swift 44 KB

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