NCManageDatabase+Metadata.swift 44 KB

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