KMURLCreatePDFWindowController.swift 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395
  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: KMLocalizedString("URL"),
  54. checkboxType: .normal)
  55. urlRadio.setTarget(self, action: #selector(typeRadioChanged(_:)))
  56. fileRadio.properties = ComponentCheckBoxProperty(size: .s,
  57. state: .normal,
  58. text: KMLocalizedString("File"),
  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: KMLocalizedString("https://pdfreaderpro.com"))
  66. inputView.delegate = self
  67. fileInputView.properties = ComponentInputProperty(size: .s,
  68. state: .normal,
  69. isError: false,
  70. isDisabled: false,
  71. placeholder: KMLocalizedString("Select File..."),
  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: KMLocalizedString("Cancel"),
  86. keepPressState: false)
  87. cancelButton.setTarget(self, action: #selector(cancelButtonClicked(_ :)))
  88. openButton.properties = ComponentButtonProperty(type: .primary,
  89. size: .s,
  90. state: .normal,
  91. buttonText: KMLocalizedString("Open"),
  92. keepPressState: false)
  93. openButton.setTarget(self, action: #selector(openButtonClicked(_ :)))
  94. }
  95. func reloadData() {
  96. urlRadio.properties.checkboxType = isFromURL ? .selected : .normal
  97. fileRadio.properties.checkboxType = isFromURL ? .normal : .selected
  98. fileInputBGView.isHidden = isFromURL ? true : false
  99. inputView.isHidden = isFromURL ? false : true
  100. cancelButton.properties.state = .normal
  101. openButton.properties.state = .normal
  102. if let filePath = choosedFilePath {
  103. fileInputView.properties.text = filePath
  104. } else {
  105. fileInputView.properties.text = ""
  106. }
  107. if isFromURL {
  108. if inputView.properties.text.count > 0 {
  109. openButton.properties.isDisabled = false
  110. } else {
  111. openButton.properties.isDisabled = true
  112. }
  113. } else {
  114. if fileInputView.properties.text.count > 0 {
  115. openButton.properties.isDisabled = false
  116. } else {
  117. openButton.properties.isDisabled = true
  118. }
  119. }
  120. inputView.reloadData()
  121. fileInputView.reloadData()
  122. fileInputAddon.reloadData()
  123. urlRadio.reloadData()
  124. fileRadio.reloadData()
  125. cancelButton.reloadData()
  126. openButton.reloadData()
  127. fileInputView.reloadData()
  128. }
  129. //MARK: - Action
  130. @objc func typeRadioChanged(_ sender: NSView) {
  131. if sender == fileRadio {
  132. isFromURL = false
  133. } else if sender == self.urlRadio {
  134. isFromURL = true
  135. }
  136. }
  137. @objc func cancelButtonClicked(_ sender: NSView) {
  138. choosedFilePath = nil
  139. inputView.properties.text = ""
  140. hideWaitting()
  141. reloadData()
  142. KMURLToPDF.shareInstance.stopLoading()
  143. handler?(nil)
  144. parentWindow?.endSheet(self.window!)
  145. }
  146. @objc func openButtonClicked(_ sender: NSView) {
  147. convertInfoToPDF()
  148. }
  149. @objc func chooseURLAction(_ sender: NSView) {
  150. let openPanel = NSOpenPanel()
  151. openPanel.allowedFileTypes = ["html", "HTML"]
  152. openPanel.allowsMultipleSelection = false
  153. openPanel.beginSheetModal(for: self.window!) { [weak self] result in
  154. if result == NSApplication.ModalResponse.OK {
  155. if let url = openPanel.url {
  156. self?.choosedFilePath = url.path
  157. self?.reloadData()
  158. }
  159. }
  160. }
  161. }
  162. func convertInfoToPDF() {
  163. if !FileManager.default.fileExists(atPath: kUrlToPDFFolderPath) {
  164. try? FileManager.default.createDirectory(atPath: kUrlToPDFFolderPath, withIntermediateDirectories: false, attributes: nil)
  165. }
  166. var url: URL?
  167. if isFromURL {
  168. var urlString = inputView.properties.text
  169. var tUrl = URL(string: urlString)
  170. if tUrl?.scheme?.count ?? 0 < 1 {
  171. if urlString.hasSuffix(".com") == false {
  172. urlString = urlString + ".com"
  173. }
  174. tUrl = URL(string: "http://\(urlString)")
  175. }
  176. url = tUrl
  177. } else {
  178. if let choosedFilePath = choosedFilePath {
  179. url = URL(fileURLWithPath: choosedFilePath)
  180. }
  181. }
  182. if let url = url {
  183. inputView.properties.text = url.absoluteString
  184. inputView.reloadData()
  185. showWaitting()
  186. let outPutPath = kUrlToPDFFolderPath
  187. if isFromURL {
  188. KMURLToPDF.shareInstance.convertLinkString(string: url.absoluteString, toPath: outPutPath) {[weak self] url, error in
  189. self?.hideWaitting()
  190. if error == nil && url?.path.isEmpty == false {
  191. self?.choosedFilePath = nil
  192. self?.inputView.properties.text = ""
  193. self?.reloadData()
  194. self?.handler?(url?.path)
  195. if let window = self?.window {
  196. self?.parentWindow?.endSheet(window)
  197. }
  198. } else {
  199. if let window = self?.window {
  200. let alert = NSAlert()
  201. alert.alertStyle = .critical
  202. alert.messageText = NSLocalizedString("Conversion Failed", comment: "")
  203. if let description = error?.localizedDescription {
  204. alert.messageText = description
  205. }
  206. alert.beginSheetModal(for: window) { [weak self] result in
  207. }
  208. }
  209. }
  210. }
  211. } else {
  212. KMURLToPDF.shareInstance.convertHtmlFileToPDF(fromPath: url.path, toPath: outPutPath) {[weak self] url, error in
  213. self?.hideWaitting()
  214. if error == nil && url?.path.isEmpty == false {
  215. self?.choosedFilePath = nil
  216. self?.inputView.properties.text = ""
  217. self?.reloadData()
  218. self?.handler?(url?.path)
  219. if let window = self?.window {
  220. self?.parentWindow?.endSheet(window)
  221. }
  222. } else {
  223. if let window = self?.window {
  224. let alert = NSAlert()
  225. alert.alertStyle = .critical
  226. alert.messageText = NSLocalizedString("Conversion Failed", comment: "")
  227. alert.beginSheetModal(for: window) { [weak self] result in
  228. }
  229. }
  230. }
  231. }
  232. }
  233. }
  234. }
  235. //MARK: - WaitingView
  236. func showWaitting() {
  237. if posterMaskView == nil {
  238. posterMaskView = KMBookletMaskView(frame: NSMakeRect(0, 0, window?.frame.size.width ?? 0, window?.frame.size.height ?? 0))
  239. }
  240. contendBox.contentView?.addSubview(posterMaskView!, positioned: .below, relativeTo: self.cancelButton)
  241. }
  242. func hideWaitting() {
  243. posterMaskView?.removeFromSuperview()
  244. }
  245. //MARK: - public
  246. func own_beginSheetModal(for window: NSWindow?, completionHandler handler: ((String?) -> Void)?) {
  247. self.window?.makeFirstResponder(nil)
  248. if window != nil {
  249. parentWindow = window
  250. window!.beginSheet(self.window!) { ModalResponse in
  251. self.handler?(nil)
  252. }
  253. choosedFilePath = nil
  254. self.handler = handler
  255. hideWaitting()
  256. reloadData()
  257. }
  258. }
  259. override func becomeFirstResponder() -> Bool {
  260. return true
  261. }
  262. func formatFloat(_ f: Float) -> String {
  263. if f.truncatingRemainder(dividingBy: 1) == 0 {
  264. return String(format: "%.0f", f)
  265. } else if (f * 10).truncatingRemainder(dividingBy: 1) == 0 {
  266. return String(format: "%.1f", f)
  267. } else {
  268. return String(format: "%.2f", f)
  269. }
  270. }
  271. func isUrl(_ urlString: String?) -> Bool {
  272. guard let urlString = urlString else { return false }
  273. // 实现方法有问题,暂不使用
  274. var url: String
  275. if urlString.count > 4, urlString.prefix(4) == "www." {
  276. url = "http://\(urlString)"
  277. } else {
  278. url = urlString
  279. }
  280. let urlRegex = "\\bhttps?://[a-zA-Z0-9\\-.]+(?::(\\d+))?(?:(?:/[a-zA-Z0-9\\-._?,'+\\&%$=~*!():@\\\\]*)+)?"
  281. let urlTest = NSPredicate(format: "SELF MATCHES %@", urlRegex)
  282. return urlTest.evaluate(with: url)
  283. }
  284. func urlValueEncode(_ str: String) -> String? {
  285. let allowedCharacterSet = CharacterSet(charactersIn: "!*'();:@&=+$,?%#[]{}").inverted
  286. return str.addingPercentEncoding(withAllowedCharacters: allowedCharacterSet)
  287. }
  288. //MARK: - MouseEvent
  289. override func mouseDown(with event: NSEvent) {
  290. super.mouseDown(with: event)
  291. window?.makeFirstResponder(nil)
  292. }
  293. }
  294. //MARK: - ComponentInputDelegate
  295. extension KMURLCreatePDFWindowController: ComponentInputDelegate {
  296. func componentInputDidChanged(inputView: ComponentInput) {
  297. reloadData()
  298. }
  299. }