FileProvider.swift 47 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099
  1. //
  2. // FileProviderExtension.swift
  3. // Files
  4. //
  5. // Created by Marino Faggiana on 26/03/18.
  6. // Copyright © 2018 TWS. All rights reserved.
  7. //
  8. // Author Marino Faggiana <m.faggiana@twsweb.it>
  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 FileProvider
  24. import UIKit
  25. import MobileCoreServices
  26. var ocNetworking: OCnetworking?
  27. var account = ""
  28. var accountUser = ""
  29. var accountUserID = ""
  30. var accountPassword = ""
  31. var accountUrl = ""
  32. var homeServerUrl = ""
  33. var directoryUser = ""
  34. // Directory
  35. var groupURL: URL?
  36. var fileProviderStorageURL: URL?
  37. var importDocumentURL: URL?
  38. var changeDocumentURL: URL?
  39. // List
  40. var listUpdateItems = [NSFileProviderItem]()
  41. var listFavoriteIdentifierRank = [String:NSNumber]()
  42. var listUpload = [String:URLSessionTask]()
  43. var timer: Timer?
  44. class FileProvider: NSFileProviderExtension {
  45. override init() {
  46. super.init()
  47. guard let activeAccount = NCManageDatabase.sharedInstance.getAccountActive() else {
  48. return
  49. }
  50. account = activeAccount.account
  51. accountUser = activeAccount.user
  52. accountUserID = activeAccount.userID
  53. accountPassword = activeAccount.password
  54. accountUrl = activeAccount.url
  55. homeServerUrl = CCUtility.getHomeServerUrlActiveUrl(activeAccount.url)
  56. directoryUser = CCUtility.getDirectoryActiveUser(activeAccount.user, activeUrl: activeAccount.url)
  57. ocNetworking = OCnetworking.init(delegate: nil, metadataNet: nil, withUser: accountUser, withUserID: accountUserID, withPassword: accountPassword, withUrl: accountUrl)
  58. groupURL = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: NCBrandOptions.sharedInstance.capabilitiesGroups)
  59. fileProviderStorageURL = groupURL!.appendingPathComponent(k_assetLocalIdentifierFileProviderStorage)
  60. importDocumentURL = groupURL!.appendingPathComponent(k_assetLocalIdentifierFileProviderStorage).appendingPathComponent(k_fileProviderStorageImportDocument)
  61. changeDocumentURL = groupURL!.appendingPathComponent(k_assetLocalIdentifierFileProviderStorage).appendingPathComponent(k_fileProviderStorageChangeDocument)
  62. // Create dir File Provider Storage
  63. do {
  64. try FileManager.default.createDirectory(atPath: fileProviderStorageURL!.path, withIntermediateDirectories: true, attributes: nil)
  65. } catch let error as NSError {
  66. NSLog("Unable to create directory \(error.debugDescription)")
  67. }
  68. // Create dir for Upload
  69. do {
  70. try FileManager.default.createDirectory(atPath: importDocumentURL!.path, withIntermediateDirectories: true, attributes: nil)
  71. } catch let error as NSError {
  72. NSLog("Unable to create directory \(error.debugDescription)")
  73. }
  74. // Create dir for change document
  75. do {
  76. try FileManager.default.createDirectory(atPath: changeDocumentURL!.path, withIntermediateDirectories: true, attributes: nil)
  77. } catch let error as NSError {
  78. NSLog("Unable to create directory \(error.debugDescription)")
  79. }
  80. if #available(iOSApplicationExtension 11.0, *) {
  81. listUpdateItems.removeAll()
  82. listFavoriteIdentifierRank = NCManageDatabase.sharedInstance.getTableMetadatasDirectoryFavoriteIdentifierRank()
  83. // Timer for upload
  84. if timer == nil {
  85. timer = Timer.init(timeInterval: 5, repeats: true, block: { (Timer) in
  86. if let metadataNet = NCManageDatabase.sharedInstance.getQueueUpload() {
  87. if metadataNet.path != nil {
  88. if let directoryID = NCManageDatabase.sharedInstance.getDirectoryID(metadataNet.serverUrl) {
  89. if let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileName = %@ AND directoryID = %@", account, metadataNet.fileName, directoryID)) {
  90. self.uploadCloud(metadataNet.fileName, serverUrl: metadataNet.serverUrl, path: metadataNet.path, identifier: NSFileProviderItemIdentifier(rawValue: metadata.fileID))
  91. }
  92. }
  93. }
  94. }
  95. })
  96. RunLoop.main.add(timer!, forMode: .defaultRunLoopMode)
  97. }
  98. } else {
  99. NSFileCoordinator().coordinate(writingItemAt: self.documentStorageURL, options: [], error: nil, byAccessor: { newURL in
  100. do {
  101. try FileManager.default.createDirectory(at: newURL, withIntermediateDirectories: true, attributes: nil)
  102. } catch let error {
  103. print("error: \(error)")
  104. }
  105. })
  106. }
  107. }
  108. // MARK: - Enumeration
  109. override func enumerator(for containerItemIdentifier: NSFileProviderItemIdentifier) throws -> NSFileProviderEnumerator {
  110. /* ONLY iOS 11*/
  111. guard #available(iOS 11, *) else {
  112. throw NSError(domain: NSCocoaErrorDomain, code: NSFileNoSuchFileError, userInfo:[:])
  113. }
  114. var maybeEnumerator: NSFileProviderEnumerator? = nil
  115. if (containerItemIdentifier == NSFileProviderItemIdentifier.rootContainer) {
  116. maybeEnumerator = FileProviderEnumerator(enumeratedItemIdentifier: containerItemIdentifier)
  117. } else if (containerItemIdentifier == NSFileProviderItemIdentifier.workingSet) {
  118. maybeEnumerator = FileProviderEnumeratorWorkingSet(enumeratedItemIdentifier: containerItemIdentifier)
  119. } else {
  120. // determine if the item is a directory or a file
  121. // - for a directory, instantiate an enumerator of its subitems
  122. // - for a file, instantiate an enumerator that observes changes to the file
  123. let item = try self.item(for: containerItemIdentifier)
  124. if item.typeIdentifier == kUTTypeFolder as String {
  125. maybeEnumerator = FileProviderEnumerator(enumeratedItemIdentifier: containerItemIdentifier)
  126. } else {
  127. maybeEnumerator = FileProviderEnumeratorFile(enumeratedItemIdentifier: containerItemIdentifier)
  128. }
  129. }
  130. guard let enumerator = maybeEnumerator else {
  131. throw NSError(domain: NSCocoaErrorDomain, code: NSFeatureUnsupportedError, userInfo:[:])
  132. }
  133. return enumerator
  134. }
  135. // MARK: - Item
  136. override func item(for identifier: NSFileProviderItemIdentifier) throws -> NSFileProviderItem {
  137. /* ONLY iOS 11*/
  138. guard #available(iOS 11, *) else {
  139. throw NSError(domain: NSCocoaErrorDomain, code: NSFileNoSuchFileError, userInfo:[:])
  140. }
  141. if identifier == .rootContainer {
  142. if let directory = NCManageDatabase.sharedInstance.getTableDirectory(predicate: NSPredicate(format: "account = %@ AND serverUrl = %@", account, homeServerUrl)) {
  143. let metadata = tableMetadata()
  144. metadata.account = account
  145. metadata.directory = true
  146. metadata.directoryID = directory.directoryID
  147. metadata.fileID = identifier.rawValue
  148. metadata.fileName = NCBrandOptions.sharedInstance.brand
  149. metadata.fileNameView = NCBrandOptions.sharedInstance.brand
  150. metadata.typeFile = k_metadataTypeFile_directory
  151. return FileProviderItem(metadata: metadata, serverUrl: homeServerUrl)
  152. }
  153. } else {
  154. if let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, identifier.rawValue)) {
  155. if let directory = NCManageDatabase.sharedInstance.getTableDirectory(predicate: NSPredicate(format: "account = %@ AND directoryID = %@", account, metadata.directoryID)) {
  156. return FileProviderItem(metadata: metadata, serverUrl: directory.serverUrl)
  157. }
  158. }
  159. }
  160. // implement the actual lookup
  161. throw NSFileProviderError(.noSuchItem)
  162. }
  163. override func urlForItem(withPersistentIdentifier identifier: NSFileProviderItemIdentifier) -> URL? {
  164. /* ONLY iOS 11*/
  165. guard #available(iOS 11, *) else {
  166. return nil
  167. }
  168. // resolve the given identifier to a file on disk
  169. guard let item = try? item(for: identifier) else {
  170. return nil
  171. }
  172. // in this implementation, all paths are structured as <base storage directory>/<item identifier>/<item file name>
  173. let manager = NSFileProviderManager.default
  174. var url = manager.documentStorageURL.appendingPathComponent(identifier.rawValue, isDirectory: true)
  175. if item.typeIdentifier == (kUTTypeFolder as String) {
  176. url = url.appendingPathComponent(item.filename, isDirectory:true)
  177. } else {
  178. url = url.appendingPathComponent(item.filename, isDirectory:false)
  179. }
  180. return url
  181. }
  182. override func persistentIdentifierForItem(at url: URL) -> NSFileProviderItemIdentifier? {
  183. // resolve the given URL to a persistent identifier using a database
  184. let pathComponents = url.pathComponents
  185. // exploit the fact that the path structure has been defined as
  186. // <base storage directory>/<item identifier>/<item file name> above
  187. assert(pathComponents.count > 2)
  188. let itemIdentifier = NSFileProviderItemIdentifier(pathComponents[pathComponents.count - 2])
  189. return itemIdentifier
  190. }
  191. // MARK: - Managing Shared Files
  192. override func providePlaceholder(at url: URL, completionHandler: @escaping (Error?) -> Void) {
  193. if #available(iOSApplicationExtension 11.0, *) {
  194. guard let identifier = persistentIdentifierForItem(at: url) else {
  195. completionHandler(NSFileProviderError(.noSuchItem))
  196. return
  197. }
  198. do {
  199. let fileProviderItem = try item(for: identifier)
  200. let placeholderURL = NSFileProviderManager.placeholderURL(for: url)
  201. try NSFileProviderManager.writePlaceholder(at: placeholderURL,withMetadata: fileProviderItem)
  202. completionHandler(nil)
  203. } catch let error {
  204. print("error: \(error)")
  205. completionHandler(error)
  206. }
  207. } else {
  208. let fileName = url.lastPathComponent
  209. let placeholderURL = NSFileProviderExtension.placeholderURL(for: self.documentStorageURL.appendingPathComponent(fileName))
  210. let fileSize = 0
  211. let metadata = [AnyHashable(URLResourceKey.fileSizeKey): fileSize]
  212. do {
  213. try NSFileProviderExtension.writePlaceholder(at: placeholderURL, withMetadata: metadata as! [URLResourceKey : Any])
  214. } catch let error {
  215. print("error: \(error)")
  216. }
  217. completionHandler(nil)
  218. }
  219. }
  220. override func startProvidingItem(at url: URL, completionHandler: @escaping ((_ error: Error?) -> Void)) {
  221. if #available(iOSApplicationExtension 11.0, *) {
  222. var fileSize : UInt64 = 0
  223. do {
  224. let attr = try FileManager.default.attributesOfItem(atPath: url.path)
  225. fileSize = attr[FileAttributeKey.size] as! UInt64
  226. } catch let error {
  227. print("Error: \(error)")
  228. }
  229. if fileSize > 0 {
  230. completionHandler(nil)
  231. return
  232. }
  233. //let fileName = url.lastPathComponent
  234. let pathComponents = url.pathComponents
  235. let identifier = NSFileProviderItemIdentifier(pathComponents[pathComponents.count - 2])
  236. //let changeDocumentPath = changeDocumentURL!.path + "/" + fileName
  237. guard let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, identifier.rawValue)) else {
  238. completionHandler(NSFileProviderError(.noSuchItem))
  239. return
  240. }
  241. guard let serverUrl = NCManageDatabase.sharedInstance.getServerUrl(metadata.directoryID) else {
  242. completionHandler(NSFileProviderError(.noSuchItem))
  243. return
  244. }
  245. let task = ocNetworking?.downloadFileNameServerUrl("\(serverUrl)/\(metadata.fileName)", fileNameLocalPath: "\(directoryUser)/\(metadata.fileID)", communication: CCNetworking.shared().sharedOCCommunicationExtensionDownload(metadata.fileName), success: { (lenght) in
  246. NCManageDatabase.sharedInstance.addLocalFile(metadata: metadata)
  247. // copy download file to url
  248. _ = self.copyFile("\(directoryUser)/\(metadata.fileID)", toPath: url.path)
  249. completionHandler(nil)
  250. }, failure: { (errorMessage, errorCode) in
  251. completionHandler(NSFileProviderError(.serverUnreachable))
  252. })
  253. if task != nil {
  254. NSFileProviderManager.default.register(task!, forItemWithIdentifier: NSFileProviderItemIdentifier(identifier.rawValue)) { (error) in }
  255. }
  256. } else {
  257. guard let fileData = try? Data(contentsOf: url) else {
  258. completionHandler(nil)
  259. return
  260. }
  261. do {
  262. _ = try fileData.write(to: url, options: NSData.WritingOptions())
  263. completionHandler(nil)
  264. } catch let error {
  265. print("error: \(error)")
  266. completionHandler(error)
  267. }
  268. }
  269. }
  270. override func itemChanged(at url: URL) {
  271. if #available(iOSApplicationExtension 11.0, *) {
  272. let fileName = url.lastPathComponent
  273. let pathComponents = url.pathComponents
  274. let changeDocumentPath = changeDocumentURL!.path + "/" + fileName
  275. let importDocumentPath = importDocumentURL!.path + "/" + fileName
  276. let metadataNet = CCMetadataNet()
  277. assert(pathComponents.count > 2)
  278. let identifier = NSFileProviderItemIdentifier(pathComponents[pathComponents.count - 2])
  279. guard let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, identifier.rawValue)) else {
  280. return
  281. }
  282. guard let serverUrl = NCManageDatabase.sharedInstance.getServerUrl(metadata.directoryID) else {
  283. return
  284. }
  285. // Refresh
  286. self.refreshEnumerator(identifier: identifier, serverUrl: serverUrl)
  287. // Copy file to Change Document & if exists on Import Document
  288. _ = self.copyFile(url.path, toPath: changeDocumentPath)
  289. if FileManager.default.fileExists(atPath: importDocumentPath) {
  290. _ = self.copyFile(url.path, toPath: importDocumentPath)
  291. }
  292. metadataNet.account = account
  293. metadataNet.assetLocalIdentifier = k_assetLocalIdentifierFileProviderStorage + identifier.rawValue
  294. metadataNet.fileName = fileName
  295. metadataNet.path = changeDocumentPath
  296. metadataNet.selector = selectorUploadFile
  297. metadataNet.selectorPost = ""
  298. metadataNet.serverUrl = serverUrl
  299. metadataNet.session = k_upload_session
  300. metadataNet.taskStatus = Int(k_taskStatusResume)
  301. _ = NCManageDatabase.sharedInstance.addQueueUpload(metadataNet: metadataNet)
  302. } else {
  303. let fileSize = (try! FileManager.default.attributesOfItem(atPath: url.path)[FileAttributeKey.size] as! NSNumber).uint64Value
  304. NSLog("[LOG] Item changed at URL %@ %lu", url as NSURL, fileSize)
  305. guard let account = NCManageDatabase.sharedInstance.getAccountActive() else {
  306. self.stopProvidingItem(at: url)
  307. return
  308. }
  309. guard let fileName = CCUtility.getFileNameExt() else {
  310. self.stopProvidingItem(at: url)
  311. return
  312. }
  313. // -------> Fix : Clear FileName for twice Office 365
  314. CCUtility.setFileNameExt("")
  315. // --------------------------------------------------
  316. if (fileName != url.lastPathComponent) {
  317. self.stopProvidingItem(at: url)
  318. return
  319. }
  320. guard let serverUrl = CCUtility.getServerUrlExt() else {
  321. self.stopProvidingItem(at: url)
  322. return
  323. }
  324. guard let directoryID = NCManageDatabase.sharedInstance.getDirectoryID(serverUrl) else {
  325. self.stopProvidingItem(at: url)
  326. return
  327. }
  328. let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "fileName == %@ AND directoryID == %@", fileName, directoryID))
  329. if metadata != nil {
  330. // Update
  331. let uploadID = k_uploadSessionID + CCUtility.createRandomString(16)
  332. let directoryUser = CCUtility.getDirectoryActiveUser(account.user, activeUrl: account.url)
  333. let destinationDirectoryUser = "\(directoryUser!)/\(uploadID)"
  334. // copy sourceURL on directoryUser
  335. _ = self.copyFile(url.path, toPath: destinationDirectoryUser)
  336. // Prepare for send Metadata
  337. metadata!.sessionID = uploadID
  338. metadata!.session = k_upload_session
  339. metadata!.sessionTaskIdentifier = Int(k_taskIdentifierWaitStart)
  340. _ = NCManageDatabase.sharedInstance.updateMetadata(metadata!)
  341. } else {
  342. // New
  343. let directoryUser = CCUtility.getDirectoryActiveUser(account.user, activeUrl: account.url)
  344. let destinationDirectoryUser = "\(directoryUser!)/\(fileName)"
  345. _ = self.copyFile(url.path, toPath: destinationDirectoryUser)
  346. CCNetworking.shared().uploadFile(fileName, serverUrl: serverUrl, assetLocalIdentifier: nil, session: k_upload_session, taskStatus: Int(k_taskStatusResume), selector: nil, selectorPost: nil, errorCode: 0, delegate: self)
  347. }
  348. self.stopProvidingItem(at: url)
  349. }
  350. }
  351. override func stopProvidingItem(at url: URL) {
  352. // Called after the last claim to the file has been released. At this point, it is safe for the file provider to remove the content file.
  353. // Care should be taken that the corresponding placeholder file stays behind after the content file has been deleted.
  354. // Called after the last claim to the file has been released. At this point, it is safe for the file provider to remove the content file.
  355. // look up whether the file has local changes
  356. let fileHasLocalChanges = false
  357. if !fileHasLocalChanges {
  358. // remove the existing file to free up space
  359. do {
  360. _ = try FileManager.default.removeItem(at: url)
  361. } catch let error {
  362. print("error: \(error)")
  363. }
  364. // write out a placeholder to facilitate future property lookups
  365. self.providePlaceholder(at: url, completionHandler: { error in
  366. // handle any error, do any necessary cleanup
  367. })
  368. }
  369. }
  370. // MARK: - Accessing Thumbnails
  371. override func fetchThumbnails(for itemIdentifiers: [NSFileProviderItemIdentifier], requestedSize size: CGSize, perThumbnailCompletionHandler: @escaping (NSFileProviderItemIdentifier, Data?, Error?) -> Void, completionHandler: @escaping (Error?) -> Void) -> Progress {
  372. /* ONLY iOS 11*/
  373. guard #available(iOS 11, *) else {
  374. return Progress(totalUnitCount:0)
  375. }
  376. let progress = Progress(totalUnitCount: Int64(itemIdentifiers.count))
  377. var counterProgress: Int64 = 0
  378. for itemIdentifier in itemIdentifiers {
  379. if let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, itemIdentifier.rawValue)) {
  380. if (metadata.typeFile == k_metadataTypeFile_image || metadata.typeFile == k_metadataTypeFile_video) {
  381. let serverUrl = NCManageDatabase.sharedInstance.getServerUrl(metadata.directoryID)
  382. let fileName = CCUtility.returnFileNamePath(fromFileName: metadata.fileName, serverUrl: serverUrl, activeUrl: accountUrl)
  383. let fileNameLocal = metadata.fileID
  384. ocNetworking?.downloadThumbnail(withDimOfThumbnail: "m", fileName: fileName, fileNameLocal: fileNameLocal, success: {
  385. do {
  386. let url = URL.init(fileURLWithPath: "\(directoryUser)/\(itemIdentifier.rawValue).ico")
  387. let data = try Data.init(contentsOf: url)
  388. perThumbnailCompletionHandler(itemIdentifier, data, nil)
  389. } catch let error {
  390. print("error: \(error)")
  391. perThumbnailCompletionHandler(itemIdentifier, nil, NSFileProviderError(.noSuchItem))
  392. }
  393. counterProgress += 1
  394. if (counterProgress == progress.totalUnitCount) {
  395. completionHandler(nil)
  396. }
  397. }, failure: { (errorMessage, errorCode) in
  398. perThumbnailCompletionHandler(itemIdentifier, nil, NSFileProviderError(.serverUnreachable))
  399. counterProgress += 1
  400. if (counterProgress == progress.totalUnitCount) {
  401. completionHandler(nil)
  402. }
  403. })
  404. } else {
  405. counterProgress += 1
  406. if (counterProgress == progress.totalUnitCount) {
  407. completionHandler(nil)
  408. }
  409. }
  410. } else {
  411. counterProgress += 1
  412. if (counterProgress == progress.totalUnitCount) {
  413. completionHandler(nil)
  414. }
  415. }
  416. }
  417. return progress
  418. }
  419. // MARK: - Actions
  420. override func createDirectory(withName directoryName: String, inParentItemIdentifier parentItemIdentifier: NSFileProviderItemIdentifier, completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void) {
  421. /* ONLY iOS 11*/
  422. guard #available(iOS 11, *) else {
  423. return
  424. }
  425. // clear list update items
  426. listUpdateItems.removeAll()
  427. var serverUrl = ""
  428. if parentItemIdentifier == .rootContainer {
  429. serverUrl = homeServerUrl
  430. } else {
  431. guard let directoryParent = NCManageDatabase.sharedInstance.getTableDirectory(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, parentItemIdentifier.rawValue)) else {
  432. completionHandler(nil, NSFileProviderError(.noSuchItem))
  433. return
  434. }
  435. serverUrl = directoryParent.serverUrl
  436. }
  437. ocNetworking?.createFolder(directoryName, serverUrl: serverUrl, account: account, success: { (fileID, date) in
  438. let metadata = tableMetadata()
  439. metadata.account = account
  440. metadata.directory = true
  441. metadata.directoryID = NCManageDatabase.sharedInstance.getDirectoryID(serverUrl)!
  442. metadata.fileID = fileID!
  443. metadata.fileName = directoryName
  444. metadata.fileNameView = directoryName
  445. metadata.typeFile = k_metadataTypeFile_directory
  446. // METADATA
  447. guard let metadataDB = NCManageDatabase.sharedInstance.addMetadata(metadata) else {
  448. completionHandler(nil, NSFileProviderError(.noSuchItem))
  449. return
  450. }
  451. // DIRECTORY
  452. guard let _ = NCManageDatabase.sharedInstance.addDirectory(encrypted: false, favorite: false, fileID: fileID!, permissions: nil, serverUrl: serverUrl + "/" + directoryName) else {
  453. completionHandler(nil, NSFileProviderError(.noSuchItem))
  454. return
  455. }
  456. let item = FileProviderItem(metadata: metadataDB, serverUrl: serverUrl)
  457. completionHandler(item, nil)
  458. }, failure: { (errorMessage, errorCode) in
  459. completionHandler(nil, NSFileProviderError(.serverUnreachable))
  460. })
  461. }
  462. override func deleteItem(withIdentifier itemIdentifier: NSFileProviderItemIdentifier, completionHandler: @escaping (Error?) -> Void) {
  463. /* ONLY iOS 11*/
  464. guard #available(iOS 11, *) else {
  465. return
  466. }
  467. // clear list update items
  468. listUpdateItems.removeAll()
  469. guard let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, itemIdentifier.rawValue)) else {
  470. completionHandler(NSFileProviderError(.noSuchItem))
  471. return
  472. }
  473. guard let serverUrl = NCManageDatabase.sharedInstance.getServerUrl(metadata.directoryID) else {
  474. completionHandler(NSFileProviderError(.noSuchItem))
  475. return
  476. }
  477. ocNetworking?.deleteFileOrFolder(metadata.fileName, serverUrl: serverUrl, success: {
  478. let fileNamePath = directoryUser + "/" + metadata.fileID
  479. do {
  480. try FileManager.default.removeItem(atPath: fileNamePath)
  481. } catch let error {
  482. print("error: \(error)")
  483. }
  484. do {
  485. try FileManager.default.removeItem(atPath: fileNamePath + ".ico")
  486. } catch let error {
  487. print("error: \(error)")
  488. }
  489. if metadata.directory {
  490. let dirForDelete = CCUtility.stringAppendServerUrl(serverUrl, addFileName: metadata.fileName)
  491. NCManageDatabase.sharedInstance.deleteDirectoryAndSubDirectory(serverUrl: dirForDelete!)
  492. }
  493. NCManageDatabase.sharedInstance.deleteLocalFile(predicate: NSPredicate(format: "fileID == %@", metadata.fileID))
  494. NCManageDatabase.sharedInstance.deleteMetadata(predicate: NSPredicate(format: "fileID == %@", metadata.fileID), clearDateReadDirectoryID: nil)
  495. completionHandler(nil)
  496. }, failure: { (errorMessage, errorCode) in
  497. if errorCode == 404 {
  498. completionHandler(nil)
  499. } else {
  500. completionHandler(NSFileProviderError(.serverUnreachable))
  501. }
  502. })
  503. }
  504. override func renameItem(withIdentifier itemIdentifier: NSFileProviderItemIdentifier, toName itemName: String, completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void) {
  505. /* ONLY iOS 11*/
  506. guard #available(iOS 11, *) else {
  507. return
  508. }
  509. // clear list update items
  510. listUpdateItems.removeAll()
  511. guard let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, itemIdentifier.rawValue)) else {
  512. completionHandler(nil, NSFileProviderError(.noSuchItem))
  513. return
  514. }
  515. guard let serverUrl = NCManageDatabase.sharedInstance.getServerUrl(metadata.directoryID) else {
  516. completionHandler(nil, NSFileProviderError(.noSuchItem))
  517. return
  518. }
  519. guard let directoryTable = NCManageDatabase.sharedInstance.getTableDirectory(predicate: NSPredicate(format: "serverUrl = %@", serverUrl)) else {
  520. completionHandler(nil, NSFileProviderError(.noSuchItem))
  521. return
  522. }
  523. // resolve the given identifier to a file on disk
  524. guard let item = try? item(for: itemIdentifier) else {
  525. completionHandler(nil, NSFileProviderError(.noSuchItem))
  526. return
  527. }
  528. let fileName = serverUrl + "/" + item.filename
  529. let fileNameTo = serverUrl + "/" + itemName
  530. ocNetworking?.moveFileOrFolder(fileName, fileNameTo: fileNameTo, success: {
  531. metadata.fileName = itemName
  532. metadata.fileNameView = itemName
  533. guard let metadata = NCManageDatabase.sharedInstance.addMetadata(metadata) else {
  534. completionHandler(nil, NSFileProviderError(.noSuchItem))
  535. return
  536. }
  537. if metadata.directory {
  538. NCManageDatabase.sharedInstance.setDirectory(serverUrl: fileName, serverUrlTo: fileNameTo, etag: nil, fileID: nil, encrypted: directoryTable.e2eEncrypted)
  539. } else {
  540. NCManageDatabase.sharedInstance.setLocalFile(fileID: metadata.fileID, date: nil, exifDate: nil, exifLatitude: nil, exifLongitude: nil, fileName: itemName)
  541. }
  542. completionHandler(item, nil)
  543. }, failure: { (errorMessage, errorCode) in
  544. completionHandler(nil, NSFileProviderError(.serverUnreachable))
  545. })
  546. print("[LOG] rename")
  547. completionHandler(nil, nil)
  548. }
  549. override func setFavoriteRank(_ favoriteRank: NSNumber?, forItemIdentifier itemIdentifier: NSFileProviderItemIdentifier, completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void) {
  550. /* ONLY iOS 11*/
  551. guard #available(iOS 11, *) else {
  552. return
  553. }
  554. completionHandler(nil, nil)
  555. return
  556. var favorite = false
  557. guard let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, itemIdentifier.rawValue)) else {
  558. completionHandler(nil, NSFileProviderError(.noSuchItem))
  559. return
  560. }
  561. guard let serverUrl = NCManageDatabase.sharedInstance.getServerUrl(metadata.directoryID) else {
  562. completionHandler(nil, NSFileProviderError(.noSuchItem))
  563. return
  564. }
  565. // Refresh Favorite Identifier Rank
  566. listFavoriteIdentifierRank = NCManageDatabase.sharedInstance.getTableMetadatasDirectoryFavoriteIdentifierRank()
  567. if favoriteRank == nil {
  568. listFavoriteIdentifierRank.removeValue(forKey: itemIdentifier.rawValue)
  569. } else {
  570. let rank = listFavoriteIdentifierRank[itemIdentifier.rawValue]
  571. if rank == nil {
  572. listFavoriteIdentifierRank[itemIdentifier.rawValue] = favoriteRank//NSNumber(value: Int64(newRank))
  573. }
  574. favorite = true
  575. }
  576. // Call the completion handler before performing any network activity or other long-running tasks. Defer these tasks to the background
  577. let item = FileProviderItem(metadata: metadata, serverUrl: serverUrl)
  578. completionHandler(item, nil)
  579. // Change Status ? Call API Nextcloud Network
  580. if (favorite == true && metadata.favorite == false) || (favorite == false && metadata.favorite == true) {
  581. DispatchQueue(label: "com.nextcloud", qos: .background, attributes: .concurrent, autoreleaseFrequency: .inherit, target: nil).async {
  582. ocNetworking?.settingFavorite(metadata.fileName, serverUrl: serverUrl, favorite: favorite, success: {
  583. // Change DB
  584. metadata.favorite = favorite
  585. _ = NCManageDatabase.sharedInstance.addMetadata(metadata)
  586. // Refresh Favorite Identifier Rank
  587. listFavoriteIdentifierRank = NCManageDatabase.sharedInstance.getTableMetadatasDirectoryFavoriteIdentifierRank()
  588. // Refresh Item
  589. self.refreshEnumerator(identifier: itemIdentifier, serverUrl: serverUrl)
  590. }, failure: { (errorMessage, errorCode) in
  591. print("errorMessage")
  592. })
  593. }
  594. }
  595. }
  596. override func setLastUsedDate(_ lastUsedDate: Date?, forItemIdentifier itemIdentifier: NSFileProviderItemIdentifier, completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void) {
  597. print("[LOG] setLastUsedDate")
  598. completionHandler(nil, nil)
  599. }
  600. override func setTagData(_ tagData: Data?, forItemIdentifier itemIdentifier: NSFileProviderItemIdentifier, completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void) {
  601. /* ONLY iOS 11*/
  602. guard #available(iOS 11, *) else {
  603. return
  604. }
  605. // Add, Remove (nil)
  606. NCManageDatabase.sharedInstance.addTag(itemIdentifier.rawValue, tagIOS: tagData)
  607. guard let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileID = %@", account, itemIdentifier.rawValue)) else {
  608. completionHandler(nil, NSFileProviderError(.noSuchItem))
  609. return
  610. }
  611. guard let serverUrl = NCManageDatabase.sharedInstance.getServerUrl(metadata.directoryID) else {
  612. completionHandler(nil, NSFileProviderError(.noSuchItem))
  613. return
  614. }
  615. let item = FileProviderItem(metadata: metadata, serverUrl: serverUrl)
  616. completionHandler(item, nil)
  617. }
  618. /*
  619. override func trashItem(withIdentifier itemIdentifier: NSFileProviderItemIdentifier, completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void) {
  620. print("[LOG] trashitem")
  621. completionHandler(nil, nil)
  622. }
  623. override func untrashItem(withIdentifier itemIdentifier: NSFileProviderItemIdentifier, toParentItemIdentifier parentItemIdentifier: NSFileProviderItemIdentifier?, completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void) {
  624. print("[LOG] untrashitem")
  625. completionHandler(nil, nil)
  626. }
  627. */
  628. override func importDocument(at fileURL: URL, toParentItemIdentifier parentItemIdentifier: NSFileProviderItemIdentifier, completionHandler: @escaping (NSFileProviderItem?, Error?) -> Void) {
  629. /* ONLY iOS 11*/
  630. guard #available(iOS 11, *) else {
  631. return
  632. }
  633. var fileName = fileURL.lastPathComponent
  634. let fileCoordinator = NSFileCoordinator()
  635. var error: NSError?
  636. var directoryPredicate: NSPredicate
  637. var size = 0 as Double
  638. var fileNamePathUpload: URL?
  639. if parentItemIdentifier == .rootContainer {
  640. directoryPredicate = NSPredicate(format: "account = %@ AND serverUrl = %@", account, homeServerUrl)
  641. } else {
  642. directoryPredicate = NSPredicate(format: "account = %@ AND fileID = %@", account, parentItemIdentifier.rawValue)
  643. }
  644. guard let directoryParent = NCManageDatabase.sharedInstance.getTableDirectory(predicate: directoryPredicate) else {
  645. completionHandler(nil, NSFileProviderError(.noSuchItem))
  646. return
  647. }
  648. let serverUrl = directoryParent.serverUrl
  649. // --------------------------------------------- Copy file here
  650. if fileURL.startAccessingSecurityScopedResource() == false {
  651. completionHandler(nil, NSFileProviderError(.noSuchItem))
  652. return
  653. }
  654. // exists with same name ? add + 1
  655. if NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileNameView = %@ AND directoryID = %@", account, fileName, directoryParent.directoryID)) != nil {
  656. var name = NSString(string: fileName).deletingPathExtension
  657. let ext = NSString(string: fileName).pathExtension
  658. let characters = Array(name)
  659. if characters.count < 2 {
  660. fileName = name + " " + "1" + "." + ext
  661. } else {
  662. let space = characters[characters.count-2]
  663. let numChar = characters[characters.count-1]
  664. var num = Int(String(numChar))
  665. if (space == " " && num != nil) {
  666. name = String(name.dropLast())
  667. num = num! + 1
  668. fileName = name + "\(num!)" + "." + ext
  669. } else {
  670. fileName = name + " " + "1" + "." + ext
  671. }
  672. }
  673. }
  674. let fileNameLocalPath = importDocumentURL!.appendingPathComponent(fileName)
  675. fileCoordinator.coordinate(readingItemAt: fileURL, options: NSFileCoordinator.ReadingOptions.withoutChanges, error: &error) { (url) in
  676. _ = self.copyFile( url.path, toPath: fileNameLocalPath.path)
  677. }
  678. fileURL.stopAccessingSecurityScopedResource()
  679. // ------------------------------------------------------------
  680. // ---------------------------------------------- Create file 0
  681. do {
  682. let attributes = try FileManager.default.attributesOfItem(atPath: fileNameLocalPath.path)
  683. size = attributes[FileAttributeKey.size] as! Double
  684. } catch let error {
  685. print("error: \(error)")
  686. }
  687. // Upload file 0 len
  688. if (size == 0) {
  689. fileNamePathUpload = fileNameLocalPath
  690. } else {
  691. fileNamePathUpload = importDocumentURL!.appendingPathComponent(fileName+".000")
  692. do {
  693. try FileManager.default.removeItem(atPath: fileNamePathUpload!.path)
  694. } catch let error {
  695. print("error: \(error)")
  696. }
  697. FileManager.default.createFile(atPath: fileNamePathUpload!.path, contents: nil, attributes: nil)
  698. }
  699. // ------------------------------------------------------------
  700. // upload
  701. _ = ocNetworking?.uploadFileNameServerUrl(serverUrl+"/"+fileName, fileNameLocalPath: fileNamePathUpload?.path, communication: CCNetworking.shared().sharedOCCommunication(), success: { (fileID, etag, date) in
  702. let metadata = tableMetadata()
  703. metadata.account = account
  704. metadata.date = date! as NSDate
  705. metadata.directory = false
  706. metadata.directoryID = directoryParent.directoryID
  707. metadata.etag = etag!
  708. metadata.fileID = fileID!
  709. metadata.fileName = fileName
  710. metadata.fileNameView = fileName
  711. metadata.size = 0
  712. CCUtility.insertTypeFileIconName(fileName, metadata: metadata)
  713. guard let metadataDB = NCManageDatabase.sharedInstance.addMetadata(metadata) else {
  714. completionHandler(nil, NSFileProviderError(.noSuchItem))
  715. return
  716. }
  717. // Create dir <base storage directory>/<item identifier>
  718. do {
  719. try FileManager.default.createDirectory(atPath: fileProviderStorageURL!.appendingPathComponent(metadata.fileID).path, withIntermediateDirectories: true, attributes: nil)
  720. } catch let error {
  721. print("error: \(error)")
  722. }
  723. // copy <base storage directory>/<item identifier>/<item file name>
  724. _ = self.copyFile(fileNameLocalPath.path, toPath: "\(fileProviderStorageURL!.appendingPathComponent(metadata.fileID).path)/\(metadata.fileNameView)")
  725. // add item
  726. let item = FileProviderItem(metadata: metadataDB, serverUrl: serverUrl)
  727. // add queue upload if size > 0
  728. if (size > 0) {
  729. let metadataNet = CCMetadataNet()
  730. metadataNet.account = account
  731. metadataNet.assetLocalIdentifier = k_assetLocalIdentifierFileProviderStorage + fileID!
  732. metadataNet.fileName = fileName
  733. metadataNet.path = importDocumentURL!.path + "/" + metadata.fileNameView
  734. metadataNet.selector = selectorUploadFile
  735. metadataNet.selectorPost = ""
  736. metadataNet.serverUrl = serverUrl
  737. metadataNet.session = k_upload_session
  738. metadataNet.taskStatus = Int(k_taskStatusResume)
  739. _ = NCManageDatabase.sharedInstance.addQueueUpload(metadataNet: metadataNet)
  740. }
  741. if (size > 0) { // MICROSOFT 365
  742. self.refreshEnumerator(identifier: item.itemIdentifier, serverUrl: serverUrl)
  743. }
  744. completionHandler(item, nil)
  745. }, failure: { (errorMessage, errorCode) in
  746. completionHandler(nil, NSFileProviderError(.serverUnreachable))
  747. })
  748. }
  749. // --------------------------------------------------------------------------------------------
  750. // MARK: - User Function
  751. // --------------------------------------------------------------------------------------------
  752. func uploadCloud(_ fileName: String, serverUrl: String, path: String, identifier: NSFileProviderItemIdentifier) {
  753. // Upload already exists
  754. if listUpload[identifier.rawValue] != nil {
  755. return
  756. }
  757. let fileNameLocalPath = directoryUser + "/" + fileName
  758. if self.copyFile(path, toPath: fileNameLocalPath) != nil {
  759. return
  760. }
  761. let task = ocNetworking?.uploadFileNameServerUrl(serverUrl+"/"+fileName, fileNameLocalPath: fileNameLocalPath, communication: CCNetworking.shared().sharedOCCommunicationExtensionUpload(fileName), success: { (fileID, etag, date) in
  762. // Remove file on queueUpload
  763. NCManageDatabase.sharedInstance.deleteQueueUpload(path: path)
  764. // Remove from dictionary
  765. listUpload.removeValue(forKey: identifier.rawValue)
  766. // Update DB
  767. if let directoryID = NCManageDatabase.sharedInstance.getDirectoryID(serverUrl) {
  768. if let metadata = NCManageDatabase.sharedInstance.getMetadata(predicate: NSPredicate(format: "account = %@ AND fileName = %@ AND directoryID = %@", account, fileName, directoryID)) {
  769. metadata.date = date! as NSDate
  770. do {
  771. let attributes = try FileManager.default.attributesOfItem(atPath: fileNameLocalPath)
  772. metadata.size = attributes[FileAttributeKey.size] as! Double
  773. } catch let error {
  774. print("error: \(error)")
  775. }
  776. _ = NCManageDatabase.sharedInstance.addMetadata(metadata)
  777. }
  778. }
  779. // Remove file *changeDocument
  780. _ = self.deleteFile(fileNameLocalPath)
  781. // Refresh
  782. self.refreshEnumerator(identifier: identifier, serverUrl: serverUrl)
  783. }, failure: { (errorMessage, errorCode) in
  784. // Remove from dictionary
  785. listUpload.removeValue(forKey: identifier.rawValue)
  786. // Refresh
  787. self.refreshEnumerator(identifier: identifier, serverUrl: serverUrl)
  788. })
  789. if task != nil {
  790. listUpload[identifier.rawValue] = task
  791. if #available(iOSApplicationExtension 11.0, *) {
  792. NSFileProviderManager.default.register(task!, forItemWithIdentifier: identifier) { (error) in }
  793. }
  794. }
  795. }
  796. func refreshEnumerator(identifier: NSFileProviderItemIdentifier, serverUrl: String) {
  797. /* ONLY iOS 11*/
  798. guard #available(iOS 11, *) else {
  799. return
  800. }
  801. let item = try? self.item(for: identifier)
  802. if item != nil {
  803. var found = false
  804. for updateItem in listUpdateItems {
  805. if updateItem.itemIdentifier.rawValue == identifier.rawValue {
  806. found = true
  807. }
  808. }
  809. if !found {
  810. listUpdateItems.append(item!)
  811. }
  812. }
  813. if serverUrl == homeServerUrl {
  814. NSFileProviderManager.default.signalEnumerator(for: .rootContainer, completionHandler: { (error) in
  815. print("send signal rootContainer")
  816. })
  817. } else if serverUrl == "WorkingSet" {
  818. NSFileProviderManager.default.signalEnumerator(for: .workingSet, completionHandler: { (error) in
  819. print("send signal workingSet")
  820. })
  821. } else {
  822. if let directory = NCManageDatabase.sharedInstance.getTableDirectory(predicate: NSPredicate(format: "account = %@ AND serverUrl = %@", account, serverUrl)) {
  823. let itemDirectory = NSFileProviderItemIdentifier(directory.fileID)
  824. NSFileProviderManager.default.signalEnumerator(for: itemDirectory, completionHandler: { (error) in
  825. print("send signal")
  826. })
  827. }
  828. }
  829. }
  830. func copyFile(_ atPath: String, toPath: String) -> Error? {
  831. var errorResult: Error?
  832. do {
  833. try FileManager.default.removeItem(atPath: toPath)
  834. } catch let error {
  835. print("error: \(error)")
  836. }
  837. do {
  838. try FileManager.default.copyItem(atPath: atPath, toPath: toPath)
  839. } catch let error {
  840. errorResult = error
  841. }
  842. return errorResult
  843. }
  844. func deleteFile(_ atPath: String) -> Error? {
  845. var errorResult: Error?
  846. do {
  847. try FileManager.default.removeItem(atPath: atPath)
  848. } catch let error {
  849. errorResult = error
  850. }
  851. return errorResult
  852. }
  853. }