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_length_lists))]
31    pub fixed_length_lists: bool,
32    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().world_include))]
33    pub world_include: bool,
34    #[cfg_attr(feature = "clap", clap(long, default_value_t = Config::default().implements))]
35    pub implements: bool,
36}
37
38impl Default for Config {
39    fn default() -> Config {
40        Config {
41            max_packages: 10,
42            max_type_size: 100,
43            max_interface_items: 10,
44            max_world_items: 10,
45            max_pkg_items: 10,
46            max_type_parts: 10,
47            max_files_per_package: 10,
48            max_resource_items: 10,
49            async_: false,
50            futures: false,
51            streams: false,
52            error_context: false,
53            fixed_length_lists: false,
54            world_include: false,
55            implements: false,
56        }
57    }
58}
59
60impl Arbitrary<'_> for Config {
61    fn arbitrary(u: &mut Unstructured<'_>) -> Result<Config> {
62        Ok(Config {
63            max_packages: u.int_in_range(1..=20)?,
64            max_files_per_package: u.int_in_range(1..=10)?,
65            max_type_size: u.int_in_range(0..=1000)?,
66            max_interface_items: u.int_in_range(0..=20)?,
67            max_world_items: u.int_in_range(0..=10)?,
68            max_pkg_items: u.int_in_range(1..=10)?,
69            max_type_parts: u.int_in_range(1..=10)?,
70            max_resource_items: u.int_in_range(0..=10)?,
71            async_: u.arbitrary()?,
72            futures: u.arbitrary()?,
73            streams: u.arbitrary()?,
74            error_context: u.arbitrary()?,
75            fixed_length_lists: u.arbitrary()?,
76            world_include: false,
77            implements: u.arbitrary()?,
78        })
79    }
80}
wit_smith/config.rs

wit_smith/
config.rs
