NCManageDatabase+Metadata.swift 46 KB

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