using Microsoft.SemanticKernel;
using Microsoft.SemanticKernel.ChatCompletion;
using System.Text.Json;

namespace ChatApi
{
    public class ChatHistoryService
    {
        private readonly Dictionary<string, ChatHistory> _keyValues = new Dictionary<string, ChatHistory>();
        public ChatHistoryService() 
        { 
        }

        public ChatHistory Get(string sessionId)
        {
            if (_keyValues.ContainsKey(sessionId))
            {
                var history = _keyValues[sessionId];
                return history;
            }
            else
            {
                var msg = new List<ChatMessageContent>();
                PromptLiliana(msg);
                string json = JsonSerializer.Serialize(msg);
                var history = new ChatHistory(JsonSerializer.Deserialize<List<ChatMessageContent>>(json));
                _keyValues[sessionId] = history;
                return _keyValues[sessionId];
            }
        }

        public ChatHistory GetSumarizer(string sessionId)
        {
            if (_keyValues.ContainsKey(sessionId))
            {
                var history = _keyValues[sessionId];
                return history;
            }
            else
            {
                var msg = new List<ChatMessageContent>();
                PromptLiliana(msg);
                string json = JsonSerializer.Serialize(msg);
                var history = new ChatHistory(JsonSerializer.Deserialize<List<ChatMessageContent>>(json));
                _keyValues[sessionId] = history;
                return _keyValues[sessionId];
            }
        }

        public void UpdateHistory(string sessionId, ChatHistory history)
        {
            _keyValues[sessionId] = history;
        }

        public void PromptLiliana(List<ChatMessageContent> msg)
        {
            msg.Add(new ChatMessageContent(AuthorRole.System, "Seu nome é LiliAna, uma assistente virtual para projetos."));
            msg.Add(new ChatMessageContent(AuthorRole.System, "Você responde sempre em português do Brasil e fala sobre detalhes de projeto, arquitetura e criação de casos de teste."));
            msg.Add(new ChatMessageContent(AuthorRole.User, "Use sempre portugues do Brasil."));
        }
    }
}