import Testing
import UIKit
@testable import Oppi

@MainActor
@Suite("ThinkingTimelineRowContentView")
struct ThinkingRowContentViewTests {
    @Test func streamingOverflowKeepsInnerScrollDisabledAndAutoFollowsTail() throws {
        // Establish bounds first (mirrors real collection view lifecycle where
        // cells have valid frames before apply() runs on content updates).
        let view = ThinkingTimelineRowContentView(configuration: ThinkingTimelineRowConfiguration(
            isDone: false, previewText: "seed", fullText: nil
        ))
        _ = fittedTimelineSize(for: view, width: 360)

        // Now grow to overflow — apply() drives followTail synchronously.
        view.configuration = ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: Array(repeating: "streaming thought line", count: 300).joined(separator: "\n"),
            fullText: nil
        )

        let scrollView = try #require(privateScrollView(in: view))
        #expect(!scrollView.isScrollEnabled)
        #expect(!scrollView.isUserInteractionEnabled)
        #expect(scrollView.contentOffset.y > 0, "Streaming overflow should tail-follow inside capped bubble")
    }

    @Test func streamingOverflowRespectsConfiguredBubbleCap() throws {
        let text = Array(repeating: "streaming thought line", count: 300).joined(separator: "\n")

        let view = ThinkingTimelineRowContentView(configuration: ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: text,
            fullText: nil,
            maxBubbleHeight: ThinkingRowHeightPolicy.defaultMaxBubbleHeight
        ))
        _ = fittedTimelineSize(for: view, width: 360)

        let bubbleHeight = try #require(privateBubbleHeightConstraintConstant(in: view))

        #expect(bubbleHeight == ThinkingRowHeightPolicy.defaultMaxBubbleHeight)
    }

    @Test func thinkingRowDoesNotInstallFloatingFullScreenButton() {
        let overflowConfig = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "",
            fullText: Array(repeating: "reasoning", count: 320).joined(separator: "\n")
        )
        let shortConfig = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "Short thought",
            fullText: nil
        )

        let overflowView = ThinkingTimelineRowContentView(configuration: overflowConfig)
        _ = fittedTimelineSize(for: overflowView, width: 360)

        let shortView = ThinkingTimelineRowContentView(configuration: shortConfig)
        _ = fittedTimelineSize(for: shortView, width: 360)

        #expect(fullScreenButton(in: overflowView) == nil)
        #expect(fullScreenButton(in: shortView) == nil)
    }

    @Test func overflowContextMenuIncludesOpenFullScreenAndCopy() throws {
        let config = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "",
            fullText: Array(repeating: "line", count: 320).joined(separator: "\n")
        )

        let view = ThinkingTimelineRowContentView(configuration: config)
        _ = fittedTimelineSize(for: view, width: 360)

        _ = try #require(privateBubbleView(in: view))
        let menu = try #require(view.contextMenuForTesting())

        #expect(timelineActionTitles(in: menu) == ["Open Full Screen", "Copy"])
    }

    @Test func overflowRegistersPinchAndDoubleTapGesturesButNoSingleTapActivation() throws {
        let config = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "",
            fullText: Array(repeating: "line", count: 320).joined(separator: "\n")
        )

        let view = ThinkingTimelineRowContentView(configuration: config)
        _ = fittedTimelineSize(for: view, width: 360)

        let bubbleView = try #require(privateBubbleView(in: view))
        let recognizers = bubbleView.gestureRecognizers ?? []
        let hasPinch = recognizers.contains { $0 is UIPinchGestureRecognizer }
        let hasDoubleTap = recognizers.contains {
            guard let tap = $0 as? UITapGestureRecognizer else { return false }
            return tap.numberOfTapsRequired == 2
        }
        let hasSingleTap = recognizers.contains {
            guard let tap = $0 as? UITapGestureRecognizer else { return false }
            return tap.numberOfTapsRequired == 1
        }

        #expect(hasPinch)
        #expect(hasDoubleTap)
        #expect(!hasSingleTap)
    }

    @Test func selectedTextEditMenuPrependsPiSubmenu() throws {
        let router = SelectedTextPiActionRouter { _ in }
        let interactionCtx = TimelineInteractionContext()
        interactionCtx.selectedTextPiRouter = router
        interactionCtx.sessionId = "session-1"
        let config = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "",
            fullText: "Alpha beta gamma",
            interactionContext: interactionCtx
        )

        let view = ThinkingTimelineRowContentView(configuration: config)
        _ = fittedTimelineSize(for: view, width: 360)

        let label = try #require(privateTextLabel(in: view))
        let copyAction = UIAction(title: "Copy") { _ in }
        let menu = try #require(view.textView(
            label,
            editMenuForTextIn: NSRange(location: 0, length: 5),
            suggestedActions: [copyAction]
        ))

        let piMenu = try #require(menu.children.first as? UIMenu)
        #expect(piMenu.title == "π")
        #expect(timelineActionTitles(in: piMenu) == ["Explain", "Do it", "Fix", "Refactor", "Add to Prompt", "New Session"])
        let copyMenuAction = try #require(menu.children.dropFirst().first as? UIAction)
        #expect(copyMenuAction.title == "Copy")
    }

    @Test func selectedTextModeKeepsOverflowFullScreenGesturesAndDisablesInlineSelection() throws {
        let router = SelectedTextPiActionRouter { _ in }
        let interactionCtx = TimelineInteractionContext()
        interactionCtx.selectedTextPiRouter = router
        interactionCtx.sessionId = "session-1"
        let config = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "",
            fullText: Array(repeating: "line", count: 320).joined(separator: "\n"),
            interactionContext: interactionCtx
        )

        let view = ThinkingTimelineRowContentView(configuration: config)
        _ = fittedTimelineSize(for: view, width: 360)

        let recognizers = timelineAllGestureRecognizers(in: view)
        let pinchGesture = try #require(recognizers.first { $0 is UIPinchGestureRecognizer })
        let doubleTapGesture = try #require(recognizers.first {
            guard let tap = $0 as? UITapGestureRecognizer else { return false }
            return tap.numberOfTapsRequired == 2
        })

        #expect(pinchGesture.isEnabled)
        #expect(doubleTapGesture.isEnabled)

        let scrollView = try #require(privateScrollView(in: view))
        #expect(!scrollView.isUserInteractionEnabled)

        let label = try #require(privateTextLabel(in: view))
        #expect(!label.isSelectable)
        #expect(fullScreenButton(in: view) == nil)
    }

    @Test func selectedTextModeAllowsInlineSelectionWhenThinkingFitsBubble() throws {
        let router = SelectedTextPiActionRouter { _ in }
        let interactionCtx = TimelineInteractionContext()
        interactionCtx.selectedTextPiRouter = router
        interactionCtx.sessionId = "session-1"
        let config = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "Short thought",
            fullText: nil,
            interactionContext: interactionCtx
        )

        let view = ThinkingTimelineRowContentView(configuration: config)
        _ = fittedTimelineSize(for: view, width: 360)

        let label = try #require(privateTextLabel(in: view))
        #expect(label.isSelectable)

        let recognizers = timelineAllGestureRecognizers(in: view)
        let pinchGesture = try #require(recognizers.first { $0 is UIPinchGestureRecognizer })
        let doubleTapGesture = try #require(recognizers.first {
            guard let tap = $0 as? UITapGestureRecognizer else { return false }
            return tap.numberOfTapsRequired == 2
        })

        #expect(!pinchGesture.isEnabled)
        #expect(!doubleTapGesture.isEnabled)
    }

    // MARK: - Streaming plain text optimization

    @Test func streamingPreservesRawMarkdownSyntax() throws {
        let config = ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: "Thinking about **bold** and `code`",
            fullText: nil
        )

        let view = ThinkingTimelineRowContentView(configuration: config)
        _ = fittedTimelineSize(for: view, width: 360)

        let label = try #require(privateTextLabel(in: view))
        // Streaming skips markdown parsing, so raw ** and ` survive in the label.
        #expect(
            label.text?.contains("**bold**") == true,
            "Streaming should preserve raw markdown syntax (no parsing)"
        )
    }

    @Test func doneStripsMarkdownSyntax() throws {
        let config = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "",
            fullText: "Thinking about **bold** and `code`"
        )

        let view = ThinkingTimelineRowContentView(configuration: config)
        _ = fittedTimelineSize(for: view, width: 360)

        let label = try #require(privateTextLabel(in: view))
        // Done state parses markdown, so ** is stripped and "bold" rendered with font traits.
        #expect(
            label.text?.contains("**bold**") != true,
            "Done state should parse markdown (no raw ** in text)"
        )
    }

    @Test func renderSignatureSkipsRedundantStreamingUpdate() throws {
        let view = ThinkingTimelineRowContentView(configuration: ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: "Initial thought",
            fullText: nil
        ))
        _ = fittedTimelineSize(for: view, width: 360)

        let label = try #require(privateTextLabel(in: view))
        let firstText = label.text

        let sig1 = try #require(privateRenderSignature(in: view))
        view.configuration = ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: "Initial thought",
            fullText: nil
        )
        let sig2 = try #require(privateRenderSignature(in: view))
        #expect(sig1 == sig2, "Render signature should not change for identical content")
        #expect(label.text == firstText)
    }

    @Test func renderSignatureChangesWhenTextGrows() throws {
        let view = ThinkingTimelineRowContentView(configuration: ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: "Short",
            fullText: nil
        ))
        _ = fittedTimelineSize(for: view, width: 360)

        let sig1 = try #require(privateRenderSignature(in: view))

        view.configuration = ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: "Short thought that grew longer",
            fullText: nil
        )
        let sig2 = try #require(privateRenderSignature(in: view))
        #expect(sig1 != sig2, "Render signature should change when text changes")
    }

    @Test func transitionFromStreamingToDoneRendersMarkdown() throws {
        let text = "Thinking about **bold** patterns"
        let view = ThinkingTimelineRowContentView(configuration: ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: text,
            fullText: nil
        ))
        _ = fittedTimelineSize(for: view, width: 360)

        let label = try #require(privateTextLabel(in: view))
        #expect(
            label.text?.contains("**bold**") == true,
            "Streaming should preserve raw markdown"
        )

        // Transition to done — markdown parsed, ** stripped.
        view.configuration = ThinkingTimelineRowConfiguration(
            isDone: true,
            previewText: "",
            fullText: text
        )
        _ = fittedTimelineSize(for: view, width: 360)
        #expect(
            label.text?.contains("**bold**") != true,
            "Done transition should parse markdown"
        )
    }

    @Test func streamingLargeTextSkipsMarkdownParsing() throws {
        // 200 lines — representative of a mid-stream thinking burst.
        // The key assertion: raw markdown survives (no parsing happened).
        let longText = (0..<200).map { "Line \($0): thinking about **patterns** and `design`" }.joined(separator: "\n")
        let config = ThinkingTimelineRowConfiguration(
            isDone: false,
            previewText: longText,
            fullText: nil
        )

        let view = ThinkingTimelineRowContentView(configuration: config)
        _ = fittedTimelineSize(for: view, width: 360)

        let label = try #require(privateTextLabel(in: view))
        #expect(
            label.text?.contains("**patterns**") == true,
            "Large streaming text must skip markdown parsing"
        )
    }
}

@MainActor
private func privateScrollView(in view: ThinkingTimelineRowContentView) -> UIScrollView? {
    Mirror(reflecting: view).children.first { $0.label == "scrollView" }?.value as? UIScrollView
}

@MainActor
private func privateBubbleView(in view: ThinkingTimelineRowContentView) -> UIView? {
    Mirror(reflecting: view).children.first { $0.label == "bubbleView" }?.value as? UIView
}

@MainActor
private func privateTextLabel(in view: ThinkingTimelineRowContentView) -> UITextView? {
    Mirror(reflecting: view).children.first { $0.label == "textLabel" }?.value as? UITextView
}

@MainActor
private func privateRenderSignature(in view: ThinkingTimelineRowContentView) -> Int? {
    Mirror(reflecting: view).children.first { $0.label == "renderSignature" }?.value as? Int
}

@MainActor
private func privateBubbleHeightConstraintConstant(in view: ThinkingTimelineRowContentView) -> CGFloat? {
    (Mirror(reflecting: view).children.first { $0.label == "bubbleHeightConstraint" }?.value as? NSLayoutConstraint)?.constant
}

@MainActor
private func fullScreenButton(in view: ThinkingTimelineRowContentView) -> UIButton? {
    timelineAllViews(in: view)
        .compactMap { $0 as? UIButton }
        .first { $0.accessibilityIdentifier == "thinking.expand-full-screen" }
}