namespace BotSharp.Plugin.SqlDriver.Hooks;

public class SqlUtilityHook : IAgentUtilityHook
{
    private const string PREFIX = "util-db-";
    private const string SQL_TABLE_DEFINITION_FN = $"{PREFIX}sql_table_definition";
    private const string VERIFY_DICTIONARY_TERM_FN = $"{PREFIX}verify_dictionary_term";
    private const string SQL_SELECT_FN = $"{PREFIX}sql_select";
    private const string SQL_EXECUTOR_FN = $"{PREFIX}sql_executor";

    public void AddUtilities(List<AgentUtility> utilities)
    {
        var items = new List<AgentUtility>
        {
            new AgentUtility
            {
                Category = "database",
                Name = "sql.tools",
                Items = [
                    new UtilityItem
                    {
                        FunctionName = SQL_TABLE_DEFINITION_FN,
                        TemplateName = $"{SQL_TABLE_DEFINITION_FN}.fn"
                    },
                    new UtilityItem
                    {
                        FunctionName = VERIFY_DICTIONARY_TERM_FN,
                        TemplateName = $"{VERIFY_DICTIONARY_TERM_FN}.fn"
                    },
                    new UtilityItem
                    {
                        FunctionName = SQL_SELECT_FN,
                        TemplateName = $"{SQL_EXECUTOR_FN}.fn"
                    }
                ]
            }
        };

        utilities.AddRange(items);
    }
}