diff options
Diffstat (limited to '')
| -rw-r--r-- | plugins.go | 64 | ||||
| -rw-r--r-- | plugins/hello.lua | 5 | ||||
| -rw-r--r-- | plugins/repology.lua | 33 |
3 files changed, 100 insertions, 2 deletions
@@ -7,6 +7,7 @@ import ( "net/url" "os" "reflect" + "strings" "github.com/ailncode/gluaxmlpath" "github.com/cjoudrey/gluahttp" @@ -197,6 +198,17 @@ func sendMessageClosure(luaState *lua.LState, client *girc.Client) func(*lua.LSt return 0 } } + +func replyToMessageClosure(luaStete *lua.LState, client *girc.Client, event girc.Event) func(*lua.LState) int { + return func(luaState *lua.LState) int { + message := luaState.CheckString(1) + + client.Cmd.Message(event.Source.Name, message) + + return 0 + } +} + func registerLuaCommand(luaState *lua.LState, appConfig *TomlConfig) func(*lua.LState) int { return func(luaState *lua.LState) int { path := luaState.CheckString(1) @@ -410,6 +422,37 @@ func millaModuleLoaderClosure(luaState *lua.LState, client *girc.Client, appConf } } +func millaModuleLoaderEventClosure(luaState *lua.LState, client *girc.Client, appConfig *TomlConfig, event girc.Event) func(*lua.LState) int { + return func(luaState *lua.LState) int { + exports := map[string]lua.LGFunction{ + "send_message": lua.LGFunction(sendMessageClosure(luaState, client)), + "reply_to": lua.LGFunction(replyToMessageClosure(luaState, client, event)), + "join_channel": lua.LGFunction(ircJoinChannelClosure(luaState, client)), + "part_channel": lua.LGFunction(ircPartChannelClosure(luaState, client)), + "send_ollama_request": lua.LGFunction(ollamaRequestClosure(luaState, appConfig)), + "send_gemini_request": lua.LGFunction(geminiRequestClosure(luaState, appConfig)), + "send_chatgpt_request": lua.LGFunction(chatGPTRequestClosure(luaState, appConfig)), + "send_or_request": lua.LGFunction(orRequestClosure(luaState, appConfig)), + "query_db": lua.LGFunction(dbQueryClosure(luaState, appConfig)), + "register_cmd": lua.LGFunction(registerLuaCommand(luaState, appConfig)), + "url_encode": lua.LGFunction(urlEncode(luaState)), + } + millaModule := luaState.SetFuncs(luaState.NewTable(), exports) + + registerStructAsLuaMetaTable[TomlConfig](luaState, millaModule, checkStruct, TomlConfig{}, "toml_config") + registerStructAsLuaMetaTable[CustomCommand](luaState, millaModule, checkStruct, CustomCommand{}, "custom_command") + registerStructAsLuaMetaTable[LogModel](luaState, millaModule, checkStruct, LogModel{}, "log_model") + registerStructAsLuaMetaTable[girc.Source](luaState, millaModule, checkStruct, girc.Source{}, "girc_source") + registerStructAsLuaMetaTable[girc.Event](luaState, millaModule, checkStruct, girc.Event{}, "girc_event") + + luaState.SetGlobal("milla", millaModule) + + luaState.Push(millaModule) + + return 1 + } +} + func RunScript(scriptPath string, client *girc.Client, appConfig *TomlConfig) { luaState := lua.NewState() defer luaState.Close() @@ -469,6 +512,15 @@ func LoadAllPlugins(appConfig *TomlConfig, client *girc.Client) { } } +func LoadAllEventPlugins(appConfig *TomlConfig, client *girc.Client) { + for _, triggeredScript := range appConfig.TriggeredScripts { + log.Print("Loading event plugin: ", triggeredScript.Path) + + go RunScript(triggeredScript.Path, client, appConfig) + registerTriggeredScripts(client, *appConfig) + } +} + func RunLuaFunc( cmd, args string, client *girc.Client, @@ -565,7 +617,7 @@ func RunTriggeredLuaFunc( appConfig.insertLState(scriptPath, luaState, cancel) - luaState.PreloadModule("milla", millaModuleLoaderClosure(luaState, client, appConfig)) + luaState.PreloadModule("milla", millaModuleLoaderEventClosure(luaState, client, appConfig, event)) gluasocket.Preload(luaState) gluaxmlpath.Preload(luaState) luaState.PreloadModule("yaml", gluayaml.Loader) @@ -629,10 +681,18 @@ func RunTriggeredLuaFunc( func registerTriggeredScripts(irc *girc.Client, appConfig TomlConfig) { for _, triggeredScript := range appConfig.TriggeredScripts { - for _, triggerType := range triggeredScript.TriggerType { + for _, triggerType := range triggeredScript.TriggerTypes { switch triggerType { case girc.PRIVMSG: irc.Handlers.AddBg(girc.PRIVMSG, func(_ *girc.Client, event girc.Event) { + if !strings.HasPrefix(event.Last(), appConfig.IrcNick+": ") { + return + } + + if appConfig.AdminOnly && !isFromAdmin(appConfig.Admins, event) { + return + } + RunTriggeredLuaFunc(triggeredScript.FuncName, triggeredScript.Path, irc, event, &appConfig) }) default: diff --git a/plugins/hello.lua b/plugins/hello.lua new file mode 100644 index 0000000..b9c5bc2 --- /dev/null +++ b/plugins/hello.lua @@ -0,0 +1,5 @@ +local milla = require("milla") + +function hello() milla.reply_to("hello") end + +milla.register_cmd("/plugins/hello.lua", "hello", "hello") diff --git a/plugins/repology.lua b/plugins/repology.lua new file mode 100644 index 0000000..f3de647 --- /dev/null +++ b/plugins/repology.lua @@ -0,0 +1,33 @@ +local milla = require("milla") +local os = require("os") +local json = require("json") + +-- /repology void_x86_64 +function repology(arg) + os.setenv("http_proxy", "http://172.17.0.1:8120") + + local http = require("http") + + local url = "https://repology.org/api/v1/repository/" .. arg .. "/problems" + + local response = http.request("GET", url) + + io.write(response.body) + + local json_response, err = json.decode(response.body) + io.write(json_response) + if err ~= nil then print(err) end + + for _, item in pairs(json_response) do + for k, v in ipairs(item) do print(k, v) end + end + + local result = "" + for key, value in pairs(json_response) do + result = result .. key .. ": " .. value .. " -- " + end + + return result +end + +milla.register_cmd("/plugins/repology.lua", "repology", "repology") |