mirror of
https://github.com/martinvonz/jj.git
synced 2025-02-10 14:26:58 +00:00
7ba1c6bdb6
It's useful for tests, scripts, and debugging to be able to use
specific config instead of the user's config. That's especially true
for our automated tests because they didn't have a place to read
config from on Windows before this patch (they read their config from
`{FOLDERID_RoamingAppData}`, which I don't think we can override in
tests).
76 lines
2.7 KiB
Rust
76 lines
2.7 KiB
Rust
// Copyright 2020 Google LLC
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use std::env;
|
|
use std::path::PathBuf;
|
|
|
|
use jujutsu::commands::dispatch;
|
|
use jujutsu::ui::Ui;
|
|
use jujutsu_lib::settings::UserSettings;
|
|
|
|
fn config_path() -> Option<PathBuf> {
|
|
if let Ok(config_path) = env::var("JJ_CONFIG") {
|
|
Some(PathBuf::from(config_path))
|
|
} else {
|
|
dirs::config_dir().map(|config_dir| config_dir.join("jj").join("config.toml"))
|
|
}
|
|
}
|
|
|
|
fn read_config() -> Result<UserSettings, config::ConfigError> {
|
|
let mut config_builder = config::Config::builder();
|
|
|
|
if let Some(config_path) = config_path() {
|
|
config_builder = config_builder.add_source(
|
|
config::File::from(config_path)
|
|
.required(false)
|
|
.format(config::FileFormat::Toml),
|
|
);
|
|
};
|
|
|
|
// TODO: Make the config from environment a separate source instead? Seems
|
|
// cleaner to separate it like that, especially if the config::Config instance
|
|
// can keep track of where the config comes from then (it doesn't seem like it
|
|
// can, however - we don't give a name or anything to the Config object).
|
|
if let Ok(value) = env::var("JJ_USER") {
|
|
config_builder = config_builder.set_override("user.name", value)?;
|
|
}
|
|
if let Ok(value) = env::var("JJ_EMAIL") {
|
|
config_builder = config_builder.set_override("user.email", value)?;
|
|
}
|
|
if let Ok(value) = env::var("JJ_TIMESTAMP") {
|
|
config_builder = config_builder.set_override("user.timestamp", value)?;
|
|
}
|
|
|
|
let config = config_builder.build()?;
|
|
Ok(UserSettings::from_config(config))
|
|
}
|
|
|
|
fn main() {
|
|
// TODO: We need to do some argument parsing here, at least for things like
|
|
// --config, and for reading user configs from the repo pointed to by
|
|
// -R.
|
|
match read_config() {
|
|
Ok(user_settings) => {
|
|
let ui = Ui::for_terminal(user_settings);
|
|
let status = dispatch(ui, &mut std::env::args_os());
|
|
std::process::exit(status);
|
|
}
|
|
Err(err) => {
|
|
let mut ui = Ui::for_terminal(UserSettings::default());
|
|
ui.write_error(&format!("Invalid config: {}\n", err))
|
|
.unwrap();
|
|
std::process::exit(1);
|
|
}
|
|
}
|
|
}
|