From 81469bbb1745a835676c3664e75a4a951223da9d Mon Sep 17 00:00:00 2001 From: terminaldweller Date: Tue, 30 Jul 2024 19:25:59 -0400 Subject: * fixed a crash when unloading a lua script * added event types, foreground and background color for watchlists * added a url_encode function to lua to encode urls. Underneath, it just calls the standard library function from golang * updated the README * the urban plugin now can take in the number of entries to return * reverted a bug where setting the http proxy for the lua http module was not working * fixed the url for the ip script so it is actually working. the current provider does not support ipv6 though --- README.md | 11 +++++++++-- main.go | 53 +++++++++++++++++++++++++++++++++++++++++++---------- plugins.go | 27 +++++++++++++++++++-------- plugins/ip.lua | 17 +++++++++-------- plugins/rss.lua | 2 -- plugins/urban.lua | 46 ++++++++++++++++++++++++++++++++++++++++------ types.go | 8 +++++++- 7 files changed, 127 insertions(+), 37 deletions(-) diff --git a/README.md b/README.md index 467a5b7..58713b1 100644 --- a/README.md +++ b/README.md @@ -272,6 +272,9 @@ Watchlists allow you to specify a list of channels to watch. The watched values watchList = ["#securityfeeds"] watchFiles = ["/watchfiles/voidbox.list"] alertChannel = "#milla_alerts" +eventTypes = ["PRIVMSG"] +fgColor = 0 +bgColor = 28 ``` ### Example Config File @@ -706,6 +709,10 @@ milla.query_db(query) milla.register_cmd(script_path, cmd_name, function_name) ``` +```lua +milla.url_encode(str) +``` + Using `register_cmd` we can register a command that will be available to run like the built-in and customs commands.
Here's an example of how to use it:
@@ -725,7 +732,7 @@ local http = require("http") -- should only return one string value function milla_get_ip(arg) local ip = arg - local response, err = http.request("GET", "http://ip-api.com/json?" .. ip) + local response, err = http.request("GET", "http://ip-api.com/json/" .. ip) if err ~= nil then print(err) end local json_response, err = json.decode(response.body) @@ -746,7 +753,7 @@ milla.register_cmd("/plugins/ip.lua", "ip", "milla_get_ip") This will allow us to do:
``` -/terra: /ip 1.1.1.1 +terra: /ip 1.1.1.1 ``` The example rss plugin, accepts a yaml file as input, reeds the provided rss feeds once, extracts the title, author name and link to the resource, sends the feed over to the `#rssfeed` irc channel and exits.
diff --git a/main.go b/main.go index 619ea63..05133c0 100644 --- a/main.go +++ b/main.go @@ -554,7 +554,12 @@ func runCommand( break } - result := RunLuaFunc(args[0], args[1], client, appConfig) + luaArgs := strings.TrimPrefix(event.Last(), appConfig.IrcNick+": ") + luaArgs = strings.TrimSpace(luaArgs) + luaArgs = strings.TrimPrefix(luaArgs, "/") + luaArgs = strings.TrimPrefix(luaArgs, args[0]) + + result := RunLuaFunc(args[0], luaArgs, client, appConfig) client.Cmd.Reply(event, result) } } @@ -640,11 +645,6 @@ func DoOllamaRequest( return "", err } - if err != nil { - - return "", err - } - defer response.Body.Close() log.Println("response body:", response.Body) @@ -1086,17 +1086,50 @@ func populateWatchListWords(appConfig *TomlConfig) { } func WatchListHandler(irc *girc.Client, appConfig TomlConfig) { - irc.Handlers.AddBg(girc.PRIVMSG, func(_ *girc.Client, event girc.Event) { + irc.Handlers.AddBg(girc.ALL_EVENTS, func(_ *girc.Client, event girc.Event) { + var isRightEventType bool + sarray := suffixarray.New([]byte(event.Last())) + if len(event.Params) == 0 { + return + } + for watchname, watchlist := range appConfig.WatchLists { for _, channel := range watchlist.WatchList { + isRightEventType = false + if channel == event.Params[0] { + + for _, eventType := range watchlist.EventTypes { + if eventType == event.Command { + isRightEventType = true + + break + } + } + + if !isRightEventType { + continue + } + for _, word := range watchlist.Words { - indexes := sarray.Lookup([]byte(word), -1) + indexes := sarray.Lookup([]byte(" "+word+" "), 1) if len(indexes) > 0 { - irc.Cmd.Message(watchlist.AlertChannel, fmt.Sprintf("%s: %s", watchname, event.Last())) - log.Printf("%s: %s", watchname, event.Last()) + nextWhitespaceIndex := strings.Index(event.Last()[indexes[0]+1:], " ") + + rewrittenMessage := + event.Last()[:indexes[0]+1] + + fmt.Sprintf("\x1b[48;5;%dm", watchlist.BGColor) + + fmt.Sprintf("\x1b[38;5;%dm", watchlist.FGColor) + + event.Last()[indexes[0]+1:indexes[0]+1+nextWhitespaceIndex] + + "\x1b[0m" + event.Last()[indexes[0]+1+nextWhitespaceIndex:] + + irc.Cmd.Message( + watchlist.AlertChannel, + fmt.Sprintf("%s: %s", watchname, rewrittenMessage)) + + log.Printf("matched from watchlist -- %s: %s", watchname, event.Last()) break } diff --git a/plugins.go b/plugins.go index c2775db..8e15976 100644 --- a/plugins.go +++ b/plugins.go @@ -197,7 +197,6 @@ func sendMessageClosure(luaState *lua.LState, client *girc.Client) func(*lua.LSt return 0 } } - func registerLuaCommand(luaState *lua.LState, appConfig *TomlConfig) func(*lua.LState) int { return func(luaState *lua.LState) int { path := luaState.CheckString(1) @@ -317,6 +316,15 @@ func dbQueryClosure(luaState *lua.LState, appConfig *TomlConfig) func(*lua.LStat } } +func urlEncode(luaState *lua.LState) func(*lua.LState) int { + return func(luaState *lua.LState) int { + URL := luaState.CheckString(1) + escapedURL := url.QueryEscape(URL) + luaState.Push(lua.LString(escapedURL)) + return 1 + } +} + func millaModuleLoaderClosure(luaState *lua.LState, client *girc.Client, appConfig *TomlConfig) func(*lua.LState) int { return func(luaState *lua.LState) int { exports := map[string]lua.LGFunction{ @@ -328,6 +336,7 @@ func millaModuleLoaderClosure(luaState *lua.LState, client *girc.Client, appConf "send_chatgpt_request": lua.LGFunction(chatGPTRequestClosure(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) @@ -425,20 +434,20 @@ func RunLuaFunc( luaState.PreloadModule("json", gopherjson.Loader) var proxyString string - switch proxyString { - case os.Getenv("ALL_PROXY"): + if os.Getenv("ALL_PROXY") != "" { proxyString = os.Getenv("ALL_PROXY") - case os.Getenv("HTTPS_PROXY"): + } else if os.Getenv("HTTPS_PROXY") != "" { proxyString = os.Getenv("HTTPS_PROXY") - case os.Getenv("HTTP_PROXY"): + } else if os.Getenv("HTTP_PROXY") != "" { proxyString = os.Getenv("HTTP_PROXY") - case os.Getenv("https_proxy"): + } else if os.Getenv("https_proxy") != "" { proxyString = os.Getenv("https_proxy") - case os.Getenv("http_proxy"): + } else if os.Getenv("http_proxy") != "" { proxyString = os.Getenv("http_proxy") - default: } + log.Print("set proxy env to:", proxyString) + proxyTransport := &http.Transport{} if proxyString != "" { @@ -465,6 +474,8 @@ func RunLuaFunc( Protect: true, } + log.Print(cmd) + log.Print(args) if err := luaState.CallByParam(funcLValue, lua.LString(args)); err != nil { log.Print("failed running lua command ...") log.Print(err) diff --git a/plugins/ip.lua b/plugins/ip.lua index 2e9afad..79ac6e2 100644 --- a/plugins/ip.lua +++ b/plugins/ip.lua @@ -2,18 +2,19 @@ local milla = require("milla") local os = require("os") local json = require("json") --- setting the proxy value before loading the http module --- this way, only this script will be using this proxy -os.setenv("ALL_PROXY", "socks5://172.17.0.1:9057") - -local http = require("http") - -- this function should be global -- one string arg that holds all args -- should only return one string value function milla_get_ip(arg) - local ip = arg - local response, err = http.request("GET", "http://ip-api.com/json?" .. ip) + -- setting the proxy value before loading the http module + -- this way, only this script will be using this proxy + os.setenv("http_proxy", "http://172.17.0.1:8120") + + local http = require("http") + + local url = "http://ip-api.com/json/" .. arg + + local response, err = http.request("GET", url) if err ~= nil then print(err) end local json_response, err = json.decode(response.body) diff --git a/plugins/rss.lua b/plugins/rss.lua index 880a0fd..111baa2 100644 --- a/plugins/rss.lua +++ b/plugins/rss.lua @@ -47,7 +47,6 @@ local function get_rss_feed(config) end path, err = xmlpath.compile("//entry/author/name") - -- local path, err = xmlpath.compile("//entry/title") if err ~= nil then milla.send_message(err, "") goto continue @@ -58,7 +57,6 @@ local function get_rss_feed(config) end path, err = xmlpath.compile("//entry/author/uri") - -- local path, err = xmlpath.compile("//entry/title") if err ~= nil then milla.send_message(err, "") goto continue diff --git a/plugins/urban.lua b/plugins/urban.lua index 08c610b..7b7e71c 100644 --- a/plugins/urban.lua +++ b/plugins/urban.lua @@ -2,15 +2,42 @@ local milla = require("milla") local os = require("os") local json = require("json") -os.setenv("ALL_PROXY", "socks5://172.17.0.1:9057") +os.setenv("ALL_PROXY", "socks5://172.17.0.1:9004") local http = require("http") -function milla_urban(arg) +function milla_urban(cli_args) + local args = {} + for i in string.gmatch(cli_args, "%S+") do table.insert(args, i) end + + for k, v in ipairs(args) do print(k, v) end + + local count = 1 + local term = "" + + local skip = false + + for i = 1, #args do + if skip then + skip = false + goto continue + end + if args[i] == "-n" then + count = tonumber(args[i + 1]) + skip = true + else + term = term .. args[i] .. " " + end + ::continue:: + end + print("Term: " .. term) + print("Count: " .. count) + local user_agent = "Mozilla/5.0 (X11; U; Linux x86_64; pl-PL; rv:1.9.2.10) Gecko/20100922 Ubuntu/10.10 (maverick) Firefox/3.6.10" - local term = arg - local url = "http://api.urbandictionary.com/v0/define?term=" .. term + + local escaped_term = milla.url_encode(term) + local url = "http://api.urbandictionary.com/v0/define?term=" .. escaped_term local response, err = http.request("GET", url, { timeout = "10s", headers = { @@ -30,10 +57,17 @@ function milla_urban(arg) local json_response, err = json.decode(response.body) if err ~= nil then print(err) end + if response.status_code ~= 200 then + return "Error: " .. response.status_code + end + local result = "" - for k, v in ipairs(json_response["list"]) do + for _, v in ipairs(json_response["list"]) do for kk, vv in pairs(v) do print(kk, vv) end - if k == 1 then result = v["definition"] end + if count > 0 then + result = result .. tostring(count) .. v["definition"] .. "----" + end + count = count - 1 end return result diff --git a/types.go b/types.go index d6f3236..85f820d 100644 --- a/types.go +++ b/types.go @@ -33,6 +33,9 @@ type WatchList struct { WatchList []string `toml:"watchList"` WatchFiles []string `toml:"watchFiles"` Words []string `toml:"watchWords"` + EventTypes []string `toml:"eventTypes"` + FGColor int `toml:"fgColor"` + BGColor int `toml:"bgColor"` } type LuaCommand struct { @@ -114,7 +117,10 @@ func (config *TomlConfig) deleteLstate(name string) { if config.LuaStates == nil { return } - config.LuaStates[name].Cancel() + + if config.LuaStates[name].Cancel != nil { + config.LuaStates[name].Cancel() + } delete(config.LuaStates, name) } -- cgit v1.2.3