using Test
using GeneralUtils
using GeneralUtils.garageS3
using AWS
using Random: randstring
using UUIDs: uuid4

@testset "garageS3.jl" begin

    # --- SimpleCredentials ---

    @testset "SimpleCredentials construction" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("AKIAIOSFODNN7EXAMPLE", "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY", "sessiontoken")
        @test creds.access_key_id == "AKIAIOSFODNN7EXAMPLE"
        @test creds.secret_key == "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY"
        @test creds.token == "sessiontoken"
    end

    @testset "SimpleCredentials AWS extensions" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key1", "secret1", "token1")
        @test AWS.credentials(creds) === creds
        @test AWS.check_credentials(creds) === creds
        @test AWS.refresh!(creds; force=false) === creds
        @test AWS.refresh!(creds) === creds
    end

    # --- GarageConfig ---

    @testset "GarageConfig construction" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
        config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com", "us-east-1", creds)
        @test config.endpoint == "https://s3-api.example.com"
        @test config.region == "us-east-1"
        @test config.credentials === creds
        @test config isa AWS.AbstractAWSConfig
    end

    @testset "GarageConfig AWS extensions" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
        config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com", "garage", creds)
        @test AWS.region(config) == "garage"
        @test AWS.credentials(config) === creds
    end

    # --- AWS.generate_service_url ---

    @testset "generate_service_url with resource starting with /" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
        config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com", "garage", creds)
        url = AWS.generate_service_url(config, "s3", "/mybucket/key")
        @test url == "https://s3-api.example.com/mybucket/key"
    end

    @testset "generate_service_url with resource NOT starting with /" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
        config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com", "garage", creds)
        url = AWS.generate_service_url(config, "s3", "mybucket/key")
        @test url == "https://s3-api.example.com/mybucket/key"
    end

    @testset "generate_service_url strips trailing slashes from endpoint" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
        config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com/", "garage", creds)
        url = AWS.generate_service_url(config, "s3", "/bucket/key")
        @test url == "https://s3-api.example.com/bucket/key"
    end

    @testset "generate_service_url with multiple trailing slashes" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
        config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com///", "garage", creds)
        url = AWS.generate_service_url(config, "s3", "/bucket/key")
        @test url == "https://s3-api.example.com/bucket/key"
    end

    @testset "generate_service_url preserves http endpoint" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
        config = GeneralUtils.garageS3.GarageConfig("http://localhost:3900", "garage", creds)
        url = AWS.generate_service_url(config, "s3", "/bucket/key")
        @test url == "http://localhost:3900/bucket/key"
    end

    @testset "generate_service_url with http endpoint and trailing slash" begin
        creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
        config = GeneralUtils.garageS3.GarageConfig("http://localhost:3900/", "garage", creds)
        url = AWS.generate_service_url(config, "s3", "/bucket/key")
        @test url == "http://localhost:3900/bucket/key"
    end

    # --- GarageStorage ---

    @testset "GarageStorage default region" begin
        storage = GarageStorage("https://s3-api.example.com", "key123", "secret123", "mybucket")
        @test storage.bucket == "mybucket"
        @test storage.config.region == "garage"
        @test storage.config.endpoint == "https://s3-api.example.com"
    end

    @testset "GarageStorage custom region" begin
        storage = GarageStorage("https://s3-api.example.com", "key123", "secret123", "mybucket"; region="custom-region")
        @test storage.config.region == "custom-region"
    end

    @testset "GarageStorage endpoint without https" begin
        storage = GarageStorage("http://localhost:3900", "key", "secret", "mybucket")
        @test storage.config.endpoint == "http://localhost:3900"
    end

    @testset "GarageStorage credentials are SimpleCredentials" begin
        storage = GarageStorage("https://s3-api.example.com", "mykey", "mysecret", "mybucket")
        @test storage.config.credentials isa GeneralUtils.garageS3.SimpleCredentials
        @test storage.config.credentials.access_key_id == "mykey"
        @test storage.config.credentials.secret_key == "mysecret"
        @test storage.config.credentials.token == ""
    end

    @testset "GarageStorage returns correct struct type" begin
        storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
        @test storage isa GarageStorage
    end

    # --- put_file callable struct ---

    @testset "put_file callable struct type is correct" begin
        storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
        uploader = put_file(storage)
        @test uploader isa put_file
    end

    @testset "put_file callable struct construction stores storage" begin
        storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
        uploader = put_file(storage)
        @test uploader.storage === storage
    end

    # --- Integration tests with real Garage server ---

    @testset "integration: full CRUD lifecycle" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        test_key = "testrunner-$(randstring(8)).json"
        test_data = "{\"status\": \"testing\", \"runner\": \"runtests\"}"

        # Test put_file(storage, key, data) - successful upload
        result = put_file(storage, test_key, test_data)
        @test result !== nothing
        @test haskey(result, :api)
        @test haskey(result, :web)
        @test occursin("testbucket", result.api)
        @test occursin("s3-api", result.api)
        @test occursin("s3-web", result.web)

        # Test get_file - retrieve the uploaded data
        downloaded = get_file(storage, test_key)
        @test downloaded !== nothing
        @test downloaded isa Vector{UInt8}
        @test String(downloaded) == test_data

        # Test list_files - verify key appears in listing
        keys = list_files(storage)
        @test test_key in keys

        # Test put_file callable struct
        test_key2 = "testrunner-callable-$(randstring(8)).json"
        callable_result = put_file(storage)(test_key2, "callable-data")
        @test callable_result !== nothing
        @test callable_result.api != ""
        @test callable_result.web != ""

        # Verify callable upload
        downloaded2 = get_file(storage, test_key2)
        @test String(downloaded2) == "callable-data"

        # Test delete_file
        delete_file(storage, test_key)
        deleted_data = get_file(storage, test_key)
        @test deleted_data === nothing

        # Cleanup callable test object
        delete_file(storage, test_key2)
        deleted_data2 = get_file(storage, test_key2)
        @test deleted_data2 === nothing
    end

    @testset "integration: nested key CRUD - single level" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        test_key = "single/slash/test-$(randstring(8)).json"
        test_data = "{\"level\": 1, \"nested\": true}"

        # Upload nested key
        result = put_file(storage, test_key, test_data)
        @test result isa NamedTuple
        @test haskey(result, :api)
        @test haskey(result, :web)
        @test occursin("single/slash/", result.api)
        @test occursin("single/slash/", result.web)

        # Download and verify
        downloaded = get_file(storage, test_key)
        @test downloaded !== nothing
        @test String(downloaded) == test_data

        # Verify in listing
        keys = list_files(storage)
        @test test_key in keys

        # Delete and verify gone
        delete_file(storage, test_key)
        @test get_file(storage, test_key) === nothing
    end

    @testset "integration: nested key CRUD - multi level" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        test_key = "deep/nested/folder/subfolder/test-$(randstring(8)).json"
        test_data = "{\"deeply\": \"nested\", \"path\": \"deep/nested/folder/subfolder/\"}"

        # Upload deeply nested key
        result = put_file(storage, test_key, test_data)
        @test result isa NamedTuple
        @test occursin("deep/nested/folder/subfolder/", result.api)
        @test occursin("s3-api", result.api)
        @test occursin("s3-web", result.web)

        # Download and verify
        downloaded = get_file(storage, test_key)
        @test downloaded !== nothing
        @test String(downloaded) == test_data

        # Verify in listing
        keys = list_files(storage)
        @test test_key in keys

        # Delete and verify gone
        delete_file(storage, test_key)
        @test get_file(storage, test_key) === nothing
    end

    @testset "integration: nested key with callable struct" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        test_key = "callable/nested/test-$(randstring(8)).json"
        test_data = "{\"callable\": true, \"nested\": true}"

        uploader = put_file(storage)
        result = uploader(test_key, test_data)
        @test result isa NamedTuple
        @test occursin("callable/nested/", result.api)

        downloaded = get_file(storage, test_key)
        @test String(downloaded) == test_data

        delete_file(storage, test_key)
        @test get_file(storage, test_key) === nothing
    end

    @testset "integration: get_file returns nothing for missing key" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )
        result = get_file(storage, "nonexistent-key-$(randstring(8)).json")
        @test result === nothing
    end

    @testset "integration: binary data upload and download" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        test_key = "testrunner-binary-$(randstring(8)).bin"
        binary_data = UInt8[0x00, 0x01, 0x02, 0xFF, 0xFE, 0xFD, 0x48, 0x65, 0x6c, 0x6c, 0x6f]

        result = put_file(storage, test_key, binary_data)
        @test result !== nothing

        downloaded = get_file(storage, test_key)
        @test downloaded == binary_data

        delete_file(storage, test_key)
        @test get_file(storage, test_key) === nothing
    end

    @testset "integration: list_files returns vector of strings" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        test_key = "testrunner-list-$(randstring(8)).txt"
        put_file(storage, test_key, "list-test-data")

        keys = list_files(storage)
        @test keys isa Vector{String}
        @test test_key in keys

        delete_file(storage, test_key)
    end

    @testset "integration: url format validation" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        test_key = "testrunner-url-$(randstring(8)).json"
        result = put_file(storage, test_key, "url-test")

        @test startswith(result.api, "https://s3-api.yiem.cc/testbucket/")
        @test occursin("/$test_key", result.api)
        @test startswith(result.web, "https://s3-web.yiem.cc/testbucket/")
        @test occursin("/$test_key", result.web)

        delete_file(storage, test_key)
    end

    # --- Lifecycle configuration tests ---

    @testset "LifecycleExpiration construction with default ID" begin
        rule = LifecycleExpiration("uploads/", 30)
        @test rule.prefix == "uploads/"
        @test rule.days == 30
        @test rule.enabled == true
        @test rule.id == "expire-uploads_-30d"
    end

    @testset "LifecycleExpiration construction with custom ID" begin
        rule = LifecycleExpiration("data/", 7; id="my-rule")
        @test rule.prefix == "data/"
        @test rule.days == 7
        @test rule.enabled == true
        @test rule.id == "my-rule"
    end

    @testset "LifecycleExpiration with disabled flag" begin
        rule = LifecycleExpiration("temp/", 1; enabled=false)
        @test rule.enabled == false
    end

    @testset "LifecycleExpiration with nested prefix" begin
        rule = LifecycleExpiration("deep/nested/", 90)
        @test rule.prefix == "deep/nested/"
    end

    @testset "build_lifecycle_xml generates valid XML" begin
        rule = LifecycleExpiration("test/", 14; id="test-rule")
        xml = build_lifecycle_xml(rule)
        @test occursin("<?xml version=\"1.0\"", xml)
        @test occursin("<LifecycleConfiguration>", xml)
        @test occursin("<ID>test-rule</ID>", xml)
        @test occursin("<Prefix>test/</Prefix>", xml)
        @test occursin("<Status>Enabled</Status>", xml)
        @test occursin("<Days>14</Days>", xml)
        @test occursin("</LifecycleConfiguration>", xml)
    end

    @testset "build_lifecycle_xml generates disabled rule XML" begin
        rule = LifecycleExpiration("temp/", 1; enabled=false)
        xml = build_lifecycle_xml(rule)
        @test occursin("<Status>Disabled</Status>", xml)
    end

    @testset "integration: lifecycle CRUD - set and get" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        # Should return nothing when no lifecycle is set
        lc = get_lifecycle(storage)
        @test lc === nothing

        # Set lifecycle
        rule = LifecycleExpiration("lifecycle-test/", 45)
        set_lifecycle_expiration(storage, rule)

        # Should return configuration after setting
        lc = get_lifecycle(storage)
        @test lc !== nothing
        @test haskey(lc, "Rule")

        # Delete lifecycle
        delete_lifecycle(storage)

        # Should return nothing after deletion
        lc = get_lifecycle(storage)
        @test lc === nothing
    end

    @testset "integration: lifecycle with nested prefix" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        rule = LifecycleExpiration("a/b/c/", 20; id="nested-rule")
        set_lifecycle_expiration(storage, rule)

        lc = get_lifecycle(storage)
        @test lc !== nothing
        @test haskey(lc, "Rule")

        delete_lifecycle(storage)
        @test get_lifecycle(storage) === nothing
    end

    @testset "integration: lifecycle enabled/disabled toggle" begin
        storage = GarageStorage(
            "https://s3-api.yiem.cc",
            "GKb080154a2e5b19100b1b2c6e",
            "a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
            "testbucket"
        )

        # Set enabled rule
        rule1 = LifecycleExpiration("toggle/", 10; enabled=true)
        set_lifecycle_expiration(storage, rule1)
        lc = get_lifecycle(storage)
        @test lc !== nothing

        # Override with disabled rule
        rule2 = LifecycleExpiration("toggle/", 10; enabled=false)
        set_lifecycle_expiration(storage, rule2)
        lc = get_lifecycle(storage)
        @test lc !== nothing

        delete_lifecycle(storage)
    end

end