Merge pull request #1 from kdssoftware/feat/init

Add LuaObfuscatorCurrent command

Author: kdssoftware

.stylua.toml

@@ -0,0 +1,6 @@
+column_width = 100
+line_endings = "Unix"
+indent_type = "Spaces"
+indent_width = 2
+quote_style = "AutoPreferSingle"
+call_parentheses = "Always"

README.md

@@ -1,2 +1,6 @@
 # lua-obfuscator.nvim
 A plugin to use https://luaobfuscator.com/ for Neovim
+
+
+## Commands
+- `LuaObfuscatorCurrent` : obfuscated 

doc/nvim-awesome-plugin.txt

@@ -0,0 +1,29 @@
+local M = {}
+
+local api = require('lua-obfuscator.api')
+local obf_commands = {LuaObfuscatorCurrent = api.obfuscateScript}
+
+local create_obf_commands = function()
+    for name, obf_api in pairs(obf_commands) do
+        vim.api.nvim_create_user_command(name, function(args)
+            local bufnr = vim.fn.bufnr("%")
+            local script_content = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)
+
+            obf_api({
+                script_content = table.concat(script_content, '\n'),
+            })
+        end, {
+            nargs = '*',
+            range = true,
+            complete = function()
+                local cmdline = vim.fn.getcmdline()
+                local _, space_count = string.gsub(cmdline, ' ', '')
+                return space_count == 1 and M.anchor_positions or {}
+            end
+        })
+    end
+end
+
+M.setup = function(opts) create_obf_commands() end
+
+return M

doc/tags

@@ -0,0 +1,47 @@
+local M = {}
+
+local call = require('lua-obfuscator.call')
+
+---- UTILS ----
+local get_range = function(opts)
+    local range
+    if opts.range == 2 then range = {start = opts.line1, _end = opts.line2} end
+    return range
+end
+
+local get_opts = function(opts)
+    opts = opts or {}
+    opts.range = get_range(opts)
+    return opts
+end
+---- END UTILS ----
+
+M.obfuscateScript = function(opts)
+    local script_content = tostring(opts.script_content)
+    if script_content == "" then
+        return vim.api.nvim_err_writeln("No content to obfuscate")
+    end
+
+    local respScriptUpload = call.newScript(script_content)
+    if respScriptUpload.error and not respScriptUpload.sessionId then
+        vim.api.nvim_err_writeln("\nError uploading script: " .. tostring(respScriptUpload.error))
+        return
+    end
+
+    local respObf = call.newObfuscateRequest(respScriptUpload.sessionId)
+    if respObf.error then
+        vim.api.nvim_err_writeln("\nError obfuscating: " .. tostring( respObf.error))
+        return
+    end
+    if not respObf.code then
+        vim.api.nvim_err_writeln("\nError obfuscating: obfuscation failed")
+        return
+    end
+    local obfuscatedCode = respObf.code
+    local current_bufnr = vim.fn.bufnr()
+
+    -- Replace the entire script with obfuscatedCode
+    vim.api.nvim_buf_set_lines(current_bufnr, 0, -1, true, {obfuscatedCode})
+end
+
+return M

lua/lua-obfuscator.lua

@@ -0,0 +1,105 @@
+local M = {}
+local json = require("lua-obfuscator.json")
+local utils = require("lua-obfuscator.utils")
+
+-- TODO port to utils 
+
+local settings = {
+    minifyAll = true,
+    virtualize = false,
+    customPlugins = {
+        SwizzleLookups = 100,
+        EncryptStrings = 100,
+        RevertAllIfStatements = 50,
+        ControlFlowFlattenAllBlocks = 75
+    }
+}
+local settingsData = {
+    MinifiyAll = settings.minifyAll,
+    Virtualize = settings.virtualize,
+    CustomPlugins = {
+        SwizzleLookups = {settings.customPlugins.SwizzleLookups},
+        EncryptStrings = {settings.customPlugins.EncryptStrings},
+        RevertAllIfStatements = {settings.customPlugins.RevertAllIfStatements},
+        ControlFlowFlattenAllBlocks = {
+            settings.customPlugins.ControlFlowFlattenAllBlocks
+        }
+    }
+}
+
+-- @param script string: Lua script to be obfuscated
+-- @return table: Returns a table with session ID and error message, both of which can be nil or string.
+M.newScript = function(script)
+    local endpoint = "https://luaobfuscator.com/api/obfuscator/newscript"
+    local headers = "-H 'Content-type: text/plain' -H 'apikey:test'"
+
+    local requestBody = json.encode({script = script})
+
+    local command = string.format('curl -X POST %s -s %s --data-raw %q',
+                                  requestBody, endpoint, headers)
+    local result = vim.fn.system(command)
+
+    if vim.v.shell_error ~= 0 then
+        return {code = nil, error = "Error executing the command."}
+    end
+
+    local response = json.decode(result)
+    if not response then
+        return {sessionId = nil, error = "Error decoding Lua response."}
+    end
+
+    if response and response.sessionId then
+        local sessionId = tostring(response.sessionId)
+        return {sessionId = sessionId, error = nil}
+    else
+        return {sessionId = nil, error = "Error decoding Lua response."}
+    end
+end
+
+-- @param sessionId string: The session ID obtained from script upload.
+-- @return table: Returns a table with obfuscated code and error message, both of which can be nil or string.
+M.newObfuscateRequest = function(sessionId)
+    local endpoint = "https://luaobfuscator.com/api/obfuscator/obfuscate/"
+
+    local headers =
+        "-H 'Content-type: text/plain' -H 'apikey: test' -H 'sessionId: " ..
+            sessionId .. "'"
+
+    local jsonData =
+        '{CustomPlugins:{ControlFlowFlattenAllBlocks:[75],SwizzleLookups:[100],EncryptStrings:[100],RevertAllIfStatements:[50]},MinifiyAll:true,Virtualize:false}'
+
+    local command = string.format('curl -X POST %s -s %s --data-raw %q',
+                                  endpoint, headers, json.decode(settingsData))
+    local result = vim.fn.system(command)
+
+    if vim.v.shell_error ~= 0 then
+        return {code = nil, error = "Error executing the command."}
+    end
+
+    local response = json.decode(result)
+    if response.title == "Unauthorized" then
+        return {
+            code = nil,
+            error = "Looks like the server is down: \n" .. tostring(result)
+        }
+    end
+
+    if string.find(result, '"code":null', 1, true) then -- could try to use response.code
+        return {code = nil, error = "failed to obfuscate: " .. tostring(result)}
+    end
+
+    -- Interpret the Lua code in the response
+    if response and type(response) == "table" and response.code then
+        return {
+            code = response.code,
+            error = nil
+        }
+    else
+        return {
+            code = nil,
+            error = "Error decoding Lua response or missing 'code' field."
+        }
+    end
+end
+
+return M