using McpDotNet.Protocol.Transport;
using McpDotNet.Protocol.Types;
using McpDotNet.Server;
using Microsoft.Extensions.Logging;
using Serilog;
using System.Dynamic;
using System.Text;
using System.Text.Json;

namespace BotSharp.PizzaBot.MCPServer
{
    internal class Program
    {
        private static async Task Main(string[] args)
        {
            Console.WriteLine("Starting server...");

            McpServerOptions options = new McpServerOptions()
            {
                ServerInfo = new Implementation() { Name = "PizzaServer", Version = "1.0.0" },
                Capabilities = new ServerCapabilities()
                {
                    Tools = new(),
                    Resources = new(),
                    Prompts = new(),
                },
                ProtocolVersion = "2024-11-05"
            };
            var loggerFactory = CreateLoggerFactory();
            McpServerFactory factory = new McpServerFactory(new StdioServerTransport("PizzaServer", loggerFactory), options, loggerFactory,
                "This is a test server with only stub functionality");
            IMcpServer server = factory.CreateServer();

            Console.WriteLine("Server object created, registering handlers.");

            #region Tools
            server.ListToolsHandler = (request, cancellationToken) =>
            {
                return Task.FromResult(new ListToolsResult()
                {
                    Tools =
                    [
                        new Tool()
                    {
                        Name = "make_payment",
                        Description = "call this function to make payment", 
                        InputSchema = new JsonSchema()
                        {
                            Type = "object",
                            Properties = new Dictionary<string, JsonSchemaProperty>()
                            {
                                ["order_number"] = new JsonSchemaProperty() { Type = "string", Description = "order number." },
                                ["total_amount"] = new JsonSchemaProperty() { Type = "string", Description = "total amount." },
                            }, 
                            Required = new List<string>() { "order_number", "total_amount" }
                        }, 
                        }
                    ]
                });
            };

            server.CallToolHandler = async (request, cancellationToken) =>
            {
                if (request.Name == "make_payment")
                {
                    if (request.Arguments is null || !request.Arguments.TryGetValue("order_number", out var order_number))
                    {
                        throw new McpServerException("Missing required argument 'order_number'");
                    }
                    if (request.Arguments is null || !request.Arguments.TryGetValue("total_amount", out var total_amount))
                    {
                        throw new McpServerException("Missing required argument 'total_amount'");
                    }
                    dynamic message = new ExpandoObject();
                    message.pepperoni_unit_price = 3.2;
                    message.cheese_unit_price = 3.5;
                    message.margherita_unit_price = 3.8;

                    // Serialize the message to JSON
                    var jso = new JsonSerializerOptions() { WriteIndented = true };
                    var jsonMessage = JsonSerializer.Serialize(message, jso);

                    return new CallToolResponse()
                    {                         
                        Content = [new Content() { Text = jsonMessage , Type = "text" }]
                    };
                }
                else
                {
                    throw new McpServerException($"Unknown tool: {request.Name}");
                }
            };
            #endregion  

            Console.WriteLine("Server initialized.");

            await server.StartAsync();

            Console.WriteLine("Server started.");

            // Run until process is stopped by the client (parent process)
            while (true)
            {
                await Task.Delay(1000);
            }
        }

        private static ILoggerFactory CreateLoggerFactory()
        {
            // Use serilog
            Log.Logger = new LoggerConfiguration()
                .MinimumLevel.Verbose() // Capture all log levels
                .WriteTo.File(Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "logs", "TestServer_.log"),
                    rollingInterval: RollingInterval.Day,
                    outputTemplate: "{Timestamp:yyyy-MM-dd HH:mm:ss.fff zzz} [{Level:u3}] {Message:lj}{NewLine}{Exception}")
                .CreateLogger();

            var logsPath = Path.Combine(AppContext.BaseDirectory, "testserver.log");
            return LoggerFactory.Create(builder =>
            {
                builder.AddSerilog();
            });
        }
    }
}