macos/writeas/ViewController.swift

//
//  ViewController.swift
//  writeas
//
//  Created by Matt Baer on 12/29/15.
//  Copyright © 2015 A Bunch Tell. All rights reserved.
//

import Cocoa

class ViewController: NSViewController, NSTextViewDelegate, NSUserNotificationCenterDelegate {

    @IBOutlet var writerText: NSTextView!
    @IBOutlet weak var writerScrollContainer: NSScrollView!

    var isNight = false

    override func viewDidLoad() {
        super.viewDidLoad()

        loadDocument()

        writerText.font = NSFont(name: Preferences.getFont().typeface, size: Preferences.getFontSize())
        writerText.isHorizontallyResizable = false
        writerText.textContainerInset = NSSize(width: 16, height: 16)
        writerText.isAutomaticQuoteSubstitutionEnabled = false
        writerText.isAutomaticDashSubstitutionEnabled = false

        NSUserNotificationCenter.default.delegate = self

        // Do any additional setup after loading the view.
        writerText.delegate = self
    }

    override func viewWillAppear() {
        super.viewWillAppear()

        self.view.window!.titlebarAppearsTransparent = true
        self.view.window!.isMovableByWindowBackground = true
        self.view.window!.titleVisibility = .hidden
        toggle(isNight: isNight)
    }

    override func viewDidAppear() {
        super.viewDidAppear()

        configureWindow()

        // Style the window
        self.view.window!.title = "Write.as"
    }

    func configureWindow() {
        // Fit textview to window size
        writerText.setFrameSize(NSSize(width: self.writerScrollContainer.frame.width - 20, height: self.writerScrollContainer.frame.height))
    }

    override var representedObject: Any? {
        didSet {
        // Update the view, if already loaded.
        }
    }

    fileprivate func getDraftPath() -> URL? {
        return URL(fileURLWithPath: Constants.draftFile)
    }

    func saveDocument() {
        // Save whatever's written
        let path = getDraftPath()
        if path == nil {
            return
        }

        //writing
        do {
            try writerText.textStorage!.string.write(to: path!, atomically: false, encoding: .utf8)
        } catch (let err) {
            print("ERROR writing: \(err)")
        }
    }

    func loadDocument() {
        let path = getDraftPath()
        if path == nil {
            return
        }
        do {
            let draftText = try String(contentsOf: path!, encoding: .utf8)
            writerText.string = draftText
        } catch (let err) {
            print("ERROR loading: \(err)")
        }
    }

    func embolden() {
        formatText(with: "**\(getSelectedText())**", offset: 2)
    }

    func emphasize() {
        formatText(with: "_\(getSelectedText())_", offset: 1)
    }

    func addLink() {
        let text = getSelectedText()
        formatText(with: "[\(text)](http://)", offset: text == "" ? 1 : (text.characters.count + 10), alwaysOffset: true)
    }

    fileprivate func getSelectedText() -> String {
        let selectedRange = writerText.selectedRange()
        return (self.writerText.textStorage?.string as! NSString).substring(with: selectedRange)
    }

    fileprivate func formatText(with: String, offset: Int) {
        formatText(with: with, offset: offset, alwaysOffset: false)
    }
    fileprivate func formatText(with: String, offset: Int, alwaysOffset: Bool) {
        let selectedRange = writerText.selectedRange()
        let selString = getSelectedText()
        if self.writerText.shouldChangeText(in: selectedRange, replacementString: with) {
            self.writerText.replaceCharacters(in: NSRange(location: selectedRange.location, length: selectedRange.length), with: with)
            self.writerText.didChangeText()
            if selString == "" || alwaysOffset {
                writerText.setSelectedRange(NSRange(location: selectedRange.location + offset, length: 0))
            }
        }
    }

    func publish() {
        saveDocument()

        DispatchQueue.global(qos: .background).async {
            let task = Process()
            task.launchPath = Bundle.main.resourcePath! + "/writeas"
            task.arguments = ["--font", Preferences.getFont().rawValue]
            let pipe = Pipe()
            task.standardInput = pipe
            task.launch()
            let fh: FileHandle = pipe.fileHandleForWriting
            fh.write(self.writerText.textStorage!.string.data(using: .utf8)!)
            fh.closeFile()
            task.waitUntilExit()

            DispatchQueue.main.async {
                if let ad = NSApplication.shared().delegate as? AppDelegate {
                    ad.filePublishItem.isEnabled = true
                }

                var notification = NSUserNotification()

                if task.terminationStatus == 0 {
                    // Successfully published
                    notification.title = "Published!"
                    notification.informativeText = "The link is copied — press ⌘+V to share it."
                } else {
                    // Something went wrong
                    notification.title = "Unable to publish."
                    notification.informativeText = "There was a problem publishing. Please check your internet and try again."
                }
                notification.soundName = NSUserNotificationDefaultSoundName

                NSUserNotificationCenter.default.deliver(notification)
            }
        }
    }

    func userNotificationCenter(_ center: NSUserNotificationCenter, shouldPresent notification: NSUserNotification) -> Bool {
        return true
    }

    func toggle(isNight: Bool) {
        let darkBG = NSColor(red:0.13, green:0.13, blue:0.13, alpha:1.0)
        if isNight {
            self.view.window!.backgroundColor = darkBG
            self.writerText.backgroundColor = darkBG
            self.writerText.textColor = NSColor.white
        } else {
            self.view.window!.backgroundColor = NSColor.white
            self.writerText.backgroundColor = NSColor.white
            self.writerText.textColor = NSColor.black
        }
        self.isNight = isNight
    }

    func adjustTextSize(increment: Bool) {
        var size = Preferences.getFontSize()
        if increment {
            size += 2
        } else {
            size -= 1
        }
        self.writerText.font = NSFont(name: Preferences.getFont().typeface, size: size)
        UserDefaults.standard.set(String(Int(size)), forKey: "editor_text_size")
        configureWindow()
    }

    func setFont(_ font: Preferences.PostFont) {
        self.writerText.font = NSFont(name: font.typeface, size: Preferences.getFontSize())
    }
}