NCViewerQuickLook.swift 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. //
  2. // NCViewerQuickLook.swift
  3. // Nextcloud
  4. //
  5. // Created by Marino Faggiana on 03/05/2020.
  6. // Copyright © 2020 Marino Faggiana. All rights reserved.
  7. //
  8. import Foundation
  9. import QuickLook
  10. @objc class NCViewerQuickLook: NSObject, QLPreviewControllerDelegate, QLPreviewControllerDataSource {
  11. let previewController = QLPreviewController()
  12. var previewItems: [PreviewItem] = []
  13. var viewController: UIViewController?
  14. @objc func quickLook(url: URL, viewController: UIViewController) {
  15. self.viewController = viewController
  16. URLSession.shared.dataTask(with: url) { data, response, error in
  17. guard let data = data, error == nil else {
  18. self.presentAlertController(with: error?.localizedDescription ?? "Failed to download the pdf!!!")
  19. return
  20. }
  21. let httpURLResponse = response as? HTTPURLResponse
  22. let mimeType = httpURLResponse?.mimeType
  23. do {
  24. // rename the temporary file or save it to the document or library directory if you want to keep the file
  25. let suggestedFilename = httpURLResponse?.suggestedFilename ?? "quicklook.pdf"
  26. var previewURL = FileManager.default.temporaryDirectory.appendingPathComponent(suggestedFilename)
  27. try data.write(to: previewURL, options: .atomic) // atomic option overwrites it if needed
  28. previewURL.hasHiddenExtension = true
  29. let previewItem = PreviewItem()
  30. previewItem.previewItemURL = previewURL
  31. self.previewItems.append(previewItem)
  32. DispatchQueue.main.async {
  33. UIApplication.shared.isNetworkActivityIndicatorVisible = false
  34. self.previewController.delegate = self
  35. self.previewController.dataSource = self
  36. self.previewController.currentPreviewItemIndex = 0
  37. self.viewController?.present(self.previewController, animated: true)
  38. }
  39. } catch {
  40. print(error)
  41. return
  42. }
  43. }.resume()
  44. UIApplication.shared.isNetworkActivityIndicatorVisible = true
  45. }
  46. func previewController(_ controller: QLPreviewController, previewItemAt index: Int) -> QLPreviewItem { previewItems[index] }
  47. func numberOfPreviewItems(in controller: QLPreviewController) -> Int {
  48. previewItems.count
  49. }
  50. func presentAlertController(with message: String) {
  51. // present your alert controller from the main thread
  52. DispatchQueue.main.async {
  53. UIApplication.shared.isNetworkActivityIndicatorVisible = false
  54. let alert = UIAlertController(title: "Alert", message: message, preferredStyle: .alert)
  55. alert.addAction(.init(title: "OK", style: .default))
  56. self.viewController?.present(alert, animated: true)
  57. }
  58. }
  59. }
  60. extension URL {
  61. var hasHiddenExtension: Bool {
  62. get { (try? resourceValues(forKeys: [.hasHiddenExtensionKey]))?.hasHiddenExtension == true }
  63. set {
  64. var resourceValues = URLResourceValues()
  65. resourceValues.hasHiddenExtension = newValue
  66. try? setResourceValues(resourceValues)
  67. }
  68. }
  69. }
  70. import QuickLook
  71. class PreviewItem: NSObject, QLPreviewItem {
  72. var previewItemURL: URL?
  73. }