vChewing-macOS/Packages/NotifierUI/Sources/NotifierUI/NotifierController.swift

// Copyright (c) 2022 and onwards The McBopomofo Authors.
//
// Permission is hereby granted, free of charge, to any person
// obtaining a copy of this software and associated documentation
// files (the "Software"), to deal in the Software without
// restriction, including without limitation the rights to use,
// copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following
// conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
// OTHER DEALINGS IN THE SOFTWARE.

import Cocoa

private protocol NotifierWindowDelegate: AnyObject {
    func windowDidBecomeClicked(_ window: NotifierWindow)
}

private class NotifierWindow: NSWindow {
    weak var clickDelegate: NotifierWindowDelegate?

    override func mouseDown(with event: NSEvent) {
        clickDelegate?.windowDidBecomeClicked(self)
    }
}

private let kWindowWidth: CGFloat = 160.0
private let kWindowHeight: CGFloat = 80.0

public class NotifierController: NSWindowController, NotifierWindowDelegate {
    private var messageTextField: NSTextField

    private var message: String = "" {
        didSet {
            let paraStyle = NSMutableParagraphStyle()
            paraStyle.setParagraphStyle(NSParagraphStyle.default)
            paraStyle.alignment = .center
            let attr: [NSAttributedString.Key: AnyObject] = [
                .foregroundColor: foregroundColor,
                .font: NSFont.systemFont(ofSize: NSFont.systemFontSize(for: .regular)),
                .paragraphStyle: paraStyle
            ]
            let attrString = NSAttributedString(string: message, attributes: attr)
            messageTextField.attributedStringValue = attrString
            let width = window?.frame.width ?? kWindowWidth
            let rect = attrString.boundingRect(with: NSSize(width: width, height: 1600), options: .usesLineFragmentOrigin)
            let height = rect.height
            let x = messageTextField.frame.origin.x
            let y = ((window?.frame.height ?? kWindowHeight) - height) / 2
            let newFrame = NSRect(x: x, y: y, width: width, height: height)
            messageTextField.frame = newFrame
        }
    }
    private var shouldStay: Bool = false
    private var backgroundColor: NSColor = .black {
        didSet {
            self.window?.backgroundColor = backgroundColor
            self.messageTextField.backgroundColor = backgroundColor
        }
    }
    private var foregroundColor: NSColor = .white {
        didSet {
            self.messageTextField.textColor = foregroundColor
        }
    }
    private var waitTimer: Timer?
    private var fadeTimer: Timer?

    private static var instanceCount = 0
    private static var lastLocation = NSPoint.zero

    @objc public static func notify(message: String, stay: Bool = false) {
        let controller = NotifierController()
        controller.message = message
        controller.shouldStay = stay
        controller.show()
    }

    private static func increaseInstanceCount() {
        instanceCount += 1
    }

    private static func decreaseInstanceCount() {
        instanceCount -= 1
        if instanceCount < 0 {
            instanceCount = 0
        }
    }

    private init() {
        let screenRect = NSScreen.main?.visibleFrame ?? NSRect.zero
        let contentRect = NSRect(x: 0, y: 0, width: kWindowWidth, height: kWindowHeight)
        var windowRect = contentRect
        windowRect.origin.x = screenRect.maxX - windowRect.width - 10
        windowRect.origin.y = screenRect.maxY - windowRect.height - 10
        let styleMask: NSWindow.StyleMask = [.borderless]
        let panel = NotifierWindow(contentRect: windowRect, styleMask: styleMask, backing: .buffered, defer: false)
        panel.level = NSWindow.Level(Int(kCGPopUpMenuWindowLevel))
        panel.hasShadow = true
        panel.backgroundColor = backgroundColor

        messageTextField = NSTextField()
        messageTextField.frame = contentRect
        messageTextField.isEditable = false
        messageTextField.isSelectable = false
        messageTextField.isBezeled = false
        messageTextField.textColor = foregroundColor
        messageTextField.drawsBackground = true
        messageTextField.backgroundColor = backgroundColor
        messageTextField.font = .systemFont(ofSize: NSFont.systemFontSize(for: .small))
        panel.contentView?.addSubview(messageTextField)

        super.init(window: panel)

        panel.clickDelegate = self
    }

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

    private func show() {
        func setStartLocation() {
            if NotifierController.instanceCount == 0 {
                return
            }
            let lastLocation = NotifierController.lastLocation
            let screenRect = NSScreen.main?.visibleFrame ?? NSRect.zero
            var windowRect = self.window?.frame ?? NSRect.zero
            windowRect.origin.x = lastLocation.x
            windowRect.origin.y = lastLocation.y - 10 - windowRect.height

            if windowRect.origin.y < screenRect.minY {
                return
            }

            self.window?.setFrame(windowRect, display: true)
        }

        func moveIn() {
            let afterRect = self.window?.frame ?? NSRect.zero
            NotifierController.lastLocation = afterRect.origin
            var beforeRect = afterRect
            beforeRect.origin.y += 10
            window?.setFrame(beforeRect, display: true)
            window?.orderFront(self)
            window?.setFrame(afterRect, display: true, animate: true)
        }

        setStartLocation()
        moveIn()
        NotifierController.increaseInstanceCount()
        waitTimer = Timer.scheduledTimer(timeInterval: shouldStay ? 5 : 1, target: self, selector: #selector(fadeOut), userInfo: nil, repeats: false)
    }

    @objc private func doFadeOut(_ timer: Timer) {
        let opacity = self.window?.alphaValue ?? 0
        if opacity <= 0 {
            self.close()
        } else {
            self.window?.alphaValue = opacity - 0.2
        }
    }

    @objc private func fadeOut() {
        waitTimer?.invalidate()
        waitTimer = nil
        NotifierController.decreaseInstanceCount()
        fadeTimer = Timer.scheduledTimer(timeInterval: 0.1, target: self, selector: #selector(doFadeOut(_:)), userInfo: nil, repeats: true)
    }

    public override func close() {
        waitTimer?.invalidate()
        waitTimer = nil
        fadeTimer?.invalidate()
        fadeTimer = nil
        super.close()
    }

    fileprivate func windowDidBecomeClicked(_ window: NotifierWindow) {
        self.fadeOut()
    }
}