diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/ColaFlow.Modules.Mcp.Application.csproj b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/ColaFlow.Modules.Mcp.Application.csproj index 6ff5bc7..789c999 100644 --- a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/ColaFlow.Modules.Mcp.Application.csproj +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/ColaFlow.Modules.Mcp.Application.csproj @@ -11,10 +11,14 @@ + + + + - + diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Handlers/ResourcesListMethodHandler.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Handlers/ResourcesListMethodHandler.cs index 4dc4da2..aed882b 100644 --- a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Handlers/ResourcesListMethodHandler.cs +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Handlers/ResourcesListMethodHandler.cs @@ -1,3 +1,4 @@ +using ColaFlow.Modules.Mcp.Application.Services; using Microsoft.Extensions.Logging; namespace ColaFlow.Modules.Mcp.Application.Handlers; @@ -8,23 +9,36 @@ namespace ColaFlow.Modules.Mcp.Application.Handlers; public class ResourcesListMethodHandler : IMcpMethodHandler { private readonly ILogger _logger; + private readonly IMcpResourceRegistry _resourceRegistry; public string MethodName => "resources/list"; - public ResourcesListMethodHandler(ILogger logger) + public ResourcesListMethodHandler( + ILogger logger, + IMcpResourceRegistry resourceRegistry) { _logger = logger; + _resourceRegistry = resourceRegistry; } public Task HandleAsync(object? @params, CancellationToken cancellationToken) { _logger.LogDebug("Handling resources/list request"); - // TODO: Implement in Story 5.5 (Core MCP Resources) - // For now, return empty list + // Get all registered resource descriptors + var descriptors = _resourceRegistry.GetResourceDescriptors(); + + _logger.LogInformation("Returning {Count} MCP resources", descriptors.Count); + var response = new { - resources = Array.Empty() + resources = descriptors.Select(d => new + { + uri = d.Uri, + name = d.Name, + description = d.Description, + mimeType = d.MimeType + }).ToArray() }; return Task.FromResult(response); diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Handlers/ResourcesReadMethodHandler.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Handlers/ResourcesReadMethodHandler.cs new file mode 100644 index 0000000..301f320 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Handlers/ResourcesReadMethodHandler.cs @@ -0,0 +1,127 @@ +using System.Text.Json; +using System.Text.RegularExpressions; +using ColaFlow.Modules.Mcp.Application.Services; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using ColaFlow.Modules.Mcp.Domain.Exceptions; +using Microsoft.Extensions.Logging; + +namespace ColaFlow.Modules.Mcp.Application.Handlers; + +/// +/// Handler for the 'resources/read' MCP method +/// +public class ResourcesReadMethodHandler : IMcpMethodHandler +{ + private readonly ILogger _logger; + private readonly IMcpResourceRegistry _resourceRegistry; + + public string MethodName => "resources/read"; + + public ResourcesReadMethodHandler( + ILogger logger, + IMcpResourceRegistry resourceRegistry) + { + _logger = logger; + _resourceRegistry = resourceRegistry; + } + + public async Task HandleAsync(object? @params, CancellationToken cancellationToken) + { + _logger.LogDebug("Handling resources/read request"); + + // Parse parameters + var paramsJson = JsonSerializer.Serialize(@params); + var request = JsonSerializer.Deserialize(paramsJson); + + if (request == null || string.IsNullOrWhiteSpace(request.Uri)) + { + throw new McpInvalidParamsException("Missing required parameter: uri"); + } + + _logger.LogInformation("Reading resource: {Uri}", request.Uri); + + // Find resource by URI + var resource = _resourceRegistry.GetResourceByUri(request.Uri); + if (resource == null) + { + throw new McpNotFoundException($"Resource not found: {request.Uri}"); + } + + // Parse URI and extract parameters + var resourceRequest = ParseResourceRequest(request.Uri, resource.Uri); + + // Get resource content + var content = await resource.GetContentAsync(resourceRequest, cancellationToken); + + // Return MCP response + var response = new + { + contents = new[] + { + new + { + uri = content.Uri, + mimeType = content.MimeType, + text = content.Text + } + } + }; + + return response; + } + + /// + /// Parse resource URI and extract path/query parameters + /// + private McpResourceRequest ParseResourceRequest(string requestUri, string templateUri) + { + var request = new McpResourceRequest { Uri = requestUri }; + + // Split URI and query string + var uriParts = requestUri.Split('?', 2); + var path = uriParts[0]; + var queryString = uriParts.Length > 1 ? uriParts[1] : string.Empty; + + // Extract path parameters from template + // Example: "colaflow://projects.get/123" with template "colaflow://projects.get/{id}" + var pattern = "^" + Regex.Escape(templateUri) + .Replace(@"\{", "{") + .Replace(@"\}", "}") + .Replace("{id}", @"(?[^/]+)") + .Replace("{projectId}", @"(?[^/]+)") + + "$"; + + var match = Regex.Match(path, pattern); + if (match.Success) + { + foreach (Group group in match.Groups) + { + if (!int.TryParse(group.Name, out _) && group.Name != "0") + { + request.UriParams[group.Name] = group.Value; + } + } + } + + // Parse query parameters + if (!string.IsNullOrEmpty(queryString)) + { + var queryPairs = queryString.Split('&'); + foreach (var pair in queryPairs) + { + var keyValue = pair.Split('=', 2); + if (keyValue.Length == 2) + { + request.QueryParams[keyValue[0]] = Uri.UnescapeDataString(keyValue[1]); + } + } + } + + return request; + } + + private class ResourceReadParams + { + public string Uri { get; set; } = string.Empty; + } +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/IssuesGetResource.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/IssuesGetResource.cs new file mode 100644 index 0000000..8f7fcb6 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/IssuesGetResource.cs @@ -0,0 +1,154 @@ +using System.Text.Json; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using ColaFlow.Modules.Mcp.Domain.Exceptions; +using ColaFlow.Modules.ProjectManagement.Application.Common.Interfaces; +using ColaFlow.Modules.ProjectManagement.Domain.Repositories; +using ColaFlow.Modules.ProjectManagement.Domain.ValueObjects; +using Microsoft.Extensions.Logging; + +namespace ColaFlow.Modules.Mcp.Application.Resources; + +/// +/// Resource: colaflow://issues.get/{id} +/// Gets detailed information about a specific issue (Epic, Story, or Task) +/// +public class IssuesGetResource : IMcpResource +{ + public string Uri => "colaflow://issues.get/{id}"; + public string Name => "Issue Details"; + public string Description => "Get detailed information about an issue (Epic/Story/Task)"; + public string MimeType => "application/json"; + + private readonly IProjectRepository _projectRepository; + private readonly ITenantContext _tenantContext; + private readonly ILogger _logger; + + public IssuesGetResource( + IProjectRepository projectRepository, + ITenantContext tenantContext, + ILogger logger) + { + _projectRepository = projectRepository; + _tenantContext = tenantContext; + _logger = logger; + } + + public async Task GetContentAsync( + McpResourceRequest request, + CancellationToken cancellationToken) + { + var tenantId = _tenantContext.GetCurrentTenantId(); + + // Extract {id} from URI parameters + if (!request.UriParams.TryGetValue("id", out var idString)) + { + throw new McpInvalidParamsException("Missing required parameter: id"); + } + + if (!Guid.TryParse(idString, out var issueIdGuid)) + { + throw new McpInvalidParamsException($"Invalid issue ID format: {idString}"); + } + + _logger.LogDebug("Fetching issue {IssueId} for tenant {TenantId}", issueIdGuid, tenantId); + + // Try to find as Epic + var epic = await _projectRepository.GetEpicByIdReadOnlyAsync(EpicId.From(issueIdGuid), cancellationToken); + if (epic != null) + { + var epicDto = new + { + id = epic.Id.Value, + type = "Epic", + name = epic.Name, + description = epic.Description, + status = epic.Status.ToString(), + priority = epic.Priority.ToString(), + createdAt = epic.CreatedAt, + updatedAt = epic.UpdatedAt, + stories = epic.Stories?.Select(s => new + { + id = s.Id.Value, + title = s.Title, + status = s.Status.ToString(), + priority = s.Priority.ToString(), + assigneeId = s.AssigneeId?.Value + }).ToList() + }; + + var json = JsonSerializer.Serialize(epicDto, new JsonSerializerOptions { WriteIndented = true }); + + return new McpResourceContent + { + Uri = request.Uri, + MimeType = MimeType, + Text = json + }; + } + + // Try to find as Story + var story = await _projectRepository.GetStoryByIdReadOnlyAsync(StoryId.From(issueIdGuid), cancellationToken); + if (story != null) + { + var storyDto = new + { + id = story.Id.Value, + type = "Story", + title = story.Title, + description = story.Description, + status = story.Status.ToString(), + priority = story.Priority.ToString(), + assigneeId = story.AssigneeId?.Value, + createdAt = story.CreatedAt, + updatedAt = story.UpdatedAt, + tasks = story.Tasks?.Select(t => new + { + id = t.Id.Value, + title = t.Title, + status = t.Status.ToString(), + priority = t.Priority.ToString(), + assigneeId = t.AssigneeId?.Value + }).ToList() + }; + + var json = JsonSerializer.Serialize(storyDto, new JsonSerializerOptions { WriteIndented = true }); + + return new McpResourceContent + { + Uri = request.Uri, + MimeType = MimeType, + Text = json + }; + } + + // Try to find as Task + var task = await _projectRepository.GetTaskByIdReadOnlyAsync(TaskId.From(issueIdGuid), cancellationToken); + if (task != null) + { + var taskDto = new + { + id = task.Id.Value, + type = "Task", + title = task.Title, + description = task.Description, + status = task.Status.ToString(), + priority = task.Priority.ToString(), + assigneeId = task.AssigneeId?.Value, + createdAt = task.CreatedAt, + updatedAt = task.UpdatedAt + }; + + var json = JsonSerializer.Serialize(taskDto, new JsonSerializerOptions { WriteIndented = true }); + + return new McpResourceContent + { + Uri = request.Uri, + MimeType = MimeType, + Text = json + }; + } + + // Not found + throw new McpNotFoundException($"Issue not found: {issueIdGuid}"); + } +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/IssuesSearchResource.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/IssuesSearchResource.cs new file mode 100644 index 0000000..15fb019 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/IssuesSearchResource.cs @@ -0,0 +1,228 @@ +using System.Text.Json; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using ColaFlow.Modules.ProjectManagement.Application.Common.Interfaces; +using ColaFlow.Modules.ProjectManagement.Domain.Repositories; +using ColaFlow.Modules.ProjectManagement.Domain.ValueObjects; +using Microsoft.Extensions.Logging; + +namespace ColaFlow.Modules.Mcp.Application.Resources; + +/// +/// Resource: colaflow://issues.search +/// Searches issues with filters (Epics, Stories, Tasks) +/// Query params: status, priority, assignee, type, project, limit, offset +/// +public class IssuesSearchResource : IMcpResource +{ + public string Uri => "colaflow://issues.search"; + public string Name => "Issues Search"; + public string Description => "Search issues with filters (status, priority, assignee, etc.)"; + public string MimeType => "application/json"; + + private readonly IProjectRepository _projectRepository; + private readonly ITenantContext _tenantContext; + private readonly ILogger _logger; + + public IssuesSearchResource( + IProjectRepository projectRepository, + ITenantContext tenantContext, + ILogger logger) + { + _projectRepository = projectRepository; + _tenantContext = tenantContext; + _logger = logger; + } + + public async Task GetContentAsync( + McpResourceRequest request, + CancellationToken cancellationToken) + { + var tenantId = _tenantContext.GetCurrentTenantId(); + + _logger.LogDebug("Searching issues for tenant {TenantId} with filters: {@Filters}", + tenantId, request.QueryParams); + + // Parse query parameters + var projectFilter = request.QueryParams.GetValueOrDefault("project"); + var statusFilter = request.QueryParams.GetValueOrDefault("status"); + var priorityFilter = request.QueryParams.GetValueOrDefault("priority"); + var typeFilter = request.QueryParams.GetValueOrDefault("type")?.ToLower(); + var assigneeFilter = request.QueryParams.GetValueOrDefault("assignee"); + var limit = int.TryParse(request.QueryParams.GetValueOrDefault("limit"), out var l) ? l : 100; + var offset = int.TryParse(request.QueryParams.GetValueOrDefault("offset"), out var o) ? o : 0; + + // Limit max results + limit = Math.Min(limit, 100); + + // Get all projects + var projects = await _projectRepository.GetAllProjectsReadOnlyAsync(cancellationToken); + + // Filter by project if specified + if (!string.IsNullOrEmpty(projectFilter) && Guid.TryParse(projectFilter, out var projectIdGuid)) + { + var projectId = ProjectId.From(projectIdGuid); + var project = await _projectRepository.GetProjectWithFullHierarchyReadOnlyAsync(projectId, cancellationToken); + projects = project != null ? new List { project } : new(); + } + else + { + // Load full hierarchy for all projects + var projectsWithHierarchy = new List(); + foreach (var p in projects) + { + var fullProject = await _projectRepository.GetProjectWithFullHierarchyReadOnlyAsync(p.Id, cancellationToken); + if (fullProject != null) + { + projectsWithHierarchy.Add(fullProject); + } + } + projects = projectsWithHierarchy; + } + + // Collect all issues (Epics, Stories, Tasks) + var allIssues = new List(); + + foreach (var project in projects) + { + if (project.Epics == null) continue; + + foreach (var epic in project.Epics) + { + // Filter Epics + if (ShouldIncludeIssue("epic", typeFilter, epic.Status.ToString(), statusFilter, + epic.Priority.ToString(), priorityFilter, null, assigneeFilter)) + { + allIssues.Add(new + { + id = epic.Id.Value, + type = "Epic", + name = epic.Name, + description = epic.Description, + status = epic.Status.ToString(), + priority = epic.Priority.ToString(), + projectId = project.Id.Value, + projectName = project.Name, + createdAt = epic.CreatedAt, + storyCount = epic.Stories?.Count ?? 0 + }); + } + + // Filter Stories + if (epic.Stories != null) + { + foreach (var story in epic.Stories) + { + if (ShouldIncludeIssue("story", typeFilter, story.Status.ToString(), statusFilter, + story.Priority.ToString(), priorityFilter, story.AssigneeId?.Value.ToString(), assigneeFilter)) + { + allIssues.Add(new + { + id = story.Id.Value, + type = "Story", + title = story.Title, + description = story.Description, + status = story.Status.ToString(), + priority = story.Priority.ToString(), + assigneeId = story.AssigneeId?.Value, + projectId = project.Id.Value, + projectName = project.Name, + epicId = epic.Id.Value, + epicName = epic.Name, + createdAt = story.CreatedAt, + taskCount = story.Tasks?.Count ?? 0 + }); + } + + // Filter Tasks + if (story.Tasks != null) + { + foreach (var task in story.Tasks) + { + if (ShouldIncludeIssue("task", typeFilter, task.Status.ToString(), statusFilter, + task.Priority.ToString(), priorityFilter, task.AssigneeId?.Value.ToString(), assigneeFilter)) + { + allIssues.Add(new + { + id = task.Id.Value, + type = "Task", + title = task.Title, + description = task.Description, + status = task.Status.ToString(), + priority = task.Priority.ToString(), + assigneeId = task.AssigneeId?.Value, + projectId = project.Id.Value, + projectName = project.Name, + storyId = story.Id.Value, + storyTitle = story.Title, + epicId = epic.Id.Value, + epicName = epic.Name, + createdAt = task.CreatedAt + }); + } + } + } + } + } + } + } + + // Apply pagination + var total = allIssues.Count; + var paginatedIssues = allIssues.Skip(offset).Take(limit).ToList(); + + var json = JsonSerializer.Serialize(new + { + issues = paginatedIssues, + total = total, + limit = limit, + offset = offset + }, new JsonSerializerOptions { WriteIndented = true }); + + _logger.LogInformation("Found {Count} issues for tenant {TenantId} (total: {Total})", + paginatedIssues.Count, tenantId, total); + + return new McpResourceContent + { + Uri = Uri, + MimeType = MimeType, + Text = json + }; + } + + private bool ShouldIncludeIssue( + string issueType, + string? typeFilter, + string status, + string? statusFilter, + string priority, + string? priorityFilter, + string? assigneeId, + string? assigneeFilter) + { + // Type filter + if (!string.IsNullOrEmpty(typeFilter) && !issueType.Equals(typeFilter, StringComparison.OrdinalIgnoreCase)) + { + return false; + } + + // Status filter + if (!string.IsNullOrEmpty(statusFilter) && !status.Equals(statusFilter, StringComparison.OrdinalIgnoreCase)) + { + return false; + } + + // Priority filter + if (!string.IsNullOrEmpty(priorityFilter) && !priority.Equals(priorityFilter, StringComparison.OrdinalIgnoreCase)) + { + return false; + } + + // Assignee filter + if (!string.IsNullOrEmpty(assigneeFilter) && assigneeId != assigneeFilter) + { + return false; + } + + return true; + } +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/ProjectsGetResource.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/ProjectsGetResource.cs new file mode 100644 index 0000000..d822778 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/ProjectsGetResource.cs @@ -0,0 +1,99 @@ +using System.Text.Json; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using ColaFlow.Modules.Mcp.Domain.Exceptions; +using ColaFlow.Modules.ProjectManagement.Application.Common.Interfaces; +using ColaFlow.Modules.ProjectManagement.Domain.Repositories; +using ColaFlow.Modules.ProjectManagement.Domain.ValueObjects; +using Microsoft.Extensions.Logging; + +namespace ColaFlow.Modules.Mcp.Application.Resources; + +/// +/// Resource: colaflow://projects.get/{id} +/// Gets detailed information about a specific project +/// +public class ProjectsGetResource : IMcpResource +{ + public string Uri => "colaflow://projects.get/{id}"; + public string Name => "Project Details"; + public string Description => "Get detailed information about a project"; + public string MimeType => "application/json"; + + private readonly IProjectRepository _projectRepository; + private readonly ITenantContext _tenantContext; + private readonly ILogger _logger; + + public ProjectsGetResource( + IProjectRepository projectRepository, + ITenantContext tenantContext, + ILogger logger) + { + _projectRepository = projectRepository; + _tenantContext = tenantContext; + _logger = logger; + } + + public async Task GetContentAsync( + McpResourceRequest request, + CancellationToken cancellationToken) + { + var tenantId = _tenantContext.GetCurrentTenantId(); + + // Extract {id} from URI parameters + if (!request.UriParams.TryGetValue("id", out var idString)) + { + throw new McpInvalidParamsException("Missing required parameter: id"); + } + + if (!Guid.TryParse(idString, out var projectIdGuid)) + { + throw new McpInvalidParamsException($"Invalid project ID format: {idString}"); + } + + var projectId = ProjectId.From(projectIdGuid); + + _logger.LogDebug("Fetching project {ProjectId} for tenant {TenantId}", projectId, tenantId); + + // Get project with full hierarchy (read-only) + var project = await _projectRepository.GetProjectWithFullHierarchyReadOnlyAsync(projectId, cancellationToken); + + if (project == null) + { + throw new McpNotFoundException($"Project not found: {projectId}"); + } + + // Map to DTO + var projectDto = new + { + id = project.Id.Value, + name = project.Name, + key = project.Key.ToString(), + description = project.Description, + status = project.Status.ToString(), + ownerId = project.OwnerId.Value, + createdAt = project.CreatedAt, + updatedAt = project.UpdatedAt, + epics = project.Epics?.Select(e => new + { + id = e.Id.Value, + name = e.Name, + description = e.Description, + status = e.Status.ToString(), + priority = e.Priority.ToString(), + createdAt = e.CreatedAt, + storyCount = e.Stories?.Count ?? 0 + }).ToList() + }; + + var json = JsonSerializer.Serialize(projectDto, new JsonSerializerOptions { WriteIndented = true }); + + _logger.LogInformation("Retrieved project {ProjectId} for tenant {TenantId}", projectId, tenantId); + + return new McpResourceContent + { + Uri = request.Uri, + MimeType = MimeType, + Text = json + }; + } +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/ProjectsListResource.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/ProjectsListResource.cs new file mode 100644 index 0000000..71e3236 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/ProjectsListResource.cs @@ -0,0 +1,73 @@ +using System.Text.Json; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using ColaFlow.Modules.ProjectManagement.Application.Common.Interfaces; +using ColaFlow.Modules.ProjectManagement.Domain.Repositories; +using Microsoft.Extensions.Logging; + +namespace ColaFlow.Modules.Mcp.Application.Resources; + +/// +/// Resource: colaflow://projects.list +/// Lists all projects in the current tenant +/// +public class ProjectsListResource : IMcpResource +{ + public string Uri => "colaflow://projects.list"; + public string Name => "Projects List"; + public string Description => "List all projects in current tenant"; + public string MimeType => "application/json"; + + private readonly IProjectRepository _projectRepository; + private readonly ITenantContext _tenantContext; + private readonly ILogger _logger; + + public ProjectsListResource( + IProjectRepository projectRepository, + ITenantContext tenantContext, + ILogger logger) + { + _projectRepository = projectRepository; + _tenantContext = tenantContext; + _logger = logger; + } + + public async Task GetContentAsync( + McpResourceRequest request, + CancellationToken cancellationToken) + { + var tenantId = _tenantContext.GetCurrentTenantId(); + + _logger.LogDebug("Fetching projects list for tenant {TenantId}", tenantId); + + // Get all projects (read-only) + var projects = await _projectRepository.GetAllProjectsReadOnlyAsync(cancellationToken); + + // Map to DTOs + var projectDtos = projects.Select(p => new + { + id = p.Id.Value, + name = p.Name, + key = p.Key.ToString(), + description = p.Description, + status = p.Status.ToString(), + createdAt = p.CreatedAt, + updatedAt = p.UpdatedAt, + epicCount = p.Epics?.Count ?? 0 + }).ToList(); + + var json = JsonSerializer.Serialize(new + { + projects = projectDtos, + total = projectDtos.Count + }, new JsonSerializerOptions { WriteIndented = true }); + + _logger.LogInformation("Retrieved {Count} projects for tenant {TenantId}", projectDtos.Count, tenantId); + + return new McpResourceContent + { + Uri = Uri, + MimeType = MimeType, + Text = json + }; + } +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/SprintsCurrentResource.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/SprintsCurrentResource.cs new file mode 100644 index 0000000..32dd787 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/SprintsCurrentResource.cs @@ -0,0 +1,86 @@ +using System.Text.Json; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using ColaFlow.Modules.Mcp.Domain.Exceptions; +using ColaFlow.Modules.ProjectManagement.Application.Common.Interfaces; +using ColaFlow.Modules.ProjectManagement.Domain.Repositories; +using Microsoft.Extensions.Logging; + +namespace ColaFlow.Modules.Mcp.Application.Resources; + +/// +/// Resource: colaflow://sprints.current +/// Gets the currently active Sprint(s) +/// +public class SprintsCurrentResource : IMcpResource +{ + public string Uri => "colaflow://sprints.current"; + public string Name => "Current Sprint"; + public string Description => "Get the currently active Sprint(s)"; + public string MimeType => "application/json"; + + private readonly IProjectRepository _projectRepository; + private readonly ITenantContext _tenantContext; + private readonly ILogger _logger; + + public SprintsCurrentResource( + IProjectRepository projectRepository, + ITenantContext tenantContext, + ILogger logger) + { + _projectRepository = projectRepository; + _tenantContext = tenantContext; + _logger = logger; + } + + public async Task GetContentAsync( + McpResourceRequest request, + CancellationToken cancellationToken) + { + var tenantId = _tenantContext.GetCurrentTenantId(); + + _logger.LogDebug("Fetching active sprints for tenant {TenantId}", tenantId); + + // Get active sprints + var activeSprints = await _projectRepository.GetActiveSprintsAsync(cancellationToken); + + if (activeSprints.Count == 0) + { + _logger.LogWarning("No active sprints found for tenant {TenantId}", tenantId); + throw new McpNotFoundException("No active sprints found"); + } + + // Map to DTOs with statistics + var sprintDtos = activeSprints.Select(sprint => new + { + id = sprint.Id.Value, + name = sprint.Name, + goal = sprint.Goal, + status = sprint.Status.ToString(), + startDate = sprint.StartDate, + endDate = sprint.EndDate, + createdAt = sprint.CreatedAt, + statistics = new + { + totalTasks = sprint.TaskIds?.Count ?? 0 + // Note: To get completed/in-progress counts, we'd need to query tasks + // For now, just return total count + } + }).ToList(); + + var json = JsonSerializer.Serialize(new + { + sprints = sprintDtos, + total = sprintDtos.Count + }, new JsonSerializerOptions { WriteIndented = true }); + + _logger.LogInformation("Retrieved {Count} active sprints for tenant {TenantId}", + sprintDtos.Count, tenantId); + + return new McpResourceContent + { + Uri = Uri, + MimeType = MimeType, + Text = json + }; + } +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/UsersListResource.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/UsersListResource.cs new file mode 100644 index 0000000..b0dc6d2 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Resources/UsersListResource.cs @@ -0,0 +1,74 @@ +using System.Text.Json; +using ColaFlow.Modules.Identity.Domain.Aggregates.Tenants; +using ColaFlow.Modules.Identity.Domain.Repositories; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using ColaFlow.Modules.ProjectManagement.Application.Common.Interfaces; +using Microsoft.Extensions.Logging; + +namespace ColaFlow.Modules.Mcp.Application.Resources; + +/// +/// Resource: colaflow://users.list +/// Lists all team members in the current tenant +/// Query params: project (optional filter by project) +/// +public class UsersListResource : IMcpResource +{ + public string Uri => "colaflow://users.list"; + public string Name => "Team Members"; + public string Description => "List all team members in current tenant"; + public string MimeType => "application/json"; + + private readonly IUserRepository _userRepository; + private readonly ITenantContext _tenantContext; + private readonly ILogger _logger; + + public UsersListResource( + IUserRepository userRepository, + ITenantContext tenantContext, + ILogger logger) + { + _userRepository = userRepository; + _tenantContext = tenantContext; + _logger = logger; + } + + public async Task GetContentAsync( + McpResourceRequest request, + CancellationToken cancellationToken) + { + var tenantId = _tenantContext.GetCurrentTenantId(); + + _logger.LogDebug("Fetching users list for tenant {TenantId}", tenantId); + + // Get all users for tenant + var users = await _userRepository.GetAllByTenantAsync(TenantId.Create(tenantId), cancellationToken); + + // Map to DTOs + var userDtos = users.Select(u => new + { + id = u.Id, + email = u.Email.Value, + fullName = u.FullName.ToString(), + status = u.Status.ToString(), + createdAt = u.CreatedAt, + avatarUrl = u.AvatarUrl, + jobTitle = u.JobTitle + }).ToList(); + + var json = JsonSerializer.Serialize(new + { + users = userDtos, + total = userDtos.Count + }, new JsonSerializerOptions { WriteIndented = true }); + + _logger.LogInformation("Retrieved {Count} users for tenant {TenantId}", userDtos.Count, tenantId); + + return new McpResourceContent + { + Uri = Uri, + MimeType = MimeType, + Text = json + }; + } +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Services/IMcpResourceRegistry.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Services/IMcpResourceRegistry.cs new file mode 100644 index 0000000..57b3019 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Services/IMcpResourceRegistry.cs @@ -0,0 +1,30 @@ +using ColaFlow.Modules.Mcp.Contracts.Resources; + +namespace ColaFlow.Modules.Mcp.Application.Services; + +/// +/// Registry for all MCP Resources +/// Manages resource discovery and routing +/// +public interface IMcpResourceRegistry +{ + /// + /// Register a resource + /// + void RegisterResource(IMcpResource resource); + + /// + /// Get all registered resources + /// + IReadOnlyList GetAllResources(); + + /// + /// Get resource by URI (supports URI templates like "colaflow://projects.get/{id}") + /// + IMcpResource? GetResourceByUri(string uri); + + /// + /// Get all resource descriptors (for resources/list method) + /// + IReadOnlyList GetResourceDescriptors(); +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Services/McpResourceRegistry.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Services/McpResourceRegistry.cs new file mode 100644 index 0000000..ae578fe --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Application/Services/McpResourceRegistry.cs @@ -0,0 +1,87 @@ +using System.Text.RegularExpressions; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using Microsoft.Extensions.Logging; + +namespace ColaFlow.Modules.Mcp.Application.Services; + +/// +/// Implementation of MCP Resource Registry +/// +public class McpResourceRegistry : IMcpResourceRegistry +{ + private readonly ILogger _logger; + private readonly Dictionary _resources = new(); + private readonly List _resourceList = new(); + + public McpResourceRegistry(ILogger logger) + { + _logger = logger; + } + + public void RegisterResource(IMcpResource resource) + { + if (_resources.ContainsKey(resource.Uri)) + { + _logger.LogWarning("Resource already registered: {Uri}. Overwriting.", resource.Uri); + } + + _resources[resource.Uri] = resource; + _resourceList.Add(resource); + + _logger.LogInformation("Registered MCP Resource: {Uri} - {Name}", resource.Uri, resource.Name); + } + + public IReadOnlyList GetAllResources() + { + return _resourceList.AsReadOnly(); + } + + public IMcpResource? GetResourceByUri(string uri) + { + // Try exact match first + if (_resources.TryGetValue(uri, out var resource)) + { + return resource; + } + + // Try matching against URI templates (e.g., "colaflow://projects.get/{id}") + foreach (var registeredResource in _resourceList) + { + if (UriMatchesTemplate(uri, registeredResource.Uri)) + { + return registeredResource; + } + } + + return null; + } + + public IReadOnlyList GetResourceDescriptors() + { + return _resourceList.Select(r => new McpResourceDescriptor + { + Uri = r.Uri, + Name = r.Name, + Description = r.Description, + MimeType = r.MimeType + }).ToList().AsReadOnly(); + } + + /// + /// Check if a URI matches a URI template + /// Example: "colaflow://projects.get/123" matches "colaflow://projects.get/{id}" + /// + private bool UriMatchesTemplate(string uri, string template) + { + // Convert template to regex pattern + // Replace {param} with regex group + var pattern = "^" + Regex.Escape(template) + .Replace(@"\{", "{") + .Replace(@"\}", "}") + .Replace("{id}", @"([^/]+)") + .Replace("{projectId}", @"([^/]+)") + + "$"; + + return Regex.IsMatch(uri, pattern); + } +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/IMcpResource.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/IMcpResource.cs new file mode 100644 index 0000000..b74cd77 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/IMcpResource.cs @@ -0,0 +1,38 @@ +namespace ColaFlow.Modules.Mcp.Contracts.Resources; + +/// +/// Interface for MCP Resources +/// Resources provide read-only data to AI agents through the MCP protocol +/// +public interface IMcpResource +{ + /// + /// Resource URI (e.g., "colaflow://projects.list") + /// + string Uri { get; } + + /// + /// Resource display name + /// + string Name { get; } + + /// + /// Resource description + /// + string Description { get; } + + /// + /// MIME type of the resource content (typically "application/json") + /// + string MimeType { get; } + + /// + /// Get resource content + /// + /// Resource request with URI and parameters + /// Cancellation token + /// Resource content + Task GetContentAsync( + McpResourceRequest request, + CancellationToken cancellationToken); +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceContent.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceContent.cs new file mode 100644 index 0000000..60aa712 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceContent.cs @@ -0,0 +1,22 @@ +namespace ColaFlow.Modules.Mcp.Contracts.Resources; + +/// +/// Content returned by an MCP Resource +/// +public class McpResourceContent +{ + /// + /// Resource URI + /// + public string Uri { get; set; } = string.Empty; + + /// + /// MIME type (typically "application/json") + /// + public string MimeType { get; set; } = "application/json"; + + /// + /// Resource content as text (JSON serialized) + /// + public string Text { get; set; } = string.Empty; +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceDescriptor.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceDescriptor.cs new file mode 100644 index 0000000..61b8885 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceDescriptor.cs @@ -0,0 +1,27 @@ +namespace ColaFlow.Modules.Mcp.Contracts.Resources; + +/// +/// Descriptor for an MCP Resource (used in resources/list) +/// +public class McpResourceDescriptor +{ + /// + /// Resource URI + /// + public string Uri { get; set; } = string.Empty; + + /// + /// Resource display name + /// + public string Name { get; set; } = string.Empty; + + /// + /// Resource description + /// + public string Description { get; set; } = string.Empty; + + /// + /// MIME type + /// + public string MimeType { get; set; } = "application/json"; +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceRequest.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceRequest.cs new file mode 100644 index 0000000..c08ecb3 --- /dev/null +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Contracts/Resources/McpResourceRequest.cs @@ -0,0 +1,22 @@ +namespace ColaFlow.Modules.Mcp.Contracts.Resources; + +/// +/// Request object for MCP Resource +/// +public class McpResourceRequest +{ + /// + /// Resource URI + /// + public string Uri { get; set; } = string.Empty; + + /// + /// URI parameters (e.g., {id} from "colaflow://projects.get/{id}") + /// + public Dictionary UriParams { get; set; } = new(); + + /// + /// Query parameters (e.g., ?status=InProgress&priority=High) + /// + public Dictionary QueryParams { get; set; } = new(); +} diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Infrastructure/ColaFlow.Modules.Mcp.Infrastructure.csproj b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Infrastructure/ColaFlow.Modules.Mcp.Infrastructure.csproj index 63212f3..3cdf0ca 100644 --- a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Infrastructure/ColaFlow.Modules.Mcp.Infrastructure.csproj +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Infrastructure/ColaFlow.Modules.Mcp.Infrastructure.csproj @@ -17,9 +17,9 @@ - - - + + + all runtime; build; native; contentfiles; analyzers; buildtransitive diff --git a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Infrastructure/Extensions/McpServiceExtensions.cs b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Infrastructure/Extensions/McpServiceExtensions.cs index 3df2d8b..e38af67 100644 --- a/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Infrastructure/Extensions/McpServiceExtensions.cs +++ b/colaflow-api/src/Modules/Mcp/ColaFlow.Modules.Mcp.Infrastructure/Extensions/McpServiceExtensions.cs @@ -1,5 +1,7 @@ using ColaFlow.Modules.Mcp.Application.Handlers; +using ColaFlow.Modules.Mcp.Application.Resources; using ColaFlow.Modules.Mcp.Application.Services; +using ColaFlow.Modules.Mcp.Contracts.Resources; using ColaFlow.Modules.Mcp.Domain.Repositories; using ColaFlow.Modules.Mcp.Infrastructure.Middleware; using ColaFlow.Modules.Mcp.Infrastructure.Persistence; @@ -34,12 +36,24 @@ public static class McpServiceExtensions // Register application services services.AddScoped(); + // Register resource registry (Singleton - shared across all requests) + services.AddSingleton(); + + // Register MCP Resources + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + services.AddScoped(); + // Register protocol handler services.AddScoped(); // Register method handlers services.AddScoped(); services.AddScoped(); + services.AddScoped(); services.AddScoped(); services.AddScoped(); @@ -57,6 +71,9 @@ public static class McpServiceExtensions /// public static IApplicationBuilder UseMcpMiddleware(this IApplicationBuilder app) { + // Initialize resource registry (register all resources) + InitializeResourceRegistry(app); + // 1. Correlation ID middleware (FIRST - needed for all subsequent logging) app.UseMiddleware(); @@ -74,4 +91,20 @@ public static class McpServiceExtensions return app; } + + /// + /// Initialize resource registry by registering all resources + /// This is called once at startup + /// + private static void InitializeResourceRegistry(IApplicationBuilder app) + { + using var scope = app.ApplicationServices.CreateScope(); + var registry = scope.ServiceProvider.GetRequiredService(); + var resources = scope.ServiceProvider.GetServices(); + + foreach (var resource in resources) + { + registry.RegisterResource(resource); + } + } } diff --git a/colaflow-api/tests/Modules/Mcp/ColaFlow.Modules.Mcp.Tests/Resources/ProjectsListResourceTests.cs b/colaflow-api/tests/Modules/Mcp/ColaFlow.Modules.Mcp.Tests/Resources/ProjectsListResourceTests.cs new file mode 100644 index 0000000..82cc437 --- /dev/null +++ b/colaflow-api/tests/Modules/Mcp/ColaFlow.Modules.Mcp.Tests/Resources/ProjectsListResourceTests.cs @@ -0,0 +1,152 @@ +using System.Text.Json; +using ColaFlow.Modules.Mcp.Application.Resources; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using ColaFlow.Modules.ProjectManagement.Application.Common.Interfaces; +using ColaFlow.Modules.ProjectManagement.Domain.Aggregates.ProjectAggregate; +using ColaFlow.Modules.ProjectManagement.Domain.Enums; +using ColaFlow.Modules.ProjectManagement.Domain.Repositories; +using ColaFlow.Modules.ProjectManagement.Domain.ValueObjects; +using FluentAssertions; +using Microsoft.Extensions.Logging; +using NSubstitute; + +namespace ColaFlow.Modules.Mcp.Tests.Resources; + +/// +/// Unit tests for ProjectsListResource +/// +public class ProjectsListResourceTests +{ + private readonly IProjectRepository _projectRepository; + private readonly ITenantContext _tenantContext; + private readonly ILogger _logger; + private readonly ProjectsListResource _sut; + private readonly Guid _tenantId = Guid.NewGuid(); + + public ProjectsListResourceTests() + { + _projectRepository = Substitute.For(); + _tenantContext = Substitute.For(); + _logger = Substitute.For>(); + + _tenantContext.GetCurrentTenantId().Returns(_tenantId); + + _sut = new ProjectsListResource(_projectRepository, _tenantContext, _logger); + } + + [Fact] + public void Uri_ReturnsCorrectValue() + { + // Assert + _sut.Uri.Should().Be("colaflow://projects.list"); + } + + [Fact] + public void Name_ReturnsCorrectValue() + { + // Assert + _sut.Name.Should().Be("Projects List"); + } + + [Fact] + public void MimeType_ReturnsApplicationJson() + { + // Assert + _sut.MimeType.Should().Be("application/json"); + } + + [Fact] + public async Task GetContentAsync_WithNoProjects_ReturnsEmptyList() + { + // Arrange + _projectRepository.GetAllProjectsReadOnlyAsync(Arg.Any()) + .Returns(new List()); + + var request = new McpResourceRequest { Uri = "colaflow://projects.list" }; + + // Act + var result = await _sut.GetContentAsync(request, CancellationToken.None); + + // Assert + result.Should().NotBeNull(); + result.Uri.Should().Be("colaflow://projects.list"); + result.MimeType.Should().Be("application/json"); + + var data = JsonSerializer.Deserialize(result.Text); + data.GetProperty("projects").GetArrayLength().Should().Be(0); + data.GetProperty("total").GetInt32().Should().Be(0); + } + + [Fact] + public async Task GetContentAsync_WithProjects_ReturnsProjectsList() + { + // Arrange + var project1 = Project.Create( + ColaFlow.Modules.ProjectManagement.Domain.ValueObjects.TenantId.From(_tenantId), + "Project Alpha", + "First project", + "ALPHA", + UserId.Create()); + + var project2 = Project.Create( + ColaFlow.Modules.ProjectManagement.Domain.ValueObjects.TenantId.From(_tenantId), + "Project Beta", + "Second project", + "BETA", + UserId.Create()); + + _projectRepository.GetAllProjectsReadOnlyAsync(Arg.Any()) + .Returns(new List { project1, project2 }); + + var request = new McpResourceRequest { Uri = "colaflow://projects.list" }; + + // Act + var result = await _sut.GetContentAsync(request, CancellationToken.None); + + // Assert + result.Should().NotBeNull(); + var data = JsonSerializer.Deserialize(result.Text); + + data.GetProperty("total").GetInt32().Should().Be(2); + + var projects = data.GetProperty("projects"); + projects.GetArrayLength().Should().Be(2); + + var firstProject = projects[0]; + firstProject.GetProperty("name").GetString().Should().Be("Project Alpha"); + firstProject.GetProperty("key").GetString().Should().Be("ALPHA"); + firstProject.GetProperty("status").GetString().Should().Be(ProjectStatus.Active.ToString()); + } + + [Fact] + public async Task GetContentAsync_CallsTenantContext() + { + // Arrange + _projectRepository.GetAllProjectsReadOnlyAsync(Arg.Any()) + .Returns(new List()); + + var request = new McpResourceRequest { Uri = "colaflow://projects.list" }; + + // Act + await _sut.GetContentAsync(request, CancellationToken.None); + + // Assert + _tenantContext.Received(1).GetCurrentTenantId(); + } + + [Fact] + public async Task GetContentAsync_CallsRepository() + { + // Arrange + _projectRepository.GetAllProjectsReadOnlyAsync(Arg.Any()) + .Returns(new List()); + + var request = new McpResourceRequest { Uri = "colaflow://projects.list" }; + + // Act + await _sut.GetContentAsync(request, CancellationToken.None); + + // Assert + await _projectRepository.Received(1).GetAllProjectsReadOnlyAsync(Arg.Any()); + } +} diff --git a/colaflow-api/tests/Modules/Mcp/ColaFlow.Modules.Mcp.Tests/Services/McpResourceRegistryTests.cs b/colaflow-api/tests/Modules/Mcp/ColaFlow.Modules.Mcp.Tests/Services/McpResourceRegistryTests.cs new file mode 100644 index 0000000..8339ebf --- /dev/null +++ b/colaflow-api/tests/Modules/Mcp/ColaFlow.Modules.Mcp.Tests/Services/McpResourceRegistryTests.cs @@ -0,0 +1,144 @@ +using ColaFlow.Modules.Mcp.Application.Services; +using ColaFlow.Modules.Mcp.Contracts.Resources; +using FluentAssertions; +using Microsoft.Extensions.Logging; +using NSubstitute; + +namespace ColaFlow.Modules.Mcp.Tests.Services; + +/// +/// Unit tests for McpResourceRegistry +/// +public class McpResourceRegistryTests +{ + private readonly ILogger _logger; + private readonly McpResourceRegistry _sut; + + public McpResourceRegistryTests() + { + _logger = Substitute.For>(); + _sut = new McpResourceRegistry(_logger); + } + + [Fact] + public void RegisterResource_AddsResourceToRegistry() + { + // Arrange + var resource = CreateMockResource("colaflow://test", "Test Resource"); + + // Act + _sut.RegisterResource(resource); + + // Assert + var resources = _sut.GetAllResources(); + resources.Should().ContainSingle(); + resources[0].Uri.Should().Be("colaflow://test"); + } + + [Fact] + public void RegisterResource_WithDuplicateUri_Overwrites() + { + // Arrange + var resource1 = CreateMockResource("colaflow://test", "Test Resource 1"); + var resource2 = CreateMockResource("colaflow://test", "Test Resource 2"); + + // Act + _sut.RegisterResource(resource1); + _sut.RegisterResource(resource2); + + // Assert + var resources = _sut.GetAllResources(); + resources.Count.Should().Be(2); // Both are in the list + var foundResource = _sut.GetResourceByUri("colaflow://test"); + foundResource!.Name.Should().Be("Test Resource 2"); // But only the last one is returned + } + + [Fact] + public void GetResourceByUri_WithExactMatch_ReturnsResource() + { + // Arrange + var resource = CreateMockResource("colaflow://projects.list", "Projects List"); + _sut.RegisterResource(resource); + + // Act + var result = _sut.GetResourceByUri("colaflow://projects.list"); + + // Assert + result.Should().NotBeNull(); + result!.Uri.Should().Be("colaflow://projects.list"); + } + + [Fact] + public void GetResourceByUri_WithTemplateMatch_ReturnsResource() + { + // Arrange + var resource = CreateMockResource("colaflow://projects.get/{id}", "Project Details"); + _sut.RegisterResource(resource); + + // Act + var result = _sut.GetResourceByUri("colaflow://projects.get/123"); + + // Assert + result.Should().NotBeNull(); + result!.Uri.Should().Be("colaflow://projects.get/{id}"); + } + + [Fact] + public void GetResourceByUri_WithNonExistentUri_ReturnsNull() + { + // Act + var result = _sut.GetResourceByUri("colaflow://nonexistent"); + + // Assert + result.Should().BeNull(); + } + + [Fact] + public void GetResourceDescriptors_ReturnsAllDescriptors() + { + // Arrange + var resource1 = CreateMockResource("colaflow://test1", "Test 1"); + var resource2 = CreateMockResource("colaflow://test2", "Test 2"); + _sut.RegisterResource(resource1); + _sut.RegisterResource(resource2); + + // Act + var descriptors = _sut.GetResourceDescriptors(); + + // Assert + descriptors.Should().HaveCount(2); + descriptors.Should().Contain(d => d.Uri == "colaflow://test1"); + descriptors.Should().Contain(d => d.Uri == "colaflow://test2"); + } + + [Fact] + public void GetAllResources_ReturnsReadOnlyList() + { + // Arrange + var resource = CreateMockResource("colaflow://test", "Test"); + _sut.RegisterResource(resource); + + // Act + var resources = _sut.GetAllResources(); + + // Assert + resources.Should().BeAssignableTo>(); + } + + private IMcpResource CreateMockResource(string uri, string name) + { + var resource = Substitute.For(); + resource.Uri.Returns(uri); + resource.Name.Returns(name); + resource.Description.Returns($"Description for {name}"); + resource.MimeType.Returns("application/json"); + resource.GetContentAsync(Arg.Any(), Arg.Any()) + .Returns(new McpResourceContent + { + Uri = uri, + MimeType = "application/json", + Text = "{}" + }); + return resource; + } +}