From 4d3f6ce2bacc158d27e182e7f4a814983f9c73ef Mon Sep 17 00:00:00 2001
From: AlejandroSuero <asuerome@gmail.com>
Date: Fri, 6 Sep 2024 15:00:42 +0200
Subject: [PATCH] feat(#13): add `:Freeze ...` complete behaviour

---
 lua/freeze/init.lua | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/lua/freeze/init.lua b/lua/freeze/init.lua
index 1b859e2..17f981f 100644
--- a/lua/freeze/init.lua
+++ b/lua/freeze/init.lua
@@ -160,10 +160,21 @@ M.start = function(args, options)
 
   local cmd = vim.tbl_extend("error", base_cmdline, {})
 
+  args = args.fargs or {}
+  local args_list = {}
+  -- Parse the arguments into a table
+  for _, v in ipairs(args) do
+    local key = vim.split(v, "=")[1]
+    local value = vim.split(v, "=")[2]
+    args_list[key] = value
+  end
+
   -- If the user gave us a language lets use it
   -- Else try to get the language from neovim's buffer filetype
   table.insert(cmd, "--language")
-  if options.language then
+  if args_list.language then
+    table.insert(cmd, args_list.language)
+  elseif options.language then
     table.insert(cmd, options.language)
   else
     table.insert(cmd, vim.bo.filetype)
@@ -188,6 +199,21 @@ M.setup = function(opts)
     desc = "convert range to code image representation",
     force = false,
     range = true,
+    nargs = "*",
+    complete = function(_, line)
+      local opts_list = vim.tbl_keys(M.allowed_opts)
+      local l = vim.split(line, "%s+")
+      -- Remove `command` from the list of options
+      for i, v in ipairs(opts_list) do
+        if v == "command" then
+          table.remove(opts_list, i)
+        end
+      end
+      table.sort(opts_list)
+      return vim.tbl_filter(function(opt)
+        return vim.startswith(opt, l[#l])
+      end, opts_list)
+    end,
   })
 end