import Foundation
import Cocoa
import SwiftUI
import GhosttyKit

/// Controller for the slide-style terminal.
class SlideTerminalController: BaseTerminalController {
    override var windowNibName: NSNib.Name? { "SlideTerminal" }

    /// The position for the slide terminal.
    let position: SlideTerminalPosition

    init(_ ghostty: Ghostty.App,
         position: SlideTerminalPosition = .top,
         baseConfig base: Ghostty.SurfaceConfiguration? = nil,
         surfaceTree tree: Ghostty.SplitNode? = nil
    ) {
        self.position = position
        super.init(ghostty, baseConfig: base, surfaceTree: tree)
    }

    required init?(coder: NSCoder) {
        fatalError("init(coder:) is not supported for this view")
    }

    // MARK: NSWindowController

    override func windowDidLoad() {
        guard let window = self.window else { return }

        // The controller is the window delegate so we can detect events such as
        // window close so we can animate out.
        window.delegate = self

        // The slide window is not restorable (yet!). "Yet" because in theory we can
        // make this restorable, but it isn't currently implemented.
        window.isRestorable = false

        // Setup our content
        window.contentView = NSHostingView(rootView: TerminalView(
            ghostty: self.ghostty,
            viewModel: self,
            delegate: self
        ))

        // Animate the window in
        slideIn()
    }

    // MARK: NSWindowDelegate

    override func windowDidResignKey(_ notification: Notification) {
        super.windowDidResignKey(notification)
        slideOut()
    }

    func windowWillResize(_ sender: NSWindow, to frameSize: NSSize) -> NSSize {
        guard let screen = NSScreen.main else { return frameSize }
        return position.restrictFrameSize(frameSize, on: screen)
    }

    // MARK: Base Controller Overrides

    override func surfaceTreeDidChange(from: Ghostty.SplitNode?, to: Ghostty.SplitNode?) {
        super.surfaceTreeDidChange(from: from, to: to)

        // If our surface tree is nil then we slide the window out.
        if (to == nil) {
            slideOut()
        }
    }

    // MARK: Slide Methods

    func slideToggle() {
        guard let window = self.window else { return }
        if (window.alphaValue > 0) {
            slideOut()
        } else {
            slideIn()
        }
    }

    func slideIn() {
        guard let window = self.window else { return }

        // Animate the window in
        slideWindowIn(window: window, from: position)

        // If our surface tree is nil then we initialize a new terminal. The surface
        // tree can be nil if for example we run "eixt" in the terminal and force a
        // slide out.
        if (surfaceTree == nil) {
            surfaceTree = .leaf(.init(ghostty.app!, baseConfig: nil))
        }
    }

    func slideOut() {
        guard let window = self.window else { return }
        slideWindowOut(window: window, to: position)
    }

    private func slideWindowIn(window: NSWindow, from position: SlideTerminalPosition) {
        guard let screen = NSScreen.main else { return }

        // Move our window off screen to the top
        position.setInitial(in: window, on: screen)

        // Move it to the visible position since animation requires this
        window.makeKeyAndOrderFront(nil)

        // Run the animation that moves our window into the proper place and makes
        // it visible.
        NSAnimationContext.runAnimationGroup { context in
            context.duration = 0.3
            context.timingFunction = .init(name: .easeIn)
            position.setFinal(in: window.animator(), on: screen)
        }
    }

    private func slideWindowOut(window: NSWindow, to position: SlideTerminalPosition) {
        guard let screen = NSScreen.main else { return }

        // Keep track of if we were the key window. If we were the key window then we
        // want to move focus to the next window so that focus is preserved somewhere
        // in the app.
        let wasKey = window.isKeyWindow

        NSAnimationContext.runAnimationGroup({ context in
            context.duration = 0.3
            context.timingFunction = .init(name: .easeIn)
            position.setInitial(in: window.animator(), on: screen)
        }, completionHandler: {
            guard wasKey else { return }
            self.focusNextWindow()
        })
    }

    private func focusNextWindow() {
        // We only want to consider windows that are visible
        let windows = NSApp.windows.filter { $0.isVisible }

        // If we have no windows there is nothing to focus.
        guard !windows.isEmpty else { return }

        // Find the current key window (the window that is currently focused)
        if let keyWindow = NSApp.keyWindow,
           let currentIndex = windows.firstIndex(of: keyWindow) {
            // Calculate the index of the next window (cycle through the list)
            let nextIndex = (currentIndex + 1) % windows.count
            let nextWindow = windows[nextIndex]

            // Make the next window key and bring it to the front
            nextWindow.makeKeyAndOrderFront(nil)
        } else {
            // If there's no key window, focus the first available window
            windows.first?.makeKeyAndOrderFront(nil)
        }
    }

    // MARK: First Responder

    @IBAction override func closeWindow(_ sender: Any) {
        // Instead of closing the window, we slide it out.
        slideOut()
    }
}