mercurial-scm/hg-stable: rust/rhg/src/main.rs annotate

annotate rust/rhg/src/main.rs @ 47412:3237ed4dcda4

rhg: split non_repo_config and `--config` loading in different functions This will help us in better handling of error caused when trying to load `--config` values. Differential Revision: https://phab.mercurial-scm.org/D10837

author	Pulkit Goyal <7895pulkit@gmail.com>
date	Mon, 07 Jun 2021 17:19:46 +0530
parents	88119fffecc8
children	6e49769b7f97

rev	line source
45536 b1cea0dc9db0 rhg: Add debug timing Antoine Cezar <antoine.cezar@octobus.net> parents: 45535 diff changeset	1 extern crate log;
46631 80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	2 use crate::ui::Ui;
45051 18f8d3b31baa rhg: add a limited `rhg root` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45050 diff changeset	3 use clap::App;
18f8d3b31baa rhg: add a limited `rhg root` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45050 diff changeset	4 use clap::AppSettings;
46555 d8730ff51d5a rhg: Add support for -R and --repository command-line arguments Simon Sapin <simon.sapin@octobus.net> parents: 46554 diff changeset	5 use clap::Arg;
45535 f17caf8f3fef rhg: add a limited `rhg debugdata` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45451 diff changeset	6 use clap::ArgMatches;
46746 1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	7 use format_bytes::{format_bytes, join};
47410 ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	8 use hg::config::{Config, ConfigSource};
46632 5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	9 use hg::repo::{Repo, RepoError};
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	10 use hg::utils::files::{get_bytes_from_os_str, get_path_from_bytes};
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	11 use hg::utils::SliceExt;
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	12 use std::ffi::OsString;
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	13 use std::path::PathBuf;
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	14 use std::process::Command;
45051 18f8d3b31baa rhg: add a limited `rhg root` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45050 diff changeset	15
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	16 mod blackbox;
45002 bacf6c7ef01b rhg: add Command trait for subcommands implemented by rhg Antoine Cezar <antoine.cezar@octobus.net> parents: 45001 diff changeset	17 mod error;
45001 cf04f62d1579 rhg: add rhg crate Antoine Cezar <antoine.cezar@octobus.net> parents: diff changeset	18 mod exitcode;
45050 513b3ef277a3 rhg: add RootCommand using hg-core FindRoot operation to prepare `hg root` Antoine Cezar <antoine.cezar@octobus.net> parents: 45002 diff changeset	19 mod ui;
45535 f17caf8f3fef rhg: add a limited `rhg debugdata` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45451 diff changeset	20 use error::CommandError;
45001 cf04f62d1579 rhg: add rhg crate Antoine Cezar <antoine.cezar@octobus.net> parents: diff changeset	21
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	22 fn main_with_result(
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	23 process_start_time: &blackbox::ProcessStartTime,
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	24 ui: &ui::Ui,
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	25 repo: Result<&Repo, &NoRepoInCwdError>,
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	26 config: &Config,
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	27 ) -> Result<(), CommandError> {
46746 1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	28 check_extensions(config)?;
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	29
45535 f17caf8f3fef rhg: add a limited `rhg debugdata` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45451 diff changeset	30 let app = App::new("rhg")
46647 4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	31 .global_setting(AppSettings::AllowInvalidUtf8)
46759 eb14264b98e8 rhg: Fall back to Python for --version Simon Sapin <simon.sapin@octobus.net> parents: 46757 diff changeset	32 .global_setting(AppSettings::DisableVersion)
45051 18f8d3b31baa rhg: add a limited `rhg root` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45050 diff changeset	33 .setting(AppSettings::SubcommandRequired)
18f8d3b31baa rhg: add a limited `rhg root` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45050 diff changeset	34 .setting(AppSettings::VersionlessSubcommands)
46647 4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	35 .arg(
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	36 Arg::with_name("repository")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	37 .help("repository root directory")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	38 .short("-R")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	39 .long("--repository")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	40 .value_name("REPO")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	41 .takes_value(true)
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	42 // Both ok: `hg -R ./foo log` or `hg log -R ./foo`
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	43 .global(true),
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	44 )
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	45 .arg(
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	46 Arg::with_name("config")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	47 .help("set/override config option (use 'section.name=value')")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	48 .long("--config")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	49 .value_name("CONFIG")
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	50 .takes_value(true)
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	51 .global(true)
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	52 // Ok: `--config section.key1=val --config section.key2=val2`
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	53 .multiple(true)
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	54 // Not ok: `--config section.key1=val section.key2=val2`
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	55 .number_of_values(1),
4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	56 )
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	57 .arg(
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	58 Arg::with_name("cwd")
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	59 .help("change working directory")
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	60 .long("--cwd")
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	61 .value_name("DIR")
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	62 .takes_value(true)
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	63 .global(true),
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	64 )
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	65 .version("0.0.1");
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	66 let app = add_subcommand_args(app);
45051 18f8d3b31baa rhg: add a limited `rhg root` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45050 diff changeset	67
46630 21d3b40b4c0e rhg: Remove error message on unsupported CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46557 diff changeset	68 let matches = app.clone().get_matches_safe()?;
46555 d8730ff51d5a rhg: Add support for -R and --repository command-line arguments Simon Sapin <simon.sapin@octobus.net> parents: 46554 diff changeset	69
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	70 let (subcommand_name, subcommand_matches) = matches.subcommand();
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	71 let run = subcommand_run_fn(subcommand_name)
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	72 .expect("unknown subcommand name from clap despite AppSettings::SubcommandRequired");
46631 80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	73 let subcommand_args = subcommand_matches
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	74 .expect("no subcommand arguments from clap despite AppSettings::SubcommandRequired");
45051 18f8d3b31baa rhg: add a limited `rhg root` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45050 diff changeset	75
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	76 let invocation = CliInvocation {
46631 80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	77 ui,
80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	78 subcommand_args,
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	79 config,
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	80 repo,
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	81 };
755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	82 let blackbox = blackbox::Blackbox::new(&invocation, process_start_time)?;
755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	83 blackbox.log_command_start();
755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	84 let result = run(&invocation);
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	85 blackbox.log_command_end(exit_code(
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	86 &result,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	87 // TODO: show a warning or combine with original error if `get_bool`
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	88 // returns an error
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	89 config
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	90 .get_bool(b"ui", b"detailed-exit-code")
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	91 .unwrap_or(false),
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	92 ));
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	93 result
46630 21d3b40b4c0e rhg: Remove error message on unsupported CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46557 diff changeset	94 }
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	95
46630 21d3b40b4c0e rhg: Remove error message on unsupported CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46557 diff changeset	96 fn main() {
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	97 // Run this first, before we find out if the blackbox extension is even
755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	98 // enabled, in order to include everything in-between in the duration
755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	99 // measurements. Reading config files can be slow if they’re on NFS.
755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	100 let process_start_time = blackbox::ProcessStartTime::now();
755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	101
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	102 env_logger::init();
46632 5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	103 let ui = ui::Ui::new();
46630 21d3b40b4c0e rhg: Remove error message on unsupported CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46557 diff changeset	104
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	105 let early_args = EarlyArgs::parse(std::env::args_os());
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	106
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	107 let initial_current_dir = early_args.cwd.map(\|cwd\| {
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	108 let cwd = get_path_from_bytes(&cwd);
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	109 std::env::current_dir()
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	110 .and_then(\|initial\| {
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	111 std::env::set_current_dir(cwd)?;
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	112 Ok(initial)
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	113 })
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	114 .unwrap_or_else(\|error\| {
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	115 exit(
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	116 &None,
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	117 &ui,
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	118 OnUnsupported::Abort,
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	119 Err(CommandError::abort(format!(
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	120 "abort: {}: '{}'",
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	121 error,
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	122 cwd.display()
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	123 ))),
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	124 false,
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	125 )
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	126 })
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	127 });
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	128
47412 3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	129 let mut non_repo_config =
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	130 Config::load_non_repo().unwrap_or_else(\|error\| {
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	131 // Normally this is decided based on config, but we don’t have that
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	132 // available. As of this writing config loading never returns an
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	133 // "unsupported" error but that is not enforced by the type system.
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	134 let on_unsupported = OnUnsupported::Abort;
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	135
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	136 exit(
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	137 &initial_current_dir,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	138 &ui,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	139 on_unsupported,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	140 Err(error.into()),
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	141 false,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	142 )
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	143 });
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	144
47412 3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	145 non_repo_config
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	146 .load_cli_args_config(early_args.config)
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	147 .unwrap_or_else(\|error\| {
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	148 exit(
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	149 &initial_current_dir,
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	150 &ui,
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	151 OnUnsupported::from_config(&ui, &non_repo_config),
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	152 Err(error.into()),
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	153 non_repo_config
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	154 .get_bool(b"ui", b"detailed-exit-code")
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	155 .unwrap_or(false),
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	156 )
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	157 });
3237ed4dcda4 rhg: split non_repo_config and `--config` loading in different functions Pulkit Goyal <7895pulkit@gmail.com> parents: 47411 diff changeset	158
46742 6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	159 if let Some(repo_path_bytes) = &early_args.repo {
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	160 lazy_static::lazy_static! {
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	161 static ref SCHEME_RE: regex::bytes::Regex =
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	162 // Same as `_matchscheme` in `mercurial/util.py`
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	163 regex::bytes::Regex::new("^[a-zA-Z0-9+.\\-]+:").unwrap();
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	164 }
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	165 if SCHEME_RE.is_match(&repo_path_bytes) {
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	166 exit(
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	167 &initial_current_dir,
46742 6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	168 &ui,
46761 bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	169 OnUnsupported::from_config(&ui, &non_repo_config),
46742 6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	170 Err(CommandError::UnsupportedFeature {
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	171 message: format_bytes!(
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	172 b"URL-like --repository {}",
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	173 repo_path_bytes
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	174 ),
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	175 }),
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	176 // TODO: show a warning or combine with original error if
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	177 // `get_bool` returns an error
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	178 non_repo_config
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	179 .get_bool(b"ui", b"detailed-exit-code")
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	180 .unwrap_or(false),
46742 6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	181 )
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	182 }
6cd9f53aaed8 rhg: Fall back to Python on --repository with an URL Simon Sapin <simon.sapin@octobus.net> parents: 46706 diff changeset	183 }
47410 ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	184 let repo_arg = early_args.repo.unwrap_or(Vec::new());
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	185 let repo_path: Option<PathBuf> = {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	186 if repo_arg.is_empty() {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	187 None
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	188 } else {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	189 let local_config = {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	190 if std::env::var_os("HGRCSKIPREPO").is_none() {
47411 88119fffecc8 rhg: look for repository in ancestors also instead of cwd only Pulkit Goyal <7895pulkit@gmail.com> parents: 47410 diff changeset	191 // TODO: handle errors from find_repo_root
88119fffecc8 rhg: look for repository in ancestors also instead of cwd only Pulkit Goyal <7895pulkit@gmail.com> parents: 47410 diff changeset	192 if let Ok(current_dir_path) = Repo::find_repo_root() {
47410 ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	193 let config_files = vec![
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	194 ConfigSource::AbsPath(
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	195 current_dir_path.join(".hg/hgrc"),
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	196 ),
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	197 ConfigSource::AbsPath(
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	198 current_dir_path.join(".hg/hgrc-not-shared"),
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	199 ),
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	200 ];
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	201 // TODO: handle errors from
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	202 // `load_from_explicit_sources`
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	203 Config::load_from_explicit_sources(config_files).ok()
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	204 } else {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	205 None
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	206 }
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	207 } else {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	208 None
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	209 }
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	210 };
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	211
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	212 let non_repo_config_val = {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	213 let non_repo_val = non_repo_config.get(b"paths", &repo_arg);
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	214 match &non_repo_val {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	215 Some(val) if val.len() > 0 => home::home_dir()
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	216 .unwrap_or_else(\|\| PathBuf::from("~"))
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	217 .join(get_path_from_bytes(val))
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	218 .canonicalize()
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	219 // TODO: handle error and make it similar to python
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	220 // implementation maybe?
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	221 .ok(),
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	222 _ => None,
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	223 }
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	224 };
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	225
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	226 let config_val = match &local_config {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	227 None => non_repo_config_val,
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	228 Some(val) => {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	229 let local_config_val = val.get(b"paths", &repo_arg);
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	230 match &local_config_val {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	231 Some(val) if val.len() > 0 => {
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	232 // presence of a local_config assures that
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	233 // current_dir
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	234 // wont result in an Error
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	235 let canpath = hg::utils::current_dir()
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	236 .unwrap()
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	237 .join(get_path_from_bytes(val))
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	238 .canonicalize();
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	239 canpath.ok().or(non_repo_config_val)
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	240 }
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	241 _ => non_repo_config_val,
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	242 }
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	243 }
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	244 };
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	245 config_val.or(Some(get_path_from_bytes(&repo_arg).to_path_buf()))
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	246 }
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	247 };
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	248
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	249 let repo_result = match Repo::find(&non_repo_config, repo_path.to_owned())
ebdef6283798 rhg: read [paths] for `--repository` value Pulkit Goyal <7895pulkit@gmail.com> parents: 46822 diff changeset	250 {
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	251 Ok(repo) => Ok(repo),
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	252 Err(RepoError::NotFound { at }) if repo_path.is_none() => {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	253 // Not finding a repo is not fatal yet, if `-R` was not given
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	254 Err(NoRepoInCwdError { cwd: at })
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	255 }
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	256 Err(error) => exit(
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	257 &initial_current_dir,
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	258 &ui,
46761 bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	259 OnUnsupported::from_config(&ui, &non_repo_config),
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	260 Err(error.into()),
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	261 // TODO: show a warning or combine with original error if
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	262 // `get_bool` returns an error
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	263 non_repo_config
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	264 .get_bool(b"ui", b"detailed-exit-code")
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	265 .unwrap_or(false),
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	266 ),
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	267 };
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	268
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	269 let config = if let Ok(repo) = &repo_result {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	270 repo.config()
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	271 } else {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	272 &non_repo_config
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	273 };
46761 bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	274 let on_unsupported = OnUnsupported::from_config(&ui, config);
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	275
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	276 let result = main_with_result(
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	277 &process_start_time,
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	278 &ui,
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	279 repo_result.as_ref(),
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	280 config,
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	281 );
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	282 exit(
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	283 &initial_current_dir,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	284 &ui,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	285 on_unsupported,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	286 result,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	287 // TODO: show a warning or combine with original error if `get_bool`
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	288 // returns an error
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	289 config
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	290 .get_bool(b"ui", b"detailed-exit-code")
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	291 .unwrap_or(false),
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	292 )
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	293 }
755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	294
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	295 fn exit_code(
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	296 result: &Result<(), CommandError>,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	297 use_detailed_exit_code: bool,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	298 ) -> i32 {
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	299 match result {
46630 21d3b40b4c0e rhg: Remove error message on unsupported CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46557 diff changeset	300 Ok(()) => exitcode::OK,
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	301 Err(CommandError::Abort {
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	302 message: _,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	303 detailed_exit_code,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	304 }) => {
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	305 if use_detailed_exit_code {
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	306 *detailed_exit_code
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	307 } else {
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	308 exitcode::ABORT
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	309 }
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	310 }
46757 b1f2c2b336ec rhg: `cat` command: print error messages for missing files Simon Sapin <simon.sapin@octobus.net> parents: 46749 diff changeset	311 Err(CommandError::Unsuccessful) => exitcode::UNSUCCESSFUL,
46513 ca3f73cc3cf4 rhg: Simplify CommandError based on its use Simon Sapin <simon.sapin@octobus.net> parents: 45938 diff changeset	312
ca3f73cc3cf4 rhg: Simplify CommandError based on its use Simon Sapin <simon.sapin@octobus.net> parents: 45938 diff changeset	313 // Exit with a specific code and no error message to let a potential
ca3f73cc3cf4 rhg: Simplify CommandError based on its use Simon Sapin <simon.sapin@octobus.net> parents: 45938 diff changeset	314 // wrapper script fallback to Python-based Mercurial.
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	315 Err(CommandError::UnsupportedFeature { .. }) => {
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	316 exitcode::UNIMPLEMENTED
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	317 }
46640 755c31a1caf9 rhg: Add support for the blackbox extension Simon Sapin <simon.sapin@octobus.net> parents: 46635 diff changeset	318 }
45001 cf04f62d1579 rhg: add rhg crate Antoine Cezar <antoine.cezar@octobus.net> parents: diff changeset	319 }
45535 f17caf8f3fef rhg: add a limited `rhg debugdata` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45451 diff changeset	320
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	321 fn exit(
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	322 initial_current_dir: &Option<PathBuf>,
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	323 ui: &Ui,
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	324 mut on_unsupported: OnUnsupported,
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	325 result: Result<(), CommandError>,
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	326 use_detailed_exit_code: bool,
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	327 ) -> ! {
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	328 if let (
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	329 OnUnsupported::Fallback { executable },
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	330 Err(CommandError::UnsupportedFeature { .. }),
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	331 ) = (&on_unsupported, &result)
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	332 {
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	333 let mut args = std::env::args_os();
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	334 let executable_path = get_path_from_bytes(&executable);
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	335 let this_executable = args.next().expect("exepcted argv[0] to exist");
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	336 if executable_path == &PathBuf::from(this_executable) {
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	337 // Avoid spawning infinitely many processes until resource
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	338 // exhaustion.
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	339 let _ = ui.write_stderr(&format_bytes!(
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	340 b"Blocking recursive fallback. The 'rhg.fallback-executable = {}' config \
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	341 points to `rhg` itself.\n",
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	342 executable
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	343 ));
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	344 on_unsupported = OnUnsupported::Abort
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	345 } else {
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	346 // `args` is now `argv[1..]` since we’ve already consumed `argv[0]`
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	347 let mut command = Command::new(executable_path);
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	348 command.args(args);
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	349 if let Some(initial) = initial_current_dir {
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	350 command.current_dir(initial);
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	351 }
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	352 let result = command.status();
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	353 match result {
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	354 Ok(status) => std::process::exit(
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	355 status.code().unwrap_or(exitcode::ABORT),
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	356 ),
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	357 Err(error) => {
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	358 let _ = ui.write_stderr(&format_bytes!(
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	359 b"tried to fall back to a '{}' sub-process but got error {}\n",
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	360 executable, format_bytes::Utf8(error)
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	361 ));
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	362 on_unsupported = OnUnsupported::Abort
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	363 }
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	364 }
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	365 }
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	366 }
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	367 exit_no_fallback(ui, on_unsupported, result, use_detailed_exit_code)
46761 bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	368 }
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	369
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	370 fn exit_no_fallback(
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	371 ui: &Ui,
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	372 on_unsupported: OnUnsupported,
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	373 result: Result<(), CommandError>,
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	374 use_detailed_exit_code: bool,
46761 bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	375 ) -> ! {
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	376 match &result {
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	377 Ok(_) => {}
46757 b1f2c2b336ec rhg: `cat` command: print error messages for missing files Simon Sapin <simon.sapin@octobus.net> parents: 46749 diff changeset	378 Err(CommandError::Unsuccessful) => {}
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	379 Err(CommandError::Abort {
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	380 message,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	381 detailed_exit_code: _,
821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	382 }) => {
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	383 if !message.is_empty() {
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	384 // Ignore errors when writing to stderr, we’re already exiting
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	385 // with failure code so there’s not much more we can do.
46744 3d692e724d06 rhg: Align config file parse error formatting with Python Simon Sapin <simon.sapin@octobus.net> parents: 46742 diff changeset	386 let _ = ui.write_stderr(&format_bytes!(b"{}\n", message));
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	387 }
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	388 }
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	389 Err(CommandError::UnsupportedFeature { message }) => {
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	390 match on_unsupported {
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	391 OnUnsupported::Abort => {
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	392 let _ = ui.write_stderr(&format_bytes!(
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	393 b"unsupported feature: {}\n",
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	394 message
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	395 ));
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	396 }
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	397 OnUnsupported::AbortSilent => {}
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	398 OnUnsupported::Fallback { .. } => unreachable!(),
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	399 }
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	400 }
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	401 }
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	402 std::process::exit(exit_code(&result, use_detailed_exit_code))
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	403 }
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	404
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	405 macro_rules! subcommands {
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	406 ($( $command: ident )+) => {
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	407 mod commands {
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	408 $(
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	409 pub mod $command;
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	410 )+
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	411 }
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	412
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	413 fn add_subcommand_args<'a, 'b>(app: App<'a, 'b>) -> App<'a, 'b> {
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	414 app
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	415 $(
46647 4e4c70401028 rhg: Use clap?s support for global CLI arguments Simon Sapin <simon.sapin@octobus.net> parents: 46640 diff changeset	416 .subcommand(commands::$command::args())
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	417 )+
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	418 }
46543 a6e4e4650bac rhg: Parse system and user configuration at program start Simon Sapin <simon.sapin@octobus.net> parents: 46513 diff changeset	419
46631 80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	420 pub type RunFn = fn(&CliInvocation) -> Result<(), CommandError>;
80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	421
80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	422 fn subcommand_run_fn(name: &str) -> Option<RunFn> {
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	423 match name {
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	424 $(
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	425 stringify!($command) => Some(commands::$command::run),
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	426 )+
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	427 _ => None,
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	428 }
45548 33ded2d3f4fc rhg: add a limited `rhg cat -r` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45543 diff changeset	429 }
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	430 };
45535 f17caf8f3fef rhg: add a limited `rhg debugdata` subcommand Antoine Cezar <antoine.cezar@octobus.net> parents: 45451 diff changeset	431 }
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	432
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	433 subcommands! {
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	434 cat
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	435 debugdata
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	436 debugrequirements
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	437 files
95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	438 root
46557 a25033eb43b5 rhg: add limited support for the `config` sub-command Simon Sapin <simon.sapin@octobus.net> parents: 46556 diff changeset	439 config
46822 c71e8d9e7f2a rhg: Initial support for the 'status' command Georges Racinet <georges.racinet@octobus.net> parents: 46820 diff changeset	440 status
46554 95d37db31479 rhg: Replace subcommand boilerplate with a macro Simon Sapin <simon.sapin@octobus.net> parents: 46553 diff changeset	441 }
46822 c71e8d9e7f2a rhg: Initial support for the 'status' command Georges Racinet <georges.racinet@octobus.net> parents: 46820 diff changeset	442
46631 80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	443 pub struct CliInvocation<'a> {
80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	444 ui: &'a Ui,
80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	445 subcommand_args: &'a ArgMatches<'a>,
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	446 config: &'a Config,
46632 5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	447 /// References inside `Result` is a bit peculiar but allow
5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	448 /// `invocation.repo?` to work out with `&CliInvocation` since this
5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	449 /// `Result` type is `Copy`.
5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	450 repo: Result<&'a Repo, &'a NoRepoInCwdError>,
5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	451 }
5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	452
5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	453 struct NoRepoInCwdError {
5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	454 cwd: PathBuf,
46631 80840b651721 rhg: Group values passed to every sub-command into a struct Simon Sapin <simon.sapin@octobus.net> parents: 46630 diff changeset	455 }
46632 5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	456
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	457 /// CLI arguments to be parsed "early" in order to be able to read
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	458 /// configuration before using Clap. Ideally we would also use Clap for this,
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	459 /// see <https://github.com/clap-rs/clap/discussions/2366>.
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	460 ///
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	461 /// These arguments are still declared when we do use Clap later, so that Clap
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	462 /// does not return an error for their presence.
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	463 struct EarlyArgs {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	464 /// Values of all `--config` arguments. (Possibly none)
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	465 config: Vec<Vec<u8>>,
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	466 /// Value of the `-R` or `--repository` argument, if any.
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	467 repo: Option<Vec<u8>>,
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	468 /// Value of the `--cwd` argument, if any.
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	469 cwd: Option<Vec<u8>>,
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	470 }
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	471
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	472 impl EarlyArgs {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	473 fn parse(args: impl IntoIterator<Item = OsString>) -> Self {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	474 let mut args = args.into_iter().map(get_bytes_from_os_str);
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	475 let mut config = Vec::new();
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	476 let mut repo = None;
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	477 let mut cwd = None;
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	478 // Use `while let` instead of `for` so that we can also call
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	479 // `args.next()` inside the loop.
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	480 while let Some(arg) = args.next() {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	481 if arg == b"--config" {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	482 if let Some(value) = args.next() {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	483 config.push(value)
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	484 }
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	485 } else if let Some(value) = arg.drop_prefix(b"--config=") {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	486 config.push(value.to_owned())
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	487 }
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	488
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	489 if arg == b"--cwd" {
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	490 if let Some(value) = args.next() {
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	491 cwd = Some(value)
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	492 }
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	493 } else if let Some(value) = arg.drop_prefix(b"--cwd=") {
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	494 cwd = Some(value.to_owned())
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	495 }
2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	496
46704 7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	497 if arg == b"--repository" \|\| arg == b"-R" {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	498 if let Some(value) = args.next() {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	499 repo = Some(value)
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	500 }
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	501 } else if let Some(value) = arg.drop_prefix(b"--repository=") {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	502 repo = Some(value.to_owned())
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	503 } else if let Some(value) = arg.drop_prefix(b"-R") {
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	504 repo = Some(value.to_owned())
7284b524b441 rhg: Make configuration available as early as possible in main() Simon Sapin <simon.sapin@octobus.net> parents: 46647 diff changeset	505 }
46632 5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	506 }
46749 2255e7eb39e5 rhg: Add support for --cwd Simon Sapin <simon.sapin@octobus.net> parents: 46747 diff changeset	507 Self { config, repo, cwd }
46632 5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	508 }
5ce2aa7c2ad5 rhg: Move `Repo` object creation into `main()` Simon Sapin <simon.sapin@octobus.net> parents: 46631 diff changeset	509 }
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	510
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	511 /// What to do when encountering some unsupported feature.
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	512 ///
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	513 /// See `HgError::UnsupportedFeature` and `CommandError::UnsupportedFeature`.
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	514 enum OnUnsupported {
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	515 /// Print an error message describing what feature is not supported,
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	516 /// and exit with code 252.
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	517 Abort,
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	518 /// Silently exit with code 252.
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	519 AbortSilent,
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	520 /// Try running a Python implementation
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	521 Fallback { executable: Vec<u8> },
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	522 }
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	523
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	524 impl OnUnsupported {
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	525 const DEFAULT: Self = OnUnsupported::Abort;
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	526
46761 bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	527 fn from_config(ui: &Ui, config: &Config) -> Self {
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	528 match config
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	529 .get(b"rhg", b"on-unsupported")
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	530 .map(\|value\| value.to_ascii_lowercase())
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	531 .as_deref()
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	532 {
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	533 Some(b"abort") => OnUnsupported::Abort,
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	534 Some(b"abort-silent") => OnUnsupported::AbortSilent,
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	535 Some(b"fallback") => OnUnsupported::Fallback {
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	536 executable: config
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	537 .get(b"rhg", b"fallback-executable")
46761 bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	538 .unwrap_or_else(\|\| {
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	539 exit_no_fallback(
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	540 ui,
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	541 Self::Abort,
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	542 Err(CommandError::abort(
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	543 "abort: 'rhg.on-unsupported=fallback' without \
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	544 'rhg.fallback-executable' set."
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	545 )),
46820 821929d59e01 rhg: add support for detailed exit code for ConfigParseError Pulkit Goyal <7895pulkit@gmail.com> parents: 46761 diff changeset	546 false,
46761 bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	547 )
bde90e9b4507 rhg: Remove `rhg.fallback-executable=hg` default configuration Simon Sapin <simon.sapin@octobus.net> parents: 46759 diff changeset	548 })
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	549 .to_owned(),
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	550 },
93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	551 None => Self::DEFAULT,
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	552 Some(_) => {
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	553 // TODO: warn about unknown config value
46706 93e9f448273c rhg: Add support for automatic fallback to Python Simon Sapin <simon.sapin@octobus.net> parents: 46705 diff changeset	554 Self::DEFAULT
46705 33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	555 }
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	556 }
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	557 }
33f2d56acc73 rhg: Add a `rhg.on-unsupported` configuration key Simon Sapin <simon.sapin@octobus.net> parents: 46704 diff changeset	558 }
46746 1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	559
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	560 const SUPPORTED_EXTENSIONS: &[&[u8]] = &[b"blackbox", b"share"];
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	561
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	562 fn check_extensions(config: &Config) -> Result<(), CommandError> {
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	563 let enabled = config.get_section_keys(b"extensions");
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	564
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	565 let mut unsupported = enabled;
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	566 for supported in SUPPORTED_EXTENSIONS {
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	567 unsupported.remove(supported);
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	568 }
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	569
46747 1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	570 if let Some(ignored_list) =
1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	571 config.get_simple_list(b"rhg", b"ignored-extensions")
1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	572 {
1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	573 for ignored in ignored_list {
1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	574 unsupported.remove(ignored);
1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	575 }
1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	576 }
1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	577
46746 1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	578 if unsupported.is_empty() {
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	579 Ok(())
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	580 } else {
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	581 Err(CommandError::UnsupportedFeature {
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	582 message: format_bytes!(
46747 1a036d33bc18 rhg: Add an allow-list of ignored extensions Simon Sapin <simon.sapin@octobus.net> parents: 46746 diff changeset	583 b"extensions: {} (consider adding them to 'rhg.ignored-extensions' config)",
46746 1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	584 join(unsupported, b", ")
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	585 ),
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	586 })
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	587 }
1bac7764ceef rhg: Fall back to Python if unsupported extensions are enabled Simon Sapin <simon.sapin@octobus.net> parents: 46744 diff changeset	588 }

Mercurial > public > mercurial-scm > hg-stable

annotate rust/rhg/src/main.rs @ 47412:3237ed4dcda4