import Foundation
import Testing
@testable import Oppi

@Suite("Workspace review models")
struct WorkspaceReviewModelsTests {

    @Test func diffHunkHeaderTextUsesUnifiedFormat() {
        let hunk = WorkspaceReviewDiffHunk(
            oldStart: 4,
            oldCount: 2,
            newStart: 4,
            newCount: 3,
            lines: []
        )

        #expect(hunk.headerText == "@@ -4,2 +4,3 @@")
    }

    @Test func diffLineKindPrefixesMatchDiffMarkers() {
        #expect(WorkspaceReviewDiffLine.Kind.context.prefix == " ")
        #expect(WorkspaceReviewDiffLine.Kind.added.prefix == "+")
        #expect(WorkspaceReviewDiffLine.Kind.removed.prefix == "-")
    }

    @Test func localDiffResponseBuildsHunksFromTexts() {
        let response = WorkspaceReviewDiffResponse.local(
            path: "Sources/App.swift",
            baselineText: "let a = 1\nlet b = 2\n",
            currentText: "let a = 1\nlet b = 3\nlet c = 4\n"
        )

        #expect(response.addedLines == 2)
        #expect(response.removedLines == 1)
        #expect(response.hunks.count == 1)

        let lines = response.hunks[0].lines
        #expect(lines.contains { $0.kind == .removed && $0.oldLine == 2 && $0.newLine == nil })
        #expect(lines.contains { $0.kind == .added && $0.oldLine == nil && $0.newLine == 2 })
        #expect(lines.contains { $0.kind == .added && $0.oldLine == nil && $0.newLine == 3 })
    }

    @Test func localDiffResponseUsesPrecomputedLines() {
        let precomputedLines = [
            DiffLine(kind: .context, text: "same"),
            DiffLine(kind: .removed, text: "before"),
            DiffLine(kind: .added, text: "after")
        ]

        let response = WorkspaceReviewDiffResponse.local(
            path: "Sources/App.swift",
            baselineText: "ignored old text",
            currentText: "ignored new text",
            precomputedLines: precomputedLines
        )

        #expect(response.addedLines == 1)
        #expect(response.removedLines == 1)
        #expect(response.hunks.count == 1)
        #expect(response.hunks[0].lines.map(\.kind) == [.context, .removed, .added])
    }

    @Test func reviewFileStatusLabelUsesGitStatusMapping() {
        let file = WorkspaceReviewFile(
            path: "README.md",
            status: "??",
            addedLines: nil,
            removedLines: nil,
            isStaged: false,
            isUnstaged: false,
            isUntracked: true,
            selectedSessionTouched: false
        )

        #expect(file.statusLabel == "Untracked")
    }

    @Test func promptTemplateQuickActionOptionHasSlashCommandProgressCopy() {
        let option = WorkspaceQuickActionOption(
            id: "prompt:grill-me",
            title: "Grill Me",
            commandName: "grill-me",
            description: "Stress-test selected files",
            argumentHint: "FILES",
            source: .prompt,
            sourceScope: "project",
            promptTemplateName: "grill-me"
        )

        #expect(option.progressTitle == "Starting /grill-me…")
    }

    @Test func reviewCommentReferenceSourceDecodesUnknownRawValueAsUnknown() throws {
        let decoded = try JSONDecoder().decode(
            ReviewCommentReferenceSource.self,
            from: Data(#""mystery_source""#.utf8)
        )

        #expect(decoded == .unknown)
        let encoded = try JSONEncoder().encode(decoded)
        #expect(String(decoding: encoded, as: UTF8.self) == #""unknown""#)
    }

    @Test func treePathCompareUsesDirectoryHierarchy() {
        let paths = [
            "Sources/Review/Row.swift",
            "README.md",
            "Sources/App.swift",
            "Sources/Review/Detail/View.swift"
        ]

        let sorted = paths.sorted { lhs, rhs in
            lhs.localizedTreePathCompare(to: rhs) == .orderedAscending
        }

        #expect(sorted == [
            "README.md",
            "Sources/App.swift",
            "Sources/Review/Detail/View.swift",
            "Sources/Review/Row.swift"
        ])
    }

    @Test func treePathCompareNormalizesRelativeSeparators() {
        #expect("./Sources\\App.swift".localizedTreePathCompare(to: "Sources/App.swift/") == .orderedSame)
    }


    @Test func fileDetailPhaseTreatsInitialNilStateAsLoading() {
        #expect(WorkspaceReviewFileDetailPhase.resolve(diff: nil, error: nil) == .loading)
    }

    @Test func fileDetailPhasePrefersLoadedContentOverStaleError() {
        let diff = WorkspaceReviewDiffResponse(
            workspaceId: "w1",
            path: "Sources/App.swift",
            baselineText: "old",
            currentText: "new",
            addedLines: 1,
            removedLines: 1,
            hunks: [],
            revisionCount: nil,
            cacheKey: nil
        )

        #expect(WorkspaceReviewFileDetailPhase.resolve(diff: diff, error: "boom") == .loaded(diff))
    }

    @Test func fileDetailToolbarKeepsActionMenuWhileAdjacentFileLoads() {
        let loaded = WorkspaceReviewFileDetailToolbarState.make(
            hasShareableContent: true,
            launchActionInFlightTitle: nil
        )
        let reloadingAfterNavigation = WorkspaceReviewFileDetailToolbarState.make(
            hasShareableContent: false,
            launchActionInFlightTitle: nil
        )

        #expect(loaded.showsShare)
        #expect(loaded.showsActionMenu)
        #expect(reloadingAfterNavigation.showsShare == false)
        #expect(reloadingAfterNavigation.showsActionMenu)
        #expect(reloadingAfterNavigation.actionMenuAccessibilityLabel == "Actions")
    }

    @Test func fileDetailToolbarDisablesActionMenuOnlyDuringLaunch() {
        let launching = WorkspaceReviewFileDetailToolbarState.make(
            hasShareableContent: true,
            launchActionInFlightTitle: "Starting /review…"
        )

        #expect(launching.showsActionMenu)
        #expect(launching.actionMenuDisabled)
    }

    @Test func commitDetailActionMenuKeepsNewSessionEnabledWhileCommitLoads() {
        let loading = CommitDetailActionMenuState.make(
            workspaceId: "w1",
            detail: nil,
            launchActionInFlightTitle: nil
        )

        #expect(loading.menuDisabled == false)
        #expect(loading.promptTemplatesDisabled)
    }

    @Test func commitDetailActionMenuEnablesPromptTemplatesAfterDetailLoads() {
        let loaded = CommitDetailActionMenuState.make(
            workspaceId: "w1",
            detail: GitCommitDetail(
                sha: "abc1234",
                message: "Test commit",
                date: "2026-07-11T00:00:00Z",
                author: "Chen",
                files: [GitCommitFileInfo(path: "Sources/App.swift", status: "M", addedLines: 1, removedLines: 0)],
                addedLines: 1,
                removedLines: 0
            ),
            launchActionInFlightTitle: nil
        )

        #expect(loaded.menuDisabled == false)
        #expect(loaded.promptTemplatesDisabled == false)
    }

    @Test func fileDetailFallsBackWhenSessionDiffIsEmptyButGitShowsLineChanges() {
        let file = makeReviewFile(addedLines: 2, removedLines: 5)
        let emptySessionDiff = makeReviewDiff(addedLines: 0, removedLines: 0, hunkCount: 0)

        #expect(WorkspaceReviewFileDetailView.shouldFallbackToWorkspaceDiff(
            sessionDiff: emptySessionDiff,
            file: file
        ))
    }

    @Test func fileDetailKeepsSessionDiffWhenItHasVisibleChanges() {
        let file = makeReviewFile(addedLines: 2, removedLines: 5)
        let sessionDiff = makeReviewDiff(addedLines: 1, removedLines: 1, hunkCount: 1)

        #expect(!WorkspaceReviewFileDetailView.shouldFallbackToWorkspaceDiff(
            sessionDiff: sessionDiff,
            file: file
        ))
    }

    private func makeReviewFile(addedLines: Int?, removedLines: Int?) -> WorkspaceReviewFile {
        WorkspaceReviewFile(
            path: "Sources/App.swift",
            status: "M",
            addedLines: addedLines,
            removedLines: removedLines,
            isStaged: false,
            isUnstaged: true,
            isUntracked: false,
            selectedSessionTouched: true
        )
    }

    @Test func quickActionNavigationCanBuildEmptySessionDestination() {
        #expect(QuickActionSessionNavDestination.empty(sessionId: "session-1") == QuickActionSessionNavDestination(
            id: "session-1",
            inputText: "",
            filePaths: [],
            fileDisplayPrefix: ""
        ))
    }

    @Test func newSessionNavigationAttachesBrowsedFile() {
        #expect(QuickActionSessionNavDestination.attaching(
            sessionId: "session-1",
            filePath: "CHANGELOG.md"
        ) == QuickActionSessionNavDestination(
            id: "session-1",
            inputText: "",
            filePaths: ["CHANGELOG.md"],
            fileDisplayPrefix: ""
        ))
    }

    @Test func newSessionNavigationAttachesSelectedFilesInDisplayOrder() {
        #expect(QuickActionSessionNavDestination.attaching(
            sessionId: "session-1",
            filePaths: ["Sources/App.swift", "README.md"]
        ) == QuickActionSessionNavDestination(
            id: "session-1",
            inputText: "",
            filePaths: ["Sources/App.swift", "README.md"],
            fileDisplayPrefix: ""
        ))
    }

    private func makeReviewDiff(
        addedLines: Int,
        removedLines: Int,
        hunkCount: Int
    ) -> WorkspaceReviewDiffResponse {
        let hunk = WorkspaceReviewDiffHunk(
            oldStart: 1,
            oldCount: 1,
            newStart: 1,
            newCount: 1,
            lines: []
        )
        return WorkspaceReviewDiffResponse(
            workspaceId: "w1",
            path: "Sources/App.swift",
            baselineText: "old",
            currentText: "new",
            addedLines: addedLines,
            removedLines: removedLines,
            hunks: Array(repeating: hunk, count: hunkCount),
            revisionCount: 1,
            cacheKey: "session"
        )
    }
}