BotSharp/src/Plugins/BotSharp.Plugin.SqlDriver/Hooks/SqlDriverAgentHook.cs

using BotSharp.Abstraction.Agents.Enums;
using BotSharp.Abstraction.Agents.Settings;

namespace BotSharp.Plugin.SqlDriver.Hooks;

public class SqlDriverAgentHook : AgentHookBase, IAgentHook
{
    private const string SQL_TABLE_DEFINITION_FN = "sql_table_definition";
    private const string VERIFY_DICTIONARY_TERM_FN = "verify_dictionary_term";
    private const string SQL_SELECT_FN = "sql_select";

    public override string SelfId => BuiltInAgentId.Planner;

    public SqlDriverAgentHook(IServiceProvider services, AgentSettings settings)
        : base(services, settings)
    {
    }

    public override void OnAgentLoaded(Agent agent)
    {
        var dbType = SqlDriverHelper.GetDatabaseType(_services);
        agent.TemplateDict["db_type"] = dbType;

        var utilityLoads = new List<AgentUtility>
        {
            new AgentUtility
            {
                Name = UtilityName.SqlTableDefinition,
                Content = new UtilityContent
                {
                    Functions = new List<UtilityFunction> { new(SQL_TABLE_DEFINITION_FN) },
                    Templates = new List<UtilityTemplate> { new($"{SQL_TABLE_DEFINITION_FN}.fn") }
                }
            },
            new AgentUtility
            {
                Name = UtilityName.SqlDictionaryLookup,
                Content = new UtilityContent
                {
                    Functions = new List<UtilityFunction> { new(VERIFY_DICTIONARY_TERM_FN) },
                    Templates = new List<UtilityTemplate> { new($"{VERIFY_DICTIONARY_TERM_FN}.fn") }
                }
            },
            new AgentUtility
            {
                Name = UtilityName.SqlExecutor,
                Content = new UtilityContent
                {
                    Functions = new List<UtilityFunction> { new(SQL_SELECT_FN), new(SQL_TABLE_DEFINITION_FN) },
                    Templates = new List<UtilityTemplate> { new($"sql_executor.fn") }
                }
            }
        };

        base.OnLoadAgentUtility(agent, utilityLoads);
        base.OnAgentLoaded(agent);
    }
}