s3cmdrun() {
    # shellcheck disable=2039,3043
    local backend accessKey secreyKey
    backend="${1}"
    accessKey="${2}"
    secreyKey="${3}"
    shift 3

    if [ "$backend" = "ceph" ]; then
        timeout -k 5 5 s3cmd \
            --access_key="${accessKey}" \
            --secret_key="${secreyKey}" \
            --host="${s3Endpoint}" \
            --host-bucket="${s3Endpoint}" \
            --no-ssl \
            "$@"
    else
        timeout -k 5 5 s3cmd \
            --access_key="${accessKey}" \
            --secret_key="${secreyKey}" \
            --host="${s3Endpoint}" \
            --host-bucket="${s3Endpoint}" \
            --ssl \
            --no-check-certificate \
            "$@"
    fi
}

test_storage_buckets() {
    # shellcheck disable=2039,3043
    local incus_backend

    incus_backend=$(storage_backend "$INCUS_DIR")

    if [ "$incus_backend" = "ceph" ]; then
        if [ -z "${INCUS_CEPH_CEPHOBJECT_RADOSGW:-}" ]; then
            # Check INCUS_CEPH_CEPHOBJECT_RADOSGW specified for ceph bucket tests.
            export TEST_UNMET_REQUIREMENT="INCUS_CEPH_CEPHOBJECT_RADOSGW not specified"
            return
        fi
    elif [ "${incus_backend}" = "linstor" ]; then
        # Skip linstor driver, as it does not support storage buckets
        export TEST_UNMET_REQUIREMENT="linstor driver does not support storage buckets"
        return 0
    fi

    poolName=$(incus profile device get default root pool)
    bucketPrefix="inc$$"

    # Check cephobject.radosgw.endpoint is required for cephobject pools.
    if [ "$incus_backend" = "ceph" ]; then
        ! incus storage create s3 cephobject || false
        incus storage create s3 cephobject cephobject.radosgw.endpoint="${INCUS_CEPH_CEPHOBJECT_RADOSGW}"
        incus storage show s3
        poolName="s3"
        s3Endpoint="${INCUS_CEPH_CEPHOBJECT_RADOSGW}"
    else
        # Create a loop device for dir pools (the test suite may otherwise run on tmpfs).
        if [ "$incus_backend" = "dir" ]; then
            configure_loop_device loop_file_1 loop_device_1
            # shellcheck disable=SC2154
            mkfs.ext4 "${loop_device_1}"
            mkdir "${TEST_DIR}/${bucketPrefix}"
            mount "${loop_device_1}" "${TEST_DIR}/${bucketPrefix}"
            losetup -d "${loop_device_1}"
            mkdir "${TEST_DIR}/${bucketPrefix}/s3"
            incus storage create s3 dir source="${TEST_DIR}/${bucketPrefix}/s3"
            poolName="s3"
        fi

        buckets_addr="127.0.0.1:$(local_tcp_port)"
        incus config set core.storage_buckets_address "${buckets_addr}"
        s3Endpoint="https://${buckets_addr}"
    fi

    # Check bucket name validation.
    ! incus storage bucket create "${poolName}" .foo || false
    ! incus storage bucket create "${poolName}" fo || false # codespell:ignore fo
    ! incus storage bucket create "${poolName}" fooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooooo || false
    ! incus storage bucket create "${poolName}" "foo bar" || false

    # Create bucket.
    initCreds=$(incus storage bucket create "${poolName}" "${bucketPrefix}.foo" user.foo=comment --description "Test description")
    initAccessKey=$(echo "${initCreds}" | awk '{ if ($2 == "access" && $3 == "key:") {print $4}}')
    initSecretKey=$(echo "${initCreds}" | awk '{ if ($2 == "secret" && $3 == "key:") {print $4}}')
    s3cmdrun "${incus_backend}" "${initAccessKey}" "${initSecretKey}" ls "s3://${bucketPrefix}.foo"

    incus storage bucket list "${poolName}" | grep -F "${bucketPrefix}.foo"
    incus storage bucket list "${poolName}" | grep -F "Test description"
    incus storage bucket show "${poolName}" "${bucketPrefix}.foo"

    # Create bucket keys.

    # Create admin key with randomly generated credentials.
    creds=$(incus storage bucket key create "${poolName}" "${bucketPrefix}.foo" admin-key --role=admin --description "Test description")
    adAccessKey=$(echo "${creds}" | awk '{ if ($1 == "Access" && $2 == "key:") {print $3}}')
    adSecretKey=$(echo "${creds}" | awk '{ if ($1 == "Secret" && $2 == "key:") {print $3}}')

    # Create read-only key with manually specified credentials.
    creds=$(incus storage bucket key create "${poolName}" "${bucketPrefix}.foo" ro-key --role=read-only --access-key="${bucketPrefix}.foo.ro" --secret-key="password")
    roAccessKey=$(echo "${creds}" | awk '{ if ($1 == "Access" && $2 == "key:") {print $3}}')
    roSecretKey=$(echo "${creds}" | awk '{ if ($1 == "Secret" && $2 == "key:") {print $3}}')

    incus storage bucket key list "${poolName}" "${bucketPrefix}.foo" | grep -F "admin-key"
    incus storage bucket key list "${poolName}" "${bucketPrefix}.foo" | grep -F "ro-key"
    incus storage bucket key list "${poolName}" "${bucketPrefix}.foo" | grep -F "Test description"
    incus storage bucket key show "${poolName}" "${bucketPrefix}.foo" admin-key
    incus storage bucket key show "${poolName}" "${bucketPrefix}.foo" ro-key

    # Test making buckets via S3 is blocked.
    ! s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" mb "s3://${bucketPrefix}.foo2" || false
    ! s3cmdrun "${incus_backend}" "${roAccessKey}" "${roSecretKey}" mb "s3://${bucketPrefix}.foo2" || false

    # Test putting a file into a bucket.
    incusTestFile="bucketfile_${bucketPrefix}.txt"
    head -c 5M /dev/urandom > "${incusTestFile}"
    s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" put "${incusTestFile}" "s3://${bucketPrefix}.foo"
    ! s3cmdrun "${incus_backend}" "${roAccessKey}" "${roSecretKey}" put "${incusTestFile}" "s3://${bucketPrefix}.foo" || false

    # Test listing bucket files via S3.
    s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" ls "s3://${bucketPrefix}.foo"
    s3cmdrun "${incus_backend}" "${roAccessKey}" "${roSecretKey}" ls "s3://${bucketPrefix}.foo"

    # Test getting a file from a bucket.
    s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" get "s3://${bucketPrefix}.foo/${incusTestFile}" "${incusTestFile}.get"
    rm "${incusTestFile}.get"
    s3cmdrun "${incus_backend}" "${roAccessKey}" "${roSecretKey}" get "s3://${bucketPrefix}.foo/${incusTestFile}" "${incusTestFile}.get"
    rm "${incusTestFile}.get"

    if [ "$incus_backend" != "ceph" ]; then
        # Test using a presigned URL.
        signedURL=$(s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" signurl "s3://${bucketPrefix}.foo/${incusTestFile}" "+60" | tr -d '[:space:]')
        signedURL="https://${signedURL#http://}"
        curl -sS -k -f "${signedURL}" -o "${incusTestFile}.signed"
        cmp "${incusTestFile}" "${incusTestFile}.signed"
        rm "${incusTestFile}.signed"

        # An expired presigned URL must be rejected.
        expiredEpoch=$(($(date +%s) - 60))
        expiredURL=$(s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" signurl "s3://${bucketPrefix}.foo/${incusTestFile}" "${expiredEpoch}" | tr -d '[:space:]')
        expiredURL="https://${expiredURL#http://}"
        ! curl -sS -k -f "${expiredURL}" -o /dev/null || false
    fi

    # Test copying a file within a bucket (S3 CopyObject).
    s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" cp "s3://${bucketPrefix}.foo/${incusTestFile}" "s3://${bucketPrefix}.foo/${incusTestFile}.copy"
    s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" ls "s3://${bucketPrefix}.foo" | grep -F "${incusTestFile}.copy"
    s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" get "s3://${bucketPrefix}.foo/${incusTestFile}.copy" "${incusTestFile}.copy.get"
    cmp "${incusTestFile}" "${incusTestFile}.copy.get"
    rm "${incusTestFile}.copy.get"
    ! s3cmdrun "${incus_backend}" "${roAccessKey}" "${roSecretKey}" cp "s3://${bucketPrefix}.foo/${incusTestFile}" "s3://${bucketPrefix}.foo/${incusTestFile}.rocopy" || false
    s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" del "s3://${bucketPrefix}.foo/${incusTestFile}.copy"

    # Test deleting a file from a bucket.
    ! s3cmdrun "${incus_backend}" "${roAccessKey}" "${roSecretKey}" del "s3://${bucketPrefix}.foo/${incusTestFile}" || false
    s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" del "s3://${bucketPrefix}.foo/${incusTestFile}"

    # Test bucket quota (except dir driver which doesn't support quotas so check that its prevented).
    if [ "$incus_backend" = "dir" ]; then
        ! incus storage bucket create "${poolName}" "${bucketPrefix}.foo2" size=1MiB || false
    else
        initCreds=$(incus storage bucket create "${poolName}" "${bucketPrefix}.foo2" size=1MiB)
        initAccessKey=$(echo "${initCreds}" | awk '{ if ($2 == "access" && $3 == "key:") {print $4}}')
        initSecretKey=$(echo "${initCreds}" | awk '{ if ($2 == "secret" && $3 == "key:") {print $4}}')
        ! s3cmdrun "${incus_backend}" "${initAccessKey}" "${initSecretKey}" put "${incusTestFile}" "s3://${bucketPrefix}.foo2" || false

        # Grow bucket quota.
        incus storage bucket set "${poolName}" "${bucketPrefix}.foo2" size=150MiB
        s3cmdrun "${incus_backend}" "${initAccessKey}" "${initSecretKey}" put "${incusTestFile}" "s3://${bucketPrefix}.foo2"
        s3cmdrun "${incus_backend}" "${initAccessKey}" "${initSecretKey}" del "s3://${bucketPrefix}.foo2/${incusTestFile}"
        incus storage bucket delete "${poolName}" "${bucketPrefix}.foo2"
    fi

    # Cleanup test file used earlier.
    rm "${incusTestFile}"

    # Test deleting bucket via s3.
    ! s3cmdrun "${incus_backend}" "${roAccessKey}" "${roSecretKey}" rb "s3://${bucketPrefix}.foo" || false
    ! s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" rb "s3://${bucketPrefix}.foo"

    # Delete bucket keys.
    incus storage bucket key delete "${poolName}" "${bucketPrefix}.foo" admin-key
    incus storage bucket key delete "${poolName}" "${bucketPrefix}.foo" ro-key
    ! incus storage bucket key list "${poolName}" "${bucketPrefix}.foo" | grep -F "admin-key" || false
    ! incus storage bucket key list "${poolName}" "${bucketPrefix}.foo" | grep -F "ro-key" || false
    ! incus storage bucket key show "${poolName}" "${bucketPrefix}.foo" admin-key || false
    ! incus storage bucket key show "${poolName}" "${bucketPrefix}.foo" ro-key || false
    ! s3cmdrun "${incus_backend}" "${adAccessKey}" "${adSecretKey}" ls "s3://${bucketPrefix}.foo" || false
    ! s3cmdrun "${incus_backend}" "${roAccessKey}" "${roSecretKey}" ls "s3://${bucketPrefix}.foo" || false

    # Delete bucket.
    incus storage bucket delete "${poolName}" "${bucketPrefix}.foo"
    ! incus storage bucket list "${poolName}" | grep -F "${bucketPrefix}.foo" || false
    ! incus storage bucket show "${poolName}" "${bucketPrefix}.foo" || false

    if [ "$incus_backend" = "ceph" ] || [ "$incus_backend" = "dir" ]; then
        incus storage delete "${poolName}"
    fi

    if [ "$incus_backend" = "dir" ]; then
        umount "${TEST_DIR}/${bucketPrefix}"
        rmdir "${TEST_DIR}/${bucketPrefix}"

        # shellcheck disable=SC2154
        deconfigure_loop_device "${loop_file_1}" "${loop_device_1}"
    fi
}

test_storage_bucket_export() {
    # shellcheck disable=2039,3043
    local incus_backend

    incus_backend=$(storage_backend "$INCUS_DIR")

    # Skip the test if we are not using a Ceph or Dir backend.
    if [ ! "$incus_backend" = "ceph" ] && [ ! "$incus_backend" = "dir" ]; then
        return
    fi

    if [ "$incus_backend" = "ceph" ]; then
        if [ -z "${INCUS_CEPH_CEPHOBJECT_RADOSGW:-}" ]; then
            # Check INCUS_CEPH_CEPHOBJECT_RADOSGW specified for ceph bucket tests.
            export TEST_UNMET_REQUIREMENT="INCUS_CEPH_CEPHOBJECT_RADOSGW not specified"
            return
        fi
    fi

    poolName=$(incus profile device get default root pool)
    bucketPrefix="inc$$"

    # Check cephobject.radosgw.endpoint is required for cephobject pools.
    if [ "$incus_backend" = "ceph" ]; then
        ! incus storage create s3 cephobject || false
        incus storage create s3 cephobject cephobject.radosgw.endpoint="${INCUS_CEPH_CEPHOBJECT_RADOSGW}"
        incus storage show s3
        poolName="s3"
        s3Endpoint="${INCUS_CEPH_CEPHOBJECT_RADOSGW}"
    else
        # Create a loop device for dir pools (the test suite may otherwise run on tmpfs).
        if [ "$incus_backend" = "dir" ]; then
            configure_loop_device loop_file_1 loop_device_1
            # shellcheck disable=SC2154
            mkfs.ext4 "${loop_device_1}"
            mkdir "${TEST_DIR}/${bucketPrefix}"
            mount "${loop_device_1}" "${TEST_DIR}/${bucketPrefix}"
            losetup -d "${loop_device_1}"
            mkdir "${TEST_DIR}/${bucketPrefix}/s3"
            # shellcheck disable=SC2034
            incus storage create s3 dir source="${TEST_DIR}/${bucketPrefix}/s3"
            poolName="s3"
        fi

        buckets_addr="127.0.0.1:$(local_tcp_port)"
        incus config set core.storage_buckets_address "${buckets_addr}"
        s3Endpoint="https://${buckets_addr}"
    fi

    # Create test bucket
    initCreds=$(incus storage bucket create "${poolName}" "${bucketPrefix}.foo" user.foo=comment)
    initAccessKey=$(echo "${initCreds}" | awk '{ if ($2 == "access" && $3 == "key:") {print $4}}')
    initSecretKey=$(echo "${initCreds}" | awk '{ if ($2 == "secret" && $3 == "key:") {print $4}}')
    s3cmdrun "${incus_backend}" "${initAccessKey}" "${initSecretKey}" ls "s3://${bucketPrefix}.foo"

    # Test putting a file into a bucket.
    incusTestFile="bucketfile_${bucketPrefix}.txt"
    echo "hello world" > "${incusTestFile}"
    s3cmdrun "${incus_backend}" "${initAccessKey}" "${initSecretKey}" put "${incusTestFile}" "s3://${bucketPrefix}.foo"

    # Export test bucket
    incus storage bucket export "${poolName}" "${bucketPrefix}.foo" "${INCUS_DIR}/testbucket.tar.gz"
    [ -f "${INCUS_DIR}/testbucket.tar.gz" ]

    # Extract storage backup tarball.
    mkdir "${INCUS_DIR}/storage-bucket-export"
    tar -xzf "${INCUS_DIR}/testbucket.tar.gz" -C "${INCUS_DIR}/storage-bucket-export"

    # Check tarball content.
    [ -f "${INCUS_DIR}/storage-bucket-export/backup/index.yaml" ]
    [ -f "${INCUS_DIR}/storage-bucket-export/backup/bucket/${incusTestFile}" ]
    [ "$(cat "${INCUS_DIR}/storage-bucket-export/backup/bucket/${incusTestFile}")" = "hello world" ]

    # Delete bucket and import exported bucket
    incus storage bucket delete "${poolName}" "${bucketPrefix}.foo"
    incus storage bucket import "${poolName}" "${INCUS_DIR}/testbucket.tar.gz" "${bucketPrefix}.bar"
    rm "${INCUS_DIR}/testbucket.tar.gz"

    # Test listing bucket files via S3.
    s3cmdrun "${incus_backend}" "${initAccessKey}" "${initSecretKey}" ls "s3://${bucketPrefix}.bar" | grep -F "${incusTestFile}"

    # Test getting admin key from bucket.
    incus storage bucket key list "${poolName}" "${bucketPrefix}.bar" | grep -F "admin"

    # Export to stdout / import from stdin
    incus storage bucket export "${poolName}" "${bucketPrefix}.bar" - > "${INCUS_DIR}/testbucket.tar.gz"
    incus storage bucket delete "${poolName}" "${bucketPrefix}.bar"
    incus storage bucket import "${poolName}" - "${bucketPrefix}.bar" < "${INCUS_DIR}/testbucket.tar.gz"

    # Clean up.
    incus storage bucket delete "${poolName}" "${bucketPrefix}.bar"
    ! incus storage bucket list "${poolName}" | grep -F "${bucketPrefix}.bar" || false
    ! incus storage bucket show "${poolName}" "${bucketPrefix}.bar" || false

    if [ "$incus_backend" = "ceph" ] || [ "$incus_backend" = "dir" ]; then
        incus storage delete "${poolName}"
    fi

    if [ "$incus_backend" = "dir" ]; then
        umount "${TEST_DIR}/${bucketPrefix}"
        rmdir "${TEST_DIR}/${bucketPrefix}"

        # shellcheck disable=SC2154
        deconfigure_loop_device "${loop_file_1}" "${loop_device_1}"
    fi

    rm -rf "${INCUS_DIR}/storage-bucket-export/"*
    rmdir "${INCUS_DIR}/storage-bucket-export"
}
