pub mod retry;

use aws_config::BehaviorVersion;
use aws_sdk_dynamodb::{
    error::{ProvideErrorMetadata, SdkError},
    operation::create_table::CreateTableError,
    types::{
        AttributeDefinition, BillingMode, KeySchemaElement, KeyType, ScalarAttributeType,
        TimeToLiveSpecification,
    },
};
use std::time::Duration;

/// Test wait timeout, generally long enough that something has probably gone wrong.
pub const TEST_WAIT: Duration = Duration::from_secs(4);

/// Config for localhost dynamodb.
pub async fn localhost_dynamodb() -> aws_sdk_dynamodb::Client {
    let conf = aws_config::defaults(BehaviorVersion::latest())
        .region("eu-west-1")
        .load()
        .await;
    let conf = aws_sdk_dynamodb::config::Builder::from(&conf)
        .endpoint_url("http://localhost:8000")
        .build();
    aws_sdk_dynamodb::Client::from_conf(conf)
}

/// Create the table, with "key" as a hash key, if it doesn't exist.
pub async fn create_lease_table(table_name: &str, client: &aws_sdk_dynamodb::Client) {
    let create_table = client
        .create_table()
        .table_name(table_name)
        .billing_mode(BillingMode::PayPerRequest)
        .attribute_definitions(
            AttributeDefinition::builder()
                .attribute_name("key")
                .attribute_type(ScalarAttributeType::S)
                .build()
                .unwrap(),
        )
        .key_schema(
            KeySchemaElement::builder()
                .attribute_name("key")
                .key_type(KeyType::Hash)
                .build()
                .unwrap(),
        )
        .send()
        .await;

    match create_table {
        Ok(_) => Ok(()),
        Err(SdkError::ServiceError(se))
            if matches!(se.err(), CreateTableError::ResourceInUseException(..)) =>
        {
            Ok(())
        }
        Err(e) => Err(e),
    }
    .expect("dynamodb create_table failed: Did you run scripts/init-test.sh ?");

    let ttl_update = client
        .update_time_to_live()
        .table_name(table_name)
        .time_to_live_specification(
            TimeToLiveSpecification::builder()
                .enabled(true)
                .attribute_name("lease_expiry")
                .build()
                .unwrap(),
        )
        .send()
        .await;
    match ttl_update {
        Ok(_) => Ok(()),
        Err(SdkError::ServiceError(se))
            if se.err().code() == Some("ValidationException")
                && se.err().message() == Some("TimeToLive is already enabled") =>
        {
            Ok(())
        }

        Err(e) => Err(e),
    }
    .expect("dynamodb ttl_update failed");
}