litecode 0.1.1

An ultra-lightweight Coding MCP server built with Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use std::borrow::Cow;

use rmcp::{
    ErrorData,
    handler::server::router::tool::{AsyncTool, ToolBase},
};

use crate::{
    error::LiteCodeError,
    schema::{TaskStopInput, TaskStopOutput},
    server::LiteCodeServer,
};

pub struct TaskStopTool;

impl ToolBase for TaskStopTool {
    type Parameter = TaskStopInput;
    type Output = TaskStopOutput;
    type Error = ErrorData;

    fn name() -> Cow<'static, str> {
        "TaskStop".into()
    }

    fn description() -> Option<Cow<'static, str>> {
        Some(
            r#"
- Stops a running background shell task by its ID
- Takes a task_id parameter identifying the task to stop
- Returns a success or failure status
- Use this tool when you need to terminate a long-running shell task
"#
            .into(),
        )
    }
}

impl AsyncTool<LiteCodeServer> for TaskStopTool {
    async fn invoke(
        service: &LiteCodeServer,
        input: Self::Parameter,
    ) -> Result<Self::Output, Self::Error> {
        let task_id = input.task_id.or(input.shell_id).ok_or_else(|| {
            LiteCodeError::invalid_input("TaskStop requires task_id or shell_id.")
        })?;

        service
            .task_manager()
            .stop_task(&task_id)
            .await
            .map_err(Into::into)
    }
}