vimr/VimR/FileBrowserComponent.swift

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

import Cocoa
import RxSwift
import PureLayout
import CocoaFontAwesome

enum FileBrowserAction {

  case open(url: URL)
  case openInNewTab(url: URL)
  case openInCurrentTab(url: URL)
  case openInHorizontalSplit(url: URL)
  case openInVerticalSplit(url: URL)
  case setAsWorkingDirectory(url: URL)
}

class FileBrowserComponent: ViewComponent {

  fileprivate let fileView: FileOutlineView
  fileprivate let fileItemService: FileItemService

  fileprivate var cwd: URL {
    get {
      return self.fileView.cwd
    }
    set {
      self.fileView.cwd = newValue
      self.innerCustomToolbar.goToParentButton.isEnabled = newValue.path != "/"
    }
  }

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

  class InnerCustomToolbar: NSView {

    fileprivate var fileBrowser: FileBrowserComponent? {
      didSet {
        self.goToParentButton.target = self.fileBrowser
      }
    }

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

    let goToParentButton = NSButton(forAutoLayout:())

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

      self.addViews()
    }

    fileprivate func addViews() {
      let goToParentIcon = NSImage.fontAwesomeIcon(name: .levelUp,
                                                   textColor: InnerToolBar.iconColor,
                                                   dimension: InnerToolBar.iconDimension)

      let goToParent = self.goToParentButton
      InnerToolBar.configureToStandardIconButton(button: goToParent, image: goToParentIcon)
      goToParent.action = #selector(FileBrowserComponent.goToParentAction)

      self.addSubview(goToParent)

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

  override var isFirstResponder: Bool {
    return self.fileView.isFirstResponder
  }

  let innerCustomToolbar = InnerCustomToolbar()

  init(source: Observable<Any>, fileItemService: FileItemService) {
    self.fileItemService = fileItemService
    self.fileView = FileOutlineView(source: source, fileItemService: fileItemService)

    super.init(source: source)

    self.innerCustomToolbar.fileBrowser = self
    self.addReactions()
  }

  override func beFirstResponder() {
    self.window?.makeFirstResponder(self.fileView)
  }

  fileprivate func addReactions() {
    self.fileView.sink
      .filter { $0 is FileOutlineViewAction }
      .map { $0 as! FileOutlineViewAction }
      .map {
        switch $0 {
        case let .open(fileItem): return FileBrowserAction.open(url: fileItem.url)
        case let .openFileInNewTab(fileItem): return FileBrowserAction.openInNewTab(url: fileItem.url)
        case let .openFileInCurrentTab(fileItem): return FileBrowserAction.openInCurrentTab(url: fileItem.url)
        case let .openFileInHorizontalSplit(fileItem): return FileBrowserAction.openInHorizontalSplit(url: fileItem.url)
        case let .openFileInVerticalSplit(fileItem): return FileBrowserAction.openInVerticalSplit(url: fileItem.url)
        case let .setAsWorkingDirectory(fileItem): return FileBrowserAction.setAsWorkingDirectory(url: fileItem.url)
        }
      }
      .subscribe(onNext: { [weak self] action in self?.publish(event: action) })
      .addDisposableTo(self.disposeBag)

    self.fileItemService.sink
        .filter { $0 is FileItemServiceChange }
        .map { $0 as! FileItemServiceChange }
        .observeOn(MainScheduler.instance)
        .subscribe(onNext: { [weak self] action in
          switch action {
          case let .childrenChanged(root, fileItem):
            guard root == self?.cwd else {
              return
            }

            self?.fileView.update(fileItem)
          }
        })
        .addDisposableTo(self.disposeBag)
  }

  override func addViews() {
    let scrollView = NSScrollView.standardScrollView()
    scrollView.borderType = .noBorder
    scrollView.documentView = self.fileView

    self.addSubview(scrollView)
    scrollView.autoPinEdgesToSuperviewEdges()
  }

  override func subscription(source: Observable<Any>) -> Disposable {
    return source
      .filter { $0 is MainWindowAction }
      .map { $0 as! MainWindowAction }
      .observeOn(MainScheduler.instance)
      .subscribe(onNext: { [weak self] action in
        switch action {
        case let .changeCwd(mainWindow: _, cwd: cwd):
          self?.cwd = cwd
//          NSLog("cwd changed to \(self.cwd) of \(mainWindow.uuid)")
          self?.fileView.reloadData()

        default:
          break
        }
      })
  }
}

// MARK: - Actions
extension FileBrowserComponent {

  func goToParentAction(_ sender: Any?) {
    self.publish(event: FileBrowserAction.setAsWorkingDirectory(url: self.cwd.parent))
  }
}