use super::error::CodingError;
use super::{
AstGrepInput, AstGrepOutput, BackgroundProcessHandle, BashInput, BashResult, EditFileArgs, EditFileResponse,
FindInput, FindOutput, GrepInput, GrepOutput, ListFilesArgs, ListFilesResult, ReadBackgroundBashOutput,
ReadFileArgs, ReadFileResult, WriteFileArgs, WriteFileResponse, edit_file_contents, execute_command_in_dir,
find_files, list_files, perform_ast_grep, perform_grep, read_background_bash, read_file_contents,
tools_trait::CodingTools, write_file_contents,
};
use std::path::PathBuf;
#[derive(Debug, Default)]
pub struct DefaultCodingTools;
impl DefaultCodingTools {
pub fn new() -> Self {
Self
}
}
impl CodingTools for DefaultCodingTools {
async fn read_file(&self, args: ReadFileArgs) -> Result<ReadFileResult, CodingError> {
read_file_contents(args).await.map_err(CodingError::from)
}
async fn write_file(&self, args: WriteFileArgs) -> Result<WriteFileResponse, CodingError> {
write_file_contents(args).await.map_err(CodingError::from)
}
async fn edit_file(&self, args: EditFileArgs) -> Result<EditFileResponse, CodingError> {
edit_file_contents(args).await.map_err(CodingError::from)
}
async fn list_files(&self, args: ListFilesArgs) -> Result<ListFilesResult, CodingError> {
list_files(args).await.map_err(CodingError::from)
}
async fn bash(&self, args: BashInput, cwd: Option<PathBuf>) -> Result<BashResult, CodingError> {
execute_command_in_dir(args, cwd.as_deref()).await.map_err(CodingError::from)
}
async fn read_background_bash(
&self,
handle: BackgroundProcessHandle,
filter: Option<String>,
) -> Result<(ReadBackgroundBashOutput, Option<BackgroundProcessHandle>), CodingError> {
read_background_bash(handle, filter).await.map_err(CodingError::from)
}
async fn grep(&self, args: GrepInput) -> Result<GrepOutput, CodingError> {
perform_grep(args).await.map_err(CodingError::from)
}
async fn ast_grep(&self, args: AstGrepInput) -> Result<AstGrepOutput, CodingError> {
perform_ast_grep(args).await.map_err(CodingError::from)
}
async fn find(&self, args: FindInput) -> Result<FindOutput, CodingError> {
find_files(args).await.map_err(CodingError::from)
}
}