aoc_leaderbot_aws_lambda_impl 3.0.0

Implementation of an Advent of Code leaderboard-watching bot running in AWS Lambda
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

#![deny(rustdoc::broken_intra_doc_links)]
#![deny(rustdoc::private_intra_doc_links)]
#![cfg_attr(coverage_nightly, feature(coverage_attribute))]

use aoc_leaderbot_aws_lib::leaderbot::storage::aws::dynamodb::DynamoDbStorage;
use aoc_leaderbot_aws_lib::leaderbot::storage::aws::dynamodb::config::table::{
    BillingModeConfig, TableConfig,
};
use aws_config::BehaviorVersion;
use clap::Parser;
use dotenvy::dotenv;

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let _ = dotenv();

    let cli = Cli::parse();

    let storage = get_storage(&cli).await;
    let table_config = get_table_config(&cli);
    storage.create_table(Some(table_config)).await?;

    Ok(())
}

async fn get_storage(cli: &Cli) -> DynamoDbStorage {
    if cli.test_endpoint_url.is_empty() {
        DynamoDbStorage::new(&cli.table_name).await
    } else {
        let config = aws_config::defaults(BehaviorVersion::latest())
            .region("ca-central-1")
            .test_credentials()
            .endpoint_url(&cli.test_endpoint_url)
            .load()
            .await;
        DynamoDbStorage::with_config(&config, &cli.table_name).await
    }
}

fn get_table_config(cli: &Cli) -> TableConfig {
    let billing_mode = match cli.billing_mode {
        cli::BillingMode::PayPerRequest => BillingModeConfig::pay_per_request(
            cli.max_read_request_units,
            cli.max_write_request_units,
        ),
        cli::BillingMode::Provisioned => {
            BillingModeConfig::provisioned(cli.read_capacity_units, cli.write_capacity_units)
        },
    };

    TableConfig { billing_mode: Some(billing_mode) }
}

use cli::Cli;

// It seems some code generated by `clap` is flagged as not covered by tests,
// but I don't know how to cover them, so I'll move the struct to a "no coverage" module.
#[cfg_attr(coverage_nightly, coverage(off))]
mod cli {
    use aoc_leaderbot_aws_lambda_impl::leaderbot::DEFAULT_DYNAMODB_TABLE_NAME;
    use aoc_leaderbot_aws_lib::leaderbot::storage::aws::dynamodb::config::table::{
        DEFAULT_READ_CAPACITY_UNITS, DEFAULT_WRITE_CAPACITY_UNITS,
    };
    use clap::ValueEnum;
    use gratte::Display;

    use super::*;

    #[derive(Debug, Copy, Clone, Display, ValueEnum)]
    pub enum BillingMode {
        PayPerRequest,
        Provisioned,
    }

    #[derive(Debug, Parser)]
    #[command(version, about = "Prepare DynamoDB table for aoc_leaderbot", long_about = None)]
    pub struct Cli {
        /// Name of DynamoDB table to use for leaderboard data
        #[arg(short, long, default_value_t = DEFAULT_DYNAMODB_TABLE_NAME.into())]
        pub table_name: String,

        /// DynamoDB billing mode
        #[arg(short, long, value_enum, default_value_t = BillingMode::Provisioned)]
        pub billing_mode: BillingMode,

        /// Max read request units in pay-per-request mode
        #[arg(long)]
        pub max_read_request_units: Option<i64>,

        /// Max write request units in pay-per-request mode
        #[arg(long)]
        pub max_write_request_units: Option<i64>,

        /// Read capacity units in provisioned mode
        #[arg(long, default_value_t = DEFAULT_READ_CAPACITY_UNITS)]
        pub read_capacity_units: i64,

        /// Write capacity units in provisioned mode
        #[arg(long, default_value_t = DEFAULT_WRITE_CAPACITY_UNITS)]
        pub write_capacity_units: i64,

        /// Test endpoint URL. Used by tests only, not shown in help
        #[arg(long, hide = true, default_value_t)]
        pub test_endpoint_url: String,
    }
}