codesnap.nvim/lua/codesnap/init.lua

local static = require("codesnap.static")
local table_utils = require("codesnap.utils.table")
local generator = require("generator")
local string_utils = require("codesnap.utils.string")
local visual_utils = require("codesnap.utils.visual")

local assets_folder = static.cwd .. "/assets"

local function get_extension()
  local file_path = vim.fn.expand("%:p")
  local file_extension = string.match(file_path, "%.([^%.]+)$")

  return file_extension
end

local main = {
  cwd = static.cwd,
  preview_switch = static.preview_switch,
}

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

local function get_config()
  local code = visual_utils.get_selected_text()
  local extension = get_extension()

  if string_utils.is_str_empty(code) then
    error("Please select code which you want to take snapshot first")
  end

  if string_utils.is_str_empty(extension) then
    error("Cannot detect current filetype")
  end

  return table_utils.merge({
    code = code,
    extension = extension,
    fonts_folder = assets_folder .. "/fonts",
    themes_folder = assets_folder .. "/themes",
    theme = "base16-onedark",
  }, static.config)
end

function main.copy_into_clipboard()
  generator.copy_into_clipboard(get_config())
  vim.cmd("delmarks <>")
  vim.notify("Save snapshot into clipboard successfully")
end

function main.save_snapshot()
  if string_utils.is_str_empty(static.config.save_path) then
    error("Cannot find save_path from config")
  end

  generator.save_snapshot(get_config())
  vim.cmd("delmarks <>")
  vim.notify("Save snapshot in " .. static.config.save_path .. " successfully")
end

return main