vimr/VimR/Workspace/WorkspaceToolButton.swift

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

import Cocoa

class WorkspaceToolButton: NSView, NSDraggingSource {

  static fileprivate let titlePadding = CGSize(width: 8, height: 2)
  static fileprivate let dummyButton = WorkspaceToolButton(title: "Dummy")

  fileprivate let title: NSAttributedString
  fileprivate var trackingArea = NSTrackingArea()
  
  required init?(coder: NSCoder) {
    fatalError("init(coder:) has not been implemented")
  }

  // MARK: - API
  static let toolUti = "com.qvacua.vimr.tool"

  static func ==(left: WorkspaceToolButton, right: WorkspaceToolButton) -> Bool {
    guard let leftTool = left.tool, let rightTool = right.tool else {
      return false
    }

    return leftTool == rightTool
  }

  var location = WorkspaceBarLocation.top
  var isSelected: Bool {
    return self.tool?.isSelected ?? false
  }

  weak var tool: WorkspaceTool?

  static func dimension() -> CGFloat {
    return self.dummyButton.intrinsicContentSize.height
  }

  static func size(forLocation loc: WorkspaceBarLocation) -> CGSize {
    switch loc {
    case .top, .bottom:
      return self.dummyButton.intrinsicContentSize
    case .right, .left:
      return CGSize(width: self.dummyButton.intrinsicContentSize.height,
                    height: self.dummyButton.intrinsicContentSize.width)
    }
  }

  init(title: String) {
    self.title = NSAttributedString(string: title, attributes: [
      NSFontAttributeName: NSFont.systemFont(ofSize: 11)
    ])

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

    self.wantsLayer = true
  }

  func highlight() {
    self.layer?.backgroundColor = NSColor.controlShadowColor.cgColor
  }

  func dehighlight() {
    self.layer?.backgroundColor = NSColor.clear.cgColor
  }
}

// MARK: - NSView
extension WorkspaceToolButton {

  override var intrinsicContentSize: NSSize {
    let titleSize = self.title.size()
    
    let padding = WorkspaceToolButton.titlePadding
    switch self.location {
    case .top, .bottom:
      return CGSize(width: titleSize.width + 2 * padding.width, height: titleSize.height + 2 * padding.height)
    case .right, .left:
      return CGSize(width: titleSize.height + 2 * padding.height, height: titleSize.width + 2 * padding.width)
    }
  }

  override func draw(_ dirtyRect: NSRect) {
    super.draw(dirtyRect)
    
    let padding = WorkspaceToolButton.titlePadding
    switch self.location {
    case .top, .bottom:
      self.title.draw(at: CGPoint(x: padding.width, y: padding.height))
    case .right:
      self.title.draw(at: CGPoint(x: padding.height, y: self.bounds.height - padding.width), angle: -CGFloat(M_PI_2))
    case .left:
      self.title.draw(at: CGPoint(x: self.bounds.width - padding.height, y: padding.width), angle: CGFloat(M_PI_2))
    }
  }

  override func updateTrackingAreas() {
    self.removeTrackingArea(self.trackingArea)

    self.trackingArea = NSTrackingArea(rect: self.bounds,
                                       options: [.mouseEnteredAndExited, .activeInActiveApp],
                                       owner: self,
                                       userInfo: nil)
    self.addTrackingArea(self.trackingArea)

    super.updateTrackingAreas()
  }

  override func mouseDown(with event: NSEvent) {
    guard let nextEvent = self.window!.nextEvent(matching: [NSLeftMouseUpMask, NSLeftMouseDraggedMask]) else {
      return
    }

    switch nextEvent.type {

    case NSLeftMouseUp:
      self.tool?.toggle()
      return

    case NSLeftMouseDragged:
      let pasteboardItem = NSPasteboardItem()
      pasteboardItem.setString(self.tool!.uuid, forType: WorkspaceToolButton.toolUti)

      let draggingItem = NSDraggingItem(pasteboardWriter: pasteboardItem)
      draggingItem.setDraggingFrame(self.bounds, contents:self.snapshot())

      self.beginDraggingSession(with: [draggingItem], event: event, source: self)
      return

    default:
      return
    }
  }

  override func mouseEntered(with _: NSEvent) {
    if self.isSelected {
      return
    }

    self.highlight()
  }

  override func mouseExited(with _: NSEvent) {
    if self.isSelected {
      return
    }

    self.dehighlight()
  }

  // Modified version of snapshot() from https://www.raywenderlich.com/136272/drag-and-drop-tutorial-for-macos
  fileprivate func snapshot() -> NSImage {
    let pdfData = self.dataWithPDF(inside: self.bounds)
    guard let image = NSImage(data: pdfData) else {
      return NSImage()
    }

    let result = NSImage()
    let rect = CGRect(origin: .zero, size: image.size)
    result.size = rect.size

    result.lockFocus()
    NSColor.controlShadowColor.set()
    NSRectFill(rect)
    image.draw(in: rect)
    result.unlockFocus()

    return result
  }
}

// MARK: - NSDraggingSource
extension WorkspaceToolButton {
  @objc(draggingSession:sourceOperationMaskForDraggingContext:)
  func draggingSession(_ session: NSDraggingSession, sourceOperationMaskFor ctc: NSDraggingContext) -> NSDragOperation {
    return .move
  }

  @objc(draggingSession:endedAtPoint:operation:)
  func draggingSession(_ session: NSDraggingSession, endedAt screenPoint: NSPoint, operation: NSDragOperation) {
    guard let pointInWindow = self.window?.convertFromScreen(CGRect(origin: screenPoint, size: .zero)) else {
      return
    }

    let pointInView = self.convert(pointInWindow, from: nil)
    // Sometimes if the drag ends, the button does not get dehighlighted.
    if !self.frame.contains(pointInView) && !(self.tool?.isSelected ?? false) {
      self.dehighlight()
    }
  }
}