Replace <anything> with the set check

Author: t-eckert

lua/obsidian/commands/set_checkbox.lua

@@ -2,9 +2,5 @@ local set_checkbox = require("obsidian.util").set_checkbox
 
 ---@param client obsidian.Client
 return function(client, data)
-  -- If no check_char is provided, default to "x"
-  if string.len(data.args) == 0 then
-    data.args = "x"
-  end
   set_checkbox(data.args)
 end

lua/obsidian/util.lua

@@ -541,14 +541,17 @@ util.set_checkbox = function(check_char, line_num)
   line_num = line_num or unpack(vim.api.nvim_win_get_cursor(0))
   local line = vim.api.nvim_buf_get_lines(0, line_num - 1, line_num, false)[1]
 
-  local checkbox_pattern = "^%s*- %[.] "
+  local checkbox_pattern = "^(%s*)- %[.%] "
+
+  local indent, rest = string.match(line, checkbox_pattern .. "(.*)")
+
+  if indent and rest then
+    -- Rebuild the line with the new check_char
+    line = indent .. "- [" .. check_char .. "] " .. rest
 
-  if string.match(line, checkbox_pattern) then
-    -- Only set the check_char if the line contains the checkbox pattern.
-    line = util.string_replace(line, "- [" .. check_char .. "]", "- [" .. check_char .. "]", 1)
+    -- Update the line in the buffer (0-indexed)
+    vim.api.nvim_buf_set_lines(0, line_num - 1, line_num, true, { line })
   end
-  -- 0-indexed
-  vim.api.nvim_buf_set_lines(0, line_num - 1, line_num, true, { line })
 end
 
 ---Determines if the given date is a working day (not weekend)