import Darwin
import Foundation
import OSLog
import SwiftUI
import MacToolsPluginKit

public final class LockScreenPluginFactory: NSObject, MacToolsPluginBundleFactory {
    public static func makeProvider(context: PluginRuntimeContext) throws -> any PluginProvider {
        LockScreenPluginProvider(context: context)
    }
}

@MainActor
private struct LockScreenPluginProvider: PluginProvider {
    let context: PluginRuntimeContext

    func makePlugins() -> [any MacToolsPlugin] {
        [LockScreenPlugin(localization: PluginLocalization(bundle: context.resourceBundle))]
    }
}

@MainActor
final class LockScreenPlugin: MacToolsPlugin, PluginPrimaryPanel {
    let metadata: PluginMetadata

    let primaryPanelDescriptor: PluginPrimaryPanelDescriptor

    var onStateChange: (() -> Void)?
    var requestPermissionGuidance: ((String) -> Void)?
    var shortcutBindingResolver: ((String) -> ShortcutBinding?)?

    private let logger = Logger(
        subsystem: Bundle.main.bundleIdentifier ?? "cc.ggbond.mactools",
        category: "LockScreenPlugin"
    )

    init(localization: PluginLocalization = PluginLocalization(bundle: .main)) {
        self.metadata = PluginMetadata(
            id: "lock-screen",
            title: localization.string("metadata.title", defaultValue: "锁定屏幕"),
            iconName: "lock",
            iconTint: Color(nsColor: .systemGray),
            order: 96,
            defaultDescription: localization.string("metadata.description", defaultValue: "立即锁定屏幕")
        )
        self.primaryPanelDescriptor = PluginPrimaryPanelDescriptor(
            controlStyle: .button,
            menuActionBehavior: .dismissBeforeHandling,
            buttonTitleProvider: { localization.string("panel.button.lock", defaultValue: "锁定") }
        )
    }

    var primaryPanelState: PluginPanelState {
        PluginPanelState(
            subtitle: metadata.defaultDescription,
            isOn: false,
            isExpanded: false,
            isEnabled: true,
            isVisible: true,
            detail: nil,
            errorMessage: nil
        )
    }

    func handleAction(_ action: PluginPanelAction) {
        guard case let .invokeAction(controlID) = action, controlID == "execute" else {
            return
        }

        lockScreen()
    }

    private func lockScreen() {
        let frameworkPath = "/System/Library/PrivateFrameworks/login.framework/login"
        guard let handle = dlopen(frameworkPath, RTLD_LAZY) else {
            logger.error("Failed to open login.framework; falling back to ScreenSaverEngine")
            startScreenSaverFallback()
            return
        }
        defer { dlclose(handle) }

        guard let symbol = dlsym(handle, "SACLockScreenImmediate") else {
            logger.error("SACLockScreenImmediate not found; falling back to ScreenSaverEngine")
            startScreenSaverFallback()
            return
        }

        typealias LockScreenFunction = @convention(c) () -> Int32
        let lockScreenImmediately = unsafeBitCast(symbol, to: LockScreenFunction.self)
        let result = lockScreenImmediately()
        if result == 0 {
            logger.info("Screen locked successfully")
        } else {
            logger.error("SACLockScreenImmediate returned \(result); falling back to ScreenSaverEngine")
            startScreenSaverFallback()
        }
    }

    private func startScreenSaverFallback() {
        let task = Process()
        task.executableURL = URL(
            fileURLWithPath: "/System/Library/CoreServices/ScreenSaverEngine.app/Contents/MacOS/ScreenSaverEngine"
        )

        do {
            try task.run()
        } catch {
            logger.error("Failed to start ScreenSaverEngine fallback: \(error.localizedDescription)")
        }
    }
}