note.nvim/ftplugin/note.lua at main · gsuuon/note.nvim · GitHub

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
if not vim.b.did_note_plugin then
  local note = require('note')
  local commands = require('note.commands')

  commands.create_buffer_commands()

  if note.config.keymap then
    commands.create_buffer_keymaps(note.config.keymap.prefix)
  end

  vim.wo.foldmethod = 'indent'

  local success, llm_prompts = pcall(require, 'llm.prompts.scopes')
  if success then
    llm_prompts.add_buffer_plugin_prompts(
      'note',
      require('note.llm.prompts')
    )
  end

  local function link_ts_highlight_defaults()
    vim.api.nvim_set_hl(0, '@note.cancelled.content', { link = "Conceal", default = true })
    vim.api.nvim_set_hl(0, '@note.done.content', { link = "SpecialComment", default = true })
    vim.api.nvim_set_hl(0, '@note.current.content', { link = "ModeMsg", default = true })
    vim.api.nvim_set_hl(0, '@note.label.content', { link = "Tag", default = true })
    vim.api.nvim_set_hl(0, '@note.cancelled.content', { link = "Conceal", default = true })
    vim.api.nvim_set_hl(0, '@note.info.content', { link = "Comment", default = true })

    vim.api.nvim_set_hl(0, '@note.pending.marker', { link = "Identifier", default = true })
    vim.api.nvim_set_hl(0, '@note.paused.marker', { link = "WarningMsg", default = true })
    vim.api.nvim_set_hl(0, '@note.cancelled.marker', { link = "Error", default = true })
    vim.api.nvim_set_hl(0, '@note.done.marker', { link = "Constant", default = true })
    vim.api.nvim_set_hl(0, '@note.current.marker', { link = "QuickFixLine", default = true })
    vim.api.nvim_set_hl(0, '@note.label.marker', { link = "Tag", default = true })
    vim.api.nvim_set_hl(0, '@note.info.marker', { link = "Comment", default = true })

    if note.config.disable_decorators == nil or note.config.disable_decorators == false then
      vim.api.nvim_set_hl(0, '@note.decorator.select', { link = "GreenSign", default = true })
      vim.api.nvim_set_hl(0, '@note.decorator.flow', { link = "AquaSign", default = true })
      vim.api.nvim_set_hl(0, '@note.decorator.warn', { link = "YellowSign", default = true })
      vim.api.nvim_set_hl(0, '@note.decorator.question', { link = "BlueSign", default = true })
    end

    vim.api.nvim_set_hl(0, '@note.link', { link = "NormalFloat", default = true })

    vim.api.nvim_set_hl(0, '@markup.heading.1.note', { link = "@markup.heading.1.markdown", default = true })
    vim.api.nvim_set_hl(0, '@markup.heading.2.note', { link = "@markup.heading.2.markdown", default = true })
    vim.api.nvim_set_hl(0, '@markup.heading.3.note', { link = "@markup.heading.3.markdown", default = true })
    vim.api.nvim_set_hl(0, '@markup.heading.4.note', { link = "@markup.heading.4.markdown", default = true })
  end

  local ok, parsers = pcall(require, "nvim-treesitter.parsers")
  if ok then
    if parsers.has_parser() then
      vim.bo.fileformat = 'unix'
      vim.opt_local.foldmethod = 'expr'
      link_ts_highlight_defaults()
    else
      vim.opt_local.foldmethod = 'indent'
    end
  end

  vim.b.did_note_plugin = true
end