added command and improved prompts

Author: Ztrizzo

doc/salesforce.txt

@@ -23,6 +23,8 @@ extension for VS Code. Out of the box commands include:
 - `SalesforceRetrieveFromOrg`: Pull the current file from the org
 - `SalesforceDiffFile`: Diff the current file against the file in the org
 - `SalesforceSetDefaultOrg`: Set the default org for the current project
+- `SalesforceCreateLightningComponent`: Create an Aura Component or LWC
+- `SalesforceCreateApex`: Create Apex trigger or class
 
 # Setup ~
 To use this plugin, you must first setup your project by running `require("salesforce").setup({})`.
@@ -61,4 +63,4 @@ Usage ~
 `require("salesforce").setup({})`
 
 
- vim:tw=78:ts=8:noet:ft=help:norl:
+ vim:tw=78:ts=8:noet:ft=help:norl:

lua/salesforce/component_generator.lua

@@ -64,15 +64,21 @@ end
 M.create_apex = function()
     local name, dir = prompt_for_name_and_dir("Apex Name: ", "Please select a directory: ")
     vim.ui.select(
-        { "class", "trigger" },
+        { "Class", "Trigger" },
         {
             prompt = "Apex file type: ",
         },
         vim.schedule_wrap(function(choice)
+            local type
+            if choice == "Class" then
+                type = "class"
+            else
+                type = "trigger"
+            end
             local command = string.format(
                 "%s apex generate %s --name %s --output-dir %s",
                 executable,
-                choice,
+                type,
                 name,
                 dir
             )

lua/salesforce/util.lua

@@ -142,10 +142,11 @@ M.send_cli_command = function(command, on_exit_callback, category, caller_name)
             end)
         end,
     })
-    if not M.current_job or not M.current_job:is_running() then
-        M.current_job = new_job
-        M.current_bufnr = vim.api.nvim_get_current_buf()
-        M.current_job:start()
+    if not M[category] or M[category].current_job or not M[category].current_job:is_running() then
+        M[category] = {}
+        M[category].current_job = new_job
+        M[category].current_bufnr = vim.api.nvim_get_current_buf()
+        M[category].current_job:start()
     else
         M.notify_command_in_progress(category)
     end

plugin/salesforce.lua

@@ -13,6 +13,7 @@ local Diff = require("salesforce.diff")
 local Config = require("salesforce.config")
 local OrgManager = require("salesforce.org_manager")
 local Debug = require("salesforce.debug")
+local ComponentGenerator = require("salesforce.component_generator")
 
 Debug:log("salesforce.lua", "Initializing Salesforce plugin commands...")
 
@@ -68,3 +69,11 @@ end, {})
 vim.api.nvim_create_user_command("SalesforceSetDefaultOrg", function()
     OrgManager:set_default_org()
 end, {})
+
+vim.api.nvim_create_user_command("SalesforceCreateLightningComponent", function()
+    ComponentGenerator:create_lightning_component()
+end, {})
+
+vim.api.nvim_create_user_command("SalesforceCreateApex", function()
+    ComponentGenerator:create_apex()
+end, {})