1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// Core CLI execution: run(), apply_ux_settings(), and parse_with_suggestions().
// These functions handle CLI argument parsing and command dispatch.
#[cfg_attr(coverage_nightly, coverage(off))]
#[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
pub async fn run(server: Arc<StatelessTemplateServer>) -> anyhow::Result<()> {
let cli = match parse_with_suggestions() {
Ok(cli) => cli,
Err(suggestion_msg) => {
eprintln!("{suggestion_msg}");
std::process::exit(2); // Exit with "misuse" code for command errors
}
};
debug!("CLI arguments parsed");
// Apply UX settings (TICKET-PMAT-6006)
apply_ux_settings(&cli);
// Handle forced mode
if let Some(commands::Mode::Mcp) = cli.mode {
info!("Forced MCP mode detected");
return crate::run_mcp_server(server).await;
}
// Use command dispatcher for improved modularity
CommandDispatcher::execute_command(cli.command, server).await
}
/// Apply UX settings from CLI flags (TICKET-PMAT-6006)
///
/// CC=3: Quiet mode + color mode checks
fn apply_ux_settings(cli: &commands::Cli) {
// Set quiet mode environment variable
if cli.quiet {
std::env::set_var("PMAT_QUIET", "1");
}
// Handle color mode
match cli.color {
commands::ColorMode::Never => {
std::env::set_var("NO_COLOR", "1");
}
commands::ColorMode::Always => {
std::env::set_var("CLICOLOR_FORCE", "1");
}
commands::ColorMode::Auto => {
// Auto mode - respect existing environment
}
}
}
/// Parse CLI with command suggestions on failure
#[cfg_attr(coverage_nightly, coverage(off))]
fn parse_with_suggestions() -> Result<Cli, String> {
use crate::utils::command_suggestions::CommandSuggester;
use clap::Parser;
// Try to parse normally first
match Cli::try_parse() {
Ok(cli) => Ok(cli),
Err(clap_error) => {
// Handle special cases where clap handles --version and --help
use clap::error::ErrorKind;
match clap_error.kind() {
ErrorKind::DisplayHelp => {
// Print the help message to stdout
print!("{clap_error}");
std::process::exit(0);
}
ErrorKind::DisplayVersion => {
// Print the version to stdout
print!("{clap_error}");
std::process::exit(0);
}
_ => {
// For other errors, provide suggestions
let args: Vec<String> = std::env::args().skip(1).collect();
let suggester = CommandSuggester::new();
// Get suggestion based on the failed arguments
if let Some(suggestion) = suggester.suggest_command(&args) {
let error_msg = format!(
"error: unrecognized subcommand\n\n{suggestion}\n\nFor more information, try 'pmat --help'"
);
Err(error_msg)
} else {
// If no suggestion, show the original clap error with examples
let examples = CommandSuggester::get_help_examples();
let error_msg = format!("{clap_error}{examples}");
Err(error_msg)
}
}
}
}
}
}