NCManageDatabase+Metadata.swift 34 KB

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