add some more stuff from brigadier

2 files changed, 109 insertions, 0 deletions

diff --git a/azalea-brigadier/src/builder/argument_builder.rs b/azalea-brigadier/src/builder/argument_builder.rs
index e69de29b..8a64a9e4 100644
--- a/azalea-brigadier/src/builder/argument_builder.rs
+++ b/azalea-brigadier/src/builder/argument_builder.rs
@@ -0,0 +1,106 @@
+use crate::{
+    command::Command,
+    redirect_modifier::RedirectModifier,
+    single_redirect_modifier::SingleRedirectModifier,
+    tree::{command_node::CommandNode, root_command_node::RootCommandNode},
+};
+
+pub struct BaseArgumentBuilder<S, T>
+where
+    T: ArgumentBuilder<S, T>,
+{
+    arguments: RootCommandNode<S>,
+    command: dyn Command<S>,
+    requirement: dyn Fn(&S) -> bool,
+    target: Option<dyn CommandNode<S>>,
+    modifier: Option<dyn RedirectModifier<S>>,
+    forks: bool,
+}
+
+pub trait ArgumentBuilder<S, T> {
+    fn this() -> T;
+    fn build(self) -> dyn CommandNode<S>;
+}
+
+impl<S, T> BaseArgumentBuilder<S, T>
+where
+    T: ArgumentBuilder<S, T>,
+{
+    pub fn then(&mut self, command: dyn CommandNode<S>) -> Result<&mut T, String> {
+        if self.target.is_some() {
+            return Err("Cannot add children to a redirected node".to_string());
+        }
+        self.command = command;
+        Ok(self)
+    }
+
+    pub fn arguments(&self) -> &Vec<dyn CommandNode<S>> {
+        &self.arguments.get_children()
+    }
+
+    pub fn executes(&mut self, command: dyn Command<S>) -> &mut T {
+        self.command = command;
+        self
+    }
+
+    pub fn command(&self) -> dyn Command<S> {
+        self.command
+    }
+
+    pub fn requires(&mut self, requirement: dyn Fn(&S) -> bool) -> &mut T {
+        self.requirement = requirement;
+        self
+    }
+
+    pub fn requirement(&self) -> dyn Fn(&S) -> bool {
+        self.requirement
+    }
+
+    pub fn redirect(&mut self, target: dyn CommandNode<S>) -> &mut T {
+        self.forward(target, None, false)
+    }
+
+    pub fn redirect_modifier(
+        &mut self,
+        target: dyn CommandNode<S>,
+        modifier: dyn SingleRedirectModifier<S>,
+    ) -> &mut T {
+        // forward(target, modifier == null ? null : o -> Collections.singleton(modifier.apply(o)), false);
+        self.forward(target, modifier.map(|m| |o| vec![m.apply(o)]), false)
+    }
+
+    pub fn fork(
+        &mut self,
+        target: dyn CommandNode<S>,
+        modifier: dyn RedirectModifier<S>,
+    ) -> &mut T {
+        self.forward(target, Some(modifier), true)
+    }
+
+    pub fn forward(
+        &mut self,
+        target: dyn CommandNode<S>,
+        modifier: Option<dyn RedirectModifier<S>>,
+        fork: bool,
+    ) -> Result<&mut T, String> {
+        if !self.arguments.get_children().is_empty() {
+            return Err("Cannot forward a node with children".to_string());
+        }
+        self.target = Some(target);
+        self.modifier = modifier;
+        self.forks = fork;
+        Ok(self)
+    }
+
+    pub fn redirect(&self) -> Option<&dyn CommandNode<S>> {
+        self.target.as_ref()
+    }
+
+    pub fn redirect_modifier(&self) -> Option<&dyn RedirectModifier<S>> {
+        self.modifier.as_ref()
+    }
+
+    pub fn is_fork(&self) -> bool {
+        self.forks
+    }
+}
diff --git a/azalea-brigadier/src/builder/mod.rs b/azalea-brigadier/src/builder/mod.rs
index e69de29b..26f2f644 100644
--- a/azalea-brigadier/src/builder/mod.rs
+++ b/azalea-brigadier/src/builder/mod.rs
@@ -0,0 +1,3 @@
+pub mod argument_builder;
+pub mod literal_argument_builder;
+pub mod required_argument_builder;