|
|
|
|
@ -1,4 +1,3 @@
|
|
|
|
|
|
|
|
|
|
use failure::Fallible;
|
|
|
|
|
use log::*;
|
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
|
@ -11,12 +10,30 @@ fn is_default<T: Default + PartialEq>(t: &T) -> bool {
|
|
|
|
|
t == &T::default()
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// ugly, but serde and structopt use different methods to define defaults
|
|
|
|
|
macro_rules! set_default {
|
|
|
|
|
($name:ident, $val:expr, $type:ty) => {
|
|
|
|
|
paste::item! {
|
|
|
|
|
fn [<def_ $name>]() -> $type {
|
|
|
|
|
$val
|
|
|
|
|
}
|
|
|
|
|
fn [<def_ $name _if>](e: &$type) -> bool {
|
|
|
|
|
e == &[<def_ $name>]()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
set_default!(cache_compression_level, 12, u32);
|
|
|
|
|
set_default!(cache_max_blob_len, 2000000, u32);
|
|
|
|
|
set_default!(max_archive_recursion, 4, i32);
|
|
|
|
|
|
|
|
|
|
#[derive(StructOpt, Debug, Deserialize, Serialize)]
|
|
|
|
|
#[structopt(rename_all = "kebab-case")]
|
|
|
|
|
pub struct RgaOptions {
|
|
|
|
|
#[structopt(rename_all = "kebab-case", set_term_width = 80)]
|
|
|
|
|
pub struct RgaArgs {
|
|
|
|
|
#[serde(default, skip_serializing_if = "is_default")]
|
|
|
|
|
#[structopt(long, help = "Disable caching of results")]
|
|
|
|
|
pub no_cache: bool,
|
|
|
|
|
pub rga_no_cache: bool,
|
|
|
|
|
|
|
|
|
|
#[serde(default, skip_serializing_if = "is_default")]
|
|
|
|
|
#[structopt(
|
|
|
|
|
@ -25,7 +42,48 @@ pub struct RgaOptions {
|
|
|
|
|
require_delimiter = true,
|
|
|
|
|
help = "Change which adapters to use and in which priority order (descending)"
|
|
|
|
|
)]
|
|
|
|
|
pub adapters: Vec<String>,
|
|
|
|
|
pub rga_adapters: Vec<String>,
|
|
|
|
|
|
|
|
|
|
#[serde(
|
|
|
|
|
default = "def_cache_max_blob_len",
|
|
|
|
|
skip_serializing_if = "def_cache_max_blob_len_if"
|
|
|
|
|
)]
|
|
|
|
|
#[structopt(
|
|
|
|
|
long,
|
|
|
|
|
default_value = "2000000",
|
|
|
|
|
help = "Max compressed size to cache",
|
|
|
|
|
long_help = "Longest byte length (after compression) to store in cache. Longer adapter outputs will not be cached and recomputed every time."
|
|
|
|
|
)]
|
|
|
|
|
pub rga_cache_max_blob_len: u32,
|
|
|
|
|
|
|
|
|
|
#[serde(
|
|
|
|
|
default = "def_cache_compression_level",
|
|
|
|
|
skip_serializing_if = "def_cache_compression_level_if"
|
|
|
|
|
)]
|
|
|
|
|
#[structopt(
|
|
|
|
|
long,
|
|
|
|
|
default_value = "12",
|
|
|
|
|
require_equals = true,
|
|
|
|
|
help = "ZSTD compression level to apply to adapter outputs before storing in cache db"
|
|
|
|
|
)]
|
|
|
|
|
pub rga_cache_compression_level: u32,
|
|
|
|
|
|
|
|
|
|
#[serde(
|
|
|
|
|
default = "def_max_archive_recursion",
|
|
|
|
|
skip_serializing_if = "def_max_archive_recursion_if"
|
|
|
|
|
)]
|
|
|
|
|
#[structopt(
|
|
|
|
|
long,
|
|
|
|
|
default_value = "4",
|
|
|
|
|
require_equals = true,
|
|
|
|
|
help = "Maximum nestedness of archives to recurse into"
|
|
|
|
|
)]
|
|
|
|
|
pub rga_max_archive_recursion: i32,
|
|
|
|
|
|
|
|
|
|
// these arguments stop the process, so don't serialize them
|
|
|
|
|
#[serde(skip)]
|
|
|
|
|
#[structopt(long, help = "List all known adapters")]
|
|
|
|
|
pub rga_list_adapters: bool,
|
|
|
|
|
|
|
|
|
|
#[serde(skip)]
|
|
|
|
|
#[structopt(long, help = "Show help for ripgrep itself")]
|
|
|
|
|
@ -34,15 +92,11 @@ pub struct RgaOptions {
|
|
|
|
|
#[serde(skip)]
|
|
|
|
|
#[structopt(long, help = "Show version of ripgrep itself")]
|
|
|
|
|
pub rg_version: bool,
|
|
|
|
|
|
|
|
|
|
#[serde(skip)]
|
|
|
|
|
#[structopt(long, help = "List all known adapters")]
|
|
|
|
|
pub list_adapters: bool,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static RGA_CONFIG: &str = "RGA_CONFIG";
|
|
|
|
|
|
|
|
|
|
pub fn parse_args<I>(args: I) -> Fallible<RgaOptions>
|
|
|
|
|
pub fn parse_args<I>(args: I) -> Fallible<RgaArgs>
|
|
|
|
|
where
|
|
|
|
|
I: IntoIterator,
|
|
|
|
|
I::Item: Into<OsString> + Clone,
|
|
|
|
|
@ -53,7 +107,7 @@ where
|
|
|
|
|
Ok(serde_json::from_str(&val)?)
|
|
|
|
|
}
|
|
|
|
|
Err(_) => {
|
|
|
|
|
let matches = RgaOptions::from_iter(args);
|
|
|
|
|
let matches = RgaArgs::from_iter(args);
|
|
|
|
|
let serialized_config = serde_json::to_string(&matches)?;
|
|
|
|
|
std::env::set_var(RGA_CONFIG, &serialized_config);
|
|
|
|
|
debug!("{}={}", RGA_CONFIG, serialized_config);
|
|
|
|
|
|
phiresky
5 years ago