Refactoring, fixing edge cases

1 files changed, 21 insertions, 2 deletions

diff --git a/lua/simple-denote/init.lua b/lua/simple-denote/init.lua
index 0d8654d..31abb58 100644
--- a/lua/simple-denote/init.lua
+++ b/lua/simple-denote/init.lua
@@ -1,15 +1,34 @@
 local M = {}
 
-local cmd = require("simple-denote.cmd")
+local api = require("simple-denote.api")
 local config = require("simple-denote.config")
 
+function M.load_cmd(options)
+  vim.api.nvim_create_user_command("Denote", function(opts)
+    if opts.fargs[1] == "note" then
+      api.note(options)
+    elseif opts.fargs[1] == "retitle" then
+      api.retitle(options)
+    elseif opts.fargs[1] == "retag" then
+      api.retag(options)
+    else
+      error("Unsupported operation " .. opts.fargs[1])
+    end
+  end, {
+    nargs = 1,
+    complete = function()
+      return { "note", "retitle", "retag" }
+    end,
+  })
+end
+
 ---@param options? table user configuration
 function M.setup(options)
   config.options = vim.tbl_deep_extend("force", config.defaults, options or {})
   if config.options.dir:sub(-1) ~= "/" then
     config.options.dir = config.options.dir .. "/"
   end
-  cmd.load_cmd(config.options)
+  M.load_cmd(config.options)
 end
 
 return M