feat: allow running commands for a specific runner

Author: jaytyrrell13

lua/static/init.lua

@@ -1,21 +1,41 @@
 local M = {}
 
 local actions = { 'build', 'serve', 'prod' }
+local runners = require('static.runners')
+local runnerNames = vim.tbl_keys(runners.runners)
 
 local execute = function(args)
   -- action == build, serve, prod
   local arguments = args.fargs
-  local action = arguments[1]
-  -- remove the first argument which is the action
-  table.remove(arguments, 1)
+  local runnerName
+  local runner
+  local action
 
-  local _, runner = require('static.runners').find()
+  -- If the second argument is the name of a runner
+  -- e.g. :Static astro build
+  if vim.tbl_contains(runnerNames, arguments[1]) then
+    runnerName = arguments[1]
+    action = arguments[2]
+
+    table.remove(arguments, 2)
+    table.remove(arguments, 1)
+  else
+    action = arguments[1]
+    table.remove(arguments, 1)
+  end
+
+  _, runner = runners.find(runnerName)
 
   if not runner then
     vim.notify('Could not determine the correct runner.', vim.log.levels.WARN)
     return
   end
 
+  if not action then
+    vim.notify('Could not determine the action.', vim.log.levels.WARN)
+    return
+  end
+
   local command = runner[action]
 
   if command == nil then
@@ -34,9 +54,35 @@ function M.setup(opts)
   config.setup(opts)
 
   vim.api.nvim_create_user_command('Static', execute, {
-    nargs = '*',
-    complete = function()
-      return actions
+    nargs = '+',
+    complete = function(_, line)
+      local parts = vim.split(vim.trim(line), '%s+')
+
+      if parts[1]:find('Static') then
+        table.remove(parts, 1)
+      end
+
+      if line:sub(-1) == ' ' then
+        parts[#parts + 1] = ''
+      end
+
+      table.remove(parts, 1)
+
+      if #parts > 0 then
+        return actions
+      end
+
+      local merged = {}
+
+      for _, value in pairs(actions) do
+        table.insert(merged, value)
+      end
+
+      for _, value in pairs(runnerNames) do
+        table.insert(merged, value)
+      end
+
+      return merged
     end,
   })
 end

lua/static/runners.lua

@@ -59,7 +59,11 @@ M.runners = {
   },
 }
 
-M.find = function()
+M.find = function(runnerName)
+  if runnerName then
+    return runnerName, M.runners[runnerName]
+  end
+
   for name, runner in pairs(M.runners) do
     local r = vim.fs.find(runner.file, {
       path = runner.path,