diff options
Diffstat (limited to '')
-rw-r--r-- | plugins.go | 27 | ||||
-rw-r--r-- | plugins/ip.lua | 17 | ||||
-rw-r--r-- | plugins/rss.lua | 2 | ||||
-rw-r--r-- | plugins/urban.lua | 46 |
4 files changed, 68 insertions, 24 deletions
@@ -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 |