JohnCoates/Aerial

//
//  InfoTableSource.swift
//  Aerial
//
//  Created by Guillaume Louel on 16/12/2019.
//  Copyright © 2019 Guillaume Louel. All rights reserved.
//

import Cocoa

class InfoTableSource: NSTableView, NSTableViewDataSource, NSTableViewDelegate {
    private var dragDropType = NSPasteboard.PasteboardType(rawValue: "private.table-row")
    var controller: OverlaysViewController?

    func setController(_ controller: OverlaysViewController) {
        self.controller = controller
    }

    override func draw(_ dirtyRect: NSRect) {
        super.draw(dirtyRect)
    }

    func numberOfRows(in tableView: NSTableView) -> Int {
        return PrefsInfo.layers.count
    }

    func tableView(_ tableView: NSTableView, heightOfRow row: Int) -> CGFloat {
        return 30
    }

    // This is where we fill each cell
    func tableView(_ tableView: NSTableView, viewFor tableColumn: NSTableColumn?, row: Int) -> NSView? {
        if tableColumn == tableView.tableColumns[0] {
            let cid = NSUserInterfaceItemIdentifier(rawValue: "InfoSourceCellID")

            if let cell = tableView.makeView(withIdentifier: cid, owner: self) as? NSTableCellView {
                cell.textField?.stringValue = PrefsInfo.layers[row].rawValue.capitalizeFirstLetter()

                let isEnabled = PrefsInfo.ofType(PrefsInfo.layers[row]).isEnabled

                // Aerial.getAccentedSymbol(<#T##named: String##String#>)
                // Aerial.getSymbol("checkmark.circle.fill")
                cell.imageView?.image = isEnabled
                ? Aerial.helper.getAccentedSymbol("checkmark.circle.fill")
                : Aerial.helper.getAccentedSymbol("circle")
                /*cell.imageView?.image = NSImage(named: isEnabled
                    ? NSImage.statusAvailableName
                    : NSImage.statusUnavailableName)*/

                return cell
            }
        } else {
            let cid = NSUserInterfaceItemIdentifier(rawValue: "InfoSourceGripID")

            if let cell = tableView.makeView(withIdentifier: cid, owner: self) as? NSTableCellView {
                cell.imageView?.image = NSImage(named: NSImage.listViewTemplateName)
                return cell
            }
        }

        return nil
    }

    // This is where selection happens
    func tableViewSelectionDidChange(_ notification: Notification) {
        let tableView = notification.object as! NSTableView
        if tableView.selectedRow < 0 {
            // controller!.resetInfoPanel()
        } else {
            controller!.drawInfoPanel(forType: PrefsInfo.layers[tableView.selectedRow])
            controller!.infoSettingsTableView.deselectAll(controller!)

        }
    }

    // MARK: - Drag 'n Drop
    func tableView(_ tableView: NSTableView, pasteboardWriterForRow row: Int) -> NSPasteboardWriting? {
        let item = NSPasteboardItem()
        item.setString(String(row), forType: self.dragDropType)
        return item
    }

    func tableView(_ tableView: NSTableView, validateDrop info: NSDraggingInfo, proposedRow row: Int, proposedDropOperation dropOperation: NSTableView.DropOperation) -> NSDragOperation {

        if dropOperation == .above {
            return .move
        }
        return []
    }

    func tableView(_ tableView: NSTableView, acceptDrop info: NSDraggingInfo, row: Int, dropOperation: NSTableView.DropOperation) -> Bool {

        var oldIndexes = [Int]()
        info.enumerateDraggingItems(options: [], for: tableView, classes: [NSPasteboardItem.self], searchOptions: [:]) { dragItem, _, _ in
            if let str = (dragItem.item as! NSPasteboardItem).string(forType: self.dragDropType), let index = Int(str) {
                oldIndexes.append(index)
            }
        }

        var oldIndexOffset = 0
        var newIndexOffset = 0

        for oldIndex in oldIndexes {
            if oldIndex < row {
                PrefsInfo.layers.move(from: oldIndex + oldIndexOffset, to: row - 1)
                oldIndexOffset -= 1
            } else {
                PrefsInfo.layers.move(from: oldIndex, to: row + newIndexOffset)
                newIndexOffset += 1
            }
        }

        tableView.reloadData()
        return true
    }
}

// Helpers to move items in array
extension Array where Element: Equatable {
    mutating func move(_ element: Element, to newIndex: Index) {
        if let oldIndex: Int = self.firstIndex(of: element) { self.move(from: oldIndex, to: newIndex) }
    }
}

extension Array {
    mutating func move(from oldIndex: Index, to newIndex: Index) {
        // Don't work for free and use swap when indices are next to each other - this
        // won't rebuild array and will be super efficient.
        if oldIndex == newIndex { return }
        if abs(newIndex - oldIndex) == 1 { return self.swapAt(oldIndex, newIndex) }
        self.insert(self.remove(at: oldIndex), at: newIndex)
    }
}

extension String {
     func capitalizeFirstLetter() -> String {
          return self.prefix(1).capitalized + dropFirst()
     }
}

extension NSTableView {
    func reloadDataKeepingSelection() {
        let selectedRowIndexes = self.selectedRowIndexes
        self.reloadData()
        self.selectRowIndexes(selectedRowIndexes, byExtendingSelection: true)
    }
}