From fefaaa12be9a8e8aed35fbef164463952e012a6e Mon Sep 17 00:00:00 2001
From: Mist <mist.zzh@gmail.com>
Date: Tue, 2 Apr 2024 18:18:37 +0800
Subject: [PATCH] [Feat] add extension as argument (#57)

---
 lua/codesnap/init.lua | 12 ++++++------
 plugin/codesnap.lua   | 16 ++++++++++------
 2 files changed, 16 insertions(+), 12 deletions(-)

diff --git a/lua/codesnap/init.lua b/lua/codesnap/init.lua
index d11dd1b..c40570f 100644
--- a/lua/codesnap/init.lua
+++ b/lua/codesnap/init.lua
@@ -23,9 +23,9 @@ function main.setup(config)
   static.config = table_utils.merge(static.config, config == nil and {} or config)
 end
 
-local function get_config()
+local function get_config(specify_extension)
   local code = visual_utils.get_selected_text()
-  local extension = get_extension()
+  local extension = specify_extension or get_extension()
 
   if string_utils.is_str_empty(code) then
     error("Please select code which you want to take snapshot first")
@@ -45,18 +45,18 @@ local function get_config()
   }, static.config)
 end
 
-function main.copy_into_clipboard()
-  generator.copy_into_clipboard(get_config())
+function main.copy_into_clipboard(extension)
+  generator.copy_into_clipboard(get_config(extension))
   vim.cmd("delmarks <>")
   vim.notify("Save snapshot into clipboard successfully")
 end
 
-function main.save_snapshot()
+function main.save_snapshot(extension)
   if string_utils.is_str_empty(static.config.save_path) then
     error("Cannot find save_path from config")
   end
 
-  generator.save_snapshot(get_config())
+  generator.save_snapshot(get_config(extension))
   vim.cmd("delmarks <>")
   vim.notify("Save snapshot in " .. static.config.save_path .. " successfully")
 end
diff --git a/plugin/codesnap.lua b/plugin/codesnap.lua
index 04fa21d..53da7d3 100644
--- a/plugin/codesnap.lua
+++ b/plugin/codesnap.lua
@@ -1,9 +1,13 @@
 local codesnap = require("codesnap")
 
-vim.api.nvim_create_user_command("CodeSnap", function()
-  codesnap.copy_into_clipboard()
-end, { nargs = "*", range = "%" })
+local function take_snapshot(take_snapshot_function)
+  return function(detail)
+    local args = detail.fargs
 
-vim.api.nvim_create_user_command("CodeSnapSave", function()
-  codesnap.save_snapshot()
-end, { nargs = "*", range = "%" })
+    take_snapshot_function(args[1])
+  end
+end
+
+vim.api.nvim_create_user_command("CodeSnap", take_snapshot(codesnap.copy_into_clipboard), { nargs = "*", range = "%" })
+
+vim.api.nvim_create_user_command("CodeSnapSave", take_snapshot(codesnap.save_snapshot), { nargs = "*", range = "%" })