uu_awk: start using clap

Author: Alonely0

Cargo.lock

@@ -44,6 +44,7 @@ bumpalo.workspace = true
 rustix = { version = "1.1.4", default-features = false, features = ["process"]}
 tracing-subscriber = { version = "0.3.23", features = ["env-filter"] }
 tracing-error = "0.2.1"
+clap = { version = "4.6.1", features = ["derive"] }
 
 [dev-dependencies]
 ctor = "0.4"
@@ -61,9 +62,6 @@ inherits = "release"
 opt-level = "z"
 strip = true
 
-[profile.dev.package.backtrace]
-opt-level = 3
-
 # A release-like profile with debug info for profiling.
 # See https://github.com/mstange/samply .
 [profile.profiling]

Cargo.toml

@@ -0,0 +1,73 @@
+// This file is part of the uutils awk package.
+//
+// For the full copyright and license information, please view the LICENSE
+// files that was distributed with this source code.
+
+use std::{ffi::OsString, path::PathBuf};
+
+use clap::Parser;
+
+#[derive(Parser, Debug)]
+#[clap(version, name = "uutils AWK")]
+#[clap(about = ::std::concat!("uutils awk ", ::std::env!("CARGO_PKG_VERSION")))]
+pub struct Args {
+    // POSIX
+    pub code: OsString,
+    #[arg(short = 'f', long)]
+    file: Option<PathBuf>,
+    #[arg(short = 'F', long)]
+    field_separator: Option<OsString>,
+    #[arg(short = 'v', long, value_parser = parse_kv)]
+    assign: Vec<(String, String)>,
+    #[arg(short = 'b', long)]
+    characters_as_bytes: bool,
+    #[arg(short = 'c', long)]
+    traditional: bool,
+    #[arg(short = 'C', long)]
+    copyright: bool,
+    #[arg(short = 'd', long)]
+    dump_variables: Option<PathBuf>,
+    #[arg(short = 'D', long)]
+    debug: Option<PathBuf>,
+    #[arg(short = 'e', long)]
+    source: Vec<u8>,
+    #[arg(short = 'E', long)]
+    exec: Option<PathBuf>,
+    #[arg(short = 'g', long)]
+    gen_pot: bool,
+    #[arg(short = 'i', long)]
+    include: Option<PathBuf>,
+    #[arg(short = 'I', long)]
+    trace: bool,
+    #[arg(short = 'l', long)]
+    load: Vec<OsString>,
+    #[arg(short = 'L', long)]
+    lint: Vec<String>,
+    #[arg(short = 'M', long)]
+    bignum: bool,
+    #[arg(short = 'n', long)]
+    non_decimal_data: bool,
+    #[arg(short = 'N', long)]
+    use_lc_numeric: bool,
+    #[arg(short = 'o', long)]
+    pretty_print: Option<PathBuf>,
+    #[arg(short = 'O', long, default_value_t = true)]
+    optimize: bool,
+    #[arg(short = 's', long = "no-optimize")]
+    no_optimize: bool,
+    #[arg(short = 'p', long)]
+    profile: Option<PathBuf>,
+    #[arg(short = 'P', long)]
+    posix: bool,
+    #[arg(short = 'r', long, default_value_t = true)]
+    re_interval: bool,
+    #[arg(short = 'S', long)]
+    sandbox: bool,
+    #[arg(short = 't', long)]
+    lint_old: bool,
+}
+
+fn parse_kv(s: &str) -> Result<(String, String), String> {
+    let (k, v) = s.split_once('=').ok_or("expected key=value")?;
+    Ok((k.to_string(), v.trim_matches(['"', '\'']).to_string()))
+}

src/cli.rs

@@ -5,13 +5,18 @@
 
 // static POSIX: bool = false;
 
+mod cli;
 mod utils;
 
 use bumpalo::Bump;
+use clap::Parser as _;
 use color_eyre::Result;
 use parser::Parser;
 
-use crate::utils::{ensure_consistent_panic, exit_err};
+use crate::{
+    cli::Args,
+    utils::{ensure_consistent_panic, exit_err},
+};
 
 fn main() {
     if let Err(e) = ensure_consistent_panic(uu_main) {
@@ -21,11 +26,12 @@ fn main() {
 
 #[tracing::instrument]
 fn uu_main() -> Result<()> {
-    let arg = std::env::args().nth(1).unwrap();
+    let args = Args::parse();
+    println!("{args:?}");
 
     let arena = Bump::with_capacity(4000); // 4KB minus metadata-ish
     let mut parser = Parser::new(&arena);
-    let ast = match parser.parse("CLI", arg.as_bytes()) {
+    let ast = match parser.parse("CLI", args.code.as_encoded_bytes()) {
         Ok(ast) => dbg!(ast),
         Err((report, source)) => {
             report.eprint(("CLI", source)).unwrap();