add get source

Author: georgevanderson

apps/framework-cli/src/framework/core/infrastructure/api_endpoint.rs

@@ -131,6 +131,12 @@ impl ApiEndpoint {
             metadata: MessageField::from_option(self.metadata.as_ref().map(|m| {
                 infrastructure_map::Metadata {
                     description: m.description.clone().unwrap_or_default(),
+                    source: MessageField::from_option(m.source.as_ref().map(|s| {
+                        infrastructure_map::SourceLocation {
+                            file: s.file.clone(),
+                            special_fields: Default::default(),
+                        }
+                    })),
                     special_fields: Default::default(),
                 }
             })),
@@ -157,6 +163,10 @@ impl ApiEndpoint {
                 } else {
                     Some(m.description)
                 },
+                source: m
+                    .source
+                    .into_option()
+                    .map(|s| super::table::SourceLocation { file: s.file }),
             }),
         }
     }

apps/framework-cli/src/framework/core/infrastructure/function_process.rs

@@ -135,6 +135,12 @@ impl FunctionProcess {
             metadata: MessageField::from_option(self.metadata.as_ref().map(|m| {
                 crate::proto::infrastructure_map::Metadata {
                     description: m.description.clone().unwrap_or_default(),
+                    source: MessageField::from_option(m.source.as_ref().map(|s| {
+                        crate::proto::infrastructure_map::SourceLocation {
+                            file: s.file.clone(),
+                            special_fields: Default::default(),
+                        }
+                    })),
                     special_fields: Default::default(),
                 }
             })),
@@ -159,6 +165,10 @@ impl FunctionProcess {
                 } else {
                     Some(m.description)
                 },
+                source: m
+                    .source
+                    .into_option()
+                    .map(|s| super::table::SourceLocation { file: s.file }),
             }),
         }
     }

apps/framework-cli/src/framework/core/infrastructure/table.rs

@@ -22,9 +22,16 @@ use serde_json::Value;
 use std::fmt;
 use std::fmt::Debug;
 
+#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Eq, Hash)]
+pub struct SourceLocation {
+    pub file: String,
+}
+
 #[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Eq, Hash)]
 pub struct Metadata {
     pub description: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none", default)]
+    pub source: Option<SourceLocation>,
 }
 
 /// Prefix for Moose-managed metadata in column comments.
@@ -197,6 +204,12 @@ impl Table {
             metadata: MessageField::from_option(self.metadata.as_ref().map(|m| {
                 infrastructure_map::Metadata {
                     description: m.description.clone().unwrap_or_default(),
+                    source: MessageField::from_option(m.source.as_ref().map(|s| {
+                        infrastructure_map::SourceLocation {
+                            file: s.file.clone(),
+                            special_fields: Default::default(),
+                        }
+                    })),
                     special_fields: Default::default(),
                 }
             })),
@@ -241,6 +254,10 @@ impl Table {
                 } else {
                     Some(m.description)
                 },
+                source: m
+                    .source
+                    .into_option()
+                    .map(|s| SourceLocation { file: s.file }),
             }),
             life_cycle: match proto.life_cycle.enum_value_or_default() {
                 ProtoLifeCycle::FULLY_MANAGED => LifeCycle::FullyManaged,

apps/framework-cli/src/framework/core/infrastructure/topic.rs

@@ -142,6 +142,12 @@ impl Topic {
             metadata: MessageField::from_option(self.metadata.as_ref().map(|m| {
                 proto::infrastructure_map::Metadata {
                     description: m.description.clone().unwrap_or_default(),
+                    source: MessageField::from_option(m.source.as_ref().map(|s| {
+                        proto::infrastructure_map::SourceLocation {
+                            file: s.file.clone(),
+                            special_fields: Default::default(),
+                        }
+                    })),
                     special_fields: Default::default(),
                 }
             })),
@@ -181,6 +187,10 @@ impl Topic {
                 } else {
                     Some(m.description)
                 },
+                source: m
+                    .source
+                    .into_option()
+                    .map(|s| super::table::SourceLocation { file: s.file }),
             }),
             life_cycle: match proto.life_cycle.enum_value_or_default() {
                 ProtoLifeCycle::FULLY_MANAGED => LifeCycle::FullyManaged,

apps/framework-cli/src/mcp/server.rs

@@ -12,7 +12,7 @@ use rmcp::{
 };
 use std::sync::Arc;
 
-use super::tools::{create_error_result, infra_map, logs};
+use super::tools::{create_error_result, get_source, infra_map, logs};
 use crate::infrastructure::redis::redis_client::RedisClient;
 
 /// Handler for the MCP server that implements the Model Context Protocol
@@ -66,7 +66,11 @@ impl ServerHandler for MooseMcpHandler {
         _context: RequestContext<RoleServer>,
     ) -> Result<ListToolsResult, ErrorData> {
         Ok(ListToolsResult {
-            tools: vec![logs::tool_definition(), infra_map::tool_definition()],
+            tools: vec![
+                logs::tool_definition(),
+                infra_map::tool_definition(),
+                get_source::tool_definition(),
+            ],
             next_cursor: None,
         })
     }
@@ -83,6 +87,11 @@ impl ServerHandler for MooseMcpHandler {
                 self.redis_client.clone(),
             )
             .await),
+            "get_source" => Ok(get_source::handle_call(
+                param.arguments.as_ref(),
+                self.redis_client.clone(),
+            )
+            .await),
             _ => Ok(create_error_result(format!("Unknown tool: {}", param.name))),
         }
     }