feat(setups): add picker module with flag-based program setup

1 files changed, 87 insertions, 0 deletions

diff --git a/mingling/src/setups/picker/structural_renderer.rs b/mingling/src/setups/picker/structural_renderer.rs
new file mode 100644
index 0000000..69f05f1
--- /dev/null
+++ b/mingling/src/setups/picker/structural_renderer.rs
@@ -0,0 +1,87 @@
+use arg_picker::{IntoPicker, PickerArg, PickerArgs};
+use mingling_core::{Program, ProgramCollect, setup::ProgramSetup};
+
+use crate::setups::picker::{
+    JSON_FLAG, JSON_PRETTY_FLAG, RON_FLAG, RON_PRETTY_FLAG, TOML_FLAG, YAML_FLAG,
+};
+
+/// Sets up the structural renderer for the program:
+///
+/// - Adds a `--renderer` global argument to specify the renderer type
+pub struct StructuralRendererSimpleSetup;
+
+impl<C> ProgramSetup<C> for StructuralRendererSimpleSetup
+where
+    C: ProgramCollect<Enum = C>,
+{
+    fn setup(self, program: &mut Program<C>) {
+        program.global_argument("--renderer", |p, renderer| {
+            p.structural_renderer_name = renderer.into();
+        });
+    }
+}
+
+/// Sets up the structural renderer for the program:
+///
+/// - Adds global flags to specify the renderer type:
+///   * `--json` for JSON output
+///   * `--json-pretty` for pretty-printed JSON output
+///   * `--yaml` for YAML output
+///   * `--toml` for TOML output
+///   * `--ron` for RON output
+///   * `--ron-pretty` for pretty-printed RON output
+pub struct StructuralRendererSetup;
+
+impl<C> ProgramSetup<C> for StructuralRendererSetup
+where
+    C: ProgramCollect<Enum = C>,
+{
+    fn setup(self, program: &mut Program<C>) {
+        let args = program.take_args();
+        let args = process_renderer_flags(args, program);
+        program.replace_args(args);
+    }
+}
+
+fn process_renderer_flags<C>(args: Vec<String>, program: &mut Program<C>) -> Vec<String>
+where
+    C: ProgramCollect<Enum = C>,
+{
+    let remains_arg = PickerArg::<PickerArgs>::new(&[], None, true);
+    let (json, json_pretty, yaml, toml, ron, ron_pretty, remains) = args
+        .pick(&JSON_FLAG)
+        .pick(&JSON_PRETTY_FLAG)
+        .pick(&YAML_FLAG)
+        .pick(&TOML_FLAG)
+        .pick(&RON_FLAG)
+        .pick(&RON_PRETTY_FLAG)
+        .pick(&remains_arg)
+        .unwrap();
+
+    #[cfg(feature = "json_serde_fmt")]
+    if *json {
+        program.structural_renderer_name = crate::StructuralRendererSetting::Json;
+    }
+    #[cfg(feature = "json_serde_fmt")]
+    if *json_pretty {
+        program.structural_renderer_name = crate::StructuralRendererSetting::JsonPretty;
+    }
+    #[cfg(feature = "yaml_serde_fmt")]
+    if *yaml {
+        program.structural_renderer_name = crate::StructuralRendererSetting::Yaml;
+    }
+    #[cfg(feature = "toml_serde_fmt")]
+    if *toml {
+        program.structural_renderer_name = crate::StructuralRendererSetting::Toml;
+    }
+    #[cfg(feature = "ron_serde_fmt")]
+    if *ron {
+        program.structural_renderer_name = crate::StructuralRendererSetting::Ron;
+    }
+    #[cfg(feature = "ron_serde_fmt")]
+    if *ron_pretty {
+        program.structural_renderer_name = crate::StructuralRendererSetting::RonPretty;
+    }
+
+    remains.into()
+}