about summary refs log tree commit diff
path: root/lua/quick_note.lua
blob: 1196faf19460d6e7439eb0e80c012f76a2b248a5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
local M = {}
local floating_win = nil

local function get_win_config()
  local width = math.floor(vim.o.columns * 0.8)
  local height = math.floor(vim.o.lines * 0.8)
  local row = math.floor((vim.o.lines - height) / 2)
  local col = math.floor((vim.o.columns - width) / 2)
  return {
    relative = "editor",
    width = width,
    height = height,
    row = row,
    col = col,
    style = "minimal",
    border = "rounded",
  }
end

function M.create_quick_note_window()
  if floating_win and vim.api.nvim_win_is_valid(floating_win) then
    vim.api.nvim_set_current_win(floating_win)
    return
  end

  local buf = vim.api.nvim_create_buf(false, true)
  floating_win = vim.api.nvim_open_win(buf, true, get_win_config())

  vim.api.nvim_buf_set_option(buf, "buftype", "acwrite")
  vim.api.nvim_buf_set_name(buf, "QuickNote")
end

function M.toggle_quick_note_window()
  if floating_win and vim.api.nvim_win_is_valid(floating_win) then
    M.close_quick_note_window()
  else
    M.create_quick_note_window()
  end
end

function M.close_quick_note_window()
  if floating_win and vim.api.nvim_win_is_valid(floating_win) then
    local buf = vim.api.nvim_win_get_buf(floating_win)
    M.prepend_to_quick_note()
    vim.api.nvim_win_close(floating_win, true)
    vim.api.nvim_buf_delete(buf, { force = true })
  end
end

function M.prepend_to_quick_note()
  local notes_dir = vim.g.notes_directory or "~/notes/"
  local filepath = vim.fn.expand(notes_dir) .. "/quick_note.md"
  local buf = vim.api.nvim_win_get_buf(floating_win)
  local content = vim.api.nvim_buf_get_lines(buf, 0, -1, false)

  -- Add a separator and timestamp before the new content
  local separator = "------------------------"
  local timestamp = os.date "%Y-%m-%d %H:%M:%S"
  table.insert(content, 1, "")
  table.insert(content, 1, separator)
  table.insert(content, 1, timestamp)
  table.insert(content, 1, separator)
  table.insert(content, 1, "")

  -- Check if quick_note.md exists, create it if it doesn't
  if vim.fn.filereadable(filepath) == 0 then
    vim.fn.writefile({}, filepath)
  end

  local file_content = vim.fn.readfile(filepath)
  for i = #file_content, 1, -1 do
    table.insert(content, 1, file_content[i])
  end

  vim.fn.writefile(content, filepath)
  print "Prepended content to quick_note.md"
end

function M.open_entire_quick_note()
  local notes_dir = vim.g.notes_directory or "~/notes/"
  local filepath = vim.fn.expand(notes_dir) .. "/quick_note.md"

  if vim.fn.filereadable(filepath) == 0 then
    print "No quick_note.md file found."
    return
  end

  vim.cmd("edit " .. filepath)
end

return M