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

use std::path::PathBuf;

use ohlcv::Database;
use tracing::instrument;

use crate::{
    config::{CoinConfig, Config},
    Error,
};

use super::root_credentials;

/// Initialize the database
///
/// # Arguments
///
/// * `config` - Optional path to the configuration file. If not provided, the
///   default configuration file will be used. This file is expected to be in
///   TOML format. The default file is `ohlcv.toml` and is expected to be in the
///   current working directory or in `/etc/ohlcv`.
///
/// # Errors
///
/// Returns an error if the database cannot be initialized or if the
/// configuration file cannot be loaded.
#[instrument]
pub async fn init(config: Option<&PathBuf>) -> Result<(), Error> {
    let mut config = Config::load(config)?;
    let creds = root_credentials(&config.database)?;
    let coins = config
        .coins
        .iter()
        .map(CoinConfig::as_coin)
        .collect::<Vec<_>>();

    config
        .database
        .init_schema(creds, coins.as_slice())
        .await
        .map_err(Error::Ohlcv)
}