feat: allow executing unsaved buffers anonymously

jonathanmorris180 · Apr 23, 2024 · a08fa1c · a08fa1c
1 parent e95d101
commit a08fa1c
Showing 1 changed file with 38 additions and 18 deletions.
diff --git a/lua/salesforce/execute_anon.lua b/lua/salesforce/execute_anon.lua
@@ -16,38 +16,56 @@ end
 local M = {}
 
 M.execute_anon = function()
-    Debug:log("execute_anon.lua", "Executing anonymous Apex script......")
+    Debug:log("execute_anon.lua", "Executing anonymous Apex script...")
+    local disallowed_extensions = { "cls", "trigger" }
     local path = vim.fn.expand("%:p")
+    local extension = vim.fn.expand("%:e")
     local file_type = vim.bo.filetype
     local default_username = OrgManager:get_default_username()
+    local file_contents = ""
+    local is_unsaved_buffer = false
 
-    if file_type == "apex" then
-        if path == "" then
-            Debug:log("execute_anon.lua", "Executing anonymous Apex script for current buffer...")
-            path = vim.fn.expand("%")
-            if not default_username then
-                Util.notify_default_org_not_set()
-                return
-            end
-        else
-            Debug:log("execute_anon.lua", "Executing anonymous Apex script for file...")
+    if file_type == "apex" and not vim.tbl_contains(disallowed_extensions, extension) then
+        if vim.fn.empty(path) == 1 then
+            Debug:log("execute_anon.lua", "File path is empty, retrieving buffer contents...")
+            is_unsaved_buffer = true
+            local bufnr = vim.api.nvim_get_current_buf()
+            local lines = vim.api.nvim_buf_get_lines(bufnr, 0, -1, false)
+            file_contents = vim.fn.join(lines, "\n")
         end
     else
-        local message = "Not an Apex script file"
+        local message = "Not a valid Apex script"
         Debug:log("execute_anon.lua", message)
         vim.notify(message, vim.log.levels.ERROR)
         return
     end
 
+    if not default_username then
+        Util.notify_default_org_not_set()
+        return
+    end
+
     Popup:create_popup({})
     Popup:write_to_popup("Executing anonymous Apex script...")
     local executable = Config:get_options().sf_executable
-    local command = string.format("%s apex run -f '%s' -o %s", executable, path, default_username)
-    Debug:log("execute_anon.lua", "Running " .. command .. "...")
+    local args = {}
+    local writer = nil
+    if not is_unsaved_buffer then
+        args = { "apex", "run", "-f", path, "-o", default_username }
+        local command =
+            string.format("%s apex run -f '%s' -o %s", executable, path, default_username)
+        Debug:log("execute_anon.lua", "Running " .. command .. "...")
+    else
+        writer = file_contents
+        args = { "apex", "run", "-o", default_username }
+        local command = string.format("%s apex run -o %s", executable, default_username)
+        Debug:log("execute_anon.lua", "Piping " .. file_contents .. " into " .. command .. "...")
+    end
     local new_job = Job:new({
         command = executable,
         env = Util.get_env(),
-        args = { "apex", "run", "-f", path, "-o", default_username },
+        args = args,
+        writer = writer,
         on_exit = function(j, code)
             vim.schedule(function()
                 if code == 0 then
@@ -63,9 +81,11 @@ M.execute_anon = function()
         on_stderr = function(_, data)
             vim.schedule(function()
                 Debug:log("execute_anon.lua", "Command stderr is: %s", data)
-                local trimmed_data = vim.trim(data)
-                if string.len(trimmed_data) > 0 then
-                    Popup:write_to_popup(data)
+                if data then
+                    local trimmed_data = vim.trim(data)
+                    if string.len(trimmed_data) > 0 then
+                        Popup:write_to_popup(data)
+                    end
                 end
             end)
         end,