wit_smith/
config.rs

1use arbitrary::{Arbitrary, Result, Unstructured};
2
3#[derive(Debug, Clone)]
4#[cfg_attr(feature = "clap", derive(clap::Parser))]
5pub struct Config {
6    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().max_packages))]
7    pub max_packages: usize,
8    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().max_type_size))]
9    pub max_type_size: usize,
10    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().max_interface_items))]
11    pub max_interface_items: usize,
12    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().max_world_items))]
13    pub max_world_items: usize,
14    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().max_pkg_items))]
15    pub max_pkg_items: usize,
16    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().max_type_parts))]
17    pub max_type_parts: usize,
18    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().max_files_per_package))]
19    pub max_files_per_package: usize,
20    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().max_resource_items))]
21    pub max_resource_items: usize,
22    #[cfg_attr(feature = "clap", clap(long = "async", default_value_t = Config::default().async_))]
23    pub async_: bool,
24    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().futures))]
25    pub futures: bool,
26    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().streams))]
27    pub streams: bool,
28    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().error_context))]
29    pub error_context: bool,
30    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().fixed_size_list))]
31    pub fixed_size_list: bool,
32    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().world_include))]
33    pub world_include: bool,
34}
35
36impl Default for Config {
37    fn default() -> Config {
38        Config {
39            max_packages: 10,
40            max_type_size: 100,
41            max_interface_items: 10,
42            max_world_items: 10,
43            max_pkg_items: 10,
44            max_type_parts: 10,
45            max_files_per_package: 10,
46            max_resource_items: 10,
47            async_: false,
48            futures: false,
49            streams: false,
50            error_context: false,
51            fixed_size_list: false,
52            world_include: false,
53        }
54    }
55}
56
57impl Arbitrary<'_> for Config {
58    fn arbitrary(u: &mut Unstructured<'_>) -> Result<Config> {
59        Ok(Config {
60            max_packages: u.int_in_range(1..=20)?,
61            max_files_per_package: u.int_in_range(1..=10)?,
62            max_type_size: u.int_in_range(0..=1000)?,
63            max_interface_items: u.int_in_range(0..=20)?,
64            max_world_items: u.int_in_range(0..=10)?,
65            max_pkg_items: u.int_in_range(1..=10)?,
66            max_type_parts: u.int_in_range(1..=10)?,
67            max_resource_items: u.int_in_range(0..=10)?,
68            async_: u.arbitrary()?,
69            futures: u.arbitrary()?,
70            streams: u.arbitrary()?,
71            error_context: u.arbitrary()?,
72            fixed_size_list: u.arbitrary()?,
73            world_include: false,
74        })
75    }
76}