FileProvider.swift 42 KB

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