import Testing
import UIKit
@testable import Oppi

@MainActor
@Suite("Tool timeline row full-screen activation")
struct ToolTimelineRowFullScreenActivationTests {
    private struct HostHarness {
        let window: UIWindow
        let host: UIViewController
    }

    @Test("bash output activation opens full screen instead of copying")
    func bashOutputActivationOpensFullScreen() throws {
        let harness = makeHostHarness()
        let host = harness.host
        let view = ToolTimelineRowContentView(configuration: makeTimelineToolConfiguration(
            expandedContent: .bash(command: "echo hi", output: "hi", unwrapped: true),
            copyCommandText: "echo hi",
            copyOutputText: "hi",
            isExpanded: true
        ))

        host.view.addSubview(view)
        view.frame = host.view.bounds
        host.view.layoutIfNeeded()

        view.performOutputActivation()

        let presented = try #require(host.presentedViewController as? FullScreenCodeViewController)
        #expect(presented.modalPresentationStyle == .pageSheet)

        host.dismiss(animated: false)
        harness.window.isHidden = true
    }

    @Test("expanded code activation opens full screen")
    func expandedCodeActivationOpensFullScreen() throws {
        let harness = makeHostHarness()
        let host = harness.host
        let view = ToolTimelineRowContentView(configuration: makeTimelineToolConfiguration(
            expandedContent: .code(text: "struct App {}", language: .swift, startLine: 1, filePath: "App.swift"),
            copyCommandText: "read App.swift",
            copyOutputText: "struct App {}",
            toolNamePrefix: "read",
            isExpanded: true
        ))

        host.view.addSubview(view)
        view.frame = host.view.bounds
        host.view.layoutIfNeeded()

        view.performExpandedActivation()

        let presented = try #require(host.presentedViewController as? FullScreenCodeViewController)
        #expect(presented.modalPresentationStyle == .pageSheet)

        host.dismiss(animated: false)
        harness.window.isHidden = true
    }

    @Test("expanded markdown activation opens full screen")
    func expandedMarkdownActivationOpensFullScreen() throws {
        let harness = makeHostHarness()
        let host = harness.host
        let view = ToolTimelineRowContentView(configuration: makeTimelineToolConfiguration(
            expandedContent: .markdown(text: "# Header\n\nBody"),
            copyOutputText: "# Header\n\nBody",
            toolNamePrefix: "read",
            isExpanded: true
        ))

        host.view.addSubview(view)
        view.frame = host.view.bounds
        host.view.layoutIfNeeded()

        view.performExpandedActivation()

        let presented = try #require(host.presentedViewController as? FullScreenCodeViewController)
        #expect(presented.modalPresentationStyle == .pageSheet)

        host.dismiss(animated: false)
        harness.window.isHidden = true
    }

    @Test("expanded large markdown activation opens full screen")
    func expandedLargeMarkdownActivationOpensFullScreen() throws {
        let largeMarkdown = (0..<1_200).map { index in
            """
            ## Section \(index)

            **Strong text \(index)** with `inline code`.
            """
        }.joined(separator: "\n\n")
        #expect(largeMarkdown.utf8.count > 64 * 1024)

        let harness = makeHostHarness()
        let host = harness.host
        let view = ToolTimelineRowContentView(configuration: makeTimelineToolConfiguration(
            expandedContent: .markdown(text: largeMarkdown),
            copyOutputText: largeMarkdown,
            toolNamePrefix: "web_fetch",
            isExpanded: true
        ))

        host.view.addSubview(view)
        view.frame = host.view.bounds
        host.view.layoutIfNeeded()

        view.performExpandedActivation()

        let presented = try #require(host.presentedViewController as? FullScreenCodeViewController)
        #expect(presented.modalPresentationStyle == .pageSheet)

        host.dismiss(animated: false)
        harness.window.isHidden = true
    }

    @Test("streaming markdown full screen content carries markdown render hint")
    func streamingMarkdownFullScreenContentCarriesMarkdownRenderHint() throws {
        let markdown = "# Streaming plan\n\nBody"
        let configuration = makeTimelineToolConfiguration(
            expandedContent: .markdown(text: markdown),
            copyOutputText: markdown,
            toolNamePrefix: "write",
            isExpanded: true,
            isDone: false
        )
        let interactionPolicy = ToolTimelineRowInteractionPolicy.forExpandedContent(
            .markdown(text: markdown),
            isDone: false
        )
        let sourceStream = SourceTraceStream(
            text: markdown,
            filePath: nil,
            isDone: false,
            finalContent: nil
        )

        let fullScreenContent = ToolTimelineRowFullScreenSupport.fullScreenContent(
            configuration: configuration,
            outputCopyText: configuration.copyOutputText,
            interactionPolicy: interactionPolicy,
            terminalStream: nil,
            sourceStream: sourceStream
        )

        guard case .liveSource(let snapshot, _) = fullScreenContent else {
            Issue.record("Expected .liveSource for streaming markdown content")
            return
        }
        guard case .markdown(let content, _, _)? = snapshot.finalContent else {
            Issue.record("Expected streaming markdown to carry a markdown render hint")
            return
        }
        #expect(content == markdown)
        #expect(!snapshot.isDone)
    }

    @Test("streaming HTML full screen content carries HTML render hint")
    func streamingHTMLFullScreenContentCarriesHTMLRenderHint() throws {
        let html = "<h1>Streaming</h1>"
        let configuration = makeTimelineToolConfiguration(
            expandedContent: .code(text: html, language: .html, startLine: 1, filePath: "report.html"),
            copyOutputText: html,
            toolNamePrefix: "write",
            isExpanded: true,
            isDone: false
        )
        let interactionPolicy = ToolTimelineRowInteractionPolicy.forExpandedContent(
            .code(text: html, language: .html, startLine: 1, filePath: "report.html"),
            isDone: false
        )
        let sourceStream = SourceTraceStream(
            text: html,
            filePath: "report.html",
            isDone: false,
            finalContent: nil
        )

        let fullScreenContent = ToolTimelineRowFullScreenSupport.fullScreenContent(
            configuration: configuration,
            outputCopyText: configuration.copyOutputText,
            interactionPolicy: interactionPolicy,
            terminalStream: nil,
            sourceStream: sourceStream
        )

        guard case .liveSource(let snapshot, _) = fullScreenContent else {
            Issue.record("Expected .liveSource for streaming HTML content")
            return
        }
        guard case .html(let content, let filePath)? = snapshot.finalContent else {
            Issue.record("Expected streaming HTML to carry an HTML render hint")
            return
        }
        #expect(content == html)
        #expect(filePath == "report.html")
        #expect(!snapshot.isDone)
    }

    @Test("expanded text activation opens full screen")
    func expandedTextActivationOpensFullScreen() throws {
        let harness = makeHostHarness()
        let host = harness.host
        let view = ToolTimelineRowContentView(configuration: makeTimelineToolConfiguration(
            expandedContent: .text(text: "Saved to journal: 2026-03-07.md", language: nil),
            copyOutputText: "Saved to journal: 2026-03-07.md",
            toolNamePrefix: "recall",
            isExpanded: true
        ))

        host.view.addSubview(view)
        view.frame = host.view.bounds
        host.view.layoutIfNeeded()

        view.performExpandedActivation()

        let presented = try #require(host.presentedViewController as? FullScreenCodeViewController)
        #expect(presented.modalPresentationStyle == .pageSheet)

        host.dismiss(animated: false)
        harness.window.isHidden = true
    }

    @Test("streaming file-like tools use live source full screen content")
    func streamingFileLikeToolsUseLiveSourceFullScreenContent() throws {
        let configuration = makeTimelineToolConfiguration(
            expandedContent: .code(
                text: "let draft = 1",
                language: .swift,
                startLine: 1,
                filePath: "Draft.swift"
            ),
            copyOutputText: "let draft = 1",
            toolNamePrefix: "write",
            isExpanded: true,
            isDone: false
        )
        let interactionPolicy = ToolTimelineRowInteractionPolicy.forExpandedContent(
            .code(text: "let draft = 1", language: .swift, startLine: 1, filePath: "Draft.swift"), isDone: false
        )
        let sourceStream = SourceTraceStream(
            text: "let draft = 1",
            filePath: "Draft.swift",
            isDone: false,
            finalContent: nil
        )

        let fullScreenContent = ToolTimelineRowFullScreenSupport.fullScreenContent(
            configuration: configuration,
            outputCopyText: configuration.copyOutputText,
            interactionPolicy: interactionPolicy,
            terminalStream: nil,
            sourceStream: sourceStream
        )

        guard case .liveSource(let snapshot, _) = fullScreenContent else {
            Issue.record("Expected .liveSource for streaming file-like content")
            return
        }
        #expect(snapshot.text == "let draft = 1")
        #expect(snapshot.filePath == "Draft.swift")
        #expect(!snapshot.isDone)
    }

    private func makeHostHarness() -> HostHarness {
        let scene = UIApplication.shared.connectedScenes
            .compactMap({ $0 as? UIWindowScene })
            .first
        let window: UIWindow
        if let scene {
            window = UIWindow(windowScene: scene)
        } else {
            fatalError("Missing UIWindowScene for ToolTimelineRowFullScreenActivationTests")
        }
        window.frame = CGRect(x: 0, y: 0, width: 390, height: 844)
        let host = UIViewController()
        window.rootViewController = host
        window.makeKeyAndVisible()
        host.loadViewIfNeeded()
        return HostHarness(window: window, host: host)
    }
}