feat: add user input handling with command parsing and validation (#143)

Author: tusharmath

Cargo.lock

@@ -19,6 +19,7 @@ serde = "1.0.217"
 serde_json = "1.0.134"
 strum = "0.26.3"
 strum_macros = "0.26.4"
+thiserror = "2.0.11"
 tokio = { version = "1.42.0", features = ["full", "test-util"] }
 tokio-stream = "0.1.17"
 uuid = { version = "1.11.0", features = [

crates/forge_domain/Cargo.toml

@@ -1,12 +1,21 @@
 use std::pin::Pin;
 
-use derive_more::derive::{Display, Error, From};
+use derive_more::derive::From;
+use thiserror::Error;
 
-#[derive(From, Debug, Display, Error)]
+#[derive(From, Debug, Error)]
 pub enum Error {
+    #[error("Tool name was not provided")]
     ToolCallMissingName,
+
+    #[error("Serde Error: {0}")]
     Serde(serde_json::Error),
-    Uuid(uuid::Error),
+
+    #[error("Invalid UUID: {0}")]
+    InvalidUuid(uuid::Error),
+
+    #[error("Invalid user command: {0}")]
+    InvalidUserCommand(String),
 }
 
 pub type Result<A> = std::result::Result<A, Error>;

crates/forge_domain/src/error.rs

@@ -21,6 +21,7 @@ mod tool_name;
 mod tool_result;
 mod tool_service;
 mod tool_usage;
+mod user;
 mod user_interaction;
 
 pub use chat_request::*;
@@ -46,4 +47,5 @@ pub use tool_name::*;
 pub use tool_result::*;
 pub use tool_service::*;
 pub use tool_usage::*;
+pub use user::*;
 pub use user_interaction::*;

crates/forge_domain/src/lib.rs

@@ -0,0 +1,97 @@
+use std::path::PathBuf;
+
+use async_trait::async_trait;
+
+use crate::error::{Error, Result};
+
+/// Represents user input types in the chat application.
+///
+/// This enum encapsulates all forms of input including:
+/// - System commands (starting with '/')
+/// - Regular chat messages
+/// - File content
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub enum Input {
+    /// End the current session and exit the application.
+    /// This can be triggered with the '/end' command.
+    End,
+    /// Start a new conversation while preserving history.
+    /// This can be triggered with the '/new' command.
+    New,
+    /// Reload the conversation with the original prompt.
+    /// This can be triggered with the '/reload' command.
+    Reload,
+    /// A regular text message from the user to be processed by the chat system.
+    /// Any input that doesn't start with '/' is treated as a message.
+    Message(String),
+}
+
+impl Input {
+    /// Returns a list of all available command strings.
+    ///
+    /// These commands are used for:
+    /// - Command validation
+    /// - Autocompletion
+    /// - Help display
+    pub fn available_commands() -> Vec<String> {
+        vec![
+            "/end".to_string(),
+            "/new".to_string(),
+            "/reload".to_string(),
+        ]
+    }
+
+    /// Parses a string input into an Input.
+    ///
+    /// This function:
+    /// - Trims whitespace from the input
+    /// - Recognizes and validates commands (starting with '/')
+    /// - Converts regular text into messages
+    ///
+    /// # Returns
+    /// - `Ok(Input)` - Successfully parsed input
+    /// - `Err` - Input was an invalid command
+    pub fn parse(input: &str) -> Result<Self> {
+        let trimmed = input.trim();
+        match trimmed {
+            "/end" => Ok(Input::End),
+            "/new" => Ok(Input::New),
+            "/reload" => Ok(Input::Reload),
+            cmd if cmd.starts_with('/') => Err(Error::InvalidUserCommand(cmd.to_string())),
+            text => Ok(Input::Message(text.to_string())),
+        }
+    }
+}
+
+/// A trait for handling user input in the application.
+///
+/// This trait defines the core functionality needed for processing
+/// user input, whether it comes from a command line interface,
+/// GUI, or file system.
+#[async_trait]
+pub trait UserInput {
+    /// Read content from a file and convert it to the input type.
+    ///
+    /// # Arguments
+    /// * `path` - The path to the file to read
+    ///
+    /// # Returns
+    /// * `Ok(Input)` - Successfully read and parsed file content
+    /// * `Err` - Failed to read or parse file
+    async fn upload<P: Into<PathBuf> + Send>(&self, path: P) -> anyhow::Result<Input>;
+
+    /// Prompts for user input with optional help text and initial value.
+    ///
+    /// # Arguments
+    /// * `help_text` - Optional help text to display with the prompt
+    /// * `initial_text` - Optional initial text to populate the input with
+    ///
+    /// # Returns
+    /// * `Ok(Input)` - Successfully processed input
+    /// * `Err` - An error occurred during input processing
+    async fn prompt(
+        &self,
+        help_text: Option<&str>,
+        initial_text: Option<&str>,
+    ) -> anyhow::Result<Input>;
+}

crates/forge_domain/src/user.rs

@@ -11,6 +11,7 @@ forge_domain = { path = "../forge_domain" }
 tokio = { version = "1.0", features = ["full"] }
 tokio-stream = "0.1.17"
 colored = "3.0.0"
+async-trait = "0.1"
 inquire = "0.7.5"
 regex = "1.10.2"
 strum = "0.26.3"