Source code
Revision control
Copy as Markdown
Other Tools
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
#![allow(unknown_lints)]
#![warn(rust_2018_idioms)]
use std::{
path::{Path, PathBuf},
sync::Arc,
};
use remote_settings::{RemoteSettingsConfig2, RemoteSettingsServer, RemoteSettingsService};
pub mod fxa_creds;
pub mod prompt;
pub fn init_logging() {
init_logging_with("info");
}
pub fn init_trace_logging() {
init_logging_with("trace");
}
pub fn init_logging_with(default_env: &str) {
tracing_support::init_from_env_with_default(default_env);
}
pub fn cli_data_dir() -> String {
ensure_cli_data_dir_exists();
data_path(None).to_string_lossy().to_string()
}
pub fn ensure_cli_data_dir_exists() {
let dir = data_path(None);
if !dir.exists() {
std::fs::create_dir(&dir).unwrap_or_else(|_| panic!("Error creating dir: {dir:?}"))
}
}
pub fn cli_data_subdir(relative_path: &str) -> String {
ensure_cli_data_dir_exists();
data_path(Some(relative_path)).to_string_lossy().to_string()
}
pub fn cli_data_path(filename: &str) -> String {
ensure_cli_data_dir_exists();
data_path(None).join(filename).to_string_lossy().to_string()
}
fn data_path(relative_path: Option<&str>) -> PathBuf {
let dir = workspace_root_dir().join(".cli-data");
match relative_path {
None => dir,
Some(relative_path) => dir.join(relative_path),
}
}
pub fn workspace_root_dir() -> PathBuf {
let cargo_output = std::process::Command::new(env!("CARGO"))
.arg("locate-project")
.arg("--workspace")
.arg("--message-format=plain")
.output()
.unwrap()
.stdout;
let cargo_toml_path = Path::new(std::str::from_utf8(&cargo_output).unwrap().trim());
cargo_toml_path.parent().unwrap().to_path_buf()
}
pub fn remote_settings_service(
rs_server: Option<RemoteSettingsServer>,
) -> Arc<RemoteSettingsService> {
let config = match rs_server {
Some(rs_server) => RemoteSettingsConfig2 {
server: Some(rs_server),
..Default::default()
},
None => RemoteSettingsConfig2::default(),
};
Arc::new(RemoteSettingsService::new(
data_path(Some("remote-settings"))
.to_string_lossy()
.to_string(),
config,
))
}