NCManageDatabase+Metadata.swift 41 KB

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