NCManageDatabase+Metadata.swift 40 KB

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