NCManageDatabase+Metadata.swift 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829
  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 NCCommunication
  26. extension NCManageDatabase {
  27. @objc func copyObject(metadata: tableMetadata) -> tableMetadata {
  28. return tableMetadata.init(value: metadata)
  29. }
  30. @objc func convertNCFileToMetadata(_ file: NCCommunicationFile, isEncrypted: Bool, account: String) -> tableMetadata {
  31. let metadata = tableMetadata()
  32. metadata.account = account
  33. metadata.checksums = file.checksums
  34. metadata.commentsUnread = file.commentsUnread
  35. metadata.contentType = file.contentType
  36. if let date = file.creationDate {
  37. metadata.creationDate = date
  38. } else {
  39. metadata.creationDate = file.date
  40. }
  41. metadata.dataFingerprint = file.dataFingerprint
  42. metadata.date = file.date
  43. metadata.directory = file.directory
  44. metadata.downloadURL = file.downloadURL
  45. metadata.e2eEncrypted = file.e2eEncrypted
  46. metadata.etag = file.etag
  47. metadata.ext = file.ext
  48. metadata.favorite = file.favorite
  49. metadata.fileId = file.fileId
  50. metadata.fileName = file.fileName
  51. metadata.fileNameView = file.fileName
  52. metadata.fileNameWithoutExt = file.fileNameWithoutExt
  53. metadata.hasPreview = file.hasPreview
  54. metadata.iconName = file.iconName
  55. metadata.livePhoto = file.livePhoto
  56. metadata.mountType = file.mountType
  57. metadata.note = file.note
  58. metadata.ocId = file.ocId
  59. metadata.ownerId = file.ownerId
  60. metadata.ownerDisplayName = file.ownerDisplayName
  61. metadata.lock = file.lock
  62. metadata.lockOwner = file.lockOwner
  63. metadata.lockOwnerDisplayName = file.lockOwnerDisplayName
  64. metadata.lockTime = file.lockTime
  65. metadata.path = file.path
  66. metadata.permissions = file.permissions
  67. metadata.quotaUsedBytes = file.quotaUsedBytes
  68. metadata.quotaAvailableBytes = file.quotaAvailableBytes
  69. metadata.richWorkspace = file.richWorkspace
  70. metadata.resourceType = file.resourceType
  71. metadata.serverUrl = file.serverUrl
  72. metadata.sharePermissionsCollaborationServices = file.sharePermissionsCollaborationServices
  73. for element in file.sharePermissionsCloudMesh {
  74. metadata.sharePermissionsCloudMesh.append(element)
  75. }
  76. for element in file.shareType {
  77. metadata.shareType.append(element)
  78. }
  79. metadata.size = file.size
  80. metadata.classFile = file.classFile
  81. if let date = file.uploadDate {
  82. metadata.uploadDate = date
  83. } else {
  84. metadata.uploadDate = file.date
  85. }
  86. metadata.urlBase = file.urlBase
  87. metadata.user = file.user
  88. metadata.userId = file.userId
  89. // E2EE find the fileName for fileNameView
  90. if isEncrypted || metadata.e2eEncrypted {
  91. if let tableE2eEncryption = NCManageDatabase.shared.getE2eEncryption(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileNameIdentifier == %@", account, file.serverUrl, file.fileName)) {
  92. metadata.fileNameView = tableE2eEncryption.fileName
  93. let results = NCCommunicationCommon.shared.getInternalType(fileName: metadata.fileNameView, mimeType: file.contentType, directory: file.directory)
  94. metadata.contentType = results.mimeType
  95. metadata.iconName = results.iconName
  96. metadata.classFile = results.classFile
  97. }
  98. }
  99. // Live Photo "DETECT"
  100. if !metadata.directory && !metadata.livePhoto && (metadata.classFile == NCCommunicationCommon.typeClassFile.video.rawValue || metadata.classFile == NCCommunicationCommon.typeClassFile.image.rawValue) {
  101. var classFile = metadata.classFile
  102. if classFile == NCCommunicationCommon.typeClassFile.image.rawValue {
  103. classFile = NCCommunicationCommon.typeClassFile.video.rawValue
  104. } else {
  105. classFile = NCCommunicationCommon.typeClassFile.image.rawValue
  106. }
  107. if getMetadata(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileNameWithoutExt == %@ AND ocId != %@ AND classFile == %@", metadata.account, metadata.serverUrl, metadata.fileNameWithoutExt, metadata.ocId, classFile)) != nil {
  108. metadata.livePhoto = true
  109. }
  110. }
  111. return metadata
  112. }
  113. @objc func convertNCCommunicationFilesToMetadatas(_ files: [NCCommunicationFile], useMetadataFolder: Bool, account: String, completion: @escaping (_ metadataFolder: tableMetadata, _ metadatasFolder: [tableMetadata], _ metadatas: [tableMetadata]) -> Void) {
  114. var counter: Int = 0
  115. var isEncrypted: Bool = false
  116. var listServerUrl: [String: Bool] = [:]
  117. var metadataFolder = tableMetadata()
  118. var metadataFolders: [tableMetadata] = []
  119. var metadatas: [tableMetadata] = []
  120. for file in files {
  121. if let key = listServerUrl[file.serverUrl] {
  122. isEncrypted = key
  123. } else {
  124. isEncrypted = CCUtility.isFolderEncrypted(file.serverUrl, e2eEncrypted: file.e2eEncrypted, account: account, urlBase: file.urlBase)
  125. listServerUrl[file.serverUrl] = isEncrypted
  126. }
  127. let metadata = convertNCFileToMetadata(file, isEncrypted: isEncrypted, account: account)
  128. if counter == 0 && useMetadataFolder {
  129. metadataFolder = tableMetadata.init(value: metadata)
  130. } else {
  131. metadatas.append(metadata)
  132. if metadata.directory {
  133. metadataFolders.append(metadata)
  134. }
  135. }
  136. counter += 1
  137. }
  138. completion(metadataFolder, metadataFolders, metadatas)
  139. }
  140. @objc func createMetadata(account: String, user: String, userId: String, fileName: String, fileNameView: String, ocId: String, serverUrl: String, urlBase: String, url: String, contentType: String, livePhoto: Bool) -> tableMetadata {
  141. let metadata = tableMetadata()
  142. let resultInternalType = NCCommunicationCommon.shared.getInternalType(fileName: fileName, mimeType: contentType, directory: false)
  143. let fileName = fileName.trimmingCharacters(in: .whitespacesAndNewlines)
  144. metadata.account = account
  145. metadata.chunk = false
  146. metadata.contentType = resultInternalType.mimeType
  147. metadata.creationDate = Date() as NSDate
  148. metadata.date = Date() as NSDate
  149. metadata.hasPreview = true
  150. metadata.iconName = resultInternalType.iconName
  151. metadata.etag = ocId
  152. metadata.ext = (fileName as NSString).pathExtension.lowercased()
  153. metadata.fileName = fileName
  154. metadata.fileNameView = fileName
  155. metadata.fileNameWithoutExt = (fileName as NSString).deletingPathExtension
  156. metadata.livePhoto = livePhoto
  157. metadata.ocId = ocId
  158. metadata.permissions = "RGDNVW"
  159. metadata.serverUrl = serverUrl
  160. metadata.classFile = resultInternalType.classFile
  161. metadata.uploadDate = Date() as NSDate
  162. metadata.url = url
  163. metadata.urlBase = urlBase
  164. metadata.user = user
  165. metadata.userId = userId
  166. if !metadata.urlBase.isEmpty, metadata.serverUrl.hasPrefix(metadata.urlBase) {
  167. metadata.path = String(metadata.serverUrl.dropFirst(metadata.urlBase.count)) + "/"
  168. }
  169. return metadata
  170. }
  171. @objc func addMetadata(_ metadata: tableMetadata) {
  172. let realm = try! Realm()
  173. do {
  174. try realm.safeWrite {
  175. realm.add(metadata, update: .all)
  176. }
  177. } catch let error {
  178. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  179. }
  180. }
  181. @objc func addMetadatas(_ metadatas: [tableMetadata]) {
  182. let realm = try! Realm()
  183. do {
  184. try realm.safeWrite {
  185. for metadata in metadatas {
  186. realm.add(metadata, update: .all)
  187. }
  188. }
  189. } catch let error {
  190. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  191. }
  192. }
  193. @objc func deleteMetadata(predicate: NSPredicate) {
  194. let realm = try! Realm()
  195. do {
  196. try realm.safeWrite {
  197. let results = realm.objects(tableMetadata.self).filter(predicate)
  198. realm.delete(results)
  199. }
  200. } catch let error {
  201. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  202. }
  203. }
  204. @objc func moveMetadata(ocId: String, serverUrlTo: String) {
  205. let realm = try! Realm()
  206. do {
  207. try realm.safeWrite {
  208. if let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first {
  209. result.serverUrl = serverUrlTo
  210. }
  211. }
  212. } catch let error {
  213. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  214. }
  215. }
  216. @objc func addMetadataServerUrl(ocId: String, serverUrl: String) {
  217. let realm = try! Realm()
  218. do {
  219. try realm.safeWrite {
  220. let results = realm.objects(tableMetadata.self).filter("ocId == %@", ocId)
  221. for result in results {
  222. result.serverUrl = serverUrl
  223. }
  224. }
  225. } catch let error {
  226. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  227. }
  228. }
  229. @objc func renameMetadata(fileNameTo: String, ocId: String) {
  230. let realm = try! Realm()
  231. do {
  232. try realm.safeWrite {
  233. if let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first {
  234. let resultsType = NCCommunicationCommon.shared.getInternalType(fileName: fileNameTo, mimeType: "", directory: result.directory)
  235. result.fileName = fileNameTo
  236. result.fileNameView = fileNameTo
  237. if result.directory {
  238. result.fileNameWithoutExt = fileNameTo
  239. result.ext = ""
  240. } else {
  241. result.fileNameWithoutExt = (fileNameTo as NSString).deletingPathExtension
  242. result.ext = resultsType.ext
  243. }
  244. result.iconName = resultsType.iconName
  245. result.contentType = resultsType.mimeType
  246. result.classFile = resultsType.classFile
  247. }
  248. }
  249. } catch let error {
  250. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  251. }
  252. }
  253. @discardableResult
  254. func updateMetadatas(_ metadatas: [tableMetadata], metadatasResult: [tableMetadata], addCompareLivePhoto: Bool = true, addExistsInLocal: Bool = false, addCompareEtagLocal: Bool = false, addDirectorySynchronized: Bool = false) -> (metadatasUpdate: [tableMetadata], metadatasLocalUpdate: [tableMetadata], metadatasDelete: [tableMetadata]) {
  255. let realm = try! Realm()
  256. var ocIdsUdate: [String] = []
  257. var ocIdsLocalUdate: [String] = []
  258. var metadatasDelete: [tableMetadata] = []
  259. var metadatasUpdate: [tableMetadata] = []
  260. var metadatasLocalUpdate: [tableMetadata] = []
  261. realm.refresh()
  262. do {
  263. try realm.safeWrite {
  264. // DELETE
  265. for metadataResult in metadatasResult {
  266. if metadatas.firstIndex(where: { $0.ocId == metadataResult.ocId }) == nil {
  267. if let result = realm.objects(tableMetadata.self).filter(NSPredicate(format: "ocId == %@", metadataResult.ocId)).first {
  268. metadatasDelete.append(tableMetadata.init(value: result))
  269. realm.delete(result)
  270. }
  271. }
  272. }
  273. // UPDATE/NEW
  274. for metadata in metadatas {
  275. if let result = metadatasResult.first(where: { $0.ocId == metadata.ocId }) {
  276. // update
  277. 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) {
  278. ocIdsUdate.append(metadata.ocId)
  279. realm.add(tableMetadata.init(value: metadata), update: .all)
  280. } else if result.status == NCGlobal.shared.metadataStatusNormal && addCompareLivePhoto && result.livePhoto != metadata.livePhoto {
  281. ocIdsUdate.append(metadata.ocId)
  282. realm.add(tableMetadata.init(value: metadata), update: .all)
  283. }
  284. } else {
  285. // new
  286. ocIdsUdate.append(metadata.ocId)
  287. realm.add(tableMetadata.init(value: metadata), update: .all)
  288. }
  289. if metadata.directory && !ocIdsUdate.contains(metadata.ocId) {
  290. let table = realm.objects(tableDirectory.self).filter(NSPredicate(format: "ocId == %@", metadata.ocId)).first
  291. if table?.etag != metadata.etag {
  292. ocIdsUdate.append(metadata.ocId)
  293. }
  294. }
  295. // Local
  296. if !metadata.directory && (addExistsInLocal || addCompareEtagLocal) {
  297. let localFile = realm.objects(tableLocalFile.self).filter(NSPredicate(format: "ocId == %@", metadata.ocId)).first
  298. if addCompareEtagLocal && localFile != nil && localFile?.etag != metadata.etag {
  299. ocIdsLocalUdate.append(metadata.ocId)
  300. }
  301. if addExistsInLocal && (localFile == nil || localFile?.etag != metadata.etag) && !ocIdsLocalUdate.contains(metadata.ocId) {
  302. ocIdsLocalUdate.append(metadata.ocId)
  303. }
  304. }
  305. }
  306. }
  307. } catch let error {
  308. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  309. }
  310. for ocId in ocIdsUdate {
  311. if let result = realm.objects(tableMetadata.self).filter(NSPredicate(format: "ocId == %@", ocId)).first {
  312. metadatasUpdate.append(tableMetadata.init(value: result))
  313. }
  314. }
  315. for ocId in ocIdsLocalUdate {
  316. if let result = realm.objects(tableMetadata.self).filter(NSPredicate(format: "ocId == %@", ocId)).first {
  317. metadatasLocalUpdate.append(tableMetadata.init(value: result))
  318. }
  319. }
  320. return (metadatasUpdate, metadatasLocalUpdate, metadatasDelete)
  321. }
  322. func setMetadataSession(ocId: String, session: String? = nil, sessionError: String? = nil, sessionSelector: String? = nil, sessionTaskIdentifier: Int? = nil, status: Int? = nil, etag: String? = nil) {
  323. let realm = try! Realm()
  324. realm.refresh()
  325. do {
  326. try realm.safeWrite {
  327. let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  328. if let session = session {
  329. result?.session = session
  330. }
  331. if let sessionError = sessionError {
  332. result?.sessionError = sessionError
  333. }
  334. if let sessionSelector = sessionSelector {
  335. result?.sessionSelector = sessionSelector
  336. }
  337. if let sessionTaskIdentifier = sessionTaskIdentifier {
  338. result?.sessionTaskIdentifier = sessionTaskIdentifier
  339. }
  340. if let status = status {
  341. result?.status = status
  342. }
  343. if let etag = etag {
  344. result?.etag = etag
  345. }
  346. }
  347. } catch let error {
  348. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  349. }
  350. }
  351. @discardableResult
  352. func setMetadataStatus(ocId: String, status: Int) -> tableMetadata? {
  353. let realm = try! Realm()
  354. var result: tableMetadata?
  355. do {
  356. try realm.safeWrite {
  357. result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  358. result?.status = status
  359. }
  360. } catch let error {
  361. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  362. }
  363. if let result = result {
  364. return tableMetadata.init(value: result)
  365. } else {
  366. return nil
  367. }
  368. }
  369. func setMetadataEtagResource(ocId: String, etagResource: String?) {
  370. let realm = try! Realm()
  371. var result: tableMetadata?
  372. guard let etagResource = etagResource else { return }
  373. do {
  374. try realm.safeWrite {
  375. result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  376. result?.etagResource = etagResource
  377. }
  378. } catch let error {
  379. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  380. }
  381. }
  382. @objc func setMetadataFavorite(ocId: String, favorite: Bool) {
  383. let realm = try! Realm()
  384. do {
  385. try realm.safeWrite {
  386. let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  387. result?.favorite = favorite
  388. }
  389. } catch let error {
  390. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  391. }
  392. }
  393. @objc func updateMetadatasFavorite(account: String, metadatas: [tableMetadata]) {
  394. let realm = try! Realm()
  395. do {
  396. try realm.safeWrite {
  397. let results = realm.objects(tableMetadata.self).filter("account == %@ AND favorite == true", account)
  398. for result in results {
  399. result.favorite = false
  400. }
  401. for metadata in metadatas {
  402. realm.add(metadata, update: .all)
  403. }
  404. }
  405. } catch let error {
  406. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  407. }
  408. }
  409. @objc func setMetadataEncrypted(ocId: String, encrypted: Bool) {
  410. let realm = try! Realm()
  411. do {
  412. try realm.safeWrite {
  413. let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first
  414. result?.e2eEncrypted = encrypted
  415. }
  416. } catch let error {
  417. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  418. }
  419. }
  420. @objc func setMetadataFileNameView(serverUrl: String, fileName: String, newFileNameView: String, account: String) {
  421. let realm = try! Realm()
  422. do {
  423. try realm.safeWrite {
  424. let result = realm.objects(tableMetadata.self).filter("account == %@ AND serverUrl == %@ AND fileName == %@", account, serverUrl, fileName).first
  425. result?.fileNameView = newFileNameView
  426. }
  427. } catch let error {
  428. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  429. }
  430. }
  431. @objc func getMetadata(predicate: NSPredicate) -> tableMetadata? {
  432. let realm = try! Realm()
  433. realm.refresh()
  434. guard let result = realm.objects(tableMetadata.self).filter(predicate).first else {
  435. return nil
  436. }
  437. return tableMetadata.init(value: result)
  438. }
  439. @objc func getMetadata(predicate: NSPredicate, sorted: String, ascending: Bool) -> tableMetadata? {
  440. let realm = try! Realm()
  441. realm.refresh()
  442. guard let result = realm.objects(tableMetadata.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending).first else {
  443. return nil
  444. }
  445. return tableMetadata.init(value: result)
  446. }
  447. @objc func getMetadatasViewer(predicate: NSPredicate, sorted: String, ascending: Bool) -> [tableMetadata]? {
  448. let realm = try! Realm()
  449. realm.refresh()
  450. let results: Results<tableMetadata>
  451. var finals: [tableMetadata] = []
  452. if (tableMetadata().objectSchema.properties.contains { $0.name == sorted }) {
  453. results = realm.objects(tableMetadata.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  454. } else {
  455. results = realm.objects(tableMetadata.self).filter(predicate)
  456. }
  457. // For Live Photo
  458. var fileNameImages: [String] = []
  459. let filtered = results.filter { $0.classFile.contains(NCCommunicationCommon.typeClassFile.image.rawValue) }
  460. filtered.forEach { print($0)
  461. let fileName = ($0.fileNameView as NSString).deletingPathExtension
  462. fileNameImages.append(fileName)
  463. }
  464. for result in results {
  465. let ext = (result.fileNameView as NSString).pathExtension.uppercased()
  466. let fileName = (result.fileNameView as NSString).deletingPathExtension
  467. if !(ext == "MOV" && fileNameImages.contains(fileName)) {
  468. finals.append(result)
  469. }
  470. }
  471. if finals.count > 0 {
  472. return Array(finals.map { tableMetadata.init(value: $0) })
  473. } else {
  474. return nil
  475. }
  476. }
  477. @objc func getMetadatas(predicate: NSPredicate) -> [tableMetadata] {
  478. let realm = try! Realm()
  479. realm.refresh()
  480. let results = realm.objects(tableMetadata.self).filter(predicate)
  481. return Array(results.map { tableMetadata.init(value: $0) })
  482. }
  483. @objc func getAdvancedMetadatas(predicate: NSPredicate, page: Int = 0, limit: Int = 0, sorted: String, ascending: Bool) -> [tableMetadata] {
  484. let realm = try! Realm()
  485. realm.refresh()
  486. var metadatas: [tableMetadata] = []
  487. let results = realm.objects(tableMetadata.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  488. if results.count > 0 {
  489. if page == 0 || limit == 0 {
  490. return Array(results.map { tableMetadata.init(value: $0) })
  491. } else {
  492. let nFrom = (page - 1) * limit
  493. let nTo = nFrom + (limit - 1)
  494. for n in nFrom...nTo {
  495. if n == results.count {
  496. break
  497. }
  498. metadatas.append(tableMetadata.init(value: results[n]))
  499. }
  500. }
  501. }
  502. return metadatas
  503. }
  504. @objc func getMetadataAtIndex(predicate: NSPredicate, sorted: String, ascending: Bool, index: Int) -> tableMetadata? {
  505. let realm = try! Realm()
  506. realm.refresh()
  507. let results = realm.objects(tableMetadata.self).filter(predicate).sorted(byKeyPath: sorted, ascending: ascending)
  508. if results.count > 0 && results.count > index {
  509. return tableMetadata.init(value: results[index])
  510. } else {
  511. return nil
  512. }
  513. }
  514. @objc func getMetadataFromOcId(_ ocId: String?) -> tableMetadata? {
  515. let realm = try! Realm()
  516. realm.refresh()
  517. guard let ocId = ocId else { return nil }
  518. guard let result = realm.objects(tableMetadata.self).filter("ocId == %@", ocId).first else { return nil }
  519. return tableMetadata.init(value: result)
  520. }
  521. @objc func getMetadataFolder(account: String, urlBase: String, serverUrl: String) -> tableMetadata? {
  522. let realm = try! Realm()
  523. realm.refresh()
  524. var serverUrl = serverUrl
  525. var fileName = ""
  526. let serverUrlHome = NCUtilityFileSystem.shared.getHomeServer(account: account)
  527. if serverUrlHome == serverUrl {
  528. fileName = "."
  529. serverUrl = ".."
  530. } else {
  531. fileName = (serverUrl as NSString).lastPathComponent
  532. serverUrl = NCUtilityFileSystem.shared.deletingLastPathComponent(account: account, serverUrl: serverUrl)
  533. }
  534. guard let result = realm.objects(tableMetadata.self).filter("account == %@ AND serverUrl == %@ AND fileName == %@", account, serverUrl, fileName).first else { return nil }
  535. return tableMetadata.init(value: result)
  536. }
  537. @objc func getTableMetadatasDirectoryFavoriteIdentifierRank(account: String) -> [String: NSNumber] {
  538. var listIdentifierRank: [String: NSNumber] = [:]
  539. let realm = try! Realm()
  540. var counter = 10 as Int64
  541. let results = realm.objects(tableMetadata.self).filter("account == %@ AND directory == true AND favorite == true", account).sorted(byKeyPath: "fileNameView", ascending: true)
  542. for result in results {
  543. counter += 1
  544. listIdentifierRank[result.ocId] = NSNumber(value: Int64(counter))
  545. }
  546. return listIdentifierRank
  547. }
  548. @objc func clearMetadatasUpload(account: String) {
  549. let realm = try! Realm()
  550. realm.refresh()
  551. do {
  552. try realm.safeWrite {
  553. let results = realm.objects(tableMetadata.self).filter("account == %@ AND (status == %d OR status == %@)", account, NCGlobal.shared.metadataStatusWaitUpload, NCGlobal.shared.metadataStatusUploadError)
  554. realm.delete(results)
  555. }
  556. } catch let error {
  557. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  558. }
  559. }
  560. @objc func readMarkerMetadata(account: String, fileId: String) {
  561. let realm = try! Realm()
  562. do {
  563. try realm.safeWrite {
  564. let results = realm.objects(tableMetadata.self).filter("account == %@ AND fileId == %@", account, fileId)
  565. for result in results {
  566. result.commentsUnread = false
  567. }
  568. }
  569. } catch let error {
  570. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  571. }
  572. }
  573. @objc func getAssetLocalIdentifiersUploaded(account: String, sessionSelector: String) -> [String] {
  574. let realm = try! Realm()
  575. realm.refresh()
  576. var assetLocalIdentifiers: [String] = []
  577. let results = realm.objects(tableMetadata.self).filter("account == %@ AND assetLocalIdentifier != '' AND deleteAssetLocalIdentifier == true AND sessionSelector == %@", account, sessionSelector)
  578. for result in results {
  579. assetLocalIdentifiers.append(result.assetLocalIdentifier)
  580. }
  581. return assetLocalIdentifiers
  582. }
  583. @objc func clearAssetLocalIdentifiers(_ assetLocalIdentifiers: [String], account: String) {
  584. let realm = try! Realm()
  585. do {
  586. try realm.safeWrite {
  587. let results = realm.objects(tableMetadata.self).filter("account == %@ AND assetLocalIdentifier IN %@", account, assetLocalIdentifiers)
  588. for result in results {
  589. result.assetLocalIdentifier = ""
  590. result.deleteAssetLocalIdentifier = false
  591. }
  592. }
  593. } catch let error {
  594. NCCommunicationCommon.shared.writeLog("Could not write to database: \(error)")
  595. }
  596. }
  597. @objc func getMetadataLivePhoto(metadata: tableMetadata) -> tableMetadata? {
  598. let realm = try! Realm()
  599. var classFile = metadata.classFile
  600. realm.refresh()
  601. if !metadata.livePhoto || !CCUtility.getLivePhoto() {
  602. return nil
  603. }
  604. if classFile == NCCommunicationCommon.typeClassFile.image.rawValue {
  605. classFile = NCCommunicationCommon.typeClassFile.video.rawValue
  606. } else {
  607. classFile = NCCommunicationCommon.typeClassFile.image.rawValue
  608. }
  609. guard let result = realm.objects(tableMetadata.self).filter(NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileNameWithoutExt == %@ AND ocId != %@ AND classFile == %@", metadata.account, metadata.serverUrl, metadata.fileNameWithoutExt, metadata.ocId, classFile)).first else {
  610. return nil
  611. }
  612. return tableMetadata.init(value: result)
  613. }
  614. func getMetadatasMedia(predicate: NSPredicate, sort: String, ascending: Bool = false) -> [tableMetadata] {
  615. let realm = try! Realm()
  616. realm.refresh()
  617. let sortProperties = [SortDescriptor(keyPath: sort, ascending: ascending), SortDescriptor(keyPath: "fileNameView", ascending: false)]
  618. let results = realm.objects(tableMetadata.self).filter(predicate).sorted(by: sortProperties)
  619. return Array(results.map { tableMetadata.init(value: $0) })
  620. }
  621. func isMetadataShareOrMounted(metadata: tableMetadata, metadataFolder: tableMetadata?) -> Bool {
  622. var isShare = false
  623. var isMounted = false
  624. if metadataFolder != nil {
  625. isShare = metadata.permissions.contains(NCGlobal.shared.permissionShared) && !metadataFolder!.permissions.contains(NCGlobal.shared.permissionShared)
  626. isMounted = metadata.permissions.contains(NCGlobal.shared.permissionMounted) && !metadataFolder!.permissions.contains(NCGlobal.shared.permissionMounted)
  627. } else if let directory = NCManageDatabase.shared.getTableDirectory(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", metadata.account, metadata.serverUrl)) {
  628. isShare = metadata.permissions.contains(NCGlobal.shared.permissionShared) && !directory.permissions.contains(NCGlobal.shared.permissionShared)
  629. isMounted = metadata.permissions.contains(NCGlobal.shared.permissionMounted) && !directory.permissions.contains(NCGlobal.shared.permissionMounted)
  630. }
  631. if isShare || isMounted {
  632. return true
  633. } else {
  634. return false
  635. }
  636. }
  637. func isDownloadMetadata(_ metadata: tableMetadata, download: Bool) -> Bool {
  638. let localFile = getTableLocalFile(predicate: NSPredicate(format: "ocId == %@", metadata.ocId))
  639. let fileSize = CCUtility.fileProviderStorageSize(metadata.ocId, fileNameView: metadata.fileNameView)
  640. if (localFile != nil || download) && (localFile?.etag != metadata.etag || fileSize == 0) {
  641. return true
  642. }
  643. return false
  644. }
  645. func getMetadataConflict(account: String, serverUrl: String, fileName: String) -> tableMetadata? {
  646. // verify exists conflict
  647. let fileNameExtension = (fileName as NSString).pathExtension.lowercased()
  648. let fileNameWithoutExtension = (fileName as NSString).deletingPathExtension
  649. var fileNameConflict = fileName
  650. if fileNameExtension == "heic" && CCUtility.getFormatCompatibility() {
  651. fileNameConflict = fileNameWithoutExtension + ".jpg"
  652. }
  653. return getMetadata(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileNameView == %@", account, serverUrl, fileNameConflict))
  654. }
  655. func getSubtitles(account: String, serverUrl: String, fileName: String) -> (all:[tableMetadata], existing:[tableMetadata]) {
  656. let realm = try! Realm()
  657. let nameOnly = (fileName as NSString).deletingPathExtension + "."
  658. var metadatas: [tableMetadata] = []
  659. let results = realm.objects(tableMetadata.self).filter("account == %@ AND serverUrl == %@ AND fileName BEGINSWITH[c] %@ AND fileName ENDSWITH[c] '.srt'", account, serverUrl, nameOnly)
  660. for result in results {
  661. if CCUtility.fileProviderStorageExists(result) {
  662. metadatas.append(result)
  663. }
  664. }
  665. return(Array(results.map { tableMetadata.init(value: $0) }), Array(metadatas.map { tableMetadata.init(value: $0) }))
  666. }
  667. }