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

annotate rust/rhg/src/main.rs @ 48076:b44e1184b7e1

rhg: fallback if `defaults` config is set for the current command Differential Revision: https://phab.mercurial-scm.org/D11381

author	Rapha?l Gom?s <rgomes@octobus.net>
date	Wed, 01 Sep 2021 18:09:35 +0200
parents	f11f233546ce
children	ba773bd99203

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

Mercurial > public > mercurial-scm > hg

annotate rust/rhg/src/main.rs @ 48076:b44e1184b7e1