Prompt.cs

using System.Collections.Generic;
using System.Diagnostics.CodeAnalysis;
using System.Reflection;
using System.Threading.Tasks;
using Waher.Networking.HTTP.JsonRpc;
using Waher.Networking.HTTP.Mcp.Model.Attributes;
using Waher.Runtime.Collections;
 
namespace Waher.Networking.HTTP.Mcp.Model.Server
{
    public class Prompt : ProtectedMethod
    {
        private Icons? icons = null;
 
        public Prompt(MethodInfo Method, string Title, string Description, 
            string IconsMethod, params KeyValuePair<string, object>[] MetaData)
            : base(Method, false)
        {
            this.Title = Title;
            this.Description = Description;
            this.IconsMethod = IconsMethod;
            this.MetaData = MetaData;
            this.ReturnAttributes = Method.ReturnParameter?.GetCustomAttribute<McpParameterAttribute>();
        }
 
        public string Title { get; }
 
        public string Description { get; }
 
        public string IconsMethod { get; }
 
        public KeyValuePair<string, object>[] MetaData { get; }
 
        public McpParameterAttribute? ReturnAttributes { get; }
 
        public async Task<Dictionary<string, object>> ToJson(HttpMcpServerResource Resource)
        {
            this.icons ??= await Tool.GetIcons(Resource, this.IconsMethod);
 
            Dictionary<string, object> Result = new Dictionary<string, object>()
            {
                { "name", this.Method.Name }
            };
 
            PopulateArguments(this.Method, Result);
 
            if (!string.IsNullOrEmpty(this.Title))
                Result.Add("title", this.Title);
 
            if (!string.IsNullOrEmpty(this.Description))
                Result.Add("description", this.Description);
 
            if (!this.icons.Empty)
                Result.Add("icons", this.icons.ToJson());
 
            if ((this.MetaData?.Length ?? 0) > 0)
            {
                Dictionary<string, object> MetaData = new Dictionary<string, object>();
 
                foreach (KeyValuePair<string, object> P in this.MetaData!)
                    MetaData[P.Key] = P.Value;
 
                Result["_meta"] = MetaData;
            }
 
            return Result;
        }
 
        private static void PopulateArguments(MethodInfo Method,
            Dictionary<string, object> Result)
        {
            ParameterInfo[] Parameters = Method.GetParameters();
 
            ChunkedList<Dictionary<string, object?>>? Arguments = null;
 
            foreach (ParameterInfo Parameter in Parameters)
            {
                Dictionary<string, object?> PromptArgument = new Dictionary<string, object?>()
                {
                    { "name", Parameter.Name },
                    { "required", !Parameter.HasDefaultValue }
                };
 
                McpParameterAttribute? Attribute = Parameter.GetCustomAttribute<McpParameterAttribute>();
                Attribute?.Annotate(PromptArgument);
 
                Arguments ??= new ChunkedList<Dictionary<string, object?>>();
                Arguments.Add(PromptArgument);
            }
 
            if (!(Arguments is null))
                Result["arguments"] = Arguments.ToArray();
        }
 
        public bool TryBuildRequest(object? Id, Dictionary<string, object?> Parameters,
            HttpRequest Request, HttpResponse Response,
            Dictionary<string, object?>? MetaData,
            [NotNullWhen(false)] out string? Reason,
            [NotNullWhen(true)] out object?[]? Arguments)
        {
            if (!this.TryBuildRequest(Id, Parameters, MetaData, out Reason, out Arguments))
                return false;
 
            if (this.RequestArgument.HasValue)
                Arguments[this.RequestArgument.Value] = Request;
 
            if (this.ResponseArgument.HasValue)
                Arguments[this.ResponseArgument.Value] = Response;
 
            if (this.IdArgument.HasValue)
                Arguments[this.IdArgument.Value] = Id;
 
            return true;
        }
 
    }
}