codesnap.nvim/lua/codesnap/init.lua

local static = require("codesnap.static")
local visual_utils = require("codesnap.utils.visual")
local table_utils = require("codesnap.utils.table")
local string_utils = require("codesnap.utils.string")
local path_utils = require("codesnap.utils.path")
local module = require("codesnap.module")
local config_module = require("codesnap.config")
local highlight_module = require("codesnap.highlight")

local main = {
  cwd = static.cwd,
  highlight_mode_config = nil,
}

-- Prepare the path of the Rust module
-- Concat lib?.extension and ?.extension to package.cpath
package.cpath = path_utils.join(";", package.cpath, module.generator_file("?"), module.generator_file("lib?"))

function main.setup(config)
  static.config = table_utils.merge_config(static.config, config == nil and {} or config)
end

-- Save snapshot to specified save_path
--- @param save_path string
function main.save(save_path)
  local generator = require("generator")

  if save_path == nil then
    error("Save path is not specified", 0)
  end

  generator.save(save_path, config_module.get_config())
  vim.cmd("delmarks <>")
  vim.notify("Save snapshot in " .. save_path .. " successfully!")
end

-- Copy snapshot into clipboard
--- @param type? string
function main.copy(type)
  local snapshot_type = type == nil and "png" or type
  local generator = require("generator")

  generator.copy(snapshot_type, config_module.get_config())
  vim.cmd("delmarks <>")
  vim.notify("The snapshot is copied into clipboard successfully!")
end

-- Generate ASCII code snapshot and copy it into clipboard
function main.copy_ascii()
  local generator = require("generator")

  generator.copy_ascii(config_module.get_config())
  vim.cmd("delmarks <>")
  vim.notify("The ASCII code snapshot is copied into clipboard successfully!")
end

-- function main.copy_into_clipboard_with_config(config)
--   require("generator").copy_into_clipboard(config)
--   vim.cmd("delmarks <>")
--   vim.notify("Save snapshot into clipboard successfully")
-- end
--
-- -- Take ASCII code snapshot into clipboard
-- function main.copy_ascii_snapshot(extension)
--   require("generator").copy_ascii(config_module.get_config(extension))
--   vim.cmd("delmarks <>")
--   vim.notify("Save snapshot into clipboard successfully")
-- end
--
-- function main.save_snapshot_with_config(config)
--   if string_utils.is_str_empty(static.config.save_path) then
--     error(
--       "If you want to save snapshot in somewhere, you should config the save_path before, refer: https://github.com/mistricky/codesnap.nvim?tab=readme-ov-file#save-the-snapshot",
--       0
--     )
--   end
--
--   local matched_extension = string.match(static.config.save_path, "%.(.+)$")
--
--   if matched_extension ~= "png" and matched_extension ~= nil then
--     error("The extension of save_path should be .png", 0)
--   end
--
--   require("generator").save_snapshot(config)
--   vim.cmd("delmarks <>")
--   ---@diagnostic disable-next-line: need-check-nil
--   vim.notify("Save snapshot in " .. config.save_path .. " successfully")
-- end
--
-- -- Take a snapshot and copy it into clipboard
-- function main.copy_into_clipboard(extension)
--   main.copy_into_clipboard_with_config(config_module.get_config(extension))
-- end
--
-- -- Take a snapshot and save it into the specified path
-- function main.save_snapshot(extension)
--   main.save_snapshot_with_config(config_module.get_config(extension))
-- end
--
-- function main.highlight_mode_copy_into_clipboard(extension)
--   main.highlight_mode_config = config_module.get_config(extension)
--
--   highlight_module.create_highlight_selector_window(
--     "copy_into_clipboard_with_config",
--     visual_utils.get_selected_lines()
--   )
-- end
--
-- function main.highlight_mode_save_snapshot(extension)
--   main.highlight_mode_config = config_module.get_config(extension)
--
--   highlight_module.create_highlight_selector_window("save_snapshot_with_config", visual_utils.get_selected_lines())
-- end

return main