yolop 0.4.0

Yolop — a terminal coding agent built on everruns-runtime
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90

//! `/worktree` — session git worktree status and auto-activation control.

use crate::worktree::WorktreeManager;
use async_trait::async_trait;
use everruns_core::capabilities::{Capability, CapabilityStatus};
use everruns_core::command::{
    CommandArg, CommandDescriptor, CommandExecutionContext, CommandResult, CommandSource,
    ExecuteCommandRequest,
};
use std::sync::Arc;

pub(crate) const WORKTREE_CAPABILITY_ID: &str = "yolop_worktree";
const WORKTREE_COMMAND_NAME: &str = "worktree";

pub(crate) struct WorktreeCapability {
    pub(crate) manager: Arc<WorktreeManager>,
}

#[async_trait]
impl Capability for WorktreeCapability {
    fn id(&self) -> &str {
        WORKTREE_CAPABILITY_ID
    }

    fn name(&self) -> &str {
        "Worktree"
    }

    fn description(&self) -> &str {
        "Show session git worktree status or disable auto-activation."
    }

    fn status(&self) -> CapabilityStatus {
        CapabilityStatus::Available
    }

    fn category(&self) -> Option<&str> {
        Some("System")
    }

    fn commands(&self) -> Vec<CommandDescriptor> {
        vec![CommandDescriptor {
            name: WORKTREE_COMMAND_NAME.to_string(),
            description: "Show worktree status, or pass `off` to disable auto-activation."
                .to_string(),
            source: CommandSource::System,
            args: vec![CommandArg {
                name: "action".to_string(),
                description: "`off` disables auto worktree creation for future turns.".to_string(),
                required: false,
                suggestions: vec!["off".to_string()],
            }],
        }]
    }

    async fn execute_command(
        &self,
        request: &ExecuteCommandRequest,
        _ctx: &CommandExecutionContext,
    ) -> everruns_core::Result<CommandResult> {
        if request.name != WORKTREE_COMMAND_NAME {
            return Err(everruns_core::AgentLoopError::config(format!(
                "{} cannot execute /{}",
                self.id(),
                request.name
            )));
        }

        let arg = request.arguments.as_deref().unwrap_or("").trim();
        let message = if arg.eq_ignore_ascii_case("off") {
            self.manager.disable_auto();
            "worktrees: auto activation disabled for this session (already-active worktrees \
             are unchanged)"
                .to_string()
        } else if arg.is_empty() {
            self.manager.status_message()
        } else {
            return Err(everruns_core::AgentLoopError::config(format!(
                "unknown /worktree argument `{arg}`; use `/worktree` or `/worktree off`"
            )));
        };

        Ok(CommandResult {
            success: true,
            message,
            error_code: None,
            error_fields: None,
        })
    }
}