import Foundation
import Testing
@testable import Hutch

private struct UpdateRepositoryInfoEnvelope: Decodable {
    let updateRepository: UpdateRepositoryInfoPayload
}

private struct UpdateRepositoryInfoPayload: Decodable {
    let id: Int
}

struct RepositorySettingsViewModelTests {

    @Test
    func saveInfoResponseDecodesMinimalRepositoryPayload() throws {
        let json = """
        {
            "data": {
                "updateRepository": {
                    "id": 42
                }
            }
        }
        """

        let decoded = try JSONDecoder().decode(
            GraphQLResponse<UpdateRepositoryInfoEnvelope>.self,
            from: Data(json.utf8)
        )

        #expect(decoded.data?.updateRepository.id == 42)
    }

    @Test
    func gitCanonicalEntityAddsMissingTilde() {
        #expect(RepositorySettingsViewModel.gitCanonicalEntity(from: "alice") == "~alice")
        #expect(RepositorySettingsViewModel.gitCanonicalEntity(from: "~alice") == "~alice")
        #expect(RepositorySettingsViewModel.gitCanonicalEntity(from: "  alice  ") == "~alice")
    }

    @Test
    func gitHeadReferenceUsesFullBranchRef() {
        #expect(RepositorySettingsViewModel.gitHeadReference(from: "main") == "refs/heads/main")
        #expect(RepositorySettingsViewModel.gitHeadReference(from: " refs/heads/dev ") == "refs/heads/dev")
    }

    @Test
    @MainActor
    func unchangedHeadIsOmittedFromSaveInput() {
        let viewModel = RepositorySettingsViewModel(
            repository: makeRepository(headName: "refs/heads/main"),
            branches: [Reference(name: "refs/heads/main", target: nil)],
            client: SRHTClient(token: "test-token")
        )

        #expect(viewModel.selectedHeadReferenceForSave() == nil)
    }

    @Test
    @MainActor
    func changedHeadUsesSelectedBranchReference() {
        let viewModel = RepositorySettingsViewModel(
            repository: makeRepository(headName: "refs/heads/main"),
            branches: [
                Reference(name: "refs/heads/main", target: nil),
                Reference(name: "refs/heads/dev", target: nil)
            ],
            client: SRHTClient(token: "test-token")
        )
        viewModel.editedHead = "dev"

        #expect(viewModel.selectedHeadReferenceForSave() == "refs/heads/dev")
    }

    @Test
    @MainActor
    func bareRepositoryOmitsHeadWhenNoBranchExists() {
        let viewModel = RepositorySettingsViewModel(
            repository: makeRepository(headName: nil),
            branches: [],
            client: SRHTClient(token: "test-token")
        )

        #expect(viewModel.selectedHeadReferenceForSave() == nil)
        viewModel.editedHead = "main"
        #expect(viewModel.selectedHeadReferenceForSave() == nil)
    }

    @MainActor
    private func makeRepository(headName: String?) -> RepositorySummary {
        RepositorySummary(
            id: 1,
            rid: "rid-1",
            service: .git,
            name: "repo",
            description: "desc",
            visibility: .public,
            updated: .now,
            owner: Entity(canonicalName: "~owner"),
            head: headName.map { Reference(name: $0, target: nil) }
        )
    }
}