1  use dotenvy_macro::dotenv;
 2  use std::env;
 3  
 4  pub enum DfxNetwork {
 5      Local,
 6      Ic,
 7  }
 8  
 9  pub fn dfx_network() -> DfxNetwork {
10      let mut dfx_network = dotenv!("DFX_NETWORK").to_string();
11      if dfx_network.is_empty() {
12          dfx_network = env::var("DFX_NETWORK").expect("DFX_NETWORK is must be set");
13      }
14  
15      match dfx_network.as_str() {
16          "local" => DfxNetwork::Local,
17          "ic" => DfxNetwork::Ic,
18          _ => panic!("Unknown dfx network: {}", dfx_network),
19      }
20  }