pict-rs 0.5.16

A simple image hosting service
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
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153

use clap::Parser;
use std::path::{Path, PathBuf};

mod commandline;
mod defaults;
mod file;
pub mod primitives;

use commandline::{Args, Output};
use config::Config;
use defaults::Defaults;

pub(crate) use commandline::Operation;
pub(crate) use file::{
    Animation, ConfigFile as Configuration, Media, ObjectStorage, OpenTelemetry, Postgres, Repo,
    Sled, Store, Tracing, Video,
};
pub(crate) use primitives::{Filesystem, LogFormat};

/// Source for pict-rs configuration when embedding as a library
pub enum ConfigSource<P, T> {
    /// A File source for pict-rs configuration
    File { path: P },
    /// An in-memory source for pict-rs configuration
    Memory { values: T },
    /// No configuration
    Empty,
}

impl<T> ConfigSource<PathBuf, T>
where
    T: serde::Serialize,
{
    /// Create a new memory source
    pub fn memory(values: T) -> Self {
        Self::Memory { values }
    }
}

impl<P> ConfigSource<P, ()>
where
    P: AsRef<Path>,
{
    /// Create a new file source
    pub fn file(path: P) -> Self {
        Self::File { path }
    }
}

impl ConfigSource<PathBuf, ()> {
    /// Create a new empty source
    pub fn empty() -> Self {
        Self::Empty
    }
}

// Instantiated configuration for the pict-rs application
#[derive(Debug)]
pub struct PictRsConfiguration {
    pub(crate) config: Configuration,
    pub(crate) operation: Operation,
}

pub(crate) fn configure_without_clap<P: AsRef<Path>, T: serde::Serialize, Q: AsRef<Path>>(
    source: ConfigSource<P, T>,
    save_to: Option<Q>,
) -> color_eyre::Result<PictRsConfiguration> {
    let config = Config::builder().add_source(config::Config::try_from(&Defaults::default())?);

    let config = match source {
        ConfigSource::Empty => config,
        ConfigSource::File { path } => config.add_source(config::File::from(path.as_ref())),
        ConfigSource::Memory { values } => config.add_source(config::Config::try_from(&values)?),
    };

    let config = if let Ok(targets) = std::env::var("RUST_LOG") {
        config.add_source(config::Config::try_from(&serde_json::json!({
            "tracing": {
                "logging": {
                    "targets": targets,
                }
            }
        }))?)
    } else {
        config
    };

    let built = config
        .add_source(
            config::Environment::with_prefix("PICTRS")
                .separator("__")
                .try_parsing(true),
        )
        .build()?;

    let operation = Operation::Run;

    let config: Configuration = built.try_deserialize()?;

    if let Some(save_to) = save_to {
        let output = toml::to_string_pretty(&config)?;
        std::fs::write(save_to, output)?;
    }

    Ok(PictRsConfiguration { config, operation })
}

pub(crate) fn configure() -> color_eyre::Result<PictRsConfiguration> {
    let Output {
        config_format,
        operation,
        save_to,
        config_file,
    } = Args::parse().into_output();

    let config = Config::builder().add_source(config::Config::try_from(&Defaults::default())?);

    let config = if let Some(config_file) = config_file {
        config.add_source(config::File::from(config_file))
    } else {
        config
    };

    let config = if let Ok(targets) = std::env::var("RUST_LOG") {
        config.add_source(config::Config::try_from(&serde_json::json!({
            "tracing": {
                "logging": {
                    "targets": targets,
                }
            }
        }))?)
    } else {
        config
    };

    let built = config
        .add_source(
            config::Environment::with_prefix("PICTRS")
                .separator("__")
                .try_parsing(true),
        )
        .add_source(config::Config::try_from(&config_format)?)
        .build()?;

    let config: Configuration = built.try_deserialize()?;

    if let Some(save_to) = save_to {
        let output = toml::to_string_pretty(&config)?;
        std::fs::write(save_to, output)?;
    }

    Ok(PictRsConfiguration { config, operation })
}