vimr/VimR/HtmlPreviewTool.swift

/**
 * Tae Won Ha - http://taewon.de - @hataewon
 * See LICENSE
 */

import Cocoa
import RxSwift
import PureLayout
import WebKit
import Swifter
import EonilFileSystemEvents

fileprivate let fileSystemEventsLatency = 2.0
fileprivate let monitorDispatchQueue = DispatchQueue.global(qos: .userInitiated)

class HtmlPreviewTool: NSView, UiComponent {

  enum Action {

    case selectHtmlFile(URL)
  }

  typealias StateType = MainWindow.State

  let innerCustomToolbar = InnerCustomToolbar()

  required init(source: Observable<StateType>, emitter: ActionEmitter, state: StateType) {
    self.emitter = emitter
    self.uuid = state.uuid

    let configuration = WKWebViewConfiguration()
    self.webview = WKWebView(frame: CGRect.zero, configuration: configuration)

    super.init(frame: .zero)
    self.configureForAutoLayout()

    self.innerCustomToolbar.htmlPreviewTool = self

    self.addViews()

    self.webview.load(URLRequest(url: URL(string: "http://apple.com")!))

    source
      .observeOn(MainScheduler.instance)
      .subscribe(onNext: { [unowned self] state in
        guard let serverUrl = state.htmlPreview.server, let htmlFileUrl = state.htmlPreview.htmlFile else {
          self.monitor = nil
          return
        }

        if serverUrl.mark == self.mark {
          return
        }

        self.monitor = FileSystemEventMonitor(pathsToWatch: [htmlFileUrl.path],
                                              latency: fileSystemEventsLatency,
                                              watchRoot: false,
                                              queue: monitorDispatchQueue)
        { [unowned self] events in
          self.webview.load(URLRequest(url: serverUrl.payload))
        }

        self.mark = serverUrl.mark
        self.webview.load(URLRequest(url: serverUrl.payload))
      })
      .addDisposableTo(self.disposeBag)
  }

  fileprivate func addViews() {
    self.webview.configureForAutoLayout()

    self.addSubview(self.webview)
    self.webview.autoPinEdgesToSuperviewEdges()
  }

  fileprivate let emitter: ActionEmitter
  fileprivate let uuid: String

  fileprivate var mark = Token()

  fileprivate let webview: WKWebView
  fileprivate var monitor: FileSystemEventMonitor?

  fileprivate let disposeBag = DisposeBag()

  required init?(coder: NSCoder) {
    fatalError("init(coder:) has not been implemented")
  }

  func selectHtmlFile(sender: Any?) {
    let panel = NSOpenPanel()
    panel.canChooseDirectories = false
    panel.allowsMultipleSelection = false
    panel.beginSheetModal(for: self.window!) { result in
      guard result == NSFileHandlingPanelOKButton else {
        return
      }

      let urls = panel.urls
      guard urls.count == 1 else {
        return
      }

      self.emitter.emit(UuidAction(uuid: self.uuid, action: Action.selectHtmlFile(urls[0])))
    }
  }
}

extension HtmlPreviewTool {

  class InnerCustomToolbar: NSView {

    fileprivate weak var htmlPreviewTool: HtmlPreviewTool? {
      didSet {
        self.selectHtmlFile.target = self.htmlPreviewTool
      }
    }

    let selectHtmlFile = NSButton(forAutoLayout: ())

    init() {
      super.init(frame: .zero)
      self.configureForAutoLayout()

      self.addViews()
    }

    fileprivate func addViews() {
      let selectHtmlFile = self.selectHtmlFile
      InnerToolBar.configureToStandardIconButton(button: selectHtmlFile, iconName: .fileCodeO)
      selectHtmlFile.toolTip = "Select the HTML file"
      selectHtmlFile.action = #selector(HtmlPreviewTool.selectHtmlFile)

      self.addSubview(selectHtmlFile)

      selectHtmlFile.autoPinEdge(toSuperviewEdge: .top)
      selectHtmlFile.autoPinEdge(toSuperviewEdge: .right)
    }

    required init?(coder: NSCoder) {
      fatalError("init(coder:) has not been implemented")
    }
  }
}