KMURLCreatePDFWindowController.swift 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387
  1. //
  2. // KMURLCreatePDFWindowController.swift
  3. // PDF Reader Pro
  4. //
  5. // Created by Niehaoyu on 2024/10/9.
  6. //
  7. import Cocoa
  8. import KMComponentLibrary
  9. let kUrlToPDFFolderPath = (try? FileManager.default.url(for: .applicationSupportDirectory, in: .userDomainMask, appropriateFor: nil, create: true).appendingPathComponent(Bundle.main.bundleIdentifier ?? "").appendingPathComponent("WebPage"))?.path ?? ""
  10. @objc protocol KMURLCreatePDFWindowControllerDelegate: AnyObject {
  11. //选择打开文件
  12. @objc optional func urlCreateWCDidChooseFileToOpen(_ vc: KMURLCreatePDFWindowController, _ fileURL: URL)
  13. }
  14. class KMURLCreatePDFWindowController: NSWindowController {
  15. @IBOutlet var contendBox: NSBox!
  16. @IBOutlet var titleLabel: NSTextField!
  17. @IBOutlet var urlRadio: ComponentRadio!
  18. @IBOutlet var fileRadio: ComponentRadio!
  19. @IBOutlet var inputView: ComponentInput!
  20. @IBOutlet var fileInputBGView: NSView!
  21. @IBOutlet var fileInputView: ComponentInput!
  22. @IBOutlet var fileInputAddon: ComponentInputAddon!
  23. @IBOutlet var cancelButton: ComponentButton!
  24. @IBOutlet var openButton: ComponentButton!
  25. private var _isFromURL: Bool = true
  26. private var parentWindow: NSWindow?
  27. private var handler: ((String?) -> Void)!
  28. private var posterMaskView: KMBookletMaskView?
  29. var choosedFilePath: String?
  30. var filePath: String?
  31. weak open var delegate: KMURLCreatePDFWindowControllerDelegate?
  32. override func windowDidLoad() {
  33. super.windowDidLoad()
  34. // Implement this method to handle any initialization after your window controller's window has been loaded from its nib file.
  35. setUpProperty()
  36. reloadData()
  37. }
  38. var isFromURL: Bool {
  39. get {
  40. return _isFromURL
  41. }
  42. set {
  43. _isFromURL = newValue
  44. reloadData()
  45. }
  46. }
  47. func setUpProperty() {
  48. titleLabel.stringValue = NSLocalizedString("New From Web Page", comment: "")
  49. titleLabel.textColor = ComponentLibrary.shared.getComponentColorFromKey("colorText/1")
  50. titleLabel.font = ComponentLibrary.shared.getFontFromKey("mac/body-m-medium")
  51. urlRadio.properties = ComponentCheckBoxProperty(size: .s,
  52. state: .normal,
  53. text: NSLocalizedString("URL", comment: ""),
  54. checkboxType: .normal)
  55. urlRadio.setTarget(self, action: #selector(typeRadioChanged(_:)))
  56. fileRadio.properties = ComponentCheckBoxProperty(size: .s,
  57. state: .normal,
  58. text: NSLocalizedString("File", comment: ""),
  59. checkboxType: .normal)
  60. fileRadio.setTarget(self, action: #selector(typeRadioChanged(_:)))
  61. inputView.properties = ComponentInputProperty(size: .s,
  62. state: .normal,
  63. isError: false,
  64. isDisabled: false,
  65. placeholder: NSLocalizedString("https://pdfreaderpro.com", comment: ""))
  66. inputView.delegate = self
  67. fileInputView.properties = ComponentInputProperty(size: .s,
  68. state: .normal,
  69. isError: false,
  70. isDisabled: false,
  71. placeholder: NSLocalizedString("Select File...", comment: ""),
  72. creatable: false)
  73. fileInputView.properties.propertyInfo.cornerRadius_topRight = 0
  74. fileInputView.properties.propertyInfo.cornerRadius_bottomRight = 0
  75. fileInputAddon.properties = ComponentInputAddonProperty(size: .s,
  76. state: .normal,
  77. addOnBefore: false,
  78. onlyRead: false,
  79. addonType: .imageWithColor,
  80. iconImage: NSImage(named: "file_icon"))
  81. fileInputAddon.setTarget(self, action: #selector(chooseURLAction(_ :)))
  82. cancelButton.properties = ComponentButtonProperty(type: .default_tertiary,
  83. size: .s,
  84. state: .normal,
  85. buttonText: NSLocalizedString("Cancel", comment: ""))
  86. cancelButton.setTarget(self, action: #selector(cancelButtonClicked(_ :)))
  87. openButton.properties = ComponentButtonProperty(type: .primary,
  88. size: .s,
  89. state: .normal,
  90. buttonText: NSLocalizedString("Open", comment: ""))
  91. openButton.setTarget(self, action: #selector(openButtonClicked(_ :)))
  92. }
  93. func reloadData() {
  94. urlRadio.properties.checkboxType = isFromURL ? .selected : .normal
  95. fileRadio.properties.checkboxType = isFromURL ? .normal : .selected
  96. fileInputBGView.isHidden = isFromURL ? true : false
  97. inputView.isHidden = isFromURL ? false : true
  98. cancelButton.properties.state = .normal
  99. openButton.properties.state = .normal
  100. if let filePath = choosedFilePath {
  101. fileInputView.properties.text = filePath
  102. } else {
  103. fileInputView.properties.text = ""
  104. }
  105. if isFromURL {
  106. if inputView.properties.text.count > 0 {
  107. openButton.properties.isDisabled = false
  108. } else {
  109. openButton.properties.isDisabled = true
  110. }
  111. } else {
  112. if fileInputView.properties.text.count > 0 {
  113. openButton.properties.isDisabled = false
  114. } else {
  115. openButton.properties.isDisabled = true
  116. }
  117. }
  118. inputView.reloadData()
  119. fileInputView.reloadData()
  120. fileInputAddon.reloadData()
  121. urlRadio.reloadData()
  122. fileRadio.reloadData()
  123. cancelButton.reloadData()
  124. openButton.reloadData()
  125. fileInputView.reloadData()
  126. }
  127. //MARK: - Action
  128. @objc func typeRadioChanged(_ sender: NSView) {
  129. if sender == fileRadio {
  130. isFromURL = false
  131. } else if sender == self.urlRadio {
  132. isFromURL = true
  133. }
  134. }
  135. @objc func cancelButtonClicked(_ sender: NSView) {
  136. choosedFilePath = nil
  137. inputView.properties.text = ""
  138. hideWaitting()
  139. reloadData()
  140. KMURLToPDF.shareInstance.stopLoading()
  141. handler?(nil)
  142. parentWindow?.endSheet(self.window!)
  143. }
  144. @objc func openButtonClicked(_ sender: NSView) {
  145. convertInfoToPDF()
  146. }
  147. @objc func chooseURLAction(_ sender: NSView) {
  148. let openPanel = NSOpenPanel()
  149. openPanel.allowedFileTypes = ["html", "HTML"]
  150. openPanel.allowsMultipleSelection = false
  151. openPanel.beginSheetModal(for: self.window!) { [weak self] result in
  152. if result == NSApplication.ModalResponse.OK {
  153. if let url = openPanel.url {
  154. self?.choosedFilePath = url.path
  155. self?.reloadData()
  156. }
  157. }
  158. }
  159. }
  160. func convertInfoToPDF() {
  161. if !FileManager.default.fileExists(atPath: kUrlToPDFFolderPath) {
  162. try? FileManager.default.createDirectory(atPath: kUrlToPDFFolderPath, withIntermediateDirectories: false, attributes: nil)
  163. }
  164. var url: URL?
  165. if isFromURL {
  166. var urlString = inputView.properties.text
  167. var tUrl = URL(string: urlString)
  168. if tUrl?.scheme?.count ?? 0 < 1 {
  169. if urlString.hasSuffix(".com") == false {
  170. urlString = urlString + ".com"
  171. }
  172. tUrl = URL(string: "http://\(urlString)")
  173. }
  174. url = tUrl
  175. } else {
  176. if let choosedFilePath = choosedFilePath {
  177. url = URL(fileURLWithPath: choosedFilePath)
  178. }
  179. }
  180. if let url = url {
  181. showWaitting()
  182. let outPutPath = kUrlToPDFFolderPath
  183. if isFromURL {
  184. KMURLToPDF.shareInstance.convertLinkString(string: url.absoluteString, toPath: outPutPath) {[weak self] url, error in
  185. self?.hideWaitting()
  186. if error == nil && url?.path.isEmpty == false {
  187. self?.choosedFilePath = nil
  188. self?.inputView.properties.text = ""
  189. self?.reloadData()
  190. self?.handler?(url?.path)
  191. if let window = self?.window {
  192. self?.parentWindow?.endSheet(window)
  193. }
  194. } else {
  195. if let window = self?.window {
  196. let alert = NSAlert()
  197. alert.alertStyle = .critical
  198. alert.messageText = NSLocalizedString("Conversion Failed", comment: "")
  199. alert.beginSheetModal(for: window) { [weak self] result in
  200. }
  201. }
  202. }
  203. }
  204. } else {
  205. KMURLToPDF.shareInstance.convertHtmlFileToPDF(fromPath: url.path, toPath: outPutPath) {[weak self] url, error in
  206. self?.hideWaitting()
  207. if error == nil && url?.path.isEmpty == false {
  208. self?.choosedFilePath = nil
  209. self?.inputView.properties.text = ""
  210. self?.reloadData()
  211. self?.handler?(url?.path)
  212. if let window = self?.window {
  213. self?.parentWindow?.endSheet(window)
  214. }
  215. } else {
  216. if let window = self?.window {
  217. let alert = NSAlert()
  218. alert.alertStyle = .critical
  219. alert.messageText = NSLocalizedString("Conversion Failed", comment: "")
  220. alert.beginSheetModal(for: window) { [weak self] result in
  221. }
  222. }
  223. }
  224. }
  225. }
  226. }
  227. }
  228. //MARK: - WaitingView
  229. func showWaitting() {
  230. if posterMaskView == nil {
  231. posterMaskView = KMBookletMaskView(frame: NSMakeRect(0, 0, window?.frame.size.width ?? 0, window?.frame.size.height ?? 0))
  232. }
  233. contendBox.contentView?.addSubview(posterMaskView!, positioned: .below, relativeTo: self.cancelButton)
  234. }
  235. func hideWaitting() {
  236. posterMaskView?.removeFromSuperview()
  237. }
  238. //MARK: - public
  239. func own_beginSheetModal(for window: NSWindow?, completionHandler handler: ((String?) -> Void)?) {
  240. self.window?.makeFirstResponder(nil)
  241. if window != nil {
  242. parentWindow = window
  243. window!.beginSheet(self.window!) { ModalResponse in
  244. self.handler?(nil)
  245. }
  246. choosedFilePath = nil
  247. self.handler = handler
  248. hideWaitting()
  249. reloadData()
  250. }
  251. }
  252. override func becomeFirstResponder() -> Bool {
  253. return true
  254. }
  255. func formatFloat(_ f: Float) -> String {
  256. if f.truncatingRemainder(dividingBy: 1) == 0 {
  257. return String(format: "%.0f", f)
  258. } else if (f * 10).truncatingRemainder(dividingBy: 1) == 0 {
  259. return String(format: "%.1f", f)
  260. } else {
  261. return String(format: "%.2f", f)
  262. }
  263. }
  264. func isUrl(_ urlString: String?) -> Bool {
  265. guard let urlString = urlString else { return false }
  266. // 实现方法有问题,暂不使用
  267. var url: String
  268. if urlString.count > 4, urlString.prefix(4) == "www." {
  269. url = "http://\(urlString)"
  270. } else {
  271. url = urlString
  272. }
  273. let urlRegex = "\\bhttps?://[a-zA-Z0-9\\-.]+(?::(\\d+))?(?:(?:/[a-zA-Z0-9\\-._?,'+\\&%$=~*!():@\\\\]*)+)?"
  274. let urlTest = NSPredicate(format: "SELF MATCHES %@", urlRegex)
  275. return urlTest.evaluate(with: url)
  276. }
  277. func urlValueEncode(_ str: String) -> String? {
  278. let allowedCharacterSet = CharacterSet(charactersIn: "!*'();:@&=+$,?%#[]{}").inverted
  279. return str.addingPercentEncoding(withAllowedCharacters: allowedCharacterSet)
  280. }
  281. //MARK: - MouseEvent
  282. override func mouseDown(with event: NSEvent) {
  283. super.mouseDown(with: event)
  284. window?.makeFirstResponder(nil)
  285. }
  286. }
  287. //MARK: - ComponentInputDelegate
  288. extension KMURLCreatePDFWindowController: ComponentInputDelegate {
  289. func componentInputDidChanged(inputView: ComponentInput) {
  290. reloadData()
  291. }
  292. }