DocumentPickerViewController.swift 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873
  1. //
  2. // DocumentPickerViewController.swift
  3. // Picker
  4. //
  5. // Created by Marino Faggiana on 27/12/16.
  6. // Copyright © 2016 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 UIKit
  24. class DocumentPickerViewController: UIDocumentPickerExtensionViewController, CCNetworkingDelegate, OCNetworkingDelegate, BKPasscodeViewControllerDelegate {
  25. // MARK: - Properties
  26. var providerDB : providerSessionDB?
  27. lazy var fileCoordinator: NSFileCoordinator = {
  28. let fileCoordinator = NSFileCoordinator()
  29. fileCoordinator.purposeIdentifier = self.parameterProviderIdentifier
  30. return fileCoordinator
  31. }()
  32. var parameterMode: UIDocumentPickerMode?
  33. var parameterOriginalURL: URL?
  34. var parameterProviderIdentifier: String!
  35. var parameterPasscodeCorrect: Bool? = false
  36. var parameterEncrypted: Bool? = false
  37. var isCryptoCloudMode: Bool? = false
  38. var metadata: CCMetadata?
  39. var recordsTableMetadata: [TableMetadata]?
  40. var titleFolder: String?
  41. var activeAccount: String?
  42. var activeUrl: String?
  43. var activeUser: String?
  44. var activePassword: String?
  45. var directoryUser: String?
  46. var serverUrl: String?
  47. var thumbnailInLoading = [String: IndexPath]()
  48. var destinationURL: URL?
  49. var passcodeFailedAttempts: UInt = 0
  50. var passcodeLockUntilDate: Date? = nil
  51. var passcodeIsPush: Bool? = false
  52. var serverUrlPush: String?
  53. lazy var networkingOperationQueue: OperationQueue = {
  54. var queue = OperationQueue()
  55. queue.name = k_netQueueName
  56. queue.maxConcurrentOperationCount = 10
  57. return queue
  58. }()
  59. let colorEncrypted = UIColor(colorLiteralRed: 241.0/255.0, green: 90.0/255.0, blue: 34.0/255.0, alpha: 1)
  60. var hud : CCHud!
  61. // MARK: - IBOutlets
  62. @IBOutlet weak var tableView: UITableView!
  63. @IBOutlet weak var toolBar: UIToolbar!
  64. @IBOutlet weak var saveButton: UIBarButtonItem!
  65. @IBOutlet weak var encryptedButton: UIBarButtonItem!
  66. // MARK: - View Life Cycle
  67. override func viewDidLoad() {
  68. providerDB = providerSessionDB.sharedInstance
  69. if let record = CCCoreData.getActiveAccount() {
  70. activeAccount = record.account!
  71. activePassword = record.password!
  72. activeUrl = record.url!
  73. activeUser = record.user!
  74. directoryUser = CCUtility.getDirectoryActiveUser(activeUser, activeUrl: activeUrl)
  75. if (self.serverUrl == nil) {
  76. self.serverUrl = CCUtility.getHomeServerUrlActiveUrl(activeUrl)
  77. } else {
  78. self.navigationItem.title = titleFolder
  79. }
  80. } else {
  81. // Close error no account return nil
  82. let deadlineTime = DispatchTime.now() + 0.1
  83. DispatchQueue.main.asyncAfter(deadline: deadlineTime) {
  84. let alert = UIAlertController(title: NSLocalizedString("_error_", comment: ""), message: NSLocalizedString("_no_active_account_", comment: ""), preferredStyle: .alert)
  85. alert.addAction(UIAlertAction(title: NSLocalizedString("_ok_", comment: ""), style: .default) { action in
  86. self.dismissGrantingAccess(to: nil)
  87. })
  88. self.present(alert, animated: true, completion: nil)
  89. }
  90. return
  91. }
  92. // MARK: - init Object
  93. CCNetworking.shared().settingDelegate(self)
  94. hud = CCHud.init(view: self.navigationController?.view)
  95. // COLOR_SEPARATOR_TABLE
  96. self.tableView.separatorColor = UIColor(colorLiteralRed: 153.0/255.0, green: 153.0/255.0, blue: 153.0/255.0, alpha: 0.2)
  97. self.tableView.tableFooterView = UIView()
  98. // Get Crypto Cloud Mode
  99. let password = CCUtility.getKeyChainPasscode(forUUID: CCUtility.getUUID())
  100. if password?.characters.count == 0 {
  101. isCryptoCloudMode = false
  102. } else {
  103. isCryptoCloudMode = true
  104. }
  105. // Managed Crypto Cloud Mode
  106. if isCryptoCloudMode == true {
  107. // Encrypted mode
  108. encryptedButton.image = UIImage(named:image_shareExtEncrypt)?.withRenderingMode(.automatic)
  109. // Color Button
  110. if parameterEncrypted == true {
  111. encryptedButton.tintColor = colorEncrypted
  112. } else {
  113. encryptedButton.tintColor = self.view.tintColor
  114. }
  115. saveButton.tintColor = encryptedButton.tintColor
  116. } else {
  117. encryptedButton.isEnabled = false
  118. encryptedButton.tintColor = UIColor.clear
  119. }
  120. readFolder()
  121. }
  122. override func viewWillAppear(_ animated: Bool) {
  123. super.viewWillAppear(animated)
  124. // (save) mode of presentation -> pass variable for pushViewController
  125. prepareForPresentation(in: parameterMode!)
  126. // String is nil or empty
  127. guard let passcode = CCUtility.getBlockCode(), !passcode.isEmpty else {
  128. return
  129. }
  130. if CCUtility.getOnlyLockDir() == false && parameterPasscodeCorrect == false {
  131. openBKPasscode("Nextcloud")
  132. }
  133. }
  134. override func viewWillDisappear(_ animated: Bool) {
  135. // remove all networking operation
  136. networkingOperationQueue.cancelAllOperations()
  137. super.viewWillDisappear(animated)
  138. }
  139. // MARK: - Overridden Instance Methods
  140. override func prepareForPresentation(in mode: UIDocumentPickerMode) {
  141. // ------------------> Settings parameter ----------------
  142. if parameterMode == nil {
  143. parameterMode = mode
  144. }
  145. // Variable for exportToService or moveToService
  146. if parameterOriginalURL == nil && originalURL != nil {
  147. parameterOriginalURL = originalURL
  148. }
  149. if parameterProviderIdentifier == nil {
  150. parameterProviderIdentifier = providerIdentifier
  151. }
  152. // -------------------------------------------------------
  153. switch mode {
  154. case .exportToService:
  155. print("Document Picker Mode : exportToService")
  156. saveButton.title = NSLocalizedString("_save_document_picker_", comment: "") // Save in this position
  157. case .moveToService:
  158. //Show confirmation button
  159. print("Document Picker Mode : moveToService")
  160. saveButton.title = NSLocalizedString("_save_document_picker_", comment: "") // Save in this position
  161. case .open:
  162. print("Document Picker Mode : open")
  163. saveButton.tintColor = UIColor.clear
  164. encryptedButton.tintColor = UIColor.clear
  165. case .import:
  166. print("Document Picker Mode : import")
  167. saveButton.tintColor = UIColor.clear
  168. encryptedButton.tintColor = UIColor.clear
  169. }
  170. }
  171. // MARK: - Read folder
  172. func readFolder() {
  173. let metadataNet = CCMetadataNet.init(account: activeAccount)!
  174. metadataNet.action = actionReadFolder
  175. metadataNet.serverUrl = self.serverUrl
  176. metadataNet.selector = selectorReadFolder
  177. let ocNetworking : OCnetworking = OCnetworking.init(delegate: self, metadataNet: metadataNet, withUser: activeUser, withPassword: activePassword, withUrl: activeUrl, activityIndicator: false, isCryptoCloudMode: isCryptoCloudMode!)
  178. networkingOperationQueue.addOperation(ocNetworking)
  179. hud.visibleIndeterminateHud()
  180. }
  181. func readFolderFailure(_ metadataNet: CCMetadataNet!, message: String!, errorCode: Int) {
  182. hud.hideHud()
  183. let alert = UIAlertController(title: NSLocalizedString("_error_", comment: ""), message: message, preferredStyle: .alert)
  184. alert.addAction(UIAlertAction(title: NSLocalizedString("_ok_", comment: ""), style: .default) { action in
  185. self.dismissGrantingAccess(to: nil)
  186. })
  187. self.present(alert, animated: true, completion: nil)
  188. }
  189. func readFolderSuccess(_ metadataNet: CCMetadataNet!, permissions: String!, rev: String!, metadatas: [Any]!) {
  190. // remove all record
  191. let predicate = NSPredicate(format: "(account == '\(activeAccount!)') AND (directoryID == '\(metadataNet.directoryID!)') AND ((session == NULL) OR (session == ''))")
  192. CCCoreData.deleteMetadata(with: predicate)
  193. for metadata in metadatas as! [CCMetadata] {
  194. // do not insert crypto file
  195. if CCUtility.isCryptoString(metadata.fileName) {
  196. continue
  197. }
  198. // Only Directory ?
  199. if (parameterMode == .moveToService || parameterMode == .exportToService) && metadata.directory == false {
  200. continue
  201. }
  202. // plist + crypto = completed ?
  203. if CCUtility.isCryptoPlistString(metadata.fileName) && metadata.directory == false {
  204. var isCryptoComplete = false
  205. for completeMetadata in metadatas as! [CCMetadata] {
  206. if completeMetadata.fileName == CCUtility.trasformedFileNamePlist(inCrypto: metadata.fileName) {
  207. isCryptoComplete = true
  208. }
  209. }
  210. if isCryptoComplete == false {
  211. continue
  212. }
  213. }
  214. // Add record
  215. CCCoreData.add(metadata, activeAccount: activeAccount, activeUrl: activeUrl, context: nil)
  216. // if plist do not exists, download it
  217. if CCUtility.isCryptoPlistString(metadata.fileName) && FileManager.default.fileExists(atPath: "\(directoryUser!)/\(metadata.fileName!)") == false {
  218. let metadataNet = CCMetadataNet.init(account: activeAccount)!
  219. metadataNet.action = actionDownloadFile
  220. metadataNet.metadata = metadata
  221. metadataNet.downloadData = false
  222. metadataNet.downloadPlist = true
  223. metadataNet.selector = selectorLoadPlist
  224. metadataNet.serverUrl = self.serverUrl
  225. metadataNet.session = k_download_session_foreground
  226. metadataNet.taskStatus = Int(k_taskStatusResume)
  227. let ocNetworking : OCnetworking = OCnetworking.init(delegate: self, metadataNet: metadataNet, withUser: activeUser, withPassword: activePassword, withUrl: activeUrl, activityIndicator: false, isCryptoCloudMode: isCryptoCloudMode!)
  228. networkingOperationQueue.addOperation(ocNetworking)
  229. }
  230. }
  231. // Get Datasource
  232. recordsTableMetadata = CCCoreData.getTableMetadata(with: NSPredicate(format: "(account == '\(activeAccount!)') AND (directoryID == '\(metadataNet.directoryID!)')"), fieldOrder: "fileName", ascending: true) as? [TableMetadata]
  233. tableView.reloadData()
  234. hud.hideHud()
  235. }
  236. // MARK: - Download Thumbnail
  237. func downloadThumbnailFailure(_ metadataNet: CCMetadataNet!, message: String!, errorCode: Int) {
  238. NSLog("[LOG] Thumbnail Error \(metadataNet.fileName) \(message) (error \(errorCode))");
  239. }
  240. func downloadThumbnailSuccess(_ metadataNet: CCMetadataNet!) {
  241. if let indexPath = thumbnailInLoading[metadataNet.fileID] {
  242. let path = "\(directoryUser!)/\(metadataNet.fileID!).ico"
  243. if FileManager.default.fileExists(atPath: path) {
  244. if let cell = tableView.cellForRow(at: indexPath) as? recordMetadataCell {
  245. cell.fileImageView.image = UIImage(contentsOfFile: path)
  246. }
  247. }
  248. }
  249. }
  250. func downloadThumbnail(_ metadata : CCMetadata) {
  251. let metadataNet = CCMetadataNet.init(account: activeAccount)!
  252. metadataNet.action = actionDownloadThumbnail
  253. metadataNet.fileID = metadata.fileID
  254. metadataNet.fileName = CCUtility.returnFileNamePath(fromFileName: metadata.fileName, serverUrl: self.serverUrl, activeUrl: activeUrl)
  255. metadataNet.fileNameLocal = metadata.fileID;
  256. metadataNet.fileNamePrint = metadata.fileNamePrint;
  257. metadataNet.options = "m";
  258. metadataNet.selector = selectorDownloadThumbnail;
  259. metadataNet.serverUrl = self.serverUrl
  260. let ocNetworking : OCnetworking = OCnetworking.init(delegate: self, metadataNet: metadataNet, withUser: activeUser, withPassword: activePassword, withUrl: activeUrl, activityIndicator: false, isCryptoCloudMode: isCryptoCloudMode!)
  261. networkingOperationQueue.addOperation(ocNetworking)
  262. }
  263. // MARK: - Download / Upload
  264. func progressTask(_ fileID: String!, serverUrl: String!, cryptated: Bool, progress: Float) {
  265. hud.progress(progress)
  266. }
  267. func cancelTransfer() {
  268. networkingOperationQueue.cancelAllOperations()
  269. }
  270. // MARK: - Download
  271. func downloadFileFailure(_ fileID: String!, serverUrl: String!, selector: String!, message: String!, errorCode: Int) {
  272. hud.hideHud()
  273. if selector == selectorLoadFileView && errorCode != -999 {
  274. let alert = UIAlertController(title: NSLocalizedString("_error_", comment: ""), message: message, preferredStyle: .alert)
  275. alert.addAction(UIAlertAction(title: NSLocalizedString("_ok_", comment: ""), style: .default) { action in
  276. NSLog("[LOG] Download Error \(fileID) \(message) (error \(errorCode))");
  277. })
  278. self.present(alert, animated: true, completion: nil)
  279. }
  280. }
  281. func downloadFileSuccess(_ fileID: String!, serverUrl: String!, selector: String!, selectorPost: String!) {
  282. hud.hideHud()
  283. let metadata = CCCoreData.getMetadataWithPreficate(NSPredicate(format: "(account == '\(activeAccount!)') AND (fileID == '\(fileID!)')"), context: nil)
  284. switch selector {
  285. case selectorLoadFileView :
  286. let sourceUrl = URL(string: "file://\(directoryUser!)/\(fileID!)".addingPercentEncoding(withAllowedCharacters: .urlPathAllowed)!)!
  287. let destinationUrl : URL! = appGroupContainerURL()?.appendingPathComponent(metadata!.fileNamePrint!)
  288. // Destination Provider
  289. do {
  290. try FileManager.default.removeItem(at: destinationUrl)
  291. } catch _ {
  292. print("file do not exists")
  293. }
  294. do {
  295. try FileManager.default.copyItem(at: sourceUrl, to: destinationUrl)
  296. } catch let error as NSError {
  297. print(error)
  298. }
  299. // Dismiss
  300. self.dismissGrantingAccess(to: destinationUrl)
  301. case selectorLoadPlist :
  302. CCCoreData.downloadFilePlist(metadata, activeAccount: activeAccount, activeUrl: activeUrl,directoryUser: directoryUser)
  303. tableView.reloadData()
  304. default :
  305. print("selector : \(selector!)")
  306. tableView.reloadData()
  307. }
  308. }
  309. // MARK: - Upload
  310. func uploadFileFailure(_ metadataNet: CCMetadataNet, fileID: String, serverUrl: String, selector: String, message: String, errorCode: NSInteger){
  311. hud.hideHud()
  312. // remove file
  313. CCCoreData.deleteMetadata(with: NSPredicate(format: "(account == '\(activeAccount!)') AND (fileID == '\(fileID)')"))
  314. if errorCode != -999 {
  315. let alert = UIAlertController(title: NSLocalizedString("_error_", comment: ""), message: message, preferredStyle: .alert)
  316. alert.addAction(UIAlertAction(title: NSLocalizedString("_ok_", comment: ""), style: .default) { action in
  317. //self.dismissGrantingAccess(to: nil)
  318. NSLog("[LOG] Download Error \(fileID) \(message) (error \(errorCode))");
  319. })
  320. self.present(alert, animated: true, completion: nil)
  321. }
  322. }
  323. func uploadFileSuccess(_ metadataNet: CCMetadataNet, fileID: String, serverUrl: String, selector: String, selectorPost: String) {
  324. hud.hideHud()
  325. dismissGrantingAccess(to: self.destinationURL)
  326. }
  327. }
  328. // MARK: - IBActions
  329. extension DocumentPickerViewController {
  330. @IBAction func encryptedButtonTapped(_ sender: AnyObject) {
  331. parameterEncrypted = !parameterEncrypted!
  332. if parameterEncrypted == true {
  333. encryptedButton.tintColor = colorEncrypted
  334. } else {
  335. encryptedButton.tintColor = self.view.tintColor
  336. }
  337. saveButton.tintColor = encryptedButton.tintColor
  338. }
  339. @IBAction func saveButtonTapped(_ sender: AnyObject) {
  340. guard let sourceURL = parameterOriginalURL else {
  341. return
  342. }
  343. switch parameterMode! {
  344. case .moveToService, .exportToService:
  345. let fileName = sourceURL.lastPathComponent
  346. let destinationURLDirectoryUser = URL(string: "file://\(directoryUser!)/\(fileName)".addingPercentEncoding(withAllowedCharacters: .urlPathAllowed)!)!
  347. //let fileSize = (try! FileManager.default.attributesOfItem(atPath: sourceURL.path)[FileAttributeKey.size] as! NSNumber).uint64Value
  348. self.destinationURL = appGroupContainerURL()?.appendingPathComponent(fileName)
  349. // copy sourceURL on directoryUser
  350. do {
  351. try FileManager.default.removeItem(at: destinationURLDirectoryUser)
  352. } catch _ {
  353. print("file do not exists")
  354. }
  355. do {
  356. try FileManager.default.copyItem(at: sourceURL, to: destinationURLDirectoryUser)
  357. } catch _ {
  358. print("file do not exists")
  359. return
  360. }
  361. fileCoordinator.coordinate(readingItemAt: sourceURL, options: .withoutChanges, error: nil, byAccessor: { [weak self] newURL in
  362. do {
  363. try FileManager.default.removeItem(at: (self?.destinationURL)!)
  364. } catch _ {
  365. print("file do not exists")
  366. }
  367. do {
  368. try FileManager.default.copyItem(at: sourceURL, to: (self?.destinationURL)!)
  369. // Upload fileName to Cloud
  370. let metadataNet = CCMetadataNet.init(account: self!.activeAccount)!
  371. metadataNet.action = actionUploadFile
  372. metadataNet.cryptated = self!.parameterEncrypted!
  373. metadataNet.fileName = fileName
  374. metadataNet.fileNamePrint = fileName
  375. metadataNet.serverUrl = self!.serverUrl
  376. metadataNet.session = k_upload_session_foreground
  377. metadataNet.taskStatus = Int(k_taskStatusResume)
  378. let ocNetworking : OCnetworking = OCnetworking.init(delegate: self!, metadataNet: metadataNet, withUser: self!.activeUser, withPassword: self!.activePassword, withUrl: self!.activeUrl, activityIndicator: false, isCryptoCloudMode: self!.isCryptoCloudMode!)
  379. self!.networkingOperationQueue.addOperation(ocNetworking)
  380. self!.hud.visibleHudTitle(NSLocalizedString("_uploading_", comment: ""), mode: MBProgressHUDMode.determinateHorizontalBar, color: self!.navigationController?.view.tintColor)
  381. self!.hud.addButtonCancel(withTarget: self, selector: "cancelTransfer")
  382. } catch _ {
  383. print("error copying file")
  384. }
  385. })
  386. default:
  387. dismiss(animated: true, completion: nil)
  388. }
  389. }
  390. func appGroupContainerURL() -> URL? {
  391. guard let groupURL = FileManager.default
  392. .containerURL(forSecurityApplicationGroupIdentifier: k_capabilitiesGroups) else {
  393. return nil
  394. }
  395. let storagePathUrl = groupURL.appendingPathComponent("File Provider Storage")
  396. let storagePath = storagePathUrl.path
  397. if !FileManager.default.fileExists(atPath: storagePath) {
  398. do {
  399. try FileManager.default.createDirectory(atPath: storagePath, withIntermediateDirectories: false, attributes: nil)
  400. } catch let error {
  401. print("error creating filepath: \(error)")
  402. return nil
  403. }
  404. }
  405. return storagePathUrl
  406. }
  407. // MARK: - Passcode
  408. func openBKPasscode(_ title : String?) {
  409. let viewController = CCBKPasscode.init()
  410. viewController.delegate = self
  411. viewController.type = BKPasscodeViewControllerCheckPasscodeType
  412. viewController.inputViewTitlePassword = true
  413. if CCUtility.getSimplyBlockCode() {
  414. viewController.passcodeStyle = BKPasscodeInputViewNumericPasscodeStyle
  415. viewController.passcodeInputView.maximumLength = 6
  416. } else {
  417. viewController.passcodeStyle = BKPasscodeInputViewNormalPasscodeStyle
  418. viewController.passcodeInputView.maximumLength = 64
  419. }
  420. let touchIDManager = BKTouchIDManager.init(keychainServiceName: BKPasscodeKeychainServiceName)
  421. touchIDManager?.promptText = NSLocalizedString("_scan_fingerprint_", comment: "")
  422. viewController.touchIDManager = touchIDManager
  423. viewController.title = title
  424. viewController.navigationItem.leftBarButtonItem = UIBarButtonItem.init(barButtonSystemItem: UIBarButtonSystemItem.cancel, target: self, action: #selector(passcodeViewCloseButtonPressed(sender:)))
  425. viewController.navigationItem.leftBarButtonItem?.tintColor = colorEncrypted
  426. let navController = UINavigationController.init(rootViewController: viewController)
  427. self.present(navController, animated: true, completion: nil)
  428. }
  429. func passcodeViewControllerNumber(ofFailedAttempts aViewController: BKPasscodeViewController!) -> UInt {
  430. return passcodeFailedAttempts
  431. }
  432. func passcodeViewControllerLock(untilDate aViewController: BKPasscodeViewController!) -> Date! {
  433. return passcodeLockUntilDate
  434. }
  435. func passcodeViewControllerDidFailAttempt(_ aViewController: BKPasscodeViewController!) {
  436. passcodeFailedAttempts += 1
  437. if passcodeFailedAttempts > 5 {
  438. var timeInterval: TimeInterval = 60
  439. if passcodeFailedAttempts > 6 {
  440. let multiplier = passcodeFailedAttempts - 6
  441. timeInterval = TimeInterval(5 * 60 * multiplier)
  442. if timeInterval > 3600 * 24 {
  443. timeInterval = 3600 * 24
  444. }
  445. }
  446. passcodeLockUntilDate = Date.init(timeIntervalSinceNow: timeInterval)
  447. }
  448. }
  449. func passcodeViewController(_ aViewController: BKPasscodeViewController!, authenticatePasscode aPasscode: String!, resultHandler aResultHandler: ((Bool) -> Void)!) {
  450. if aPasscode == CCUtility.getBlockCode() {
  451. passcodeLockUntilDate = nil
  452. passcodeFailedAttempts = 0
  453. aResultHandler(true)
  454. } else {
  455. aResultHandler(false)
  456. }
  457. }
  458. public func passcodeViewController(_ aViewController: BKPasscodeViewController!, didFinishWithPasscode aPasscode: String!) {
  459. parameterPasscodeCorrect = true
  460. aViewController.dismiss(animated: true, completion: nil)
  461. if self.passcodeIsPush == true {
  462. performSegue()
  463. }
  464. }
  465. func passcodeViewCloseButtonPressed(sender :Any) {
  466. dismiss(animated: true, completion: {
  467. if self.passcodeIsPush == false {
  468. self.dismissGrantingAccess(to: nil)
  469. }
  470. })
  471. }
  472. }
  473. // MARK: - UITableViewDelegate
  474. extension DocumentPickerViewController: UITableViewDelegate {
  475. func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
  476. return 50
  477. }
  478. }
  479. // MARK: - UITableViewDataSource
  480. extension DocumentPickerViewController: UITableViewDataSource {
  481. func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
  482. return recordsTableMetadata?.count ?? 0
  483. }
  484. func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
  485. let cell = tableView.dequeueReusableCell(withIdentifier: "Cell", for: indexPath) as! recordMetadataCell
  486. cell.separatorInset = UIEdgeInsetsMake(0, 60, 0, 0)
  487. let recordTableMetadata = recordsTableMetadata?[(indexPath as NSIndexPath).row]
  488. let metadata = CCCoreData.insertEntity(in: recordTableMetadata)!
  489. // File Image View
  490. let filePath = "\(directoryUser!)/\(metadata.fileID!).ico"
  491. if FileManager.default.fileExists(atPath: filePath) {
  492. cell.fileImageView.image = UIImage(contentsOfFile: filePath)
  493. } else {
  494. cell.fileImageView.image = UIImage(named: metadata.iconName!)
  495. if metadata.thumbnailExists && metadata.directory == false {
  496. downloadThumbnail(metadata)
  497. thumbnailInLoading[metadata.fileID] = indexPath
  498. }
  499. }
  500. // File Name
  501. cell.fileName.text = metadata.fileNamePrint
  502. // Status Image View
  503. let lockServerUrl = CCUtility.stringAppendServerUrl(self.serverUrl!, addServerUrl: metadata.fileNameData)
  504. var passcode: String? = CCUtility.getBlockCode()
  505. if passcode == nil {
  506. passcode = ""
  507. }
  508. if metadata.directory && CCCoreData.isDirectoryLock(lockServerUrl, activeAccount: activeAccount) && (passcode?.characters.count)! > 0 {
  509. cell.StatusImageView.image = UIImage(named: image_passcode)
  510. } else {
  511. cell.StatusImageView.image = nil
  512. }
  513. return cell
  514. }
  515. func tableView(_ tableView: UITableView, didSelectRowAt indexPath: IndexPath) {
  516. let recordTableMetadata = recordsTableMetadata?[(indexPath as NSIndexPath).row]
  517. tableView.deselectRow(at: indexPath, animated: true)
  518. // Error passcode ?
  519. if (recordTableMetadata?.errorPasscode == 1) {
  520. return
  521. }
  522. self.metadata = CCCoreData.insertEntity(in: recordTableMetadata)!
  523. if recordTableMetadata!.directory == 0 {
  524. if FileManager.default.fileExists(atPath: "\(directoryUser!)/\(self.metadata?.fileID!)") {
  525. downloadFileSuccess(self.metadata?.fileID!, serverUrl: self.serverUrl!, selector: selectorLoadFileView, selectorPost: nil)
  526. } else {
  527. // Download file
  528. let metadataNet = CCMetadataNet.init(account: activeAccount)!
  529. metadataNet.action = actionDownloadFile
  530. metadataNet.downloadData = true
  531. metadataNet.downloadPlist = false
  532. metadataNet.metadata = metadata
  533. metadataNet.selector = selectorLoadFileView
  534. metadataNet.serverUrl = self.serverUrl
  535. metadataNet.session = k_download_session_foreground
  536. metadataNet.taskStatus = Int(k_taskStatusResume)
  537. let ocNetworking : OCnetworking = OCnetworking.init(delegate: self, metadataNet: metadataNet, withUser: activeUser, withPassword: activePassword, withUrl: activeUrl, activityIndicator: false, isCryptoCloudMode: self.isCryptoCloudMode!)
  538. networkingOperationQueue.addOperation(ocNetworking)
  539. hud.visibleHudTitle(NSLocalizedString("_loading_", comment: ""), mode: MBProgressHUDMode.determinateHorizontalBar, color: self.navigationController?.view.tintColor)
  540. hud.addButtonCancel(withTarget: self, selector: "cancelTransfer")
  541. }
  542. } else {
  543. var dir : String! = self.metadata?.fileName!
  544. if (self.metadata?.cryptated)! {
  545. dir = CCUtility.trasformedFileNamePlist(inCrypto: self.metadata?.fileName)
  546. }
  547. serverUrlPush = CCUtility.stringAppendServerUrl(self.serverUrl!, addServerUrl: dir)
  548. var passcode: String? = CCUtility.getBlockCode()
  549. if passcode == nil {
  550. passcode = ""
  551. }
  552. if CCCoreData.isDirectoryLock(serverUrlPush, activeAccount: activeAccount) && (passcode?.characters.count)! > 0 {
  553. self.passcodeIsPush = true
  554. openBKPasscode(self.metadata?.fileNamePrint)
  555. } else {
  556. performSegue()
  557. }
  558. }
  559. }
  560. func performSegue() {
  561. let nextViewController = self.storyboard?.instantiateViewController(withIdentifier: "DocumentPickerViewController") as! DocumentPickerViewController
  562. nextViewController.parameterMode = parameterMode
  563. nextViewController.parameterOriginalURL = parameterOriginalURL
  564. nextViewController.parameterProviderIdentifier = parameterProviderIdentifier
  565. nextViewController.parameterPasscodeCorrect = parameterPasscodeCorrect
  566. nextViewController.parameterEncrypted = parameterEncrypted
  567. nextViewController.serverUrl = serverUrlPush
  568. nextViewController.titleFolder = self.metadata?.fileNamePrint
  569. self.navigationController?.pushViewController(nextViewController, animated: true)
  570. }
  571. }
  572. // MARK: - Class UITableViewCell
  573. class recordMetadataCell: UITableViewCell {
  574. @IBOutlet weak var fileImageView: UIImageView!
  575. @IBOutlet weak var StatusImageView: UIImageView!
  576. @IBOutlet weak var fileName : UILabel!
  577. }
  578. // MARK: - Class providerSession
  579. class providerSessionDB {
  580. class var sharedInstance : providerSessionDB {
  581. struct Static {
  582. static let instance = providerSessionDB()
  583. }
  584. return Static.instance
  585. }
  586. private init() {
  587. let dirGroup = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: k_capabilitiesGroups)
  588. let pathDB = dirGroup?.appendingPathComponent(appDatabase).appendingPathComponent("cryptocloud")
  589. MagicalRecord.setupCoreDataStackWithAutoMigratingSqliteStore(at: pathDB!)
  590. MagicalRecord.setLoggingLevel(MagicalRecordLoggingLevel.off)
  591. }
  592. }