story-kit: merge 267_story_mcp_update_story_tool_should_support_front_matter_fields

server/src/http/mcp.rs

@@ -19,6 +19,7 @@ use poem::web::Data;
 use poem::{Body, Request, Response};
 use serde::{Deserialize, Serialize};
 use serde_json::{json, Value};
+use std::collections::HashMap;
 use std::fs;
 use std::sync::Arc;
 
@@ -638,7 +639,7 @@ fn handle_tools_list(id: Option<Value>) -> JsonRpcResponse {
                 },
                 {
                     "name": "update_story",
-                    "description": "Update the user story text and/or description of an existing story file. Replaces the content of the '## User Story' and/or '## Description' section in place. Auto-commits via the filesystem watcher.",
+                    "description": "Update an existing story file. Can replace the '## User Story' and/or '## Description' section content, and/or set YAML front matter fields (e.g. agent, manual_qa). Auto-commits via the filesystem watcher.",
                     "inputSchema": {
                         "type": "object",
                         "properties": {
@@ -653,6 +654,17 @@ fn handle_tools_list(id: Option<Value>) -> JsonRpcResponse {
                             "description": {
                                 "type": "string",
                                 "description": "New description text to replace the '## Description' section content"
+                            },
+                            "agent": {
+                                "type": "string",
+                                "description": "Set or change the 'agent' YAML front matter field"
+                            },
+                            "front_matter": {
+                                "type": "object",
+                                "description": "Arbitrary YAML front matter key-value pairs to set or update",
+                                "additionalProperties": {
+                                    "type": "string"
+                                }
                             }
                         },
                         "required": ["story_id"]
@@ -1544,8 +1556,24 @@ fn tool_update_story(args: &Value, ctx: &AppContext) -> Result<String, String> {
     let user_story = args.get("user_story").and_then(|v| v.as_str());
     let description = args.get("description").and_then(|v| v.as_str());
 
+    // Collect front matter fields: explicit `agent` param + arbitrary `front_matter` object.
+    let mut front_matter: HashMap<String, String> = HashMap::new();
+    if let Some(agent) = args.get("agent").and_then(|v| v.as_str()) {
+        front_matter.insert("agent".to_string(), agent.to_string());
+    }
+    if let Some(obj) = args.get("front_matter").and_then(|v| v.as_object()) {
+        for (k, v) in obj {
+            let val = match v {
+                Value::String(s) => s.clone(),
+                other => other.to_string(),
+            };
+            front_matter.insert(k.clone(), val);
+        }
+    }
+    let front_matter_opt = if front_matter.is_empty() { None } else { Some(&front_matter) };
+
     let root = ctx.state.get_project_root()?;
-    update_story_in_file(&root, story_id, user_story, description)?;
+    update_story_in_file(&root, story_id, user_story, description, front_matter_opt)?;
 
     Ok(format!("Updated story '{story_id}'."))
 }