Define a cataloger and a fetcher.

Author: artob

Cargo.lock

@@ -19,8 +19,8 @@ publish = true
 [features]
 default = ["all", "cli", "std"]
 all = ["pretty", "tracing"]
-cli = ["asimov-module/cli", "std", "dep:clap", "dep:clientele"]
-std = ["asimov-module/std", "clap?/std", "clientele?/std"]
+cli = ["asimov-module/cli", "dogma/clap", "std", "dep:clap", "dep:clientele"]
+std = ["asimov-module/std", "clap?/std", "clientele?/std", "dogma/std"]
 unstable = []
 
 # Optional features:
@@ -29,6 +29,7 @@ tracing = ["asimov-module/tracing", "clientele?/tracing"]
 
 [dependencies]
 asimov-module = { version = "25.0.0-dev.21", default-features = true }
+dogma = { version = "0.1.11", default-features = false, features = ["all"] }
 hex = "0.4"
 libaes = "0.7"
 pbkdf2 = "0.12"
@@ -58,6 +59,16 @@ opt-level = "z"
 strip = true
 lto = "thin"
 
+[[bin]]
+name = "asimov-signal-cataloger"
+path = "src/cataloger/main.rs"
+required-features = ["cli"]
+
+[[bin]]
+name = "asimov-signal-fetcher"
+path = "src/fetcher/main.rs"
+required-features = ["cli"]
+
 [[bin]]
 name = "asimov-signal-reader"
 path = "src/reader/main.rs"

Cargo.toml

@@ -0,0 +1,90 @@
+// This is free and unencumbered software released into the public domain.
+
+#[cfg(not(feature = "std"))]
+compile_error!("asimov-signal-cataloger requires the 'std' feature");
+
+use asimov_module::SysexitsError::{self, *};
+use clap::Parser;
+use clientele::StandardOptions;
+use dogma::{Uri, UriScheme, UriValueParser};
+use std::error::Error;
+
+/// asimov-signal-cataloger
+#[derive(Debug, Parser)]
+#[command(arg_required_else_help = false)]
+struct Options {
+    #[clap(flatten)]
+    flags: StandardOptions,
+
+    /// Limit the number of resources to catalog.
+    #[arg(value_name = "COUNT", short = 'n', long)]
+    limit: Option<usize>,
+
+    /// Set the output format [default: jsonl] [possible values: jsonl]
+    #[arg(value_name = "FORMAT", short = 'o', long)]
+    output: Option<String>,
+
+    /// The Signal URL to catalog (e.g., `sgnl://signal.me/#p/...`)
+    #[arg(value_name = "URL", value_parser = UriValueParser::new(&[
+        UriScheme::Other("sgnl".into()),
+    ]))]
+    url: Uri<'static>,
+}
+
+pub fn main() -> Result<SysexitsError, Box<dyn Error>> {
+    // Load environment variables from `.env`:
+    asimov_module::dotenv().ok();
+
+    // Expand wildcards and @argfiles:
+    let args = asimov_module::args_os()?;
+
+    // Parse command-line options:
+    let options = Options::parse_from(args);
+
+    // Handle the `--version` flag:
+    if options.flags.version {
+        println!("{} {}", env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
+        return Ok(EX_OK);
+    }
+
+    // Handle the `--license` flag:
+    if options.flags.license {
+        print!("{}", include_str!("../../UNLICENSE"));
+        return Ok(EX_OK);
+    }
+
+    // Locate the program path:
+    let mut program_path = std::env::current_exe()?;
+    program_path.set_file_name("asimov-signal-reader");
+    if cfg!(windows) {
+        program_path.set_extension("exe");
+    }
+
+    // Construct the program arguments:
+    let mut program_args = vec![format!(
+        "--output={}",
+        options.output.as_deref().unwrap_or("jsonl")
+    )];
+    if options.flags.verbose > 0 {
+        program_args.push(format!("-{}", "v".repeat(options.flags.verbose as usize)));
+    }
+    if options.flags.debug {
+        program_args.push("-d".into());
+    }
+
+    // On Unix-like systems, use `execvp(3)` to replace the current process:
+    #[cfg(unix)]
+    {
+        use std::{os::unix::process::CommandExt, process::Command};
+        let error = Command::new(&program_path).args(program_args).exec();
+        return Err(format!("failed to exec: {}", error).into());
+    }
+
+    // On Windows or when `execvp(3)` is not available, spawn a subprocess:
+    #[cfg(not(unix))]
+    {
+        use std::process::{Command, exit};
+        let status = Command::new(&program_path).args(program_args).status()?;
+        exit(status.code().unwrap_or(SysexitsError::EX_UNAVAILABLE));
+    }
+}

src/cataloger/main.rs

@@ -0,0 +1,86 @@
+// This is free and unencumbered software released into the public domain.
+
+#[cfg(not(feature = "std"))]
+compile_error!("asimov-signal-fetcher requires the 'std' feature");
+
+use asimov_module::SysexitsError::{self, *};
+use clap::Parser;
+use clientele::StandardOptions;
+use dogma::{Uri, UriScheme, UriValueParser};
+use std::error::Error;
+
+/// asimov-signal-fetcher
+#[derive(Debug, Parser)]
+#[command(arg_required_else_help = false)]
+struct Options {
+    #[clap(flatten)]
+    flags: StandardOptions,
+
+    /// Set the output format [default: jsonl] [possible values: jsonl]
+    #[arg(value_name = "FORMAT", short = 'o', long)]
+    output: Option<String>,
+
+    /// The Signal URL to fetch (e.g., `sgnl://signal.me/#p/...`)
+    #[arg(value_name = "URL", value_parser = UriValueParser::new(&[
+        UriScheme::Other("sgnl".into()),
+    ]))]
+    url: Uri<'static>,
+}
+
+pub fn main() -> Result<SysexitsError, Box<dyn Error>> {
+    // Load environment variables from `.env`:
+    asimov_module::dotenv().ok();
+
+    // Expand wildcards and @argfiles:
+    let args = asimov_module::args_os()?;
+
+    // Parse command-line options:
+    let options = Options::parse_from(args);
+
+    // Handle the `--version` flag:
+    if options.flags.version {
+        println!("{} {}", env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
+        return Ok(EX_OK);
+    }
+
+    // Handle the `--license` flag:
+    if options.flags.license {
+        print!("{}", include_str!("../../UNLICENSE"));
+        return Ok(EX_OK);
+    }
+
+    // Locate the program path:
+    let mut program_path = std::env::current_exe()?;
+    program_path.set_file_name("asimov-signal-reader");
+    if cfg!(windows) {
+        program_path.set_extension("exe");
+    }
+
+    // Construct the program arguments:
+    let mut program_args = vec![format!(
+        "--output={}",
+        options.output.as_deref().unwrap_or("jsonl")
+    )];
+    if options.flags.verbose > 0 {
+        program_args.push(format!("-{}", "v".repeat(options.flags.verbose as usize)));
+    }
+    if options.flags.debug {
+        program_args.push("-d".into());
+    }
+
+    // On Unix-like systems, use `execvp(3)` to replace the current process:
+    #[cfg(unix)]
+    {
+        use std::{os::unix::process::CommandExt, process::Command};
+        let error = Command::new(&program_path).args(program_args).exec();
+        return Err(format!("failed to exec: {}", error).into());
+    }
+
+    // On Windows or when `execvp(3)` is not available, spawn a subprocess:
+    #[cfg(not(unix))]
+    {
+        use std::process::{Command, exit};
+        let status = Command::new(&program_path).args(program_args).status()?;
+        exit(status.code().unwrap_or(SysexitsError::EX_UNAVAILABLE));
+    }
+}