42 lines
1.4 KiB
C#
42 lines
1.4 KiB
C#
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);
|
|
}
|
|
}
|