k2db-api-server 0.1.1

Single-binary Rust server for the k2db API
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
107
108

// SPDX-FileCopyrightText: 2026 Alexander R. Croft
// SPDX-License-Identifier: MIT

use clap::{Args, Parser, Subcommand};

#[derive(Debug, Parser)]
#[command(name = "k2db-api")]
#[command(about = "Single-binary Rust server for the k2db API")]
pub struct Cli {
    #[command(subcommand)]
    pub command: Command,
}

#[derive(Debug, Subcommand)]
pub enum Command {
    Serve(BootstrapArgs),
    Init(BootstrapArgs),
    Recover(BootstrapArgs),
    Keys(KeysArgs),
    Config(ConfigArgs),
}

#[derive(Debug, Clone, Args)]
pub struct BootstrapArgs {
    #[arg(long)]
    pub mongo_uri: Option<String>,
    #[arg(long)]
    pub mongo_uri_env: Option<String>,
    #[arg(long)]
    pub bootstrap_token: Option<String>,
    #[arg(long)]
    pub bootstrap_token_env: Option<String>,
    #[arg(long, default_value = "k2_system")]
    pub system_db_name: String,
    #[arg(long)]
    pub listen_host: Option<String>,
    #[arg(long)]
    pub listen_port: Option<u16>,
    #[arg(long)]
    pub ownership_mode: Option<String>,
    #[arg(long)]
    pub slow_query_ms: Option<u64>,
    #[arg(long)]
    pub seed_key_name: Option<String>,
    #[arg(long)]
    pub seed_key_database: Option<String>,
    #[arg(long = "seed-key-permission")]
    pub seed_key_permissions: Vec<String>,
}

#[derive(Debug, Clone, Subcommand)]
pub enum KeysCommand {
    Create(CreateKeyArgs),
    Revoke(RevokeKeyArgs),
    List,
}

#[derive(Debug, Clone, Args)]
pub struct KeysArgs {
    #[command(flatten)]
    pub bootstrap: BootstrapArgs,
    #[command(subcommand)]
    pub command: KeysCommand,
}

#[derive(Debug, Clone, Args)]
pub struct CreateKeyArgs {
    #[arg(long)]
    pub name: String,
    #[arg(long)]
    pub database: String,
    #[arg(long = "permission")]
    pub permissions: Vec<String>,
}

#[derive(Debug, Clone, Args)]
pub struct RevokeKeyArgs {
    #[arg(long)]
    pub key_id: String,
}

#[derive(Debug, Clone, Subcommand)]
pub enum ConfigCommand {
    Get,
    Set(SetConfigArgs),
}

#[derive(Debug, Clone, Args)]
pub struct ConfigArgs {
    #[command(flatten)]
    pub bootstrap: BootstrapArgs,
    #[command(subcommand)]
    pub command: ConfigCommand,
}

#[derive(Debug, Clone, Args)]
pub struct SetConfigArgs {
    #[arg(long)]
    pub listen_host: Option<String>,
    #[arg(long)]
    pub listen_port: Option<u16>,
    #[arg(long)]
    pub ownership_mode: Option<String>,
    #[arg(long)]
    pub slow_query_ms: Option<u64>,
    #[arg(long, default_value_t = false)]
    pub clear_slow_query_ms: bool,
}