NCManageDatabase+Metadata.swift 37 KB

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