args.rs

use clap::builder::Styles;
use clap::builder::styling::AnsiColor;
use clap::{ArgAction, Parser, Subcommand};
use pdfium_render::prelude::PdfPageIndex;
use paperjet::options::{
	ColorMode, Finishing, MediaSize, MediaSource, MediaType, Orientation, Quality, SidesMode,
};
use std::ffi::c_int;
use std::path::PathBuf;

mod headings {
	pub const DEVICES: &str = "Devices";
	pub const PDF_MANIP: &str = "File transformations (PDF only)";
	pub const PRINT_OPTS: &str = "Printing options";
	pub const MISC: &str = "Miscellaneous";
}

#[derive(Parser)]
#[command(version, disable_version_flag = true, disable_help_flag = true, styles = help_style())]
pub struct Args {
	#[command(subcommand)]
	pub command: Command,

	// /// Output more information to the console.
	// #[arg(short, long, global = true, help_heading = headings::MISC)]
	// pub verbose: bool,
	/// Print version.
	#[arg(long, action = ArgAction::Version, help_heading = headings::MISC)]
	pub version: Option<bool>,

	/// Print help.
	#[arg(short, long, global = true, action = ArgAction::Help, help_heading = headings::MISC)]
	pub help: Option<bool>,
}

#[derive(Subcommand)]
#[command(disable_help_subcommand = true)]
pub enum Command {
	/// Lists available printers.
	List,
	/// Displays information about a printer.
	Display(DisplayArgs),
	/// Submits one or more documents for printing.
	///
	/// This command supports extensive configuration of options such as the amount of copies,
	/// paper size, orientation, and others, listed below.
	/// Support is up to a particular device - unsupported options or option values will be rejected
	/// and the printing will not commence.
	Print(PrintArgs),
}

#[derive(clap::Args)]
pub struct DisplayArgs {
	/// The ID (as determined by the `list` command) or the name of the printer.
	pub criteria: String,
	/// Display all options of the printer.
	#[arg(short, long)]
	pub options: bool,
}

#[derive(Debug, clap::Args)]
pub struct PrintArgs {
	/// Paths to the files to be printed.
	///
	/// File extensions, types, or contents are not validated.
	/// Support will be determined by the device's driver.
	#[arg(value_name = "files", required = true, num_args = 1..)]
	pub paths: Vec<PathBuf>,

	/// The ID of the printer that should print the files, as per the `list` command.
	///
	/// The ID refers to the position in the list output by the `list` command.
	/// If not specified, the default printer is used.
	#[arg(short, long, value_name = "ID", group = "printer", help_heading = headings::DEVICES)]
	pub printer_id: Option<usize>,

	/// The name of the printer that should print the files.
	///
	/// Use the `list` command to get a list of available printers and their names.
	#[arg(
		short = 'n',
		long,
		value_name = "NAME",
		group = "printer",
		help_heading = headings::DEVICES
	)]
	pub printer_name: Option<String>,

	/// The first page of the document to be printed (omits previous pages), counting from 1.
	#[arg(short, long, value_name = "PAGE", help_heading = headings::PDF_MANIP)]
	pub from: Option<PdfPageIndex>,

	/// The last page of the document to be printed (omits following pages).
	#[arg(short, long, value_name = "PAGE", help_heading = headings::PDF_MANIP)]
	pub to: Option<PdfPageIndex>,

	/// Split the document in two to simulate duplex mode (printing on both sides).
	#[arg(short, long, help_heading = headings::PDF_MANIP)]
	pub duplex: bool,

	/// Amount of copies [default: 1]
	///
	/// In case of multiple files, this option applies to each of them.
	#[arg(
		short,
		long,
		value_parser = clap::value_parser!(c_int).range(1..),
		help_heading = headings::PRINT_OPTS
	)]
	pub copies: Option<c_int>,

	/// Finishing processes to be performed by the printer.
	#[arg(long, value_delimiter = ',', help_heading = headings::PRINT_OPTS)]
	pub finishings: Option<Vec<Finishing>>,

	/// Size of the media, most often paper size.
	#[arg(short, long, help_heading = headings::PRINT_OPTS)]
	pub size: Option<MediaSize>,

	/// Source where the media is pulled from.
	#[arg(short = 'r', long, help_heading = headings::PRINT_OPTS)]
	pub source: Option<MediaSource>,

	/// Type of media.
	#[arg(long, value_name = "TYPE", help_heading = headings::PRINT_OPTS)]
	pub media_type: Option<MediaType>,

	/// Number of document pages per media side [default: 1]
	#[arg(
		short = 'u',
		long,
		value_name = "PAGES",
		value_parser = clap::value_parser!(c_int).range(1..),
		help_heading = headings::PRINT_OPTS
	)]
	pub number_up: Option<c_int>,

	/// Orientation of document pages.
	#[arg(short, long, help_heading = headings::PRINT_OPTS)]
	pub orientation: Option<Orientation>,

	/// Determines whether the printer should use color or monochrome ink.
	#[arg(short = 'm', long, help_heading = headings::PRINT_OPTS)]
	pub color_mode: Option<ColorMode>,

	/// The quality of the resulting print.
	#[arg(short, long, value_name = "MODE", help_heading = headings::PRINT_OPTS)]
	pub quality: Option<Quality>,

	/// Determines whether only one or both sides of the media should be printed on.
	#[arg(short = 'i', value_name = "MODE", long, help_heading = headings::PRINT_OPTS)]
	pub sides_mode: Option<SidesMode>,
}

impl Args {
	pub fn parse() -> Self {
		<Self as Parser>::parse()
	}
}

fn help_style() -> Styles {
	Styles::styled()
		.context(AnsiColor::White.on_default().dimmed())
		.placeholder(AnsiColor::White.on_default().dimmed())
}