aboutsummaryrefslogblamecommitdiff
path: root/src/config/options.rs
blob: b8dccf74467c18eb1e4455dfb5b84f232fd66a70 (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11










                                                                              
                                      
                           





                                                            


                              
                                      
                           


                                              
 

                                                        


                                
                                      
                             



                                                          


                                                
                       


                               


                 



                                                                                  
 





                             
 


                                                                    
                                                     


                                                                                               
 






                             
use anyhow::Result;
use serde::Deserialize;

use crate::config::RuntimeConfig;

// This code is inspired by the Trunk crate (https://github.com/thedodd/trunk)

// This file parses the options that can be declared in the environment.
// runtime.rs applies business logic and builds RuntimeConfig structs.

/// Base configuration options
#[derive(Clone, Default, Deserialize)]
pub struct ConfigOptsBase {
  /// This node's private IP address [default: None]
  pub private_ip: Option<String>,
  /// Expiration time for IGD rules [default: 60]
  pub expiration_time: Option<u16>,
  /// Refresh time for IGD and Firewall rules [default: 300]
  pub refresh_time: Option<u16>,
}

/// ACME configuration options
#[derive(Clone, Default, Deserialize)]
pub struct ConfigOptsAcme {
  /// Whether ACME is enabled [default: false]
  #[serde(default)]
  pub enable: bool,

  /// The default domain holder's e-mail [default: None]
  pub email: Option<String>,
}

/// Consul configuration options
#[derive(Clone, Default, Deserialize)]
pub struct ConfigOptsConsul {
  /// Consul's node name [default: None]
  pub node_name: Option<String>,
  /// Consul's REST URL [default: "http://127.0.0.1:8500"]
  pub url: Option<String>,
}

/// Model of all potential configuration options
pub struct ConfigOpts {
  pub base: ConfigOptsBase,
  pub acme: ConfigOptsAcme,
  pub consul: ConfigOptsConsul,
}

impl ConfigOpts {
  pub fn from_env() -> Result<RuntimeConfig> {
    let base: ConfigOptsBase = envy::prefixed("DIPLONAT_").from_env()?;
    let consul: ConfigOptsConsul = envy::prefixed("DIPLONAT_CONSUL_").from_env()?;
    let acme: ConfigOptsAcme = envy::prefixed("DIPLONAT_ACME_").from_env()?;

    RuntimeConfig::new(Self {
      base: base,
      consul: consul,
      acme: acme,
    })
  }

  // Currently only used in tests
  #[allow(dead_code)]
  pub fn from_iter<Iter: Clone>(iter: Iter) -> Result<RuntimeConfig>
  where Iter: IntoIterator<Item = (String, String)> {
    let base: ConfigOptsBase = envy::prefixed("DIPLONAT_").from_iter(iter.clone())?;
    let consul: ConfigOptsConsul = envy::prefixed("DIPLONAT_CONSUL_").from_iter(iter.clone())?;
    let acme: ConfigOptsAcme = envy::prefixed("DIPLONAT_ACME_").from_iter(iter.clone())?;

    RuntimeConfig::new(Self {
      base: base,
      consul: consul,
      acme: acme,
    })
  }
}