skytool 0.1.0-pre.2

an experimental API client for BlueSky / ATProto
Documentation 
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

use clap::{Args, ValueEnum};
use rand_chacha::ChaCha20Rng;
use rand_core::{CryptoRngCore, SeedableRng};
use rand_hc::Hc128Rng;
use rand_seeder::Seeder;

#[derive(Debug, Clone, Default, clap::Args, serde::Deserialize, serde::Serialize, schemars::JsonSchema)]
#[group(id = "rng")]
#[remain::sorted]
#[rustfmt::skip]
pub(crate) struct RngOptions {
  #[arg(
    env  = "RNG_ALGORITHM",
    long = "rng-algorithm",
    id   = "rng.algorithm",
    value_name    = "ALGORITHM",
    default_value = "chacha20",
    help_heading = "RNG Options",
  )]
  pub(crate) algorithm: RngAlgorithm,
  #[arg(
    env  = "RNG_SEED",
    long = "rng-seed",
    id   = "rng.seed",
    value_name = "STRING",
    help_heading = "RNG Options",
  )]
  pub(crate) seed: Option<String>,
}

#[derive(Debug, Clone, Copy, Default, clap::ValueEnum, serde::Deserialize, serde::Serialize, schemars::JsonSchema)]
#[clap(rename_all = "lowercase")]
#[remain::sorted]
pub(crate) enum RngAlgorithm {
  #[default]
  ChaCha20,
  Hc128,
}

impl RngOptions {
  pub(crate) fn to_crypto_rng(&self) -> Box<dyn CryptoRngCore> {
    let Self { algorithm, seed } = self;

    match algorithm {
      RngAlgorithm::ChaCha20 => Box::new(match seed {
        Some(seed) => Seeder::from(&seed).make_rng(),
        None => ChaCha20Rng::from_entropy(),
      }),
      RngAlgorithm::Hc128 => Box::new(match seed {
        Some(seed) => Seeder::from(&seed).make_rng(),
        None => Hc128Rng::from_entropy(),
      }),
    }
  }
}