dave
845b85e7a7
cargo fmt without --all fails with "Failed to find targets" in workspace repos. This was blocking every story's gates. Also ran cargo fmt --all to fix all existing formatting issues. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
81 lines
2.6 KiB
Rust
81 lines
2.6 KiB
Rust
//! Agent worktree management — creates and configures git worktrees for agents.
|
|
use crate::config::ProjectConfig;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
use super::AgentPool;
|
|
|
|
impl AgentPool {
|
|
/// Create a worktree for the given story using the server port (writes .mcp.json).
|
|
pub async fn create_worktree(
|
|
&self,
|
|
project_root: &Path,
|
|
story_id: &str,
|
|
) -> Result<crate::worktree::WorktreeInfo, String> {
|
|
let config = ProjectConfig::load(project_root)?;
|
|
crate::worktree::create_worktree(project_root, story_id, &config, self.port).await
|
|
}
|
|
|
|
/// Get project root helper.
|
|
pub fn get_project_root(&self, state: &crate::state::SessionState) -> Result<PathBuf, String> {
|
|
state.get_project_root()
|
|
}
|
|
}
|
|
|
|
/// Return the active pipeline stage directory name for `story_id`, or `None` if the
|
|
/// story is not in any active stage (`2_current/`, `3_qa/`, `4_merge/`).
|
|
pub(super) fn find_active_story_stage(
|
|
_project_root: &Path,
|
|
story_id: &str,
|
|
) -> Option<&'static str> {
|
|
if let Ok(Some(item)) = crate::pipeline_state::read_typed(story_id)
|
|
&& item.stage.is_active()
|
|
{
|
|
return Some(item.stage.dir_name());
|
|
}
|
|
None
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::find_active_story_stage;
|
|
|
|
#[test]
|
|
fn find_active_story_stage_detects_current() {
|
|
crate::db::ensure_content_store();
|
|
crate::db::write_item_with_content("10_story_test", "2_current", "---\nname: Test\n---\n");
|
|
let tmp = tempfile::tempdir().unwrap();
|
|
assert_eq!(
|
|
find_active_story_stage(tmp.path(), "10_story_test"),
|
|
Some("2_current")
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn find_active_story_stage_detects_qa() {
|
|
crate::db::ensure_content_store();
|
|
crate::db::write_item_with_content("11_story_test", "3_qa", "---\nname: Test\n---\n");
|
|
let tmp = tempfile::tempdir().unwrap();
|
|
assert_eq!(
|
|
find_active_story_stage(tmp.path(), "11_story_test"),
|
|
Some("3_qa")
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn find_active_story_stage_detects_merge() {
|
|
crate::db::ensure_content_store();
|
|
crate::db::write_item_with_content("12_story_test", "4_merge", "---\nname: Test\n---\n");
|
|
let tmp = tempfile::tempdir().unwrap();
|
|
assert_eq!(
|
|
find_active_story_stage(tmp.path(), "12_story_test"),
|
|
Some("4_merge")
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn find_active_story_stage_returns_none_for_unknown_story() {
|
|
let tmp = tempfile::tempdir().unwrap();
|
|
assert_eq!(find_active_story_stage(tmp.path(), "99_nonexistent"), None);
|
|
}
|
|
}
|