1 files changed, 39 insertions, 0 deletions
diff --git a/plugins/robtex.lua b/plugins/robtex.lua
new file mode 100644
index 0000000..79b7b63
--- /dev/null
+++ b/plugins/robtex.lua
@@ -0,0 +1,39 @@
+local milla = require("milla")
+local os = require("os")
+
+-- https://www.robtex.com/api/
+function robtex(cli_args)
+    local args = {}
+    for i in string.gmatch(cli_args, "%S+") do table.insert(args, i) end
+
+    os.setenv("http_proxy", "http://172.17.0.1:8120")
+
+    local http = require("http")
+
+    local url_ipquery = "https://freeapi.robtex.com/ipquery/"
+    local url_asquery = "https://freeapi.robtex.com/asquery/"
+    local url_pdns_forward = "https://freeapi.robtex.com/pdns/forward/"
+    local url_rpns_reverse = "https://freeapi.robtex.com/pdns/reverse"
+
+    local url
+
+    if args[1] == "ipquery" then
+        url = url_ipquery .. args[2]
+    elseif args[1] == "asquery" then
+        url = url_asquery .. args[2]
+    elseif args[1] == "pdns_forward" then
+        url = url_pdns_forward .. args[2]
+    elseif args[1] == "pdns_reverse" then
+        url = url_rpns_reverse .. args[2]
+    else
+        return "Invalid command"
+    end
+
+    local response = http.request("GET", url)
+
+    io.write(response.body)
+
+    return response.body
+end
+
+milla.register_cmd("/plugins/robtex.lua", "robtex", "robtex")