a3s 0.8.2

use std::sync::Arc;

use a3s_boot::{controller, BootResponse, Result as BootResult};

use super::service::WorkspaceService;

pub(super) struct WorkspaceController {
    service: Arc<WorkspaceService>,
}

impl WorkspaceController {
    pub(super) fn new(service: Arc<WorkspaceService>) -> Self {
        Self { service }
    }
}

#[controller("/v1/workspace")]
impl WorkspaceController {
    #[get("/default-root")]
    async fn workspace_default_root(&self) -> BootResult<serde_json::Value> {
        Ok(self.service.default_root())
    }

    #[get("/readiness")]
    async fn inspect_workspace_readiness(
        &self,
        #[query("workspaceRoot")] workspace_root: Option<String>,
    ) -> BootResult<serde_json::Value> {
        self.service.inspect_readiness(workspace_root, false).await
    }

    #[post("/readiness")]
    async fn ensure_workspace_readiness(
        &self,
        #[query("workspaceRoot")] workspace_root: Option<String>,
    ) -> BootResult<serde_json::Value> {
        self.service.inspect_readiness(workspace_root, true).await
    }

    #[post("/init-agent")]
    async fn init_workspace_agent(
        &self,
        #[body] request: serde_json::Value,
    ) -> BootResult<serde_json::Value> {
        self.service.init_agent(request).await
    }

    #[post("/actions/init-prompt")]
    async fn workspace_init_prompt(
        &self,
        #[body] request: serde_json::Value,
    ) -> BootResult<serde_json::Value> {
        self.service.init_prompt(request).await
    }

    #[post("/mkdir")]
    async fn create_workspace_dir(
        &self,
        #[body] request: serde_json::Value,
    ) -> BootResult<serde_json::Value> {
        self.service.create_dir(request).await
    }

    #[post("/write")]
    async fn write_workspace_file(
        &self,
        #[body] request: serde_json::Value,
    ) -> BootResult<serde_json::Value> {
        self.service.write_file(request).await
    }

    #[post("/write-binary")]
    async fn write_workspace_binary_file(
        &self,
        #[body] request: serde_json::Value,
    ) -> BootResult<serde_json::Value> {
        self.service.write_binary_file(request).await
    }

    #[get("/read")]
    async fn read_workspace_file(
        &self,
        #[query("path")] path: String,
    ) -> BootResult<serde_json::Value> {
        self.service.read_file(path).await
    }

    #[get("/read-binary", raw)]
    async fn read_workspace_binary_file(
        &self,
        #[query("path")] path: String,
    ) -> BootResult<BootResponse> {
        let body = self.service.read_binary_file(path).await?;
        Ok(BootResponse::new(200, body).with_content_type("application/octet-stream"))
    }

    #[get("/exists")]
    async fn workspace_path_exists(
        &self,
        #[query("path")] path: String,
    ) -> BootResult<serde_json::Value> {
        self.service.path_exists(path).await
    }

    #[delete("/delete")]
    async fn delete_workspace_path(
        &self,
        #[query("path")] path: String,
    ) -> BootResult<serde_json::Value> {
        self.service.delete_path(path).await
    }

    #[get("/read-dir")]
    async fn read_workspace_dir(
        &self,
        #[query("path")] path: String,
    ) -> BootResult<Vec<serde_json::Value>> {
        self.service.read_dir(path).await
    }

    #[post("/rename")]
    async fn rename_workspace_path(
        &self,
        #[body] request: serde_json::Value,
    ) -> BootResult<serde_json::Value> {
        self.service.rename_path(request).await
    }

    #[post("/copy")]
    async fn copy_workspace_path(
        &self,
        #[body] request: serde_json::Value,
    ) -> BootResult<serde_json::Value> {
        self.service.copy_path(request).await
    }

    #[get("/git-status")]
    async fn workspace_git_status(
        &self,
        #[query("rootPath")] root_path: Option<String>,
    ) -> BootResult<serde_json::Value> {
        Ok(self.service.git_status(root_path))
    }

    #[allow(clippy::too_many_arguments)]
    #[get("/search")]
    async fn search_workspace_files(
        &self,
        #[query("rootPath")] root_path: String,
        #[query("query")] query: String,
        #[query("caseSensitive")] case_sensitive: Option<bool>,
        #[query("useRegex")] use_regex: Option<bool>,
        #[query("matchWholeWord")] match_whole_word: Option<bool>,
        #[query("includePattern")] include_pattern: Option<String>,
        #[query("excludePattern")] exclude_pattern: Option<String>,
        #[query("maxResults")] max_results: Option<usize>,
    ) -> BootResult<Vec<serde_json::Value>> {
        self.service
            .search_files(
                root_path,
                query,
                super::service::WorkspaceSearchOptions {
                    case_sensitive: case_sensitive.unwrap_or(false),
                    use_regex: use_regex.unwrap_or(false),
                    match_whole_word: match_whole_word.unwrap_or(false),
                    include_pattern,
                    exclude_pattern,
                    max_results: max_results.unwrap_or(1000),
                },
            )
            .await
    }

    #[post("/replace")]
    async fn replace_workspace_files(
        &self,
        #[body] request: serde_json::Value,
    ) -> BootResult<serde_json::Value> {
        self.service.replace_in_files(request).await
    }
}