using BotSharp.Abstraction.Agents.Models;
using BotSharp.Abstraction.Repositories;
using System.IO;

namespace BotSharp.Core.Agents.Services;

public partial class AgentService
{
    public async Task UpdateAgent(Agent agent)
    {
        var db = _services.GetRequiredService<IBotSharpRepository>();

        db.Transaction<IBotSharpTable>(delegate
        {
            var record = (from a in db.Agent
                          join ua in db.UserAgent on a.Id equals ua.AgentId
                          join u in db.User on ua.UserId equals u.Id
                          where (ua.UserId == _user.Id || u.ExternalId == _user.Id) && 
                            a.Id == agent.Id
                          select a).First();

            record.Name = agent.Name;
            record.Description = agent.Description;
            record.UpdatedTime = DateTime.UtcNow;
        });

        // Save instruction to file
        var dir = GetAgentDataDir(agent.Id);
        var instructionFile = Path.Combine(dir, "instruction.txt");
        File.WriteAllText(instructionFile, agent.Instruction);

        var samplesFile = Path.Combine(dir, "samples.txt");
        File.WriteAllText(samplesFile, agent.Samples);
    }
}