Skip to main content

devboy_executor/
executor.rs

1use devboy_core::types::ChatType;
2use devboy_core::{
3    AddStructureRowsInput, AssignToSprintInput, CreateCommentInput, CreateIssueInput,
4    CreateMergeRequestInput, CreatePageParams, CreateStructureInput, Error, GetChatsParams,
5    GetForestOptions, GetMessagesParams, GetPipelineInput, GetStructureValuesInput,
6    GetUsersOptions, IssueFilter, IssueProvider, JobLogMode, JobLogOptions, KnowledgeBaseProvider,
7    ListCustomFieldsParams, ListPagesParams, ListProjectVersionsParams, MeetingFilter,
8    MeetingNotesProvider, MergeRequestProvider, MessengerProvider, MoveStructureRowsInput,
9    MrFilter, PipelineProvider, Result, SaveStructureViewInput, SearchKbParams,
10    SearchMessagesParams, SendMessageParams, SprintState, StructureRowItem, StructureViewColumn,
11    ToolCategory, UpdateIssueInput, UpdatePageParams, UpsertProjectVersionInput,
12};
13use serde::Deserialize;
14use serde_json::Value;
15use tracing::debug;
16
17use crate::context::AdditionalContext;
18use crate::factory;
19use crate::output::{ResultMeta, ToolOutput};
20use devboy_core::ToolEnricher;
21
22/// Maximum file size for upload / download asset operations (10 MB).
23const MAX_FILE_SIZE: usize = 10 * 1024 * 1024;
24
25/// Parse `tools/call` args into a typed `Params` struct, turning
26/// deserialisation failures into `Error::InvalidData` so callers see
27/// the exact field that went wrong instead of the tool silently
28/// running with defaults (the previous `unwrap_or_default()` path).
29///
30/// `Value::Null` is accepted as "no arguments" and yields `T::default()`
31/// — the MCP spec allows a null `arguments` field for tools whose
32/// params are all optional, so rejecting null here would break those
33/// call sites. Actual content is validated by `serde_json::from_value`.
34fn parse_tool_params<T>(args: &Value, tool: &str) -> Result<T>
35where
36    T: Default + serde::de::DeserializeOwned,
37{
38    if args.is_null() {
39        return Ok(T::default());
40    }
41    serde_json::from_value(args.clone())
42        .map_err(|e| Error::InvalidData(format!("invalid {tool} params: {e}")))
43}
44
45/// Deserialize a value that can be either a string or a number into Option<String>.
46/// Enricher may transform priority "high" → 2 (number), but executor needs String.
47fn deserialize_string_or_number<'de, D>(
48    deserializer: D,
49) -> std::result::Result<Option<String>, D::Error>
50where
51    D: serde::Deserializer<'de>,
52{
53    let value: Option<Value> = Option::deserialize(deserializer)?;
54    Ok(value.map(|v| match v {
55        Value::String(s) => s,
56        Value::Number(n) => n.to_string(),
57        other => other.to_string(),
58    }))
59}
60
61/// Tool execution engine.
62///
63/// Manages enrichers and dispatches tool calls to providers.
64/// Stateless per call — provider is created from `AdditionalContext` each time.
65pub struct Executor {
66    enrichers: Vec<Box<dyn ToolEnricher>>,
67    asset_manager: Option<devboy_assets::AssetManager>,
68}
69
70impl Executor {
71    pub fn new() -> Self {
72        Self {
73            enrichers: Vec::new(),
74            asset_manager: None,
75        }
76    }
77
78    /// Configure an optional local asset cache for download/delete operations.
79    pub fn with_asset_manager(mut self, mgr: devboy_assets::AssetManager) -> Self {
80        self.asset_manager = Some(mgr);
81        self
82    }
83
84    /// Register an enricher (provider, pipeline, or custom).
85    /// Enrichers are applied in registration order.
86    pub fn add_enricher(&mut self, enricher: Box<dyn ToolEnricher>) {
87        self.enrichers.push(enricher);
88    }
89
90    /// List available tools with enriched schemas.
91    ///
92    /// 1. Starts with base tool definitions
93    /// 2. Keeps only tools whose category is supported by at least one enricher
94    /// 3. Applies schema enrichment from enrichers that support each tool's category
95    pub fn list_tools(&self) -> Vec<crate::tools::ToolDefinition> {
96        let mut tools = crate::tools::base_tool_definitions();
97
98        // Collect all supported categories from enrichers
99        let supported_categories: std::collections::HashSet<devboy_core::ToolCategory> = self
100            .enrichers
101            .iter()
102            .flat_map(|e| e.supported_categories().iter().copied())
103            .collect();
104
105        // Keep only tools whose category is supported
106        tools.retain(|t| supported_categories.contains(&t.category));
107
108        // Apply schema enrichment from enrichers that support each tool's category
109        for enricher in &self.enrichers {
110            let cats = enricher.supported_categories();
111            for tool in &mut tools {
112                if cats.contains(&tool.category) {
113                    enricher.enrich_schema(&tool.name, &mut tool.input_schema);
114                }
115            }
116        }
117
118        tools
119    }
120
121    /// Execute a tool with the given arguments and context.
122    ///
123    /// Flow:
124    /// 1. Pre-execute: enrichers transform args
125    /// 2. Create provider from context (cheap, stack-allocated)
126    /// 3. Dispatch tool call to provider method
127    /// 4. Post-execute: enrichers transform output
128    /// 5. Return typed ToolOutput
129    pub async fn execute(
130        &self,
131        tool: &str,
132        args: Value,
133        ctx: &AdditionalContext,
134    ) -> Result<ToolOutput> {
135        let mut args = args;
136
137        // Pre-execute: enrichers transform args
138        // Look up tool category from base definitions for matching
139        let tool_category = crate::tools::base_tool_definitions()
140            .iter()
141            .find(|t| t.name == tool)
142            .map(|t| t.category);
143        for enricher in &self.enrichers {
144            if let Some(cat) = tool_category
145                && enricher.supported_categories().contains(&cat)
146            {
147                enricher.transform_args(tool, &mut args);
148            }
149        }
150
151        debug!(
152            tool = tool,
153            provider = ctx.provider.provider_name(),
154            "executing tool"
155        );
156
157        // Dispatch based on tool category
158        let output = if tool_category == Some(ToolCategory::MeetingNotes) {
159            let provider = factory::create_meeting_notes_provider(&ctx.provider)?;
160            dispatch_meeting_tool(tool, &args, provider.as_ref()).await?
161        } else if tool_category == Some(ToolCategory::KnowledgeBase) {
162            let provider =
163                factory::create_knowledge_base_provider(&ctx.provider, ctx.proxy.as_ref())?;
164            dispatch_knowledge_base_tool(tool, &args, provider.as_ref()).await?
165        } else if tool_category == Some(ToolCategory::Messenger) {
166            let provider = factory::create_messenger_provider(&ctx.provider)?;
167            dispatch_messenger_tool(tool, &args, provider.as_ref()).await?
168        } else {
169            let provider = factory::create_provider(&ctx.provider, ctx.proxy.as_ref())?;
170            dispatch_tool(tool, &args, provider.as_ref(), self.asset_manager.as_ref()).await?
171        };
172
173        Ok(output)
174    }
175
176    /// Execute a tool with a pre-created Provider (for MCP server).
177    /// Enrichers are applied if configured.
178    pub async fn execute_direct(
179        &self,
180        tool: &str,
181        args: Value,
182        provider: &dyn devboy_core::Provider,
183    ) -> Result<ToolOutput> {
184        let mut args = args;
185        // Apply enricher transforms (same as execute())
186        let tool_category = Self::tool_category(tool);
187        for enricher in &self.enrichers {
188            if let Some(cat) = tool_category
189                && enricher.supported_categories().contains(&cat)
190            {
191                enricher.transform_args(tool, &mut args);
192            }
193        }
194        dispatch_tool(tool, &args, provider, self.asset_manager.as_ref()).await
195    }
196
197    /// Execute a meeting tool with a pre-created MeetingNotesProvider.
198    pub async fn execute_direct_meeting(
199        &self,
200        tool: &str,
201        args: Value,
202        provider: &dyn MeetingNotesProvider,
203    ) -> Result<ToolOutput> {
204        let mut args = args;
205        let tool_category = Self::tool_category(tool);
206        for enricher in &self.enrichers {
207            if let Some(cat) = tool_category
208                && enricher.supported_categories().contains(&cat)
209            {
210                enricher.transform_args(tool, &mut args);
211            }
212        }
213        dispatch_meeting_tool(tool, &args, provider).await
214    }
215
216    /// Execute a knowledge base tool with a pre-created KnowledgeBaseProvider.
217    pub async fn execute_direct_knowledge_base(
218        &self,
219        tool: &str,
220        args: Value,
221        provider: &dyn KnowledgeBaseProvider,
222    ) -> Result<ToolOutput> {
223        let mut args = args;
224        let tool_category = Self::tool_category(tool);
225        for enricher in &self.enrichers {
226            if let Some(cat) = tool_category
227                && enricher.supported_categories().contains(&cat)
228            {
229                enricher.transform_args(tool, &mut args);
230            }
231        }
232        dispatch_knowledge_base_tool(tool, &args, provider).await
233    }
234
235    /// Execute a messenger tool with a pre-created MessengerProvider.
236    pub async fn execute_direct_messenger(
237        &self,
238        tool: &str,
239        args: Value,
240        provider: &dyn MessengerProvider,
241    ) -> Result<ToolOutput> {
242        let mut args = args;
243        let tool_category = Self::tool_category(tool);
244        for enricher in &self.enrichers {
245            if let Some(cat) = tool_category
246                && enricher.supported_categories().contains(&cat)
247            {
248                enricher.transform_args(tool, &mut args);
249            }
250        }
251        dispatch_messenger_tool(tool, &args, provider).await
252    }
253
254    /// Get the tool category for a tool name, if known.
255    pub fn tool_category(tool: &str) -> Option<ToolCategory> {
256        crate::tools::base_tool_definitions()
257            .iter()
258            .find(|t| t.name == tool)
259            .map(|t| t.category)
260    }
261}
262
263impl Default for Executor {
264    fn default() -> Self {
265        Self::new()
266    }
267}
268
269// --- Knowledge base tool dispatch ---
270
271/// Dispatch a knowledge base tool call.
272async fn dispatch_knowledge_base_tool(
273    tool: &str,
274    args: &Value,
275    provider: &dyn KnowledgeBaseProvider,
276) -> Result<ToolOutput> {
277    match tool {
278        "get_knowledge_base_spaces" => execute_get_knowledge_base_spaces(provider).await,
279        "list_knowledge_base_pages" => execute_list_knowledge_base_pages(provider, args).await,
280        "get_knowledge_base_page" => execute_get_knowledge_base_page(provider, args).await,
281        "create_knowledge_base_page" => execute_create_knowledge_base_page(provider, args).await,
282        "update_knowledge_base_page" => execute_update_knowledge_base_page(provider, args).await,
283        "search_knowledge_base" => execute_search_knowledge_base(provider, args).await,
284        _ => Err(Error::NotFound(format!(
285            "unknown knowledge base tool: {tool}"
286        ))),
287    }
288}
289
290// --- Knowledge base tool handlers ---
291
292async fn execute_get_knowledge_base_spaces(
293    provider: &dyn KnowledgeBaseProvider,
294) -> Result<ToolOutput> {
295    let result = provider.get_spaces().await?;
296    let meta = ResultMeta {
297        pagination: result.pagination,
298        sort_info: result.sort_info,
299    };
300    Ok(ToolOutput::KnowledgeBaseSpaces(result.items, Some(meta)))
301}
302
303#[derive(Deserialize)]
304#[serde(rename_all = "camelCase")]
305struct ListKnowledgeBasePagesParams {
306    space_key: String,
307    limit: Option<u32>,
308    offset: Option<u32>,
309    cursor: Option<String>,
310    search: Option<String>,
311    parent_id: Option<String>,
312}
313
314async fn execute_list_knowledge_base_pages(
315    provider: &dyn KnowledgeBaseProvider,
316    args: &Value,
317) -> Result<ToolOutput> {
318    let params: ListKnowledgeBasePagesParams =
319        serde_json::from_value(args.clone()).map_err(|e| {
320            Error::InvalidData(format!("invalid list_knowledge_base_pages params: {e}"))
321        })?;
322    let result = provider
323        .list_pages(ListPagesParams {
324            space_key: params.space_key,
325            limit: params.limit,
326            offset: params.offset,
327            cursor: params.cursor,
328            search: params.search,
329            parent_id: params.parent_id,
330        })
331        .await?;
332    let meta = ResultMeta {
333        pagination: result.pagination,
334        sort_info: result.sort_info,
335    };
336    Ok(ToolOutput::KnowledgeBasePages(result.items, Some(meta)))
337}
338
339#[derive(Deserialize)]
340#[serde(rename_all = "camelCase")]
341struct GetKnowledgeBasePageParams {
342    page_id: String,
343}
344
345async fn execute_get_knowledge_base_page(
346    provider: &dyn KnowledgeBaseProvider,
347    args: &Value,
348) -> Result<ToolOutput> {
349    let params: GetKnowledgeBasePageParams = serde_json::from_value(args.clone())
350        .map_err(|e| Error::InvalidData(format!("invalid get_knowledge_base_page params: {e}")))?;
351    let page = provider.get_page(&params.page_id).await?;
352    Ok(ToolOutput::KnowledgeBasePage(Box::new(page)))
353}
354
355#[derive(Deserialize)]
356#[serde(rename_all = "camelCase")]
357struct CreateKnowledgeBasePageParams {
358    space_key: String,
359    title: String,
360    content: String,
361    #[serde(default)]
362    content_type: Option<String>,
363    parent_id: Option<String>,
364    #[serde(default)]
365    labels: Vec<String>,
366}
367
368async fn execute_create_knowledge_base_page(
369    provider: &dyn KnowledgeBaseProvider,
370    args: &Value,
371) -> Result<ToolOutput> {
372    let params: CreateKnowledgeBasePageParams =
373        serde_json::from_value(args.clone()).map_err(|e| {
374            Error::InvalidData(format!("invalid create_knowledge_base_page params: {e}"))
375        })?;
376    let page = provider
377        .create_page(CreatePageParams {
378            space_key: params.space_key,
379            title: params.title,
380            content: params.content,
381            content_type: params.content_type,
382            parent_id: params.parent_id,
383            labels: params.labels,
384        })
385        .await?;
386    Ok(ToolOutput::KnowledgeBasePageSummary(Box::new(page)))
387}
388
389#[derive(Deserialize)]
390#[serde(rename_all = "camelCase")]
391struct UpdateKnowledgeBasePageParams {
392    page_id: String,
393    #[serde(default)]
394    title: Option<String>,
395    #[serde(default)]
396    content: Option<String>,
397    #[serde(default)]
398    content_type: Option<String>,
399    version: Option<u32>,
400    #[serde(default)]
401    labels: Option<Vec<String>>,
402    parent_id: Option<String>,
403}
404
405async fn execute_update_knowledge_base_page(
406    provider: &dyn KnowledgeBaseProvider,
407    args: &Value,
408) -> Result<ToolOutput> {
409    let params: UpdateKnowledgeBasePageParams =
410        serde_json::from_value(args.clone()).map_err(|e| {
411            Error::InvalidData(format!("invalid update_knowledge_base_page params: {e}"))
412        })?;
413    let page = provider
414        .update_page(UpdatePageParams {
415            page_id: params.page_id,
416            title: params.title,
417            content: params.content,
418            content_type: params.content_type,
419            version: params.version,
420            labels: params.labels,
421            parent_id: params.parent_id,
422        })
423        .await?;
424    Ok(ToolOutput::KnowledgeBasePageSummary(Box::new(page)))
425}
426
427#[derive(Deserialize)]
428#[serde(rename_all = "camelCase")]
429struct SearchKnowledgeBaseParams {
430    query: String,
431    space_key: Option<String>,
432    cursor: Option<String>,
433    limit: Option<u32>,
434    #[serde(default)]
435    raw_query: bool,
436}
437
438async fn execute_search_knowledge_base(
439    provider: &dyn KnowledgeBaseProvider,
440    args: &Value,
441) -> Result<ToolOutput> {
442    let params: SearchKnowledgeBaseParams = serde_json::from_value(args.clone())
443        .map_err(|e| Error::InvalidData(format!("invalid search_knowledge_base params: {e}")))?;
444    let result = provider
445        .search(SearchKbParams {
446            query: params.query,
447            space_key: params.space_key,
448            cursor: params.cursor,
449            limit: params.limit,
450            raw_query: params.raw_query,
451        })
452        .await?;
453    let meta = ResultMeta {
454        pagination: result.pagination,
455        sort_info: result.sort_info,
456    };
457    Ok(ToolOutput::KnowledgeBasePages(result.items, Some(meta)))
458}
459
460// --- Messenger tool dispatch ---
461
462/// Dispatch a messenger tool call.
463async fn dispatch_messenger_tool(
464    tool: &str,
465    args: &Value,
466    provider: &dyn MessengerProvider,
467) -> Result<ToolOutput> {
468    match tool {
469        "get_messenger_chats" => execute_get_messenger_chats(provider, args).await,
470        "get_chat_messages" => execute_get_chat_messages(provider, args).await,
471        "search_chat_messages" => execute_search_chat_messages(provider, args).await,
472        "send_message" => execute_send_message(provider, args).await,
473        _ => Err(Error::NotFound(format!("unknown messenger tool: {tool}"))),
474    }
475}
476
477// --- Messenger tool handlers ---
478
479#[derive(Deserialize, Default)]
480struct GetMessengerChatsParams {
481    search: Option<String>,
482    chat_type: Option<ChatType>,
483    limit: Option<u32>,
484    cursor: Option<String>,
485    include_inactive: Option<bool>,
486}
487
488async fn execute_get_messenger_chats(
489    provider: &dyn MessengerProvider,
490    args: &Value,
491) -> Result<ToolOutput> {
492    let params: GetMessengerChatsParams = parse_tool_params(args, "get_messenger_chats")?;
493    let request = GetChatsParams {
494        search: params.search,
495        chat_type: params.chat_type,
496        limit: params.limit,
497        cursor: params.cursor,
498        include_inactive: params.include_inactive,
499    };
500    let result = provider.get_chats(request).await?;
501    let meta = ResultMeta {
502        pagination: result.pagination,
503        sort_info: result.sort_info,
504    };
505    Ok(ToolOutput::MessengerChats(result.items, Some(meta)))
506}
507
508#[derive(Deserialize)]
509struct GetChatMessagesParams {
510    chat_id: String,
511    limit: Option<u32>,
512    cursor: Option<String>,
513    thread_id: Option<String>,
514    since: Option<String>,
515    until: Option<String>,
516}
517
518async fn execute_get_chat_messages(
519    provider: &dyn MessengerProvider,
520    args: &Value,
521) -> Result<ToolOutput> {
522    let params: GetChatMessagesParams = serde_json::from_value(args.clone())
523        .map_err(|e| Error::InvalidData(format!("missing 'chat_id' parameter: {e}")))?;
524    let request = GetMessagesParams {
525        chat_id: params.chat_id,
526        limit: params.limit,
527        cursor: params.cursor,
528        thread_id: params.thread_id,
529        since: params.since,
530        until: params.until,
531    };
532    let result = provider.get_messages(request).await?;
533    let meta = ResultMeta {
534        pagination: result.pagination,
535        sort_info: result.sort_info,
536    };
537    Ok(ToolOutput::MessengerMessages(result.items, Some(meta)))
538}
539
540#[derive(Deserialize)]
541struct SearchChatMessagesParams {
542    query: String,
543    chat_id: Option<String>,
544    limit: Option<u32>,
545    cursor: Option<String>,
546    since: Option<String>,
547    until: Option<String>,
548}
549
550async fn execute_search_chat_messages(
551    provider: &dyn MessengerProvider,
552    args: &Value,
553) -> Result<ToolOutput> {
554    let params: SearchChatMessagesParams = serde_json::from_value(args.clone())
555        .map_err(|e| Error::InvalidData(format!("missing 'query' parameter: {e}")))?;
556    let request = SearchMessagesParams {
557        query: params.query,
558        chat_id: params.chat_id,
559        limit: params.limit,
560        cursor: params.cursor,
561        since: params.since,
562        until: params.until,
563    };
564    let result = provider.search_messages(request).await?;
565    let meta = ResultMeta {
566        pagination: result.pagination,
567        sort_info: result.sort_info,
568    };
569    Ok(ToolOutput::MessengerMessages(result.items, Some(meta)))
570}
571
572#[derive(Deserialize)]
573struct SendMessengerMessageParams {
574    chat_id: String,
575    text: String,
576    thread_id: Option<String>,
577    reply_to_id: Option<String>,
578}
579
580async fn execute_send_message(
581    provider: &dyn MessengerProvider,
582    args: &Value,
583) -> Result<ToolOutput> {
584    let params: SendMessengerMessageParams = serde_json::from_value(args.clone())
585        .map_err(|e| Error::InvalidData(format!("invalid send_message params: {e}")))?;
586    let request = SendMessageParams {
587        chat_id: params.chat_id,
588        text: params.text,
589        thread_id: params.thread_id,
590        reply_to_id: params.reply_to_id,
591        attachments: vec![],
592    };
593    let message = provider.send_message(request).await?;
594    Ok(ToolOutput::SingleMessage(Box::new(message)))
595}
596
597// --- Tool dispatch ---
598
599/// Dispatch a tool call to the appropriate provider method.
600async fn dispatch_tool(
601    tool: &str,
602    args: &Value,
603    provider: &dyn devboy_core::Provider,
604    asset_manager: Option<&devboy_assets::AssetManager>,
605) -> Result<ToolOutput> {
606    match tool {
607        // Issue tools
608        "get_issues" => execute_get_issues(provider, args).await,
609        "get_issue" => execute_get_issue(provider, args).await,
610        "get_issue_comments" => execute_get_issue_comments(provider, args).await,
611        "get_issue_relations" => execute_get_issue_relations(provider, args).await,
612        "create_issue" => execute_create_issue(provider, args).await,
613        "update_issue" => execute_update_issue(provider, args).await,
614        "add_issue_comment" => execute_add_issue_comment(provider, args).await,
615
616        // Merge request tools
617        "get_merge_requests" => execute_get_merge_requests(provider, args).await,
618        "get_merge_request" => execute_get_merge_request(provider, args).await,
619        "get_merge_request_discussions" => {
620            execute_get_merge_request_discussions(provider, args).await
621        }
622        "get_merge_request_diffs" => execute_get_merge_request_diffs(provider, args).await,
623        "create_merge_request" => execute_create_merge_request(provider, args).await,
624        "create_merge_request_comment" => {
625            execute_create_merge_request_comment(provider, args).await
626        }
627
628        // Pipeline tools
629        "get_pipeline" => execute_get_pipeline(provider, args).await,
630        "get_job_logs" => execute_get_job_logs(provider, args).await,
631
632        // Status / user / link tools
633        "get_available_statuses" => execute_get_available_statuses(provider).await,
634        "get_users" => execute_get_users(provider, args).await,
635        "link_issues" => execute_link_issues(provider, args).await,
636        "unlink_issues" => execute_unlink_issues(provider, args).await,
637
638        // Epic tools (issue-based with "epic" label convention)
639        "get_epics" => execute_get_epics(provider, args).await,
640        "create_epic" => execute_create_epic(provider, args).await,
641        "update_epic" => execute_update_epic(provider, args).await,
642
643        // MR update
644        "update_merge_request" => execute_update_merge_request(provider, args).await,
645
646        // Asset tools
647        "get_assets" => execute_get_assets(provider, args).await,
648        "upload_asset" => execute_upload_asset(provider, args).await,
649        "download_asset" => execute_download_asset(provider, args, asset_manager).await,
650        "delete_asset" => execute_delete_asset(provider, args, asset_manager).await,
651
652        // Jira Structure tools
653        "get_structures" => execute_get_structures(provider).await,
654        "get_structure_forest" => execute_get_structure_forest(provider, args).await,
655        "add_structure_rows" => execute_add_structure_rows(provider, args).await,
656        "move_structure_rows" => execute_move_structure_rows(provider, args).await,
657        "remove_structure_row" => execute_remove_structure_row(provider, args).await,
658        "get_structure_values" => execute_get_structure_values(provider, args).await,
659        "get_structure_views" => execute_get_structure_views(provider, args).await,
660        "save_structure_view" => execute_save_structure_view(provider, args).await,
661        "create_structure" => execute_create_structure(provider, args).await,
662
663        // Project versions / fixVersion (issue #238)
664        "list_project_versions" => execute_list_project_versions(provider, args).await,
665        "upsert_project_version" => execute_upsert_project_version(provider, args).await,
666
667        // Agile / Sprint (issue #198)
668        "get_board_sprints" => execute_get_board_sprints(provider, args).await,
669        "assign_to_sprint" => execute_assign_to_sprint(provider, args).await,
670
671        // Custom-field discovery
672        "get_custom_fields" => execute_get_custom_fields(provider, args).await,
673
674        _ => Err(Error::NotFound(format!("unknown tool: {tool}"))),
675    }
676}
677
678/// Dispatch a meeting notes tool call.
679async fn dispatch_meeting_tool(
680    tool: &str,
681    args: &Value,
682    provider: &dyn MeetingNotesProvider,
683) -> Result<ToolOutput> {
684    match tool {
685        "get_meeting_notes" => execute_get_meeting_notes(provider, args).await,
686        "get_meeting_transcript" => execute_get_meeting_transcript(provider, args).await,
687        "search_meeting_notes" => execute_search_meeting_notes(provider, args).await,
688        _ => Err(Error::NotFound(format!("unknown meeting tool: {tool}"))),
689    }
690}
691
692// --- Meeting notes tool handlers ---
693
694#[derive(Deserialize, Default)]
695struct GetMeetingNotesParams {
696    from_date: Option<String>,
697    to_date: Option<String>,
698    participants: Option<Vec<String>>,
699    host_email: Option<String>,
700    limit: Option<u32>,
701    offset: Option<u32>,
702}
703
704async fn execute_get_meeting_notes(
705    provider: &dyn MeetingNotesProvider,
706    args: &Value,
707) -> Result<ToolOutput> {
708    let params: GetMeetingNotesParams = parse_tool_params(args, "get_meeting_notes")?;
709    let filter = MeetingFilter {
710        keyword: None,
711        from_date: params.from_date,
712        to_date: params.to_date,
713        participants: params.participants,
714        host_email: params.host_email,
715        limit: params.limit,
716        skip: params.offset,
717    };
718    let result = provider.get_meetings(filter).await?;
719    let meta = ResultMeta {
720        pagination: result.pagination,
721        sort_info: result.sort_info,
722    };
723    Ok(ToolOutput::MeetingNotes(result.items, Some(meta)))
724}
725
726#[derive(Deserialize)]
727struct GetMeetingTranscriptParams {
728    meeting_id: String,
729}
730
731async fn execute_get_meeting_transcript(
732    provider: &dyn MeetingNotesProvider,
733    args: &Value,
734) -> Result<ToolOutput> {
735    let params: GetMeetingTranscriptParams = serde_json::from_value(args.clone())
736        .map_err(|e| Error::InvalidData(format!("invalid params: {e}")))?;
737    let transcript = provider.get_transcript(&params.meeting_id).await?;
738    Ok(ToolOutput::MeetingTranscript(Box::new(transcript)))
739}
740
741#[derive(Deserialize)]
742struct SearchMeetingNotesParams {
743    query: String,
744    from_date: Option<String>,
745    to_date: Option<String>,
746    participants: Option<Vec<String>>,
747    host_email: Option<String>,
748    limit: Option<u32>,
749    offset: Option<u32>,
750}
751
752async fn execute_search_meeting_notes(
753    provider: &dyn MeetingNotesProvider,
754    args: &Value,
755) -> Result<ToolOutput> {
756    let params: SearchMeetingNotesParams = serde_json::from_value(args.clone())
757        .map_err(|e| Error::InvalidData(format!("invalid params: {e}")))?;
758    let filter = MeetingFilter {
759        keyword: None,
760        from_date: params.from_date,
761        to_date: params.to_date,
762        participants: params.participants,
763        host_email: params.host_email,
764        limit: params.limit,
765        skip: params.offset,
766    };
767    let result = provider.search_meetings(&params.query, filter).await?;
768    let meta = ResultMeta {
769        pagination: result.pagination,
770        sort_info: result.sort_info,
771    };
772    Ok(ToolOutput::MeetingNotes(result.items, Some(meta)))
773}
774
775// --- Issue tool handlers ---
776
777#[derive(Deserialize, Default)]
778struct GetIssuesParams {
779    state: Option<String>,
780    #[serde(rename = "stateCategory")]
781    state_category: Option<String>,
782    search: Option<String>,
783    labels: Option<Vec<String>>,
784    #[serde(rename = "labelsOperator")]
785    labels_operator: Option<String>,
786    assignee: Option<String>,
787    limit: Option<u32>,
788    offset: Option<u32>,
789    sort_by: Option<String>,
790    sort_order: Option<String>,
791    #[serde(rename = "projectKey")]
792    project_key: Option<String>,
793    #[serde(rename = "nativeQuery")]
794    native_query: Option<String>,
795    /// Token budget for response size control (consumed by format layer via execute_and_format).
796    #[allow(dead_code)]
797    budget: Option<usize>,
798}
799
800async fn execute_get_issues(
801    provider: &dyn devboy_core::Provider,
802    args: &Value,
803) -> Result<ToolOutput> {
804    let params: GetIssuesParams = parse_tool_params(args, "get_issues")?;
805    let filter = IssueFilter {
806        state: params.state,
807        state_category: params.state_category,
808        search: params.search,
809        labels: params.labels,
810        labels_operator: params.labels_operator,
811        assignee: params.assignee,
812        limit: params.limit.or(Some(20)),
813        offset: params.offset,
814        sort_by: params.sort_by,
815        sort_order: params.sort_order,
816        project_key: params.project_key,
817        native_query: params.native_query,
818    };
819    let result = provider.get_issues(filter).await?;
820    let meta = ResultMeta {
821        pagination: result.pagination,
822        sort_info: result.sort_info,
823    };
824    Ok(ToolOutput::Issues(result.items, Some(meta)))
825}
826
827#[derive(Deserialize)]
828struct KeyParam {
829    key: String,
830    /// Token budget for response size control (consumed by format layer via execute_and_format).
831    #[serde(default)]
832    #[allow(dead_code)]
833    budget: Option<usize>,
834}
835
836#[derive(Deserialize)]
837struct GetIssueParams {
838    key: String,
839    #[serde(default = "default_true", rename = "includeComments")]
840    include_comments: bool,
841    #[serde(default = "default_true", rename = "includeRelations")]
842    include_relations: bool,
843    #[serde(default)]
844    #[allow(dead_code)]
845    budget: Option<usize>,
846}
847
848fn default_true() -> bool {
849    true
850}
851
852async fn execute_get_issue(
853    provider: &dyn devboy_core::Provider,
854    args: &Value,
855) -> Result<ToolOutput> {
856    let params: GetIssueParams = serde_json::from_value(args.clone())
857        .map_err(|e| Error::InvalidData(format!("missing 'key' parameter: {e}")))?;
858    let issue = provider.get_issue(&params.key).await?;
859
860    // If no extras requested, return just the issue
861    if !params.include_comments && !params.include_relations {
862        return Ok(ToolOutput::SingleIssue(Box::new(issue)));
863    }
864
865    // Build a composite JSON with issue + optional comments/relations
866    let mut result = serde_json::to_value(&issue).unwrap_or_default();
867    let mut has_extras = false;
868
869    if params.include_comments
870        && let Ok(comments_result) = provider.get_comments(&params.key).await
871    {
872        result["comments"] = serde_json::to_value(&comments_result.items).unwrap_or_default();
873        result["comments_count"] = serde_json::json!(comments_result.items.len());
874        has_extras = true;
875    }
876
877    if params.include_relations
878        && let Ok(relations) = provider.get_issue_relations(&params.key).await
879    {
880        result["relations"] = serde_json::to_value(&relations).unwrap_or_default();
881        if issue.subtasks.is_empty() && !relations.subtasks.is_empty() {
882            result["subtasks"] = serde_json::to_value(&relations.subtasks).unwrap_or_default();
883        }
884        result["subtasks_count"] =
885            serde_json::json!(issue.subtasks.len().max(relations.subtasks.len()));
886        has_extras = true;
887    }
888
889    // If no extras were actually fetched, return simple issue
890    if !has_extras {
891        return Ok(ToolOutput::SingleIssue(Box::new(issue)));
892    }
893
894    Ok(ToolOutput::Text(
895        serde_json::to_string_pretty(&result).unwrap_or_default(),
896    ))
897}
898
899async fn execute_get_issue_comments(
900    provider: &dyn devboy_core::Provider,
901    args: &Value,
902) -> Result<ToolOutput> {
903    let params: KeyParam = serde_json::from_value(args.clone())
904        .map_err(|e| Error::InvalidData(format!("missing 'key' parameter: {e}")))?;
905    let result = provider.get_comments(&params.key).await?;
906    let meta = ResultMeta {
907        pagination: result.pagination,
908        sort_info: result.sort_info,
909    };
910    Ok(ToolOutput::Comments(result.items, Some(meta)))
911}
912
913async fn execute_get_issue_relations(
914    provider: &dyn devboy_core::Provider,
915    args: &Value,
916) -> Result<ToolOutput> {
917    let params: KeyParam = serde_json::from_value(args.clone())
918        .map_err(|e| Error::InvalidData(format!("missing 'key' parameter: {e}")))?;
919    let relations = provider.get_issue_relations(&params.key).await?;
920    Ok(ToolOutput::Relations(Box::new(relations)))
921}
922
923#[derive(Deserialize)]
924struct CreateIssueParams {
925    title: String,
926    description: Option<String>,
927    #[serde(default)]
928    labels: Vec<String>,
929    #[serde(default)]
930    assignees: Vec<String>,
931    #[serde(default, deserialize_with = "deserialize_string_or_number")]
932    priority: Option<String>,
933    #[serde(alias = "parentId")]
934    parent: Option<String>,
935    markdown: Option<bool>,
936    #[serde(rename = "projectId")]
937    project_id: Option<String>,
938    #[serde(rename = "issueType")]
939    issue_type: Option<String>,
940    /// Jira component names (issue #197). Serde rejects non-string-array
941    /// input instead of silently dropping entries (Copilot review on PR #205).
942    #[serde(default)]
943    components: Vec<String>,
944    /// Jira fix-version names. Same shape and semantics as `components`.
945    #[serde(default, rename = "fixVersions")]
946    fix_versions: Vec<String>,
947    /// Jira parent epic key. Resolved via the field-id lookup so callers
948    /// don't need to know the instance's `customfield_*` number.
949    #[serde(default, rename = "epicKey")]
950    epic_key: Option<String>,
951    /// Jira sprint id. Numeric agile-board sprint id.
952    #[serde(default, rename = "sprintId")]
953    sprint_id: Option<i64>,
954    /// Jira Epic Name. Required by Server/DC + Cloud company-managed
955    /// when `issueType == "Epic"`.
956    #[serde(default, rename = "epicName")]
957    epic_name: Option<String>,
958}
959
960async fn execute_create_issue(
961    provider: &dyn devboy_core::Provider,
962    args: &Value,
963) -> Result<ToolOutput> {
964    let params: CreateIssueParams = serde_json::from_value(args.clone())
965        .map_err(|e| Error::InvalidData(format!("invalid create_issue params: {e}")))?;
966    let custom_fields = args.get("customFields").cloned();
967    let input = CreateIssueInput {
968        title: params.title,
969        description: params.description,
970        labels: params.labels,
971        assignees: params.assignees,
972        priority: params.priority,
973        parent: params.parent,
974        markdown: params.markdown.unwrap_or(true),
975        project_id: params.project_id,
976        issue_type: params.issue_type,
977        custom_fields,
978        components: params.components,
979        fix_versions: params.fix_versions,
980        epic_key: params.epic_key,
981        sprint_id: params.sprint_id,
982        epic_name: params.epic_name,
983    };
984    let issue = provider.create_issue(input).await?;
985
986    // Set custom fields via separate API call (ClickUp uses Array format)
987    if let Some(cf) = args.get("customFields").and_then(|v| v.as_array())
988        && !cf.is_empty()
989        && let Err(e) = provider.set_custom_fields(&issue.key, cf).await
990    {
991        tracing::warn!(error = %e, "Failed to set custom fields on created issue");
992    }
993
994    Ok(ToolOutput::SingleIssue(Box::new(issue)))
995}
996
997#[derive(Deserialize)]
998struct UpdateIssueParams {
999    key: String,
1000    title: Option<String>,
1001    description: Option<String>,
1002    state: Option<String>,
1003    /// Provider-specific status name (#288). For ClickUp: any custom
1004    /// status from `get_available_statuses` (e.g. "in progress",
1005    /// "review"). Currently a no-op for other providers.
1006    #[serde(default)]
1007    status: Option<String>,
1008    labels: Option<Vec<String>>,
1009    assignees: Option<Vec<String>>,
1010    #[serde(default, deserialize_with = "deserialize_string_or_number")]
1011    priority: Option<String>,
1012    #[serde(rename = "parentId")]
1013    parent_id: Option<String>,
1014    markdown: Option<bool>,
1015    /// Jira component names (issue #197). `None` (key absent) leaves
1016    /// components untouched; `Some([])` clears all; `Some([...])` replaces.
1017    /// Serde-parsed so non-array / non-string input errors fast.
1018    #[serde(default)]
1019    components: Option<Vec<String>>,
1020    /// Jira fix-version names. Same shape and semantics as `components`.
1021    #[serde(default, rename = "fixVersions")]
1022    fix_versions: Option<Vec<String>>,
1023    /// Jira parent epic key.
1024    #[serde(default, rename = "epicKey")]
1025    epic_key: Option<String>,
1026    /// Jira sprint id.
1027    #[serde(default, rename = "sprintId")]
1028    sprint_id: Option<i64>,
1029    /// Jira Epic Name (Epic-typed issues).
1030    #[serde(default, rename = "epicName")]
1031    epic_name: Option<String>,
1032}
1033
1034async fn execute_update_issue(
1035    provider: &dyn devboy_core::Provider,
1036    args: &Value,
1037) -> Result<ToolOutput> {
1038    let params: UpdateIssueParams = serde_json::from_value(args.clone())
1039        .map_err(|e| Error::InvalidData(format!("invalid update_issue params: {e}")))?;
1040    let custom_fields = args.get("customFields").cloned();
1041    let input = UpdateIssueInput {
1042        title: params.title,
1043        description: params.description,
1044        state: params.state,
1045        status: params.status,
1046        labels: params.labels,
1047        assignees: params.assignees,
1048        priority: params.priority,
1049        parent_id: params.parent_id,
1050        markdown: params.markdown.unwrap_or(true),
1051        custom_fields,
1052        components: params.components,
1053        fix_versions: params.fix_versions,
1054        epic_key: params.epic_key,
1055        sprint_id: params.sprint_id,
1056        epic_name: params.epic_name,
1057    };
1058    let key = params.key;
1059    let issue = provider.update_issue(&key, input).await?;
1060
1061    // Set custom fields via separate API call (ClickUp uses Array format)
1062    if let Some(cf) = args.get("customFields").and_then(|v| v.as_array())
1063        && !cf.is_empty()
1064        && let Err(e) = provider.set_custom_fields(&key, cf).await
1065    {
1066        tracing::warn!(error = %e, "Failed to set custom fields on updated issue");
1067    }
1068    Ok(ToolOutput::SingleIssue(Box::new(issue)))
1069}
1070
1071#[derive(Deserialize)]
1072struct AddCommentParams {
1073    key: String,
1074    body: String,
1075    #[serde(default)]
1076    attachments: Vec<AttachmentParam>,
1077}
1078
1079#[derive(Deserialize)]
1080struct AttachmentParam {
1081    /// Base64-encoded file content
1082    #[serde(rename = "fileData")]
1083    file_data: String,
1084    /// Filename (e.g., "screenshot.png")
1085    filename: String,
1086}
1087
1088async fn execute_add_issue_comment(
1089    provider: &dyn devboy_core::Provider,
1090    args: &Value,
1091) -> Result<ToolOutput> {
1092    let params: AddCommentParams = serde_json::from_value(args.clone())
1093        .map_err(|e| Error::InvalidData(format!("invalid add_issue_comment params: {e}")))?;
1094
1095    let mut body = params.body.clone();
1096    let mut uploaded = 0;
1097    let mut upload_errors = Vec::new();
1098
1099    // Validate attachment limits
1100    const MAX_ATTACHMENTS: usize = 10;
1101
1102    if params.attachments.len() > MAX_ATTACHMENTS {
1103        return Err(Error::InvalidData(format!(
1104            "Too many attachments: {} (max {})",
1105            params.attachments.len(),
1106            MAX_ATTACHMENTS
1107        )));
1108    }
1109
1110    // Upload attachments and append links to comment body
1111    for att in &params.attachments {
1112        use base64::Engine;
1113        let data = match base64::engine::general_purpose::STANDARD.decode(&att.file_data) {
1114            Ok(d) => d,
1115            Err(e) => {
1116                upload_errors.push(format!("{}: decode error: {}", att.filename, e));
1117                continue;
1118            }
1119        };
1120
1121        if data.len() > MAX_FILE_SIZE {
1122            upload_errors.push(format!(
1123                "{}: file too large ({} bytes, max {})",
1124                att.filename,
1125                data.len(),
1126                MAX_FILE_SIZE
1127            ));
1128            continue;
1129        }
1130
1131        match provider
1132            .upload_attachment(&params.key, &att.filename, &data)
1133            .await
1134        {
1135            Ok(url) => {
1136                if !url.is_empty() {
1137                    body.push_str(&format!("\n\n[{}]({})", att.filename, url));
1138                }
1139                uploaded += 1;
1140            }
1141            Err(e) => {
1142                upload_errors.push(format!("{}: {}", att.filename, e));
1143            }
1144        }
1145    }
1146
1147    let comment = devboy_core::IssueProvider::add_comment(provider, &params.key, &body).await?;
1148
1149    let mut msg = format!("Comment added to {} (id: {})", params.key, comment.id);
1150    if uploaded > 0 {
1151        msg.push_str(&format!(", {} attachment(s) uploaded", uploaded));
1152    }
1153    if !upload_errors.is_empty() {
1154        msg.push_str(&format!(
1155            ", {} attachment error(s): {}",
1156            upload_errors.len(),
1157            upload_errors.join("; ")
1158        ));
1159    }
1160    Ok(ToolOutput::Text(msg))
1161}
1162
1163// --- Merge request tool handlers ---
1164
1165#[derive(Deserialize, Default)]
1166struct GetMergeRequestsParams {
1167    state: Option<String>,
1168    author: Option<String>,
1169    labels: Option<Vec<String>>,
1170    source_branch: Option<String>,
1171    target_branch: Option<String>,
1172    limit: Option<u32>,
1173    offset: Option<u32>,
1174    sort_by: Option<String>,
1175    sort_order: Option<String>,
1176    /// Token budget for response size control (consumed by format layer via execute_and_format).
1177    #[allow(dead_code)]
1178    budget: Option<usize>,
1179}
1180
1181async fn execute_get_merge_requests(
1182    provider: &dyn devboy_core::Provider,
1183    args: &Value,
1184) -> Result<ToolOutput> {
1185    let params: GetMergeRequestsParams = parse_tool_params(args, "get_merge_requests")?;
1186    let filter = MrFilter {
1187        state: params.state,
1188        source_branch: params.source_branch,
1189        target_branch: params.target_branch,
1190        author: params.author,
1191        labels: params.labels,
1192        limit: params.limit.or(Some(20)),
1193        offset: params.offset,
1194        sort_by: params.sort_by,
1195        sort_order: params.sort_order,
1196    };
1197    let result = provider.get_merge_requests(filter).await?;
1198    let meta = ResultMeta {
1199        pagination: result.pagination,
1200        sort_info: result.sort_info,
1201    };
1202    Ok(ToolOutput::MergeRequests(result.items, Some(meta)))
1203}
1204
1205async fn execute_get_merge_request(
1206    provider: &dyn devboy_core::Provider,
1207    args: &Value,
1208) -> Result<ToolOutput> {
1209    let params: KeyParam = serde_json::from_value(args.clone())
1210        .map_err(|e| Error::InvalidData(format!("missing 'key' parameter: {e}")))?;
1211    let mr = provider.get_merge_request(&params.key).await?;
1212    Ok(ToolOutput::SingleMergeRequest(Box::new(mr)))
1213}
1214
1215async fn execute_get_merge_request_discussions(
1216    provider: &dyn devboy_core::Provider,
1217    args: &Value,
1218) -> Result<ToolOutput> {
1219    let params: KeyParam = serde_json::from_value(args.clone())
1220        .map_err(|e| Error::InvalidData(format!("missing 'key' parameter: {e}")))?;
1221    let result = provider.get_discussions(&params.key).await?;
1222    let meta = ResultMeta {
1223        pagination: result.pagination,
1224        sort_info: result.sort_info,
1225    };
1226    Ok(ToolOutput::Discussions(result.items, Some(meta)))
1227}
1228
1229async fn execute_get_merge_request_diffs(
1230    provider: &dyn devboy_core::Provider,
1231    args: &Value,
1232) -> Result<ToolOutput> {
1233    let params: KeyParam = serde_json::from_value(args.clone())
1234        .map_err(|e| Error::InvalidData(format!("missing 'key' parameter: {e}")))?;
1235    let result = provider.get_diffs(&params.key).await?;
1236    let meta = ResultMeta {
1237        pagination: result.pagination,
1238        sort_info: result.sort_info,
1239    };
1240    Ok(ToolOutput::Diffs(result.items, Some(meta)))
1241}
1242
1243#[derive(Deserialize)]
1244struct CreateMergeRequestParams {
1245    title: String,
1246    description: Option<String>,
1247    source_branch: String,
1248    target_branch: String,
1249    #[serde(default)]
1250    draft: bool,
1251    #[serde(default)]
1252    labels: Vec<String>,
1253    #[serde(default)]
1254    reviewers: Vec<String>,
1255}
1256
1257async fn execute_create_merge_request(
1258    provider: &dyn devboy_core::Provider,
1259    args: &Value,
1260) -> Result<ToolOutput> {
1261    let params: CreateMergeRequestParams = serde_json::from_value(args.clone())
1262        .map_err(|e| Error::InvalidData(format!("invalid create_merge_request params: {e}")))?;
1263    let input = CreateMergeRequestInput {
1264        title: params.title,
1265        description: params.description,
1266        source_branch: params.source_branch,
1267        target_branch: params.target_branch,
1268        draft: params.draft,
1269        labels: params.labels,
1270        reviewers: params.reviewers,
1271    };
1272    let mr = provider.create_merge_request(input).await?;
1273    Ok(ToolOutput::SingleMergeRequest(Box::new(mr)))
1274}
1275
1276#[derive(Deserialize)]
1277struct CreateMrCommentParams {
1278    #[serde(alias = "mrKey")]
1279    key: String,
1280    body: String,
1281    #[serde(alias = "filePath")]
1282    file_path: Option<String>,
1283    line: Option<u32>,
1284    #[serde(alias = "lineType")]
1285    line_type: Option<String>,
1286    #[serde(alias = "commitSha")]
1287    commit_sha: Option<String>,
1288    #[serde(alias = "discussionId")]
1289    discussion_id: Option<String>,
1290}
1291
1292async fn execute_create_merge_request_comment(
1293    provider: &dyn devboy_core::Provider,
1294    args: &Value,
1295) -> Result<ToolOutput> {
1296    let params: CreateMrCommentParams = serde_json::from_value(args.clone()).map_err(|e| {
1297        Error::InvalidData(format!("invalid create_merge_request_comment params: {e}"))
1298    })?;
1299
1300    let position = params.file_path.map(|fp| devboy_core::CodePosition {
1301        file_path: fp,
1302        line: params.line.unwrap_or(1),
1303        line_type: params.line_type.unwrap_or_else(|| "new".into()),
1304        commit_sha: params.commit_sha,
1305    });
1306
1307    let input = CreateCommentInput {
1308        body: params.body,
1309        position,
1310        discussion_id: params.discussion_id,
1311    };
1312
1313    let comment = MergeRequestProvider::add_comment(provider, &params.key, input).await?;
1314    Ok(ToolOutput::Text(format!(
1315        "Comment added to {} (id: {})",
1316        params.key, comment.id
1317    )))
1318}
1319
1320// --- Pipeline tool handlers ---
1321
1322#[derive(Deserialize, Default)]
1323struct GetPipelineParams {
1324    branch: Option<String>,
1325    #[serde(rename = "mrKey")]
1326    mr_key: Option<String>,
1327    #[serde(rename = "includeFailedLogs")]
1328    include_failed_logs: Option<bool>,
1329}
1330
1331async fn execute_get_pipeline(
1332    provider: &dyn devboy_core::Provider,
1333    args: &Value,
1334) -> Result<ToolOutput> {
1335    let params: GetPipelineParams = parse_tool_params(args, "get_pipeline")?;
1336    let input = GetPipelineInput {
1337        branch: params.branch,
1338        mr_key: params.mr_key,
1339        include_failed_logs: params.include_failed_logs.unwrap_or(true),
1340    };
1341    let pipeline = PipelineProvider::get_pipeline(provider, input).await?;
1342    Ok(ToolOutput::Pipeline(Box::new(pipeline)))
1343}
1344
1345#[derive(Deserialize)]
1346struct GetJobLogsParams {
1347    #[serde(rename = "jobId")]
1348    job_id: String,
1349    pattern: Option<String>,
1350    context: Option<usize>,
1351    #[serde(rename = "maxMatches")]
1352    max_matches: Option<usize>,
1353    offset: Option<usize>,
1354    limit: Option<usize>,
1355    full: Option<bool>,
1356}
1357
1358async fn execute_get_job_logs(
1359    provider: &dyn devboy_core::Provider,
1360    args: &Value,
1361) -> Result<ToolOutput> {
1362    let params: GetJobLogsParams = serde_json::from_value(args.clone())
1363        .map_err(|e| Error::InvalidData(format!("invalid get_job_logs params: {e}")))?;
1364
1365    // Clamp limit to max 1000 as declared in schema
1366    let clamped_limit = params.limit.map(|l| l.min(1000));
1367
1368    let mode = if let Some(pattern) = params.pattern {
1369        JobLogMode::Search {
1370            pattern,
1371            context: params.context.unwrap_or(5).min(50),
1372            max_matches: params.max_matches.unwrap_or(20).min(100),
1373        }
1374    } else if let Some(true) = params.full {
1375        JobLogMode::Full {
1376            max_lines: clamped_limit.unwrap_or(1000),
1377        }
1378    } else if params.offset.is_some() || clamped_limit.is_some() {
1379        JobLogMode::Paginated {
1380            offset: params.offset.unwrap_or(0),
1381            limit: clamped_limit.unwrap_or(200),
1382        }
1383    } else {
1384        JobLogMode::Smart
1385    };
1386
1387    let options = JobLogOptions { mode };
1388    let log_output = PipelineProvider::get_job_logs(provider, &params.job_id, options).await?;
1389    Ok(ToolOutput::JobLog(Box::new(log_output)))
1390}
1391
1392// --- Status / User / Link tool handlers ---
1393
1394async fn execute_get_available_statuses(
1395    provider: &dyn devboy_core::Provider,
1396) -> Result<ToolOutput> {
1397    let result = IssueProvider::get_statuses(provider).await?;
1398    let meta = ResultMeta {
1399        pagination: result.pagination,
1400        sort_info: result.sort_info,
1401    };
1402    Ok(ToolOutput::Statuses(result.items, Some(meta)))
1403}
1404
1405#[derive(Deserialize, Default)]
1406struct GetUsersParams {
1407    user_id: Option<String>,
1408    project_key: Option<String>,
1409    search: Option<String>,
1410    include_inactive: Option<bool>,
1411    start_at: Option<u32>,
1412    max_results: Option<u32>,
1413}
1414
1415async fn execute_get_users(
1416    provider: &dyn devboy_core::Provider,
1417    args: &Value,
1418) -> Result<ToolOutput> {
1419    let params: GetUsersParams = parse_tool_params(args, "get_users")?;
1420    let options = GetUsersOptions {
1421        user_id: params.user_id,
1422        project_key: params.project_key,
1423        search: params.search,
1424        include_inactive: params.include_inactive,
1425        start_at: params.start_at,
1426        max_results: params.max_results,
1427    };
1428    let result = IssueProvider::get_users(provider, options).await?;
1429    let meta = ResultMeta {
1430        pagination: result.pagination,
1431        sort_info: result.sort_info,
1432    };
1433    Ok(ToolOutput::Users(result.items, Some(meta)))
1434}
1435
1436#[derive(Deserialize)]
1437struct LinkIssuesParams {
1438    #[serde(alias = "sourceIssueKey", alias = "issueKey1")]
1439    source_key: String,
1440    #[serde(alias = "targetIssueKey", alias = "issueKey2")]
1441    target_key: String,
1442    #[serde(alias = "linkType")]
1443    link_type: String,
1444}
1445
1446async fn execute_link_issues(
1447    provider: &dyn devboy_core::Provider,
1448    args: &Value,
1449) -> Result<ToolOutput> {
1450    let params: LinkIssuesParams = serde_json::from_value(args.clone())
1451        .map_err(|e| Error::InvalidData(format!("invalid link_issues params: {e}")))?;
1452    IssueProvider::link_issues(
1453        provider,
1454        &params.source_key,
1455        &params.target_key,
1456        &params.link_type,
1457    )
1458    .await?;
1459    Ok(ToolOutput::Text(format!(
1460        "Linked {} -> {} (type: {})",
1461        params.source_key, params.target_key, params.link_type
1462    )))
1463}
1464
1465async fn execute_unlink_issues(
1466    provider: &dyn devboy_core::Provider,
1467    args: &Value,
1468) -> Result<ToolOutput> {
1469    let params: LinkIssuesParams = serde_json::from_value(args.clone())
1470        .map_err(|e| Error::InvalidData(format!("invalid unlink_issues params: {e}")))?;
1471    IssueProvider::unlink_issues(
1472        provider,
1473        &params.source_key,
1474        &params.target_key,
1475        &params.link_type,
1476    )
1477    .await?;
1478    Ok(ToolOutput::Text(format!(
1479        "Unlinked {} -> {} (type: {})",
1480        params.source_key, params.target_key, params.link_type
1481    )))
1482}
1483
1484// --- Epic tool handlers ---
1485
1486#[derive(Deserialize, Default)]
1487struct GetEpicsParams {
1488    state: Option<String>,
1489    search: Option<String>,
1490    assignee: Option<String>,
1491    #[serde(rename = "goalId")]
1492    goal_id: Option<String>,
1493    limit: Option<u32>,
1494    offset: Option<u32>,
1495}
1496
1497/// Extract goal ID (G1-G9) from issue labels/tags.
1498fn extract_goal_id(labels: &[String]) -> Option<String> {
1499    labels.iter().find_map(|l| {
1500        let lower = l.to_lowercase();
1501        if lower.len() == 2
1502            && lower.starts_with('g')
1503            && lower.chars().nth(1).is_some_and(|c| c.is_ascii_digit())
1504        {
1505            Some(lower.to_uppercase())
1506        } else {
1507            None
1508        }
1509    })
1510}
1511
1512/// Calculate epic progress from subtasks.
1513fn epic_progress(subtasks: &[devboy_core::Issue]) -> serde_json::Value {
1514    let total = subtasks.len();
1515    let completed = subtasks.iter().filter(|s| s.state == "closed").count();
1516    let percentage = if total > 0 {
1517        (completed as f64 / total as f64 * 100.0).round() as u32
1518    } else {
1519        0
1520    };
1521    serde_json::json!({
1522        "total_subtasks": total,
1523        "completed_subtasks": completed,
1524        "percentage": percentage,
1525    })
1526}
1527
1528async fn execute_get_epics(
1529    provider: &dyn devboy_core::Provider,
1530    args: &Value,
1531) -> Result<ToolOutput> {
1532    let params: GetEpicsParams = parse_tool_params(args, "get_epics")?;
1533    let filter = IssueFilter {
1534        state: params.state,
1535        state_category: None,
1536        search: params.search,
1537        labels: Some(vec!["epic".to_string()]),
1538        labels_operator: None,
1539        assignee: params.assignee,
1540        limit: params.limit.or(Some(50)),
1541        offset: params.offset,
1542        sort_by: None,
1543        sort_order: None,
1544        project_key: None,
1545        native_query: None,
1546    };
1547    let result = provider.get_issues(filter).await?;
1548    let mut epics = result.items;
1549
1550    // Filter by goalId if provided
1551    if let Some(ref goal) = params.goal_id {
1552        let goal_lower = goal.to_lowercase();
1553        epics.retain(|e| e.labels.iter().any(|l| l.to_lowercase() == goal_lower));
1554    }
1555
1556    // Enrich each epic with goal ID and progress
1557    let enriched: Vec<serde_json::Value> = epics
1558        .iter()
1559        .map(|epic| {
1560            let mut v = serde_json::to_value(epic).unwrap_or_default();
1561            v["goal_id"] = serde_json::json!(extract_goal_id(&epic.labels));
1562            v["progress"] = epic_progress(&epic.subtasks);
1563            v
1564        })
1565        .collect();
1566
1567    Ok(ToolOutput::Text(
1568        serde_json::to_string_pretty(&enriched).unwrap_or_default(),
1569    ))
1570}
1571
1572#[derive(Deserialize)]
1573struct CreateEpicParams {
1574    title: String,
1575    description: Option<String>,
1576    #[serde(rename = "goalId")]
1577    goal_id: Option<String>,
1578    #[serde(default)]
1579    labels: Vec<String>,
1580    #[serde(default)]
1581    assignees: Vec<String>,
1582    #[serde(default, deserialize_with = "deserialize_string_or_number")]
1583    priority: Option<String>,
1584    markdown: Option<bool>,
1585}
1586
1587async fn execute_create_epic(
1588    provider: &dyn devboy_core::Provider,
1589    args: &Value,
1590) -> Result<ToolOutput> {
1591    let params: CreateEpicParams = serde_json::from_value(args.clone())
1592        .map_err(|e| Error::InvalidData(format!("invalid create_epic params: {e}")))?;
1593
1594    // Ensure "epic" label is included
1595    let mut labels = params.labels;
1596    if !labels.iter().any(|l| l.eq_ignore_ascii_case("epic")) {
1597        labels.push("epic".to_string());
1598    }
1599
1600    // Add goal tag if goalId provided (e.g., "G1" → tag "g1")
1601    if let Some(ref goal) = params.goal_id {
1602        let goal_tag = goal.to_lowercase();
1603        if !labels.iter().any(|l| l.to_lowercase() == goal_tag) {
1604            labels.push(goal_tag);
1605        }
1606    }
1607
1608    let input = CreateIssueInput {
1609        title: params.title,
1610        description: params.description,
1611        labels,
1612        assignees: params.assignees,
1613        priority: params.priority,
1614        parent: None,
1615        markdown: params.markdown.unwrap_or(true),
1616        project_id: None,
1617        issue_type: None,
1618        custom_fields: args.get("customFields").cloned(),
1619        components: Vec::new(),
1620        fix_versions: Vec::new(),
1621        epic_key: None,
1622        sprint_id: None,
1623        epic_name: None,
1624    };
1625    let issue = provider.create_issue(input).await?;
1626
1627    // Set custom fields via separate API call (ClickUp uses Array format)
1628    if let Some(cf) = args.get("customFields").and_then(|v| v.as_array())
1629        && !cf.is_empty()
1630        && let Err(e) = provider.set_custom_fields(&issue.key, cf).await
1631    {
1632        tracing::warn!(error = %e, "Failed to set custom fields on created epic");
1633    }
1634
1635    Ok(ToolOutput::SingleIssue(Box::new(issue)))
1636}
1637
1638#[derive(Deserialize)]
1639struct UpdateEpicParams {
1640    #[serde(alias = "epicKey")]
1641    key: String,
1642    title: Option<String>,
1643    description: Option<String>,
1644    state: Option<String>,
1645    /// Provider-specific status name (#288). Same semantics as
1646    /// `UpdateIssueParams::status` — epics are stored as ClickUp tasks,
1647    /// so the same custom-status workflow applies (e.g. "in progress",
1648    /// "review", "complete").
1649    #[serde(default)]
1650    status: Option<String>,
1651    #[serde(rename = "goalId")]
1652    goal_id: Option<String>,
1653    labels: Option<Vec<String>>,
1654    assignees: Option<Vec<String>>,
1655    #[serde(default, deserialize_with = "deserialize_string_or_number")]
1656    priority: Option<String>,
1657    markdown: Option<bool>,
1658}
1659
1660async fn execute_update_epic(
1661    provider: &dyn devboy_core::Provider,
1662    args: &Value,
1663) -> Result<ToolOutput> {
1664    let params: UpdateEpicParams = serde_json::from_value(args.clone())
1665        .map_err(|e| Error::InvalidData(format!("invalid update_epic params: {e}")))?;
1666
1667    // Handle goal tag transition: if goalId is changing, update labels
1668    let labels = if let Some(ref new_goal) = params.goal_id {
1669        // Fetch current issue to get existing labels
1670        let current = provider.get_issue(&params.key).await?;
1671        let mut labels: Vec<String> = current
1672            .labels
1673            .iter()
1674            // Remove old goal tags (g1-g9)
1675            .filter(|l| {
1676                let lower = l.to_lowercase();
1677                !(lower.len() == 2
1678                    && lower.starts_with('g')
1679                    && lower.chars().nth(1).is_some_and(|c| c.is_ascii_digit()))
1680            })
1681            .cloned()
1682            .collect();
1683
1684        // Add new goal tag
1685        let goal_tag = new_goal.to_lowercase();
1686        if !labels.iter().any(|l| l.to_lowercase() == goal_tag) {
1687            labels.push(goal_tag);
1688        }
1689
1690        // Merge with explicitly provided labels
1691        if let Some(extra) = params.labels {
1692            for l in extra {
1693                if !labels
1694                    .iter()
1695                    .any(|existing| existing.eq_ignore_ascii_case(&l))
1696                {
1697                    labels.push(l);
1698                }
1699            }
1700        }
1701        Some(labels)
1702    } else {
1703        params.labels
1704    };
1705
1706    let input = UpdateIssueInput {
1707        title: params.title,
1708        description: params.description,
1709        state: params.state,
1710        status: params.status,
1711        labels,
1712        assignees: params.assignees,
1713        priority: params.priority,
1714        parent_id: None,
1715        markdown: params.markdown.unwrap_or(true),
1716        custom_fields: args.get("customFields").cloned(),
1717        components: None,
1718        fix_versions: None,
1719        epic_key: None,
1720        sprint_id: None,
1721        epic_name: None,
1722    };
1723    let key = params.key;
1724    let issue = provider.update_issue(&key, input).await?;
1725
1726    // Set custom fields via separate API call (ClickUp uses Array format)
1727    if let Some(cf) = args.get("customFields").and_then(|v| v.as_array())
1728        && !cf.is_empty()
1729        && let Err(e) = provider.set_custom_fields(&key, cf).await
1730    {
1731        tracing::warn!(error = %e, "Failed to set custom fields on updated epic");
1732    }
1733
1734    Ok(ToolOutput::SingleIssue(Box::new(issue)))
1735}
1736
1737/// List of all tool names supported by the executor.
1738pub const SUPPORTED_TOOLS: &[&str] = &[
1739    "get_issues",
1740    "get_issue",
1741    "get_issue_comments",
1742    "get_issue_relations",
1743    "create_issue",
1744    "update_issue",
1745    "add_issue_comment",
1746    "get_merge_requests",
1747    "get_merge_request",
1748    "get_merge_request_discussions",
1749    "get_merge_request_diffs",
1750    "create_merge_request",
1751    "create_merge_request_comment",
1752    "update_merge_request",
1753    "get_pipeline",
1754    "get_job_logs",
1755    "get_available_statuses",
1756    "get_users",
1757    "link_issues",
1758    "unlink_issues",
1759    "get_epics",
1760    "create_epic",
1761    "update_epic",
1762    "get_meeting_notes",
1763    "get_meeting_transcript",
1764    "search_meeting_notes",
1765    // Knowledge base tools
1766    "get_knowledge_base_spaces",
1767    "list_knowledge_base_pages",
1768    "get_knowledge_base_page",
1769    "create_knowledge_base_page",
1770    "update_knowledge_base_page",
1771    "search_knowledge_base",
1772    // Messenger tools
1773    "get_messenger_chats",
1774    "get_chat_messages",
1775    "search_chat_messages",
1776    "send_message",
1777    // Asset tools
1778    "get_assets",
1779    "upload_asset",
1780    "download_asset",
1781    "delete_asset",
1782];
1783
1784// =============================================================================
1785// Update Merge Request handler
1786// =============================================================================
1787
1788#[derive(Deserialize)]
1789struct UpdateMergeRequestParams {
1790    key: String,
1791    #[serde(default)]
1792    title: Option<String>,
1793    #[serde(default)]
1794    description: Option<String>,
1795    #[serde(default)]
1796    state: Option<String>,
1797    #[serde(default)]
1798    labels: Option<Vec<String>>,
1799    #[serde(default)]
1800    draft: Option<bool>,
1801}
1802
1803async fn execute_update_merge_request(
1804    provider: &dyn devboy_core::Provider,
1805    args: &Value,
1806) -> Result<ToolOutput> {
1807    let params: UpdateMergeRequestParams = serde_json::from_value(args.clone())?;
1808    debug!(key = %params.key, "update_merge_request");
1809
1810    let input = devboy_core::UpdateMergeRequestInput {
1811        title: params.title,
1812        description: params.description,
1813        state: params.state,
1814        labels: params.labels,
1815        draft: params.draft,
1816    };
1817
1818    let mr = MergeRequestProvider::update_merge_request(provider, &params.key, input).await?;
1819    Ok(ToolOutput::SingleMergeRequest(Box::new(mr)))
1820}
1821
1822// =============================================================================
1823// Asset tool handlers
1824// =============================================================================
1825
1826#[derive(Deserialize)]
1827struct GetAssetsParams {
1828    /// "issue" or "mr"
1829    context_type: String,
1830    /// Issue key (e.g. "DEV-123") or MR key (e.g. "mr#42")
1831    key: String,
1832}
1833
1834async fn execute_get_assets(
1835    provider: &dyn devboy_core::Provider,
1836    args: &Value,
1837) -> Result<ToolOutput> {
1838    let params: GetAssetsParams = serde_json::from_value(args.clone())?;
1839    debug!(context_type = %params.context_type, key = %params.key, "get_assets");
1840
1841    let assets = match params.context_type.as_str() {
1842        "issue" => IssueProvider::get_issue_attachments(provider, &params.key).await?,
1843        "mr" | "merge_request" | "pull_request" => {
1844            MergeRequestProvider::get_mr_attachments(provider, &params.key).await?
1845        }
1846        other => {
1847            return Err(Error::InvalidData(format!(
1848                "unsupported context_type: '{other}', expected 'issue' or 'mr'"
1849            )));
1850        }
1851    };
1852
1853    let capabilities =
1854        serde_json::to_value(IssueProvider::asset_capabilities(provider)).unwrap_or_default();
1855    let count = assets.len();
1856    let attachments: Vec<serde_json::Value> = assets
1857        .into_iter()
1858        .map(|a| serde_json::to_value(a).unwrap_or_default())
1859        .collect();
1860    Ok(ToolOutput::AssetList {
1861        attachments,
1862        count,
1863        capabilities,
1864    })
1865}
1866
1867#[derive(Deserialize)]
1868struct UploadAssetParams {
1869    /// "issue" or "mr"
1870    context_type: String,
1871    key: String,
1872    filename: String,
1873    /// Base64-encoded file data
1874    #[serde(rename = "fileData")]
1875    file_data: String,
1876}
1877
1878async fn execute_upload_asset(
1879    provider: &dyn devboy_core::Provider,
1880    args: &Value,
1881) -> Result<ToolOutput> {
1882    let params: UploadAssetParams = serde_json::from_value(args.clone())?;
1883    debug!(context_type = %params.context_type, key = %params.key, filename = %params.filename, "upload_asset");
1884
1885    let data = base64_decode(&params.file_data)?;
1886
1887    if data.len() > MAX_FILE_SIZE {
1888        return Err(Error::InvalidData(format!(
1889            "file '{}' is {} bytes, max allowed is {} bytes",
1890            params.filename,
1891            data.len(),
1892            MAX_FILE_SIZE,
1893        )));
1894    }
1895
1896    let size = data.len();
1897    let url = match params.context_type.as_str() {
1898        "issue" => {
1899            IssueProvider::upload_attachment(provider, &params.key, &params.filename, &data).await?
1900        }
1901        other => {
1902            return Err(Error::InvalidData(format!(
1903                "upload not supported for context_type: '{other}', use 'issue'"
1904            )));
1905        }
1906    };
1907
1908    Ok(ToolOutput::AssetUploaded {
1909        url,
1910        filename: params.filename,
1911        size,
1912    })
1913}
1914
1915#[derive(Deserialize)]
1916struct DownloadAssetParams {
1917    /// "issue" or "mr"
1918    context_type: String,
1919    key: String,
1920    /// Asset identifier (provider-specific)
1921    asset_id: String,
1922}
1923
1924async fn execute_download_asset(
1925    provider: &dyn devboy_core::Provider,
1926    args: &Value,
1927    asset_manager: Option<&devboy_assets::AssetManager>,
1928) -> Result<ToolOutput> {
1929    let params: DownloadAssetParams = serde_json::from_value(args.clone())?;
1930    debug!(context_type = %params.context_type, key = %params.key, asset_id = %params.asset_id, "download_asset");
1931
1932    // Check local cache first.
1933    if let Some(mgr) = asset_manager
1934        && let Ok(Some(resolved)) = mgr.get(&params.asset_id)
1935    {
1936        return Ok(ToolOutput::AssetDownloaded {
1937            asset_id: params.asset_id,
1938            size: resolved.asset.size as usize,
1939            local_path: Some(resolved.absolute_path.to_string_lossy().into_owned()),
1940            data: None,
1941            cached: true,
1942        });
1943    }
1944
1945    // Not cached — download from provider.
1946    let bytes = match params.context_type.as_str() {
1947        "issue" => {
1948            IssueProvider::download_attachment(provider, &params.key, &params.asset_id).await?
1949        }
1950        "mr" | "merge_request" | "pull_request" => {
1951            MergeRequestProvider::download_mr_attachment(provider, &params.key, &params.asset_id)
1952                .await?
1953        }
1954        other => {
1955            return Err(Error::InvalidData(format!(
1956                "unsupported context_type: '{other}', expected 'issue' or 'mr'"
1957            )));
1958        }
1959    };
1960
1961    // Store in cache if available.
1962    if let Some(mgr) = asset_manager {
1963        let context = match params.context_type.as_str() {
1964            "mr" | "merge_request" | "pull_request" => devboy_core::AssetContext::MergeRequest {
1965                mr_id: params.key.clone(),
1966            },
1967            _ => devboy_core::AssetContext::Issue {
1968                key: params.key.clone(),
1969            },
1970        };
1971        let filename = devboy_core::filename_from_url(&params.asset_id);
1972        match mgr.store(devboy_assets::StoreRequest {
1973            context,
1974            asset_id: Some(&params.asset_id),
1975            filename: &filename,
1976            mime_type: None,
1977            remote_url: None,
1978            data: &bytes,
1979        }) {
1980            Ok(cached) => {
1981                let abs = mgr.cache_dir().join(&cached.local_path);
1982                return Ok(ToolOutput::AssetDownloaded {
1983                    asset_id: cached.id,
1984                    size: cached.size as usize,
1985                    local_path: Some(abs.to_string_lossy().into_owned()),
1986                    data: None,
1987                    cached: true,
1988                });
1989            }
1990            Err(e) => {
1991                tracing::warn!(?e, "failed to cache asset, returning base64 fallback");
1992            }
1993        }
1994    }
1995
1996    // Fallback: return base64-encoded content.
1997    if bytes.len() > MAX_FILE_SIZE {
1998        return Err(Error::InvalidData(format!(
1999            "downloaded attachment is {} bytes, max allowed for base64 response is {} bytes",
2000            bytes.len(),
2001            MAX_FILE_SIZE,
2002        )));
2003    }
2004
2005    let encoded = base64_encode(&bytes);
2006    Ok(ToolOutput::AssetDownloaded {
2007        asset_id: params.asset_id,
2008        size: bytes.len(),
2009        local_path: None,
2010        data: Some(encoded),
2011        cached: false,
2012    })
2013}
2014
2015#[derive(Deserialize)]
2016struct DeleteAssetParams {
2017    key: String,
2018    asset_id: String,
2019}
2020
2021async fn execute_delete_asset(
2022    provider: &dyn devboy_core::Provider,
2023    args: &Value,
2024    asset_manager: Option<&devboy_assets::AssetManager>,
2025) -> Result<ToolOutput> {
2026    let params: DeleteAssetParams = serde_json::from_value(args.clone())?;
2027    debug!(key = %params.key, asset_id = %params.asset_id, "delete_asset");
2028
2029    IssueProvider::delete_attachment(provider, &params.key, &params.asset_id).await?;
2030
2031    // Evict from local cache so stale files aren't served.
2032    if let Some(mgr) = asset_manager
2033        && let Err(e) = mgr.delete(&params.asset_id)
2034    {
2035        tracing::warn!(?e, asset_id = %params.asset_id, "failed to evict deleted asset from cache");
2036    }
2037
2038    let message = format!(
2039        "Attachment '{}' deleted from {}",
2040        params.asset_id, params.key
2041    );
2042    Ok(ToolOutput::AssetDeleted {
2043        asset_id: params.asset_id,
2044        message,
2045    })
2046}
2047
2048/// Maximum base64 encoded length for MAX_FILE_SIZE bytes.
2049const MAX_BASE64_LEN: usize = (MAX_FILE_SIZE / 3 + 1) * 4 + 4;
2050
2051/// Decode base64 with standard or URL-safe alphabet, rejecting
2052/// oversized inputs *before* allocating the decoded buffer.
2053fn base64_decode(input: &str) -> Result<Vec<u8>> {
2054    let trimmed = input.trim();
2055    if trimmed.len() > MAX_BASE64_LEN {
2056        return Err(Error::InvalidData(format!(
2057            "base64 input too large ({} chars), max decoded size is {} bytes",
2058            trimmed.len(),
2059            MAX_FILE_SIZE,
2060        )));
2061    }
2062    use base64::Engine;
2063    base64::engine::general_purpose::STANDARD
2064        .decode(trimmed)
2065        .or_else(|_| base64::engine::general_purpose::URL_SAFE.decode(trimmed))
2066        .map_err(|e| Error::InvalidData(format!("invalid base64: {e}")))
2067}
2068
2069/// Encode bytes as standard base64.
2070fn base64_encode(data: &[u8]) -> String {
2071    use base64::Engine;
2072    base64::engine::general_purpose::STANDARD.encode(data)
2073}
2074
2075// =============================================================================
2076// Jira Structure tool handlers
2077// =============================================================================
2078
2079async fn execute_get_structures(provider: &dyn devboy_core::Provider) -> Result<ToolOutput> {
2080    let result = provider.get_structures().await?;
2081    let meta = ResultMeta {
2082        pagination: result.pagination,
2083        sort_info: result.sort_info,
2084    };
2085    Ok(ToolOutput::Structures(result.items, Some(meta)))
2086}
2087
2088#[derive(Deserialize)]
2089#[serde(rename_all = "camelCase")]
2090struct GetStructureForestParams {
2091    structure_id: u64,
2092    offset: Option<u64>,
2093    limit: Option<u64>,
2094}
2095
2096async fn execute_get_structure_forest(
2097    provider: &dyn devboy_core::Provider,
2098    args: &Value,
2099) -> Result<ToolOutput> {
2100    let params: GetStructureForestParams = serde_json::from_value(args.clone())
2101        .map_err(|e| Error::InvalidData(format!("missing 'structureId': {e}")))?;
2102    let forest = provider
2103        .get_structure_forest(
2104            params.structure_id,
2105            GetForestOptions {
2106                offset: params.offset,
2107                limit: Some(params.limit.unwrap_or(200)),
2108            },
2109        )
2110        .await?;
2111    Ok(ToolOutput::StructureForest(Box::new(forest)))
2112}
2113
2114#[derive(Deserialize)]
2115#[serde(rename_all = "camelCase")]
2116struct AddStructureRowsParams {
2117    structure_id: u64,
2118    items: Vec<Value>,
2119    under: Option<u64>,
2120    after: Option<u64>,
2121    forest_version: Option<u64>,
2122}
2123
2124/// Turn a single `items[]` entry from `add_structure_rows` into a
2125/// `StructureRowItem`. The tool schema can only express a list of
2126/// strings, so callers wanting to set `item_type` or nested fields
2127/// are forced to pass JSON inside a string. Accept both:
2128///
2129/// - bare string → `{ item_id: s, item_type: None }`
2130/// - JSON object (either as a real object or a string that parses as
2131///   one) → `serde_json::from_value`
2132///
2133/// Malformed input surfaces as `InvalidData` rather than silently
2134/// dropping values through `unwrap_or_default()`.
2135fn parse_structure_row_item(v: Value) -> Result<StructureRowItem> {
2136    if let Some(s) = v.as_str() {
2137        if let Ok(parsed) = serde_json::from_str::<Value>(s)
2138            && parsed.is_object()
2139        {
2140            return serde_json::from_value(parsed)
2141                .map_err(|e| Error::InvalidData(format!("invalid structure row item JSON: {e}")));
2142        }
2143        return Ok(StructureRowItem {
2144            item_id: s.to_string(),
2145            item_type: None,
2146        });
2147    }
2148    serde_json::from_value(v)
2149        .map_err(|e| Error::InvalidData(format!("invalid structure row item: {e}")))
2150}
2151
2152/// Turn a `columns[]` entry from `get_structure_values` /
2153/// `save_structure_view` into a `StructureViewColumn`. Same dual
2154/// shape as `parse_structure_row_item`: bare string means
2155/// `{ field: Some(s) }`, anything else (or a JSON-object string) is
2156/// deserialised as a full spec. Errors propagate.
2157fn parse_structure_column_spec(v: Value) -> Result<StructureViewColumn> {
2158    if let Some(s) = v.as_str() {
2159        if let Ok(parsed) = serde_json::from_str::<Value>(s)
2160            && parsed.is_object()
2161        {
2162            return serde_json::from_value(parsed).map_err(|e| {
2163                Error::InvalidData(format!("invalid structure column spec JSON: {e}"))
2164            });
2165        }
2166        return Ok(StructureViewColumn {
2167            field: Some(s.to_string()),
2168            ..Default::default()
2169        });
2170    }
2171    serde_json::from_value(v)
2172        .map_err(|e| Error::InvalidData(format!("invalid structure column spec: {e}")))
2173}
2174
2175async fn execute_add_structure_rows(
2176    provider: &dyn devboy_core::Provider,
2177    args: &Value,
2178) -> Result<ToolOutput> {
2179    let params: AddStructureRowsParams = serde_json::from_value(args.clone())
2180        .map_err(|e| Error::InvalidData(format!("invalid add_structure_rows params: {e}")))?;
2181
2182    let items: Vec<StructureRowItem> = params
2183        .items
2184        .into_iter()
2185        .map(parse_structure_row_item)
2186        .collect::<Result<Vec<_>>>()?;
2187
2188    let result = provider
2189        .add_structure_rows(
2190            params.structure_id,
2191            AddStructureRowsInput {
2192                items,
2193                under: params.under,
2194                after: params.after,
2195                forest_version: params.forest_version,
2196            },
2197        )
2198        .await?;
2199    Ok(ToolOutput::ForestModified(result))
2200}
2201
2202#[derive(Deserialize)]
2203#[serde(rename_all = "camelCase")]
2204struct MoveStructureRowsParams {
2205    structure_id: u64,
2206    row_ids: Vec<u64>,
2207    under: Option<u64>,
2208    after: Option<u64>,
2209    forest_version: Option<u64>,
2210}
2211
2212async fn execute_move_structure_rows(
2213    provider: &dyn devboy_core::Provider,
2214    args: &Value,
2215) -> Result<ToolOutput> {
2216    let params: MoveStructureRowsParams = serde_json::from_value(args.clone())
2217        .map_err(|e| Error::InvalidData(format!("invalid move_structure_rows params: {e}")))?;
2218    let result = provider
2219        .move_structure_rows(
2220            params.structure_id,
2221            MoveStructureRowsInput {
2222                row_ids: params.row_ids,
2223                under: params.under,
2224                after: params.after,
2225                forest_version: params.forest_version,
2226            },
2227        )
2228        .await?;
2229    Ok(ToolOutput::ForestModified(result))
2230}
2231
2232#[derive(Deserialize)]
2233#[serde(rename_all = "camelCase")]
2234struct RemoveStructureRowParams {
2235    structure_id: u64,
2236    row_id: u64,
2237}
2238
2239async fn execute_remove_structure_row(
2240    provider: &dyn devboy_core::Provider,
2241    args: &Value,
2242) -> Result<ToolOutput> {
2243    let params: RemoveStructureRowParams = serde_json::from_value(args.clone())
2244        .map_err(|e| Error::InvalidData(format!("invalid remove_structure_row params: {e}")))?;
2245    provider
2246        .remove_structure_row(params.structure_id, params.row_id)
2247        .await?;
2248    Ok(ToolOutput::Text(format!(
2249        "Row {} removed from structure {}",
2250        params.row_id, params.structure_id
2251    )))
2252}
2253
2254#[derive(Deserialize)]
2255#[serde(rename_all = "camelCase")]
2256struct GetStructureValuesParams {
2257    structure_id: u64,
2258    rows: Vec<u64>,
2259    columns: Vec<Value>,
2260}
2261
2262async fn execute_get_structure_values(
2263    provider: &dyn devboy_core::Provider,
2264    args: &Value,
2265) -> Result<ToolOutput> {
2266    let params: GetStructureValuesParams = serde_json::from_value(args.clone())
2267        .map_err(|e| Error::InvalidData(format!("invalid get_structure_values params: {e}")))?;
2268
2269    let columns: Vec<StructureViewColumn> = params
2270        .columns
2271        .into_iter()
2272        .map(parse_structure_column_spec)
2273        .collect::<Result<Vec<_>>>()?;
2274
2275    let result = provider
2276        .get_structure_values(GetStructureValuesInput {
2277            structure_id: params.structure_id,
2278            rows: params.rows,
2279            columns,
2280        })
2281        .await?;
2282    Ok(ToolOutput::StructureValues(Box::new(result)))
2283}
2284
2285#[derive(Deserialize)]
2286#[serde(rename_all = "camelCase")]
2287struct GetStructureViewsParams {
2288    structure_id: u64,
2289    view_id: Option<u64>,
2290}
2291
2292async fn execute_get_structure_views(
2293    provider: &dyn devboy_core::Provider,
2294    args: &Value,
2295) -> Result<ToolOutput> {
2296    let params: GetStructureViewsParams = serde_json::from_value(args.clone())
2297        .map_err(|e| Error::InvalidData(format!("invalid get_structure_views params: {e}")))?;
2298    let views = provider
2299        .get_structure_views(params.structure_id, params.view_id)
2300        .await?;
2301    Ok(ToolOutput::StructureViews(views, None))
2302}
2303
2304#[derive(Deserialize)]
2305#[serde(rename_all = "camelCase")]
2306struct SaveStructureViewParams {
2307    id: Option<u64>,
2308    structure_id: u64,
2309    name: String,
2310    columns: Option<Vec<Value>>,
2311    group_by: Option<String>,
2312    sort_by: Option<String>,
2313    filter: Option<String>,
2314}
2315
2316async fn execute_save_structure_view(
2317    provider: &dyn devboy_core::Provider,
2318    args: &Value,
2319) -> Result<ToolOutput> {
2320    let params: SaveStructureViewParams = serde_json::from_value(args.clone())
2321        .map_err(|e| Error::InvalidData(format!("invalid save_structure_view params: {e}")))?;
2322
2323    let columns: Option<Vec<StructureViewColumn>> = params
2324        .columns
2325        .map(|cols| {
2326            cols.into_iter()
2327                .map(parse_structure_column_spec)
2328                .collect::<Result<Vec<_>>>()
2329        })
2330        .transpose()?;
2331
2332    let view = provider
2333        .save_structure_view(SaveStructureViewInput {
2334            id: params.id,
2335            structure_id: params.structure_id,
2336            name: params.name,
2337            columns,
2338            group_by: params.group_by,
2339            sort_by: params.sort_by,
2340            filter: params.filter,
2341        })
2342        .await?;
2343    Ok(ToolOutput::StructureViews(vec![view], None))
2344}
2345
2346#[derive(Deserialize)]
2347struct CreateStructureParams {
2348    name: String,
2349    description: Option<String>,
2350}
2351
2352async fn execute_create_structure(
2353    provider: &dyn devboy_core::Provider,
2354    args: &Value,
2355) -> Result<ToolOutput> {
2356    let params: CreateStructureParams = serde_json::from_value(args.clone())
2357        .map_err(|e| Error::InvalidData(format!("missing 'name': {e}")))?;
2358    let structure = provider
2359        .create_structure(CreateStructureInput {
2360            name: params.name,
2361            description: params.description,
2362        })
2363        .await?;
2364    Ok(ToolOutput::Structures(vec![structure], None))
2365}
2366
2367// =============================================================================
2368// Project versions / fixVersion handlers (issue #238)
2369// =============================================================================
2370
2371/// Tri-state filter for `released` / `archived` — accepts the strings
2372/// `"true"`, `"false"`, `"all"` (default `"all"` → no filter).
2373fn parse_tri_filter(s: Option<&str>) -> Result<Option<bool>> {
2374    match s.map(str::trim).map(str::to_ascii_lowercase).as_deref() {
2375        None | Some("") | Some("all") | Some("any") => Ok(None),
2376        Some("true") | Some("yes") | Some("1") => Ok(Some(true)),
2377        Some("false") | Some("no") | Some("0") => Ok(Some(false)),
2378        Some(other) => Err(Error::InvalidData(format!(
2379            "expected 'true' | 'false' | 'all', got '{other}'"
2380        ))),
2381    }
2382}
2383
2384/// Validate that a string is an ISO 8601 calendar date in `YYYY-MM-DD`
2385/// form. Jira accepts that exact shape on `releaseDate`/`startDate`
2386/// payloads; anything else (timestamps, slashes, locale formats) gets
2387/// rejected with a 400 by the server, so catch it client-side with a
2388/// clear error pointing at the offending field.
2389fn validate_iso_date(field: &str, value: &str) -> Result<()> {
2390    let bytes = value.as_bytes();
2391    let shape_ok = bytes.len() == 10
2392        && bytes[4] == b'-'
2393        && bytes[7] == b'-'
2394        && bytes[..4].iter().all(u8::is_ascii_digit)
2395        && bytes[5..7].iter().all(u8::is_ascii_digit)
2396        && bytes[8..].iter().all(u8::is_ascii_digit);
2397    if !shape_ok {
2398        return Err(Error::InvalidData(format!(
2399            "{field} must be an ISO 8601 calendar date (YYYY-MM-DD), got '{value}'"
2400        )));
2401    }
2402    let month: u32 = value[5..7].parse().unwrap();
2403    let day: u32 = value[8..10].parse().unwrap();
2404    if !(1..=12).contains(&month) || !(1..=31).contains(&day) {
2405        return Err(Error::InvalidData(format!(
2406            "{field} = '{value}' is not a valid calendar date"
2407        )));
2408    }
2409    Ok(())
2410}
2411
2412#[derive(Deserialize, Default)]
2413#[serde(rename_all = "camelCase")]
2414struct ListProjectVersionsArgs {
2415    project: Option<String>,
2416    released: Option<String>,
2417    archived: Option<String>,
2418    limit: Option<u32>,
2419    include_issue_count: Option<bool>,
2420}
2421
2422async fn execute_list_project_versions(
2423    provider: &dyn devboy_core::Provider,
2424    args: &Value,
2425) -> Result<ToolOutput> {
2426    let params: ListProjectVersionsArgs = parse_tool_params(args, "list_project_versions")?;
2427
2428    // Paper 1 / TrimTree defaults: hide archived noise + cap at 20 most
2429    // recent. Defaults only apply when the caller omits the field —
2430    // explicit `"all"` must round-trip as `None` (no filter).
2431    let archived = match params.archived.as_deref() {
2432        None => Some(false),
2433        Some(s) => parse_tri_filter(Some(s))?,
2434    };
2435    let released = match params.released.as_deref() {
2436        None => None,
2437        Some(s) => parse_tri_filter(Some(s))?,
2438    };
2439    // `limit: 0` would round-trip to a useless empty list — reject up
2440    // front instead of letting the schema's `min: 1` get bypassed by a
2441    // raw call site (Codex review on PR #239).
2442    if let Some(0) = params.limit {
2443        return Err(Error::InvalidData(
2444            "limit must be at least 1 (use the default by omitting the field)".into(),
2445        ));
2446    }
2447    let limit = params.limit.unwrap_or(20).min(200);
2448
2449    let result = provider
2450        .list_project_versions(ListProjectVersionsParams {
2451            project: params.project.unwrap_or_default(),
2452            released,
2453            archived,
2454            limit: Some(limit),
2455            include_issue_count: params.include_issue_count.unwrap_or(false),
2456        })
2457        .await?;
2458
2459    let meta = ResultMeta {
2460        pagination: result.pagination,
2461        sort_info: result.sort_info,
2462    };
2463    Ok(ToolOutput::ProjectVersions(result.items, Some(meta)))
2464}
2465
2466#[derive(Deserialize)]
2467#[serde(rename_all = "camelCase")]
2468struct UpsertProjectVersionArgs {
2469    project: Option<String>,
2470    name: String,
2471    description: Option<String>,
2472    start_date: Option<String>,
2473    release_date: Option<String>,
2474    released: Option<bool>,
2475    archived: Option<bool>,
2476}
2477
2478async fn execute_upsert_project_version(
2479    provider: &dyn devboy_core::Provider,
2480    args: &Value,
2481) -> Result<ToolOutput> {
2482    let params: UpsertProjectVersionArgs = serde_json::from_value(args.clone())
2483        .map_err(|e| Error::InvalidData(format!("invalid upsert_project_version params: {e}")))?;
2484
2485    // Codex review on PR #239 — validate inputs before they cross the
2486    // wire so the failure points at the parameter, not at "Jira said 400".
2487    if let Some(ref d) = params.start_date {
2488        validate_iso_date("startDate", d)?;
2489    }
2490    if let Some(ref d) = params.release_date {
2491        validate_iso_date("releaseDate", d)?;
2492    }
2493
2494    let version = provider
2495        .upsert_project_version(UpsertProjectVersionInput {
2496            project: params.project.unwrap_or_default(),
2497            name: params.name,
2498            description: params.description,
2499            start_date: params.start_date,
2500            release_date: params.release_date,
2501            released: params.released,
2502            archived: params.archived,
2503        })
2504        .await?;
2505
2506    Ok(ToolOutput::SingleProjectVersion(Box::new(version)))
2507}
2508
2509#[derive(Deserialize, Default)]
2510#[serde(rename_all = "camelCase")]
2511struct GetBoardSprintsArgs {
2512    board_id: u64,
2513    /// Optional state filter: `active`, `future`, `closed`, or `all`
2514    /// (default `all`).
2515    state: Option<String>,
2516}
2517
2518async fn execute_get_board_sprints(
2519    provider: &dyn devboy_core::Provider,
2520    args: &Value,
2521) -> Result<ToolOutput> {
2522    let params: GetBoardSprintsArgs = parse_tool_params(args, "get_board_sprints")?;
2523    let state = match params.state.as_deref() {
2524        None | Some("all") => SprintState::All,
2525        Some("active") => SprintState::Active,
2526        Some("future") => SprintState::Future,
2527        Some("closed") => SprintState::Closed,
2528        Some(other) => {
2529            return Err(Error::InvalidData(format!(
2530                "invalid sprint state `{other}` — expected one of: active, future, closed, all"
2531            )));
2532        }
2533    };
2534
2535    let result = provider.get_board_sprints(params.board_id, state).await?;
2536    let meta = ResultMeta {
2537        pagination: result.pagination,
2538        sort_info: result.sort_info,
2539    };
2540    Ok(ToolOutput::Sprints(result.items, Some(meta)))
2541}
2542
2543#[derive(Deserialize, Default)]
2544#[serde(rename_all = "camelCase")]
2545struct AssignToSprintArgs {
2546    sprint_id: u64,
2547    issue_keys: Vec<String>,
2548}
2549
2550async fn execute_assign_to_sprint(
2551    provider: &dyn devboy_core::Provider,
2552    args: &Value,
2553) -> Result<ToolOutput> {
2554    let params: AssignToSprintArgs = parse_tool_params(args, "assign_to_sprint")?;
2555    if params.issue_keys.is_empty() {
2556        return Err(Error::InvalidData(
2557            "issueKeys must contain at least one issue key".into(),
2558        ));
2559    }
2560    let count = params.issue_keys.len();
2561    provider
2562        .assign_to_sprint(AssignToSprintInput {
2563            sprint_id: params.sprint_id,
2564            issue_keys: params.issue_keys,
2565        })
2566        .await?;
2567    Ok(ToolOutput::Text(format!(
2568        "Moved {count} issue(s) to sprint {}.",
2569        params.sprint_id
2570    )))
2571}
2572
2573#[derive(Deserialize, Default)]
2574#[serde(rename_all = "camelCase")]
2575struct GetCustomFieldsArgs {
2576    project: Option<String>,
2577    issue_type: Option<String>,
2578    search: Option<String>,
2579    limit: Option<u32>,
2580}
2581
2582async fn execute_get_custom_fields(
2583    provider: &dyn devboy_core::Provider,
2584    args: &Value,
2585) -> Result<ToolOutput> {
2586    let params: GetCustomFieldsArgs = parse_tool_params(args, "get_custom_fields")?;
2587    if let Some(0) = params.limit {
2588        return Err(Error::InvalidData(
2589            "limit must be at least 1 (use the default by omitting the field)".into(),
2590        ));
2591    }
2592    let result = provider
2593        .list_custom_fields(ListCustomFieldsParams {
2594            project: params.project,
2595            issue_type: params.issue_type,
2596            search: params.search,
2597            limit: params.limit,
2598        })
2599        .await?;
2600    let meta = ResultMeta {
2601        pagination: result.pagination,
2602        sort_info: result.sort_info,
2603    };
2604    Ok(ToolOutput::CustomFields(result.items, Some(meta)))
2605}
2606
2607#[cfg(test)]
2608mod tests {
2609    use super::*;
2610    use async_trait::async_trait;
2611    use devboy_core::{
2612        Comment, CreateMergeRequestInput, Discussion, FileDiff, Issue, IssueLink, IssueProvider,
2613        IssueRelations, KbPage, KbPageContent, KbSpace, KnowledgeBaseProvider, MergeRequest,
2614        MergeRequestProvider, Provider, User,
2615    };
2616
2617    // --- Mock Provider ---
2618
2619    struct MockProvider;
2620
2621    fn sample_issue() -> Issue {
2622        Issue {
2623            key: "gh#1".into(),
2624            title: "Test Issue".into(),
2625            description: Some("Body".into()),
2626            state: "open".into(),
2627            source: "mock".into(),
2628            priority: None,
2629            labels: vec!["bug".into()],
2630            author: None,
2631            assignees: vec![],
2632            url: Some("https://example.com/1".into()),
2633            created_at: Some("2024-01-01T00:00:00Z".into()),
2634            updated_at: Some("2024-01-02T00:00:00Z".into()),
2635            attachments_count: None,
2636            parent: None,
2637            subtasks: vec![],
2638            custom_fields: std::collections::HashMap::new(),
2639        }
2640    }
2641
2642    fn sample_mr() -> MergeRequest {
2643        MergeRequest {
2644            key: "pr#1".into(),
2645            title: "Test PR".into(),
2646            description: Some("PR body".into()),
2647            state: "open".into(),
2648            source: "mock".into(),
2649            source_branch: "feature".into(),
2650            target_branch: "main".into(),
2651            author: None,
2652            assignees: vec![],
2653            reviewers: vec![],
2654            labels: vec![],
2655            draft: false,
2656            url: Some("https://example.com/pr/1".into()),
2657            created_at: Some("2024-01-01T00:00:00Z".into()),
2658            updated_at: Some("2024-01-02T00:00:00Z".into()),
2659        }
2660    }
2661
2662    fn sample_comment() -> Comment {
2663        Comment {
2664            id: "c1".into(),
2665            body: "Test comment".into(),
2666            author: None,
2667            created_at: None,
2668            updated_at: None,
2669            position: None,
2670        }
2671    }
2672
2673    fn sample_discussion() -> Discussion {
2674        Discussion {
2675            id: "d1".into(),
2676            resolved: false,
2677            resolved_by: None,
2678            comments: vec![sample_comment()],
2679            position: None,
2680        }
2681    }
2682
2683    fn sample_diff() -> FileDiff {
2684        FileDiff {
2685            file_path: "src/main.rs".into(),
2686            old_path: None,
2687            new_file: false,
2688            deleted_file: false,
2689            renamed_file: false,
2690            diff: "+added\n-removed".into(),
2691            additions: Some(1),
2692            deletions: Some(1),
2693        }
2694    }
2695
2696    fn sample_kb_space() -> KbSpace {
2697        KbSpace {
2698            id: "space-1".into(),
2699            key: "ENG".into(),
2700            name: "Engineering".into(),
2701            ..Default::default()
2702        }
2703    }
2704
2705    fn sample_kb_page() -> KbPage {
2706        KbPage {
2707            id: "page-1".into(),
2708            title: "Architecture".into(),
2709            space_key: Some("ENG".into()),
2710            ..Default::default()
2711        }
2712    }
2713
2714    fn sample_kb_page_content() -> KbPageContent {
2715        KbPageContent {
2716            page: sample_kb_page(),
2717            content: "<p>body</p>".into(),
2718            content_type: "storage".into(),
2719            ancestors: vec![],
2720            labels: vec!["docs".into()],
2721        }
2722    }
2723
2724    #[async_trait]
2725    impl IssueProvider for MockProvider {
2726        async fn get_issues(
2727            &self,
2728            _filter: IssueFilter,
2729        ) -> devboy_core::Result<devboy_core::ProviderResult<Issue>> {
2730            Ok(vec![sample_issue()].into())
2731        }
2732        async fn get_issue(&self, _key: &str) -> devboy_core::Result<Issue> {
2733            Ok(sample_issue())
2734        }
2735        async fn create_issue(
2736            &self,
2737            _input: devboy_core::CreateIssueInput,
2738        ) -> devboy_core::Result<Issue> {
2739            Ok(sample_issue())
2740        }
2741        async fn update_issue(
2742            &self,
2743            _key: &str,
2744            _input: devboy_core::UpdateIssueInput,
2745        ) -> devboy_core::Result<Issue> {
2746            Ok(sample_issue())
2747        }
2748        async fn get_comments(
2749            &self,
2750            _key: &str,
2751        ) -> devboy_core::Result<devboy_core::ProviderResult<Comment>> {
2752            Ok(vec![sample_comment()].into())
2753        }
2754        async fn add_comment(&self, _key: &str, _body: &str) -> devboy_core::Result<Comment> {
2755            Ok(sample_comment())
2756        }
2757        async fn get_issue_relations(&self, _key: &str) -> devboy_core::Result<IssueRelations> {
2758            Ok(IssueRelations {
2759                parent: Some(sample_issue()),
2760                subtasks: vec![sample_issue()],
2761                blocks: vec![IssueLink {
2762                    issue: sample_issue(),
2763                    link_type: "Blocks".into(),
2764                }],
2765                ..Default::default()
2766            })
2767        }
2768        async fn get_structures(
2769            &self,
2770        ) -> devboy_core::Result<devboy_core::ProviderResult<devboy_core::Structure>> {
2771            Ok(vec![sample_structure()].into())
2772        }
2773        async fn get_structure_forest(
2774            &self,
2775            structure_id: u64,
2776            _options: devboy_core::GetForestOptions,
2777        ) -> devboy_core::Result<devboy_core::StructureForest> {
2778            Ok(sample_forest(structure_id))
2779        }
2780        async fn add_structure_rows(
2781            &self,
2782            _structure_id: u64,
2783            input: devboy_core::AddStructureRowsInput,
2784        ) -> devboy_core::Result<devboy_core::ForestModifyResult> {
2785            Ok(devboy_core::ForestModifyResult {
2786                version: 2,
2787                affected_count: input.items.len(),
2788            })
2789        }
2790        async fn move_structure_rows(
2791            &self,
2792            _structure_id: u64,
2793            input: devboy_core::MoveStructureRowsInput,
2794        ) -> devboy_core::Result<devboy_core::ForestModifyResult> {
2795            Ok(devboy_core::ForestModifyResult {
2796                version: 3,
2797                affected_count: input.row_ids.len(),
2798            })
2799        }
2800        async fn remove_structure_row(
2801            &self,
2802            _structure_id: u64,
2803            _row_id: u64,
2804        ) -> devboy_core::Result<()> {
2805            Ok(())
2806        }
2807        async fn get_structure_values(
2808            &self,
2809            input: devboy_core::GetStructureValuesInput,
2810        ) -> devboy_core::Result<devboy_core::StructureValues> {
2811            Ok(devboy_core::StructureValues {
2812                structure_id: input.structure_id,
2813                values: vec![],
2814            })
2815        }
2816        async fn get_structure_views(
2817            &self,
2818            structure_id: u64,
2819            _view_id: Option<u64>,
2820        ) -> devboy_core::Result<Vec<devboy_core::StructureView>> {
2821            Ok(vec![sample_view(structure_id)])
2822        }
2823        async fn save_structure_view(
2824            &self,
2825            input: devboy_core::SaveStructureViewInput,
2826        ) -> devboy_core::Result<devboy_core::StructureView> {
2827            Ok(devboy_core::StructureView {
2828                id: input.id.unwrap_or(99),
2829                name: input.name,
2830                structure_id: input.structure_id,
2831                ..Default::default()
2832            })
2833        }
2834        async fn create_structure(
2835            &self,
2836            input: devboy_core::CreateStructureInput,
2837        ) -> devboy_core::Result<devboy_core::Structure> {
2838            Ok(devboy_core::Structure {
2839                id: 42,
2840                name: input.name,
2841                description: input.description,
2842            })
2843        }
2844        async fn list_project_versions(
2845            &self,
2846            params: devboy_core::ListProjectVersionsParams,
2847        ) -> devboy_core::Result<devboy_core::ProviderResult<devboy_core::ProjectVersion>> {
2848            // Echo applied filters back through the data so dispatch
2849            // tests can pin behaviour without sniffing call args.
2850            let mut name = format!(
2851                "v-released={:?}-archived={:?}-limit={:?}-expand={}",
2852                params.released, params.archived, params.limit, params.include_issue_count
2853            );
2854            if !params.project.is_empty() {
2855                name.push_str(&format!("-project={}", params.project));
2856            }
2857            Ok(vec![devboy_core::ProjectVersion {
2858                id: "1".into(),
2859                project: if params.project.is_empty() {
2860                    "MOCK".into()
2861                } else {
2862                    params.project
2863                },
2864                name,
2865                description: Some("desc".into()),
2866                start_date: None,
2867                release_date: Some("2026-01-01".into()),
2868                released: false,
2869                archived: false,
2870                overdue: None,
2871                issue_count: Some(0),
2872                unresolved_issue_count: None,
2873                source: "mock".into(),
2874            }]
2875            .into())
2876        }
2877        async fn upsert_project_version(
2878            &self,
2879            input: devboy_core::UpsertProjectVersionInput,
2880        ) -> devboy_core::Result<devboy_core::ProjectVersion> {
2881            Ok(devboy_core::ProjectVersion {
2882                id: "777".into(),
2883                project: if input.project.is_empty() {
2884                    "MOCK".into()
2885                } else {
2886                    input.project
2887                },
2888                name: input.name,
2889                description: input.description,
2890                start_date: input.start_date,
2891                release_date: input.release_date,
2892                released: input.released.unwrap_or(false),
2893                archived: input.archived.unwrap_or(false),
2894                overdue: None,
2895                issue_count: None,
2896                unresolved_issue_count: None,
2897                source: "mock".into(),
2898            })
2899        }
2900        async fn get_board_sprints(
2901            &self,
2902            board_id: u64,
2903            state: devboy_core::SprintState,
2904        ) -> devboy_core::Result<devboy_core::ProviderResult<devboy_core::Sprint>> {
2905            // Echo applied filters into the name so dispatch tests can
2906            // pin behaviour without sniffing call args.
2907            Ok(vec![devboy_core::Sprint {
2908                id: 1,
2909                name: format!("sprint-board={board_id}-state={state:?}"),
2910                state: "active".into(),
2911                origin_board_id: Some(board_id),
2912                start_date: None,
2913                end_date: None,
2914                goal: None,
2915            }]
2916            .into())
2917        }
2918        async fn assign_to_sprint(
2919            &self,
2920            _input: devboy_core::AssignToSprintInput,
2921        ) -> devboy_core::Result<()> {
2922            Ok(())
2923        }
2924        async fn list_custom_fields(
2925            &self,
2926            params: devboy_core::ListCustomFieldsParams,
2927        ) -> devboy_core::Result<devboy_core::ProviderResult<devboy_core::CustomFieldDescriptor>>
2928        {
2929            // Return a few fixed entries so dispatch tests can pin
2930            // filter and limit behaviour.
2931            let mut all = vec![
2932                devboy_core::CustomFieldDescriptor {
2933                    id: "customfield_10014".into(),
2934                    name: "Epic Link".into(),
2935                    field_type: "any".into(),
2936                    description: None,
2937                    native: None,
2938                },
2939                devboy_core::CustomFieldDescriptor {
2940                    id: "customfield_10011".into(),
2941                    name: "Epic Name".into(),
2942                    field_type: "string".into(),
2943                    description: None,
2944                    native: None,
2945                },
2946                devboy_core::CustomFieldDescriptor {
2947                    id: "customfield_10020".into(),
2948                    name: "Sprint".into(),
2949                    field_type: "array".into(),
2950                    description: None,
2951                    native: None,
2952                },
2953            ];
2954            if let Some(needle) = params.search.as_deref().map(str::to_lowercase) {
2955                all.retain(|f| f.name.to_lowercase().contains(&needle));
2956            }
2957            let total = all.len() as u32;
2958            let limit = params.limit.unwrap_or(50);
2959            if (limit as usize) < all.len() {
2960                all.truncate(limit as usize);
2961            }
2962            let pagination = devboy_core::Pagination {
2963                offset: 0,
2964                limit,
2965                total: Some(total),
2966                has_more: (all.len() as u32) < total,
2967                next_cursor: None,
2968            };
2969            Ok(devboy_core::ProviderResult::new(all).with_pagination(pagination))
2970        }
2971        fn provider_name(&self) -> &'static str {
2972            "mock"
2973        }
2974    }
2975
2976    #[async_trait]
2977    impl MergeRequestProvider for MockProvider {
2978        async fn get_merge_requests(
2979            &self,
2980            _filter: MrFilter,
2981        ) -> devboy_core::Result<devboy_core::ProviderResult<MergeRequest>> {
2982            Ok(vec![sample_mr()].into())
2983        }
2984        async fn get_merge_request(&self, _key: &str) -> devboy_core::Result<MergeRequest> {
2985            Ok(sample_mr())
2986        }
2987        async fn get_discussions(
2988            &self,
2989            _key: &str,
2990        ) -> devboy_core::Result<devboy_core::ProviderResult<Discussion>> {
2991            Ok(vec![sample_discussion()].into())
2992        }
2993        async fn get_diffs(
2994            &self,
2995            _key: &str,
2996        ) -> devboy_core::Result<devboy_core::ProviderResult<FileDiff>> {
2997            Ok(vec![sample_diff()].into())
2998        }
2999        async fn add_comment(
3000            &self,
3001            _key: &str,
3002            _input: CreateCommentInput,
3003        ) -> devboy_core::Result<Comment> {
3004            Ok(sample_comment())
3005        }
3006        async fn create_merge_request(
3007            &self,
3008            _input: CreateMergeRequestInput,
3009        ) -> devboy_core::Result<MergeRequest> {
3010            Ok(sample_mr())
3011        }
3012        fn provider_name(&self) -> &'static str {
3013            "mock"
3014        }
3015    }
3016
3017    #[async_trait]
3018    impl devboy_core::PipelineProvider for MockProvider {
3019        fn provider_name(&self) -> &'static str {
3020            "mock"
3021        }
3022    }
3023
3024    #[async_trait]
3025    impl KnowledgeBaseProvider for MockProvider {
3026        fn provider_name(&self) -> &'static str {
3027            "mock"
3028        }
3029
3030        async fn get_spaces(&self) -> devboy_core::Result<devboy_core::ProviderResult<KbSpace>> {
3031            Ok(vec![sample_kb_space()].into())
3032        }
3033
3034        async fn list_pages(
3035            &self,
3036            _params: ListPagesParams,
3037        ) -> devboy_core::Result<devboy_core::ProviderResult<KbPage>> {
3038            Ok(vec![sample_kb_page()].into())
3039        }
3040
3041        async fn get_page(&self, _page_id: &str) -> devboy_core::Result<KbPageContent> {
3042            Ok(sample_kb_page_content())
3043        }
3044
3045        async fn create_page(
3046            &self,
3047            _params: devboy_core::CreatePageParams,
3048        ) -> devboy_core::Result<KbPage> {
3049            Ok(sample_kb_page())
3050        }
3051
3052        async fn update_page(
3053            &self,
3054            _params: devboy_core::UpdatePageParams,
3055        ) -> devboy_core::Result<KbPage> {
3056            Ok(sample_kb_page())
3057        }
3058
3059        async fn search(
3060            &self,
3061            _params: SearchKbParams,
3062        ) -> devboy_core::Result<devboy_core::ProviderResult<KbPage>> {
3063            Ok(vec![sample_kb_page()].into())
3064        }
3065    }
3066
3067    #[async_trait]
3068    impl Provider for MockProvider {
3069        async fn get_current_user(&self) -> devboy_core::Result<User> {
3070            Ok(User {
3071                id: "1".into(),
3072                username: "test".into(),
3073                name: None,
3074                email: None,
3075                avatar_url: None,
3076            })
3077        }
3078    }
3079
3080    // --- Tests ---
3081
3082    #[test]
3083    fn test_executor_new() {
3084        let executor = Executor::new();
3085        assert!(executor.enrichers.is_empty());
3086    }
3087
3088    #[test]
3089    fn test_supported_tools_contains_all() {
3090        assert!(SUPPORTED_TOOLS.contains(&"get_issues"));
3091        assert!(SUPPORTED_TOOLS.contains(&"get_merge_requests"));
3092        assert!(SUPPORTED_TOOLS.contains(&"create_merge_request_comment"));
3093        assert!(SUPPORTED_TOOLS.contains(&"get_meeting_notes"));
3094        assert!(SUPPORTED_TOOLS.contains(&"get_meeting_transcript"));
3095        assert!(SUPPORTED_TOOLS.contains(&"search_meeting_notes"));
3096        assert!(SUPPORTED_TOOLS.contains(&"get_knowledge_base_spaces"));
3097        assert!(SUPPORTED_TOOLS.contains(&"list_knowledge_base_pages"));
3098        assert!(SUPPORTED_TOOLS.contains(&"get_knowledge_base_page"));
3099        assert!(SUPPORTED_TOOLS.contains(&"create_knowledge_base_page"));
3100        assert!(SUPPORTED_TOOLS.contains(&"update_knowledge_base_page"));
3101        assert!(SUPPORTED_TOOLS.contains(&"search_knowledge_base"));
3102        assert!(SUPPORTED_TOOLS.contains(&"get_messenger_chats"));
3103        assert!(SUPPORTED_TOOLS.contains(&"get_chat_messages"));
3104        assert!(SUPPORTED_TOOLS.contains(&"search_chat_messages"));
3105        assert!(SUPPORTED_TOOLS.contains(&"send_message"));
3106        assert_eq!(SUPPORTED_TOOLS.len(), 40);
3107    }
3108
3109    #[tokio::test]
3110    async fn test_dispatch_get_knowledge_base_spaces() {
3111        let provider = MockProvider;
3112        let result =
3113            dispatch_knowledge_base_tool("get_knowledge_base_spaces", &Value::Null, &provider)
3114                .await
3115                .unwrap();
3116        assert!(matches!(result, ToolOutput::KnowledgeBaseSpaces(v, _) if v.len() == 1));
3117    }
3118
3119    #[tokio::test]
3120    async fn test_dispatch_list_knowledge_base_pages() {
3121        let provider = MockProvider;
3122        let args = serde_json::json!({"spaceKey": "ENG", "limit": 10});
3123        let result = dispatch_knowledge_base_tool("list_knowledge_base_pages", &args, &provider)
3124            .await
3125            .unwrap();
3126        assert!(matches!(result, ToolOutput::KnowledgeBasePages(v, _) if v.len() == 1));
3127    }
3128
3129    #[tokio::test]
3130    async fn test_dispatch_get_knowledge_base_page() {
3131        let provider = MockProvider;
3132        let args = serde_json::json!({"pageId": "page-1"});
3133        let result = dispatch_knowledge_base_tool("get_knowledge_base_page", &args, &provider)
3134            .await
3135            .unwrap();
3136        assert!(matches!(result, ToolOutput::KnowledgeBasePage(_)));
3137    }
3138
3139    #[tokio::test]
3140    async fn test_dispatch_create_knowledge_base_page() {
3141        let provider = MockProvider;
3142        let args = serde_json::json!({
3143            "spaceKey": "ENG",
3144            "title": "New Page",
3145            "content": "<p>body</p>",
3146            "contentType": "storage",
3147            "labels": ["docs"]
3148        });
3149        let result = dispatch_knowledge_base_tool("create_knowledge_base_page", &args, &provider)
3150            .await
3151            .unwrap();
3152        assert!(matches!(result, ToolOutput::KnowledgeBasePageSummary(_)));
3153    }
3154
3155    #[tokio::test]
3156    async fn test_dispatch_update_knowledge_base_page() {
3157        let provider = MockProvider;
3158        let args = serde_json::json!({
3159            "pageId": "page-1",
3160            "title": "Updated",
3161            "content": "<p>new body</p>",
3162            "version": 2
3163        });
3164        let result = dispatch_knowledge_base_tool("update_knowledge_base_page", &args, &provider)
3165            .await
3166            .unwrap();
3167        assert!(matches!(result, ToolOutput::KnowledgeBasePageSummary(_)));
3168    }
3169
3170    #[tokio::test]
3171    async fn test_dispatch_search_knowledge_base() {
3172        let provider = MockProvider;
3173        let args = serde_json::json!({"query": "architecture", "spaceKey": "ENG"});
3174        let result = dispatch_knowledge_base_tool("search_knowledge_base", &args, &provider)
3175            .await
3176            .unwrap();
3177        assert!(matches!(result, ToolOutput::KnowledgeBasePages(v, _) if v.len() == 1));
3178    }
3179
3180    // --- Issue tool dispatch tests ---
3181
3182    #[tokio::test]
3183    async fn test_dispatch_get_issues() {
3184        let provider = MockProvider;
3185        let args = serde_json::json!({"state": "open", "limit": 10});
3186        let result = dispatch_tool("get_issues", &args, &provider, None)
3187            .await
3188            .unwrap();
3189        assert!(matches!(result, ToolOutput::Issues(v, _) if v.len() == 1));
3190    }
3191
3192    #[tokio::test]
3193    async fn test_dispatch_get_issues_empty_args() {
3194        let provider = MockProvider;
3195        let result = dispatch_tool("get_issues", &Value::Null, &provider, None)
3196            .await
3197            .unwrap();
3198        assert!(matches!(result, ToolOutput::Issues(_, _)));
3199    }
3200
3201    #[tokio::test]
3202    async fn test_dispatch_get_issues_invalid_params_are_rejected() {
3203        // Regression for #188: before parse_tool_params the executor
3204        // silently accepted `{"state": 42}` by falling through to
3205        // default() and ran the tool without any filter. Now it must
3206        // surface the deserialisation error instead.
3207        let provider = MockProvider;
3208        let args = serde_json::json!({"state": 42});
3209        let err = dispatch_tool("get_issues", &args, &provider, None)
3210            .await
3211            .unwrap_err();
3212        assert!(
3213            matches!(err, devboy_core::Error::InvalidData(ref msg) if msg.contains("get_issues")),
3214            "expected InvalidData referencing get_issues, got {err:?}"
3215        );
3216    }
3217
3218    #[tokio::test]
3219    async fn test_dispatch_get_merge_requests_invalid_params_rejected() {
3220        let provider = MockProvider;
3221        let args = serde_json::json!({"limit": "not-a-number"});
3222        let err = dispatch_tool("get_merge_requests", &args, &provider, None)
3223            .await
3224            .unwrap_err();
3225        assert!(
3226            matches!(err, devboy_core::Error::InvalidData(ref msg) if msg.contains("get_merge_requests")),
3227            "expected InvalidData referencing get_merge_requests, got {err:?}"
3228        );
3229    }
3230
3231    #[tokio::test]
3232    async fn test_dispatch_get_pipeline_invalid_params_rejected() {
3233        let provider = MockProvider;
3234        let args = serde_json::json!({"includeFailedLogs": "yes"});
3235        let err = dispatch_tool("get_pipeline", &args, &provider, None)
3236            .await
3237            .unwrap_err();
3238        assert!(
3239            matches!(err, devboy_core::Error::InvalidData(ref msg) if msg.contains("get_pipeline")),
3240            "expected InvalidData referencing get_pipeline, got {err:?}"
3241        );
3242    }
3243
3244    #[test]
3245    fn parse_tool_params_null_yields_default() {
3246        #[derive(Debug, Default, serde::Deserialize)]
3247        struct P {
3248            #[allow(dead_code)]
3249            x: Option<String>,
3250        }
3251        let _: P = parse_tool_params(&Value::Null, "test").expect("null → default");
3252    }
3253
3254    #[test]
3255    fn parse_tool_params_empty_object_yields_default() {
3256        // MCP clients often send `{}` for "no arguments"; the helper
3257        // must accept it alongside `null`.
3258        #[derive(Debug, Default, serde::Deserialize)]
3259        struct P {
3260            #[allow(dead_code)]
3261            x: Option<String>,
3262        }
3263        let _: P = parse_tool_params(&serde_json::json!({}), "test").expect("{} → default");
3264    }
3265
3266    #[test]
3267    fn parse_tool_params_invalid_maps_to_invalid_data() {
3268        #[derive(Debug, Default, serde::Deserialize)]
3269        struct P {
3270            #[allow(dead_code)]
3271            n: u32,
3272        }
3273        let err = parse_tool_params::<P>(&serde_json::json!({"n": "nope"}), "tool-x").unwrap_err();
3274        assert!(
3275            matches!(err, devboy_core::Error::InvalidData(ref msg) if msg.contains("tool-x")),
3276            "expected InvalidData(tool-x), got {err:?}"
3277        );
3278    }
3279
3280    #[tokio::test]
3281    async fn test_dispatch_get_issue() {
3282        let provider = MockProvider;
3283        // With includeComments/includeRelations defaulting to true, returns composite Text
3284        let args = serde_json::json!({"key": "gh#1"});
3285        let result = dispatch_tool("get_issue", &args, &provider, None)
3286            .await
3287            .unwrap();
3288        assert!(matches!(result, ToolOutput::Text(_)));
3289
3290        // Without extras, returns SingleIssue
3291        let args =
3292            serde_json::json!({"key": "gh#1", "includeComments": false, "includeRelations": false});
3293        let result = dispatch_tool("get_issue", &args, &provider, None)
3294            .await
3295            .unwrap();
3296        assert!(matches!(result, ToolOutput::SingleIssue(_)));
3297    }
3298
3299    #[tokio::test]
3300    async fn test_dispatch_get_issue_missing_key() {
3301        let provider = MockProvider;
3302        let result = dispatch_tool("get_issue", &serde_json::json!({}), &provider, None).await;
3303        assert!(result.is_err());
3304    }
3305
3306    #[tokio::test]
3307    async fn test_dispatch_get_issue_comments() {
3308        let provider = MockProvider;
3309        let args = serde_json::json!({"key": "gh#1"});
3310        let result = dispatch_tool("get_issue_comments", &args, &provider, None)
3311            .await
3312            .unwrap();
3313        assert!(matches!(result, ToolOutput::Comments(v, _) if v.len() == 1));
3314    }
3315
3316    #[tokio::test]
3317    async fn test_dispatch_create_issue() {
3318        let provider = MockProvider;
3319        let args =
3320            serde_json::json!({"title": "New issue", "description": "Body", "labels": ["bug"]});
3321        let result = dispatch_tool("create_issue", &args, &provider, None)
3322            .await
3323            .unwrap();
3324        assert!(matches!(result, ToolOutput::SingleIssue(_)));
3325    }
3326
3327    #[test]
3328    fn create_issue_params_accepts_parent_id_alias() {
3329        let args = serde_json::json!({ "title": "t", "parentId": "DEV-799" });
3330        let params: CreateIssueParams = serde_json::from_value(args).unwrap();
3331        assert_eq!(params.parent.as_deref(), Some("DEV-799"));
3332    }
3333
3334    #[test]
3335    fn create_issue_params_still_accepts_parent() {
3336        let args = serde_json::json!({ "title": "t", "parent": "DEV-799" });
3337        let params: CreateIssueParams = serde_json::from_value(args).unwrap();
3338        assert_eq!(params.parent.as_deref(), Some("DEV-799"));
3339    }
3340
3341    #[tokio::test]
3342    async fn test_dispatch_update_issue() {
3343        let provider = MockProvider;
3344        let args = serde_json::json!({"key": "gh#1", "title": "Updated"});
3345        let result = dispatch_tool("update_issue", &args, &provider, None)
3346            .await
3347            .unwrap();
3348        assert!(matches!(result, ToolOutput::SingleIssue(_)));
3349    }
3350
3351    #[tokio::test]
3352    async fn test_dispatch_add_issue_comment() {
3353        let provider = MockProvider;
3354        let args = serde_json::json!({"key": "gh#1", "body": "A comment"});
3355        let result = dispatch_tool("add_issue_comment", &args, &provider, None)
3356            .await
3357            .unwrap();
3358        assert!(matches!(result, ToolOutput::Text(ref t) if t.contains("Comment added")));
3359    }
3360
3361    #[tokio::test]
3362    async fn test_dispatch_get_issue_relations() {
3363        let provider = MockProvider;
3364        let args = serde_json::json!({"key": "gh#1"});
3365        let result = dispatch_tool("get_issue_relations", &args, &provider, None)
3366            .await
3367            .unwrap();
3368        match result {
3369            ToolOutput::Relations(relations) => {
3370                assert!(relations.parent.is_some());
3371                assert_eq!(relations.subtasks.len(), 1);
3372                assert_eq!(relations.blocks.len(), 1);
3373            }
3374            other => panic!("Expected Relations, got {:?}", other),
3375        }
3376    }
3377
3378    #[tokio::test]
3379    async fn test_dispatch_get_issue_relations_missing_key() {
3380        let provider = MockProvider;
3381        let result = dispatch_tool(
3382            "get_issue_relations",
3383            &serde_json::json!({}),
3384            &provider,
3385            None,
3386        )
3387        .await;
3388        assert!(result.is_err());
3389    }
3390
3391    // --- MR tool dispatch tests ---
3392
3393    #[tokio::test]
3394    async fn test_dispatch_get_merge_requests() {
3395        let provider = MockProvider;
3396        let args = serde_json::json!({"state": "open", "limit": 5});
3397        let result = dispatch_tool("get_merge_requests", &args, &provider, None)
3398            .await
3399            .unwrap();
3400        assert!(matches!(result, ToolOutput::MergeRequests(v, _) if v.len() == 1));
3401    }
3402
3403    #[tokio::test]
3404    async fn test_dispatch_get_merge_requests_empty_args() {
3405        let provider = MockProvider;
3406        let result = dispatch_tool("get_merge_requests", &Value::Null, &provider, None)
3407            .await
3408            .unwrap();
3409        assert!(matches!(result, ToolOutput::MergeRequests(_, _)));
3410    }
3411
3412    #[tokio::test]
3413    async fn test_dispatch_get_merge_request() {
3414        let provider = MockProvider;
3415        let args = serde_json::json!({"key": "pr#1"});
3416        let result = dispatch_tool("get_merge_request", &args, &provider, None)
3417            .await
3418            .unwrap();
3419        assert!(matches!(result, ToolOutput::SingleMergeRequest(_)));
3420    }
3421
3422    #[tokio::test]
3423    async fn test_dispatch_get_merge_request_discussions() {
3424        let provider = MockProvider;
3425        let args = serde_json::json!({"key": "pr#1"});
3426        let result = dispatch_tool("get_merge_request_discussions", &args, &provider, None)
3427            .await
3428            .unwrap();
3429        assert!(matches!(result, ToolOutput::Discussions(v, _) if v.len() == 1));
3430    }
3431
3432    #[tokio::test]
3433    async fn test_dispatch_get_merge_request_diffs() {
3434        let provider = MockProvider;
3435        let args = serde_json::json!({"key": "pr#1"});
3436        let result = dispatch_tool("get_merge_request_diffs", &args, &provider, None)
3437            .await
3438            .unwrap();
3439        assert!(matches!(result, ToolOutput::Diffs(v, _) if v.len() == 1));
3440    }
3441
3442    #[tokio::test]
3443    async fn test_dispatch_create_merge_request() {
3444        let provider = MockProvider;
3445        let args = serde_json::json!({
3446            "title": "New PR",
3447            "source_branch": "feature",
3448            "target_branch": "main",
3449            "draft": false
3450        });
3451        let result = dispatch_tool("create_merge_request", &args, &provider, None)
3452            .await
3453            .unwrap();
3454        assert!(matches!(result, ToolOutput::SingleMergeRequest(_)));
3455    }
3456
3457    #[tokio::test]
3458    async fn test_dispatch_create_merge_request_comment_general() {
3459        let provider = MockProvider;
3460        let args = serde_json::json!({"key": "pr#1", "body": "LGTM"});
3461        let result = dispatch_tool("create_merge_request_comment", &args, &provider, None)
3462            .await
3463            .unwrap();
3464        assert!(matches!(result, ToolOutput::Text(ref t) if t.contains("Comment added")));
3465    }
3466
3467    #[tokio::test]
3468    async fn test_dispatch_create_merge_request_comment_inline() {
3469        let provider = MockProvider;
3470        let args = serde_json::json!({
3471            "key": "pr#1",
3472            "body": "Fix this line",
3473            "file_path": "src/main.rs",
3474            "line": 42,
3475            "line_type": "new",
3476            "commit_sha": "abc123"
3477        });
3478        let result = dispatch_tool("create_merge_request_comment", &args, &provider, None)
3479            .await
3480            .unwrap();
3481        assert!(matches!(result, ToolOutput::Text(ref t) if t.contains("Comment added")));
3482    }
3483
3484    #[test]
3485    fn test_create_merge_request_comment_params_accept_camel_case() {
3486        let args = serde_json::json!({
3487            "mrKey": "mr#566",
3488            "body": "reply",
3489            "filePath": "src/main.rs",
3490            "line": 12,
3491            "lineType": "new",
3492            "commitSha": "abc123",
3493            "discussionId": "788adb16c57805c9a5d59272c944cddea381a605"
3494        });
3495
3496        let params: CreateMrCommentParams = serde_json::from_value(args).unwrap();
3497        assert_eq!(params.key, "mr#566");
3498        assert_eq!(params.file_path.as_deref(), Some("src/main.rs"));
3499        assert_eq!(params.line_type.as_deref(), Some("new"));
3500        assert_eq!(params.commit_sha.as_deref(), Some("abc123"));
3501        assert_eq!(
3502            params.discussion_id.as_deref(),
3503            Some("788adb16c57805c9a5d59272c944cddea381a605")
3504        );
3505    }
3506
3507    #[test]
3508    fn test_create_merge_request_comment_params_still_accept_snake_case() {
3509        // Belt-and-suspenders: the camelCase aliases must not break the
3510        // original snake_case payload shape, which the MCP schema
3511        // declares and which some callers (our own skills included)
3512        // send today.
3513        let args = serde_json::json!({
3514            "key": "mr#566",
3515            "body": "reply",
3516            "file_path": "src/main.rs",
3517            "line": 12,
3518            "line_type": "new",
3519            "commit_sha": "abc123",
3520            "discussion_id": "788adb16c57805c9a5d59272c944cddea381a605"
3521        });
3522
3523        let params: CreateMrCommentParams = serde_json::from_value(args).unwrap();
3524        assert_eq!(params.key, "mr#566");
3525        assert_eq!(params.file_path.as_deref(), Some("src/main.rs"));
3526        assert_eq!(params.line_type.as_deref(), Some("new"));
3527        assert_eq!(params.commit_sha.as_deref(), Some("abc123"));
3528        assert_eq!(
3529            params.discussion_id.as_deref(),
3530            Some("788adb16c57805c9a5d59272c944cddea381a605")
3531        );
3532    }
3533
3534    #[tokio::test]
3535    async fn test_dispatch_create_merge_request_comment_accepts_camel_case_args() {
3536        // End-to-end: the executor's `dispatch_tool` path must accept
3537        // the same camelCase payload that real MCP clients (and our
3538        // skills) send, otherwise the alias would only help direct
3539        // `from_value` callers.
3540        let provider = MockProvider;
3541        let args = serde_json::json!({
3542            "mrKey": "mr#1",
3543            "body": "threaded reply",
3544            "discussionId": "abc123"
3545        });
3546        let result = dispatch_tool("create_merge_request_comment", &args, &provider, None)
3547            .await
3548            .unwrap();
3549        assert!(matches!(result, ToolOutput::Text(ref t) if t.contains("Comment added")));
3550    }
3551
3552    #[tokio::test]
3553    async fn test_dispatch_unknown_tool() {
3554        let provider = MockProvider;
3555        let result = dispatch_tool("nonexistent_tool", &Value::Null, &provider, None).await;
3556        assert!(result.is_err());
3557    }
3558
3559    // --- Executor enricher integration ---
3560
3561    #[tokio::test]
3562    async fn test_executor_enricher_transforms_args() {
3563        use devboy_core::{ToolEnricher, ToolSchema};
3564
3565        struct TestEnricher;
3566        impl ToolEnricher for TestEnricher {
3567            fn supported_categories(&self) -> &[devboy_core::ToolCategory] {
3568                &[devboy_core::ToolCategory::IssueTracker]
3569            }
3570            fn enrich_schema(&self, _tool: &str, _schema: &mut ToolSchema) {}
3571            fn transform_args(&self, _tool: &str, args: &mut Value) {
3572                if let Some(obj) = args.as_object_mut() {
3573                    obj.insert("transformed".into(), Value::Bool(true));
3574                }
3575            }
3576        }
3577
3578        let mut executor = Executor::new();
3579        executor.add_enricher(Box::new(TestEnricher));
3580        assert_eq!(executor.enrichers.len(), 1);
3581    }
3582
3583    // --- Pipeline dispatch tests ---
3584
3585    #[tokio::test]
3586    async fn test_dispatch_get_pipeline_unsupported() {
3587        let provider = MockProvider;
3588        let args = serde_json::json!({"branch": "main"});
3589        let result = dispatch_tool("get_pipeline", &args, &provider, None).await;
3590        // MockProvider doesn't implement get_pipeline → ProviderUnsupported
3591        assert!(result.is_err());
3592    }
3593
3594    #[tokio::test]
3595    async fn test_dispatch_get_job_logs_unsupported() {
3596        let provider = MockProvider;
3597        let args = serde_json::json!({"jobId": "123"});
3598        let result = dispatch_tool("get_job_logs", &args, &provider, None).await;
3599        assert!(result.is_err());
3600    }
3601
3602    #[tokio::test]
3603    async fn test_dispatch_get_pipeline_with_mr_key() {
3604        let provider = MockProvider;
3605        let args = serde_json::json!({"mrKey": "pr#1", "includeFailedLogs": false});
3606        let result = dispatch_tool("get_pipeline", &args, &provider, None).await;
3607        assert!(result.is_err());
3608    }
3609
3610    #[tokio::test]
3611    async fn test_dispatch_get_job_logs_with_pattern() {
3612        let provider = MockProvider;
3613        let args = serde_json::json!({"jobId": "123", "pattern": "ERROR", "context": 3});
3614        let result = dispatch_tool("get_job_logs", &args, &provider, None).await;
3615        assert!(result.is_err());
3616    }
3617
3618    #[tokio::test]
3619    async fn test_dispatch_get_job_logs_paginated() {
3620        let provider = MockProvider;
3621        let args = serde_json::json!({"jobId": "123", "offset": 10, "limit": 50});
3622        let result = dispatch_tool("get_job_logs", &args, &provider, None).await;
3623        assert!(result.is_err());
3624    }
3625
3626    #[tokio::test]
3627    async fn test_dispatch_get_job_logs_full() {
3628        let provider = MockProvider;
3629        let args = serde_json::json!({"jobId": "123", "full": true});
3630        let result = dispatch_tool("get_job_logs", &args, &provider, None).await;
3631        assert!(result.is_err());
3632    }
3633
3634    #[test]
3635    fn test_executor_default() {
3636        let executor = Executor::default();
3637        assert!(executor.enrichers.is_empty());
3638    }
3639
3640    // --- Status / User / Link / Epic dispatch tests ---
3641
3642    #[tokio::test]
3643    async fn test_dispatch_get_available_statuses_unsupported() {
3644        let provider = MockProvider;
3645        let result = dispatch_tool("get_available_statuses", &Value::Null, &provider, None).await;
3646        // MockProvider returns ProviderUnsupported for get_statuses
3647        assert!(result.is_err());
3648    }
3649
3650    #[tokio::test]
3651    async fn test_dispatch_get_users_unsupported() {
3652        let provider = MockProvider;
3653        let args = serde_json::json!({"search": "test"});
3654        let result = dispatch_tool("get_users", &args, &provider, None).await;
3655        // MockProvider uses default impl which returns ProviderUnsupported
3656        assert!(result.is_err());
3657    }
3658
3659    #[tokio::test]
3660    async fn test_dispatch_link_issues_unsupported() {
3661        let provider = MockProvider;
3662        let args = serde_json::json!({
3663            "source_key": "gh#1",
3664            "target_key": "gh#2",
3665            "link_type": "blocks"
3666        });
3667        let result = dispatch_tool("link_issues", &args, &provider, None).await;
3668        assert!(result.is_err());
3669    }
3670
3671    #[tokio::test]
3672    async fn test_dispatch_get_epics() {
3673        let provider = MockProvider;
3674        let args = serde_json::json!({"state": "open", "limit": 10});
3675        let result = dispatch_tool("get_epics", &args, &provider, None)
3676            .await
3677            .unwrap();
3678        // Returns enriched JSON with goal_id and progress
3679        assert!(matches!(result, ToolOutput::Text(_)));
3680    }
3681
3682    #[tokio::test]
3683    async fn test_dispatch_get_epics_empty_args() {
3684        let provider = MockProvider;
3685        let result = dispatch_tool("get_epics", &Value::Null, &provider, None)
3686            .await
3687            .unwrap();
3688        assert!(matches!(result, ToolOutput::Text(_)));
3689    }
3690
3691    #[tokio::test]
3692    async fn test_dispatch_create_epic() {
3693        let provider = MockProvider;
3694        let args = serde_json::json!({"title": "New Epic", "description": "Epic description"});
3695        let result = dispatch_tool("create_epic", &args, &provider, None)
3696            .await
3697            .unwrap();
3698        assert!(matches!(result, ToolOutput::SingleIssue(_)));
3699    }
3700
3701    #[tokio::test]
3702    async fn test_dispatch_update_epic() {
3703        let provider = MockProvider;
3704        let args = serde_json::json!({"key": "gh#1", "title": "Updated Epic"});
3705        let result = dispatch_tool("update_epic", &args, &provider, None)
3706            .await
3707            .unwrap();
3708        assert!(matches!(result, ToolOutput::SingleIssue(_)));
3709    }
3710
3711    #[tokio::test]
3712    async fn test_dispatch_link_issues_missing_params() {
3713        let provider = MockProvider;
3714        let args = serde_json::json!({"source_key": "gh#1"});
3715        let result = dispatch_tool("link_issues", &args, &provider, None).await;
3716        assert!(result.is_err());
3717    }
3718
3719    // --- Mock MeetingNotesProvider tests ---
3720
3721    struct MockMeetingProvider;
3722
3723    #[async_trait]
3724    impl MeetingNotesProvider for MockMeetingProvider {
3725        fn provider_name(&self) -> &'static str {
3726            "mock_meetings"
3727        }
3728
3729        async fn get_meetings(
3730            &self,
3731            _filter: MeetingFilter,
3732        ) -> devboy_core::Result<devboy_core::ProviderResult<devboy_core::MeetingNote>> {
3733            Ok(vec![devboy_core::MeetingNote {
3734                id: "m1".into(),
3735                title: "Test Meeting".into(),
3736                ..Default::default()
3737            }]
3738            .into())
3739        }
3740
3741        async fn get_transcript(
3742            &self,
3743            meeting_id: &str,
3744        ) -> devboy_core::Result<devboy_core::MeetingTranscript> {
3745            Ok(devboy_core::MeetingTranscript {
3746                meeting_id: meeting_id.to_string(),
3747                title: Some("Test Transcript".into()),
3748                sentences: vec![devboy_core::TranscriptSentence {
3749                    speaker_id: "s1".into(),
3750                    speaker_name: Some("Alice".into()),
3751                    text: "Hello".into(),
3752                    start_time: 0.0,
3753                    end_time: 1.0,
3754                }],
3755            })
3756        }
3757
3758        async fn search_meetings(
3759            &self,
3760            _query: &str,
3761            _filter: MeetingFilter,
3762        ) -> devboy_core::Result<devboy_core::ProviderResult<devboy_core::MeetingNote>> {
3763            Ok(vec![devboy_core::MeetingNote {
3764                id: "m2".into(),
3765                title: "Search Result Meeting".into(),
3766                ..Default::default()
3767            }]
3768            .into())
3769        }
3770    }
3771
3772    #[tokio::test]
3773    async fn test_dispatch_get_meeting_notes() {
3774        let provider = MockMeetingProvider;
3775        let args = serde_json::json!({"from_date": "2025-01-01", "limit": 10});
3776        let result = dispatch_meeting_tool("get_meeting_notes", &args, &provider)
3777            .await
3778            .unwrap();
3779        match result {
3780            ToolOutput::MeetingNotes(meetings, _) => {
3781                assert_eq!(meetings.len(), 1);
3782                assert_eq!(meetings[0].title, "Test Meeting");
3783            }
3784            other => panic!("Expected MeetingNotes, got {:?}", other),
3785        }
3786    }
3787
3788    #[tokio::test]
3789    async fn test_dispatch_get_meeting_transcript() {
3790        let provider = MockMeetingProvider;
3791        let args = serde_json::json!({"meeting_id": "m1"});
3792        let result = dispatch_meeting_tool("get_meeting_transcript", &args, &provider)
3793            .await
3794            .unwrap();
3795        match result {
3796            ToolOutput::MeetingTranscript(transcript) => {
3797                assert_eq!(transcript.meeting_id, "m1");
3798                assert_eq!(transcript.sentences.len(), 1);
3799                assert_eq!(transcript.sentences[0].speaker_name, Some("Alice".into()));
3800            }
3801            other => panic!("Expected MeetingTranscript, got {:?}", other),
3802        }
3803    }
3804
3805    #[tokio::test]
3806    async fn test_dispatch_search_meeting_notes() {
3807        let provider = MockMeetingProvider;
3808        let args = serde_json::json!({"query": "sprint", "limit": 5});
3809        let result = dispatch_meeting_tool("search_meeting_notes", &args, &provider)
3810            .await
3811            .unwrap();
3812        match result {
3813            ToolOutput::MeetingNotes(meetings, _) => {
3814                assert_eq!(meetings.len(), 1);
3815                assert_eq!(meetings[0].title, "Search Result Meeting");
3816            }
3817            other => panic!("Expected MeetingNotes, got {:?}", other),
3818        }
3819    }
3820
3821    #[tokio::test]
3822    async fn test_dispatch_unknown_meeting_tool() {
3823        let provider = MockMeetingProvider;
3824        let result = dispatch_meeting_tool("nonexistent_tool", &Value::Null, &provider).await;
3825        assert!(result.is_err());
3826    }
3827
3828    // =========================================================================
3829    // Structure tool dispatch tests
3830    // =========================================================================
3831
3832    fn sample_structure() -> devboy_core::Structure {
3833        devboy_core::Structure {
3834            id: 1,
3835            name: "Q1 Plan".into(),
3836            description: Some("Quarter 1 planning".into()),
3837        }
3838    }
3839
3840    fn sample_forest(structure_id: u64) -> devboy_core::StructureForest {
3841        devboy_core::StructureForest {
3842            version: 1,
3843            structure_id,
3844            tree: vec![devboy_core::StructureNode {
3845                row_id: 100,
3846                item_id: Some("PROJ-1".into()),
3847                item_type: Some("issue".into()),
3848                children: vec![],
3849            }],
3850            total_count: Some(1),
3851        }
3852    }
3853
3854    fn sample_view(structure_id: u64) -> devboy_core::StructureView {
3855        devboy_core::StructureView {
3856            id: 10,
3857            name: "Default".into(),
3858            structure_id,
3859            ..Default::default()
3860        }
3861    }
3862
3863    #[tokio::test]
3864    async fn test_dispatch_get_structures() {
3865        let provider = MockProvider;
3866        let result = dispatch_tool("get_structures", &Value::Null, &provider, None)
3867            .await
3868            .unwrap();
3869        assert!(matches!(result, ToolOutput::Structures(ref items, _) if items.len() == 1));
3870        assert_eq!(result.type_name(), "structures");
3871    }
3872
3873    #[tokio::test]
3874    async fn test_dispatch_get_structure_forest() {
3875        let provider = MockProvider;
3876        let args = serde_json::json!({"structureId": 1});
3877        let result = dispatch_tool("get_structure_forest", &args, &provider, None)
3878            .await
3879            .unwrap();
3880        assert!(matches!(result, ToolOutput::StructureForest(_)));
3881        assert_eq!(result.type_name(), "structure_forest");
3882    }
3883
3884    #[tokio::test]
3885    async fn test_dispatch_get_structure_forest_missing_id() {
3886        let provider = MockProvider;
3887        let result = dispatch_tool("get_structure_forest", &Value::Null, &provider, None).await;
3888        assert!(result.is_err());
3889    }
3890
3891    #[tokio::test]
3892    async fn test_dispatch_add_structure_rows() {
3893        let provider = MockProvider;
3894        let args = serde_json::json!({
3895            "structureId": 1,
3896            "items": ["PROJ-1", "PROJ-2"],
3897            "under": 100
3898        });
3899        let result = dispatch_tool("add_structure_rows", &args, &provider, None)
3900            .await
3901            .unwrap();
3902        match result {
3903            ToolOutput::ForestModified(r) => {
3904                assert_eq!(r.version, 2);
3905                assert_eq!(r.affected_count, 2);
3906            }
3907            _ => panic!("expected ForestModified"),
3908        }
3909    }
3910
3911    #[tokio::test]
3912    async fn test_dispatch_move_structure_rows() {
3913        let provider = MockProvider;
3914        let args = serde_json::json!({
3915            "structureId": 1,
3916            "rowIds": [100, 101],
3917            "under": 200
3918        });
3919        let result = dispatch_tool("move_structure_rows", &args, &provider, None)
3920            .await
3921            .unwrap();
3922        assert!(matches!(result, ToolOutput::ForestModified(_)));
3923    }
3924
3925    #[tokio::test]
3926    async fn test_dispatch_remove_structure_row() {
3927        let provider = MockProvider;
3928        let args = serde_json::json!({"structureId": 1, "rowId": 100});
3929        let result = dispatch_tool("remove_structure_row", &args, &provider, None)
3930            .await
3931            .unwrap();
3932        assert!(matches!(result, ToolOutput::Text(_)));
3933    }
3934
3935    #[tokio::test]
3936    async fn test_dispatch_get_structure_values() {
3937        let provider = MockProvider;
3938        let args = serde_json::json!({
3939            "structureId": 1,
3940            "rows": [100],
3941            "columns": ["summary", {"field": "status"}]
3942        });
3943        let result = dispatch_tool("get_structure_values", &args, &provider, None)
3944            .await
3945            .unwrap();
3946        assert!(matches!(result, ToolOutput::StructureValues(_)));
3947    }
3948
3949    #[tokio::test]
3950    async fn test_dispatch_get_structure_views() {
3951        let provider = MockProvider;
3952        let args = serde_json::json!({"structureId": 1});
3953        let result = dispatch_tool("get_structure_views", &args, &provider, None)
3954            .await
3955            .unwrap();
3956        assert!(matches!(result, ToolOutput::StructureViews(views, _) if views.len() == 1));
3957    }
3958
3959    #[tokio::test]
3960    async fn test_dispatch_save_structure_view() {
3961        let provider = MockProvider;
3962        let args = serde_json::json!({
3963            "structureId": 1,
3964            "name": "Sprint View"
3965        });
3966        let result = dispatch_tool("save_structure_view", &args, &provider, None)
3967            .await
3968            .unwrap();
3969        assert!(
3970            matches!(result, ToolOutput::StructureViews(views, _) if views[0].name == "Sprint View")
3971        );
3972    }
3973
3974    #[tokio::test]
3975    async fn test_dispatch_create_structure() {
3976        let provider = MockProvider;
3977        let args = serde_json::json!({"name": "New Structure", "description": "Test"});
3978        let result = dispatch_tool("create_structure", &args, &provider, None)
3979            .await
3980            .unwrap();
3981        match result {
3982            ToolOutput::Structures(items, _) => {
3983                assert_eq!(items[0].name, "New Structure");
3984                assert_eq!(items[0].id, 42);
3985            }
3986            _ => panic!("expected Structures"),
3987        }
3988    }
3989
3990    // -------------------------------------------------------------------
3991    // Project versions / fixVersion (issue #238)
3992    // -------------------------------------------------------------------
3993
3994    #[tokio::test]
3995    async fn test_dispatch_list_project_versions_applies_paper_defaults() {
3996        // No filter args → archived defaults to false, limit to 20.
3997        let provider = MockProvider;
3998        let result = dispatch_tool(
3999            "list_project_versions",
4000            &serde_json::json!({}),
4001            &provider,
4002            None,
4003        )
4004        .await
4005        .unwrap();
4006        match result {
4007            ToolOutput::ProjectVersions(items, _) => {
4008                let echoed = &items[0].name;
4009                assert!(echoed.contains("released=None"), "got {echoed}");
4010                assert!(echoed.contains("archived=Some(false)"), "got {echoed}");
4011                assert!(echoed.contains("limit=Some(20)"), "got {echoed}");
4012                assert!(echoed.contains("expand=false"), "got {echoed}");
4013            }
4014            other => panic!("expected ProjectVersions, got {other:?}"),
4015        }
4016    }
4017
4018    #[tokio::test]
4019    async fn test_dispatch_list_project_versions_explicit_filters_override_defaults() {
4020        let provider = MockProvider;
4021        let args = serde_json::json!({
4022            "project": "PROJ",
4023            "released": "true",
4024            "archived": "all",
4025            "limit": 5,
4026            "includeIssueCount": true,
4027        });
4028        let result = dispatch_tool("list_project_versions", &args, &provider, None)
4029            .await
4030            .unwrap();
4031        match result {
4032            ToolOutput::ProjectVersions(items, _) => {
4033                let echoed = &items[0].name;
4034                assert!(echoed.contains("released=Some(true)"), "got {echoed}");
4035                assert!(echoed.contains("archived=None"), "got {echoed}");
4036                assert!(echoed.contains("limit=Some(5)"), "got {echoed}");
4037                assert!(echoed.contains("expand=true"), "got {echoed}");
4038                assert_eq!(items[0].project, "PROJ");
4039            }
4040            other => panic!("expected ProjectVersions, got {other:?}"),
4041        }
4042    }
4043
4044    #[tokio::test]
4045    async fn test_dispatch_list_project_versions_rejects_unknown_filter() {
4046        let provider = MockProvider;
4047        let err = dispatch_tool(
4048            "list_project_versions",
4049            &serde_json::json!({"released": "maybe"}),
4050            &provider,
4051            None,
4052        )
4053        .await
4054        .unwrap_err();
4055        assert!(
4056            matches!(err, devboy_core::Error::InvalidData(ref m) if m.contains("'maybe'")),
4057            "expected InvalidData about 'maybe', got {err:?}"
4058        );
4059    }
4060
4061    #[tokio::test]
4062    async fn test_dispatch_upsert_project_version_returns_single() {
4063        let provider = MockProvider;
4064        let args = serde_json::json!({
4065            "project": "PROJ",
4066            "name": "3.18.0",
4067            "description": "release notes",
4068            "released": true,
4069            "releaseDate": "2026-05-01",
4070        });
4071        let result = dispatch_tool("upsert_project_version", &args, &provider, None)
4072            .await
4073            .unwrap();
4074        match result {
4075            ToolOutput::SingleProjectVersion(v) => {
4076                assert_eq!(v.name, "3.18.0");
4077                assert_eq!(v.project, "PROJ");
4078                assert!(v.released);
4079                assert_eq!(v.release_date.as_deref(), Some("2026-05-01"));
4080                assert_eq!(v.description.as_deref(), Some("release notes"));
4081            }
4082            other => panic!("expected SingleProjectVersion, got {other:?}"),
4083        }
4084    }
4085
4086    #[tokio::test]
4087    async fn test_dispatch_upsert_project_version_requires_name() {
4088        let provider = MockProvider;
4089        let err = dispatch_tool(
4090            "upsert_project_version",
4091            &serde_json::json!({"project": "PROJ"}),
4092            &provider,
4093            None,
4094        )
4095        .await
4096        .unwrap_err();
4097        assert!(matches!(err, devboy_core::Error::InvalidData(_)));
4098    }
4099
4100    #[test]
4101    fn parse_tri_filter_accepts_canonical_strings() {
4102        assert_eq!(parse_tri_filter(None).unwrap(), None);
4103        assert_eq!(parse_tri_filter(Some("all")).unwrap(), None);
4104        assert_eq!(parse_tri_filter(Some("True")).unwrap(), Some(true));
4105        assert_eq!(parse_tri_filter(Some("false")).unwrap(), Some(false));
4106        assert_eq!(parse_tri_filter(Some("yes")).unwrap(), Some(true));
4107        assert_eq!(parse_tri_filter(Some("0")).unwrap(), Some(false));
4108        assert!(parse_tri_filter(Some("maybe")).is_err());
4109    }
4110
4111    #[test]
4112    fn validate_iso_date_accepts_yyyy_mm_dd() {
4113        assert!(validate_iso_date("releaseDate", "2026-05-04").is_ok());
4114        assert!(validate_iso_date("releaseDate", "2026-12-31").is_ok());
4115    }
4116
4117    #[test]
4118    fn validate_iso_date_rejects_other_shapes() {
4119        // Wrong shape
4120        assert!(validate_iso_date("releaseDate", "2026/05/04").is_err());
4121        assert!(validate_iso_date("releaseDate", "2026-5-4").is_err());
4122        assert!(validate_iso_date("releaseDate", "2026-05-04T00:00:00Z").is_err());
4123        assert!(validate_iso_date("releaseDate", "tomorrow").is_err());
4124        // Out-of-range month / day
4125        assert!(validate_iso_date("releaseDate", "2026-13-01").is_err());
4126        assert!(validate_iso_date("releaseDate", "2026-00-15").is_err());
4127        assert!(validate_iso_date("releaseDate", "2026-05-32").is_err());
4128    }
4129
4130    #[tokio::test]
4131    async fn test_dispatch_upsert_project_version_rejects_bad_date() {
4132        let provider = MockProvider;
4133        let err = dispatch_tool(
4134            "upsert_project_version",
4135            &serde_json::json!({"name": "3.18.0", "releaseDate": "next friday"}),
4136            &provider,
4137            None,
4138        )
4139        .await
4140        .unwrap_err();
4141        assert!(
4142            matches!(err, devboy_core::Error::InvalidData(ref m) if m.contains("releaseDate")),
4143            "expected InvalidData about releaseDate, got {err:?}"
4144        );
4145    }
4146
4147    #[tokio::test]
4148    async fn test_dispatch_list_project_versions_rejects_zero_limit() {
4149        let provider = MockProvider;
4150        let err = dispatch_tool(
4151            "list_project_versions",
4152            &serde_json::json!({"limit": 0}),
4153            &provider,
4154            None,
4155        )
4156        .await
4157        .unwrap_err();
4158        assert!(
4159            matches!(err, devboy_core::Error::InvalidData(ref m) if m.contains("limit")),
4160            "expected InvalidData about limit, got {err:?}"
4161        );
4162    }
4163
4164    // -------------------------------------------------------------------
4165    // Agile / Sprint dispatch (issue #198)
4166    // -------------------------------------------------------------------
4167
4168    #[tokio::test]
4169    async fn test_dispatch_get_board_sprints_default_state_is_all() {
4170        let provider = MockProvider;
4171        let result = dispatch_tool(
4172            "get_board_sprints",
4173            &serde_json::json!({"boardId": 7}),
4174            &provider,
4175            None,
4176        )
4177        .await
4178        .unwrap();
4179        match result {
4180            ToolOutput::Sprints(items, _) => {
4181                assert_eq!(items.len(), 1);
4182                assert!(items[0].name.contains("board=7"), "got {}", items[0].name);
4183                assert!(items[0].name.contains("state=All"), "got {}", items[0].name);
4184            }
4185            other => panic!("expected Sprints, got {other:?}"),
4186        }
4187    }
4188
4189    #[tokio::test]
4190    async fn test_dispatch_get_board_sprints_state_filter_round_trips() {
4191        let provider = MockProvider;
4192        let result = dispatch_tool(
4193            "get_board_sprints",
4194            &serde_json::json!({"boardId": 9, "state": "active"}),
4195            &provider,
4196            None,
4197        )
4198        .await
4199        .unwrap();
4200        match result {
4201            ToolOutput::Sprints(items, _) => {
4202                assert!(
4203                    items[0].name.contains("state=Active"),
4204                    "got {}",
4205                    items[0].name
4206                );
4207            }
4208            other => panic!("expected Sprints, got {other:?}"),
4209        }
4210    }
4211
4212    #[tokio::test]
4213    async fn test_dispatch_get_board_sprints_rejects_unknown_state() {
4214        let provider = MockProvider;
4215        let err = dispatch_tool(
4216            "get_board_sprints",
4217            &serde_json::json!({"boardId": 1, "state": "wat"}),
4218            &provider,
4219            None,
4220        )
4221        .await
4222        .unwrap_err();
4223        assert!(
4224            matches!(err, devboy_core::Error::InvalidData(ref m) if m.contains("wat")),
4225            "expected InvalidData mentioning the bad value, got {err:?}"
4226        );
4227    }
4228
4229    #[tokio::test]
4230    async fn test_dispatch_assign_to_sprint_returns_text_summary() {
4231        let provider = MockProvider;
4232        let result = dispatch_tool(
4233            "assign_to_sprint",
4234            &serde_json::json!({
4235                "sprintId": 42,
4236                "issueKeys": ["PROJ-1", "PROJ-2"],
4237            }),
4238            &provider,
4239            None,
4240        )
4241        .await
4242        .unwrap();
4243        match result {
4244            ToolOutput::Text(msg) => {
4245                assert!(msg.contains("2 issue"), "got {msg}");
4246                assert!(msg.contains("42"), "got {msg}");
4247            }
4248            other => panic!("expected Text, got {other:?}"),
4249        }
4250    }
4251
4252    #[tokio::test]
4253    async fn test_dispatch_assign_to_sprint_rejects_empty_issue_keys() {
4254        let provider = MockProvider;
4255        let err = dispatch_tool(
4256            "assign_to_sprint",
4257            &serde_json::json!({"sprintId": 1, "issueKeys": []}),
4258            &provider,
4259            None,
4260        )
4261        .await
4262        .unwrap_err();
4263        assert!(
4264            matches!(err, devboy_core::Error::InvalidData(ref m) if m.contains("issueKeys")),
4265            "expected InvalidData about issueKeys, got {err:?}"
4266        );
4267    }
4268
4269    // -------------------------------------------------------------------
4270    // get_custom_fields dispatch
4271    // -------------------------------------------------------------------
4272
4273    #[tokio::test]
4274    async fn test_dispatch_get_custom_fields_returns_all_entries_by_default() {
4275        let provider = MockProvider;
4276        let result = dispatch_tool("get_custom_fields", &serde_json::json!({}), &provider, None)
4277            .await
4278            .unwrap();
4279        match result {
4280            ToolOutput::CustomFields(items, _) => {
4281                assert_eq!(items.len(), 3);
4282                let names: Vec<_> = items.iter().map(|f| f.name.as_str()).collect();
4283                assert!(names.contains(&"Epic Link"));
4284                assert!(names.contains(&"Sprint"));
4285            }
4286            other => panic!("expected CustomFields, got {other:?}"),
4287        }
4288    }
4289
4290    #[tokio::test]
4291    async fn test_dispatch_get_custom_fields_search_filters_by_substring() {
4292        let provider = MockProvider;
4293        let result = dispatch_tool(
4294            "get_custom_fields",
4295            &serde_json::json!({"search": "epic"}),
4296            &provider,
4297            None,
4298        )
4299        .await
4300        .unwrap();
4301        match result {
4302            ToolOutput::CustomFields(items, _) => {
4303                assert_eq!(items.len(), 2);
4304                for f in items {
4305                    assert!(f.name.to_lowercase().contains("epic"));
4306                }
4307            }
4308            other => panic!("expected CustomFields, got {other:?}"),
4309        }
4310    }
4311
4312    #[tokio::test]
4313    async fn test_dispatch_get_custom_fields_rejects_zero_limit() {
4314        let provider = MockProvider;
4315        let err = dispatch_tool(
4316            "get_custom_fields",
4317            &serde_json::json!({"limit": 0}),
4318            &provider,
4319            None,
4320        )
4321        .await
4322        .unwrap_err();
4323        assert!(
4324            matches!(err, devboy_core::Error::InvalidData(ref m) if m.contains("limit")),
4325            "expected InvalidData about limit, got {err:?}"
4326        );
4327    }
4328
4329    // -------------------------------------------------------------------
4330    // Regression: structure-specific arg parsing
4331    // -------------------------------------------------------------------
4332
4333    #[test]
4334    fn parse_row_item_bare_string_becomes_item_id() {
4335        let item = parse_structure_row_item(serde_json::json!("PROJ-1")).unwrap();
4336        assert_eq!(item.item_id, "PROJ-1");
4337        assert!(item.item_type.is_none());
4338    }
4339
4340    #[test]
4341    fn parse_row_item_json_object_string_parses_fields() {
4342        let item = parse_structure_row_item(serde_json::json!(
4343            "{\"item_id\":\"PROJ-2\",\"item_type\":\"issue\"}"
4344        ))
4345        .unwrap();
4346        assert_eq!(item.item_id, "PROJ-2");
4347        assert_eq!(item.item_type.as_deref(), Some("issue"));
4348    }
4349
4350    #[test]
4351    fn parse_row_item_malformed_json_object_is_error() {
4352        // Valid JSON object but fields do not match StructureRowItem.
4353        let err = parse_structure_row_item(serde_json::json!("{\"wrong\":true}")).unwrap_err();
4354        assert!(matches!(err, Error::InvalidData(_)));
4355    }
4356
4357    #[test]
4358    fn parse_column_spec_bare_string_sets_field() {
4359        let col = parse_structure_column_spec(serde_json::json!("summary")).unwrap();
4360        assert_eq!(col.field.as_deref(), Some("summary"));
4361        assert!(col.formula.is_none());
4362    }
4363
4364    #[test]
4365    fn parse_column_spec_formula_json_string_parses() {
4366        let col = parse_structure_column_spec(serde_json::json!(
4367            "{\"formula\":\"SUM(\\\"Story Points\\\")\"}"
4368        ))
4369        .unwrap();
4370        assert!(col.field.is_none());
4371        assert_eq!(col.formula.as_deref(), Some("SUM(\"Story Points\")"));
4372    }
4373
4374    #[test]
4375    fn parse_column_spec_object_value_is_deserialised() {
4376        // A real JSON object (not a stringified one) should also work.
4377        let col = parse_structure_column_spec(serde_json::json!({"field": "status", "width": 120}))
4378            .unwrap();
4379        assert_eq!(col.field.as_deref(), Some("status"));
4380        assert_eq!(col.width, Some(120));
4381    }
4382}