1 files changed, 76 insertions, 0 deletions
diff --git a/crates/shirabe-symfony-console/src/completion/completion_suggestions.rs b/crates/shirabe-symfony-console/src/completion/completion_suggestions.rs
new file mode 100644
index 00000000..dd9235ae
--- /dev/null
+++ b/crates/shirabe-symfony-console/src/completion/completion_suggestions.rs
@@ -0,0 +1,76 @@
+//! ref: composer/vendor/symfony/console/Completion/CompletionSuggestions.php
+
+use crate::completion::suggestion::Suggestion;
+use crate::input::input_option::InputOption;
+
+/// PHP union type `string|Suggestion` used by `suggestValue`/`suggestValues`.
+#[derive(Debug)]
+pub enum StringOrSuggestion {
+    String(String),
+    Suggestion(Suggestion),
+}
+
+/// Stores all completion suggestions for the current input.
+#[derive(Debug)]
+pub struct CompletionSuggestions {
+    value_suggestions: Vec<Suggestion>,
+    option_suggestions: Vec<std::rc::Rc<InputOption>>,
+}
+
+impl Default for CompletionSuggestions {
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl CompletionSuggestions {
+    pub fn new() -> Self {
+        Self {
+            value_suggestions: vec![],
+            option_suggestions: vec![],
+        }
+    }
+
+    /// Add a suggested value for an input option or argument.
+    pub fn suggest_value(&mut self, value: StringOrSuggestion) -> &mut Self {
+        self.value_suggestions.push(match value {
+            StringOrSuggestion::Suggestion(value) => value,
+            StringOrSuggestion::String(value) => Suggestion::new(value),
+        });
+
+        self
+    }
+
+    /// Add multiple suggested values at once for an input option or argument.
+    pub fn suggest_values(&mut self, values: Vec<StringOrSuggestion>) -> &mut Self {
+        for value in values {
+            self.suggest_value(value);
+        }
+
+        self
+    }
+
+    /// Add a suggestion for an input option name.
+    pub fn suggest_option(&mut self, option: std::rc::Rc<InputOption>) -> &mut Self {
+        self.option_suggestions.push(option);
+
+        self
+    }
+
+    /// Add multiple suggestions for input option names at once.
+    pub fn suggest_options(&mut self, options: Vec<std::rc::Rc<InputOption>>) -> &mut Self {
+        for option in options {
+            self.suggest_option(option);
+        }
+
+        self
+    }
+
+    pub fn get_option_suggestions(&self) -> &Vec<std::rc::Rc<InputOption>> {
+        &self.option_suggestions
+    }
+
+    pub fn get_value_suggestions(&self) -> &Vec<Suggestion> {
+        &self.value_suggestions
+    }
+}