-- lazy.nvim 安装配置
{
  "nvim-treesitter/nvim-treesitter",
  build = ":TSUpdate",
  config = function()
    local ts = require("nvim-treesitter.configs")
    ts.setup({
      -- 安装这些语言的解析器
      ensure_installed = {
        "lua", "python", "javascript",
        "typescript", "rust", "go",
        "html", "css", "json",
      },
      -- 自动安装缺失的解析器
      auto_install = true,
      -- 启用语法高亮
      highlight = {
        enable = true,
        -- 对大文件禁用（性能保护）
        disable = function(lang, buf)
          local max_filesize = 100 * 1024
          local ok, stats = pcall(
            vim.loop.fs_stat,
            vim.api.nvim_buf_get_name(buf)
          )
          if ok and stats and
            stats.size > max_filesize then
              return true
          end
        end,
      },
      -- 启用增量选择
      incremental_selection = {
        enable = true,
        keymaps = {
          init_selection = "gnn",
          node_incremental = "grn",
          scope_incremental = "grc",
          node_decremental = "grm",
        },
      },
      -- 启用代码折叠
      indent = { enable = true },
    })
  end,
}

-- 在 Neovim 中探索语法树节点
local parsers = require("nvim-treesitter.parsers")
local ts_utils = require("nvim-treesitter.ts_utils")

-- 获取当前 buffer 的语法树根节点
local buf = vim.api.nvim_get_current_buf()
local parser = vim.treesitter.get_parser(buf, "lua")
local tree = parser:parse()[1]
local root = tree:root()

-- 遍历根节点的直接子节点
for node in root:iter_children() do
  print(node:type())     -- "function_declaration"
  print(node:named())    -- true（命名节点）
  print(node:child_count()) -- 子节点数量

  -- 获取节点的文本范围
  local sr, sc = node:start()  -- 起始行、列
  local er, ec = node:end_()   -- 结束行、列
  print(f"Range: {sr}:{sc} - {er}:{ec}")
end

-- 常见的 Lua 节点类型：
-- function_declaration, function_name
-- parameters, identifier, string
-- if_statement, for_statement, while_loop
-- binary_expression, unary_expression
-- table_constructor, field

; highlights.scm — Lua 语言高亮查询
;
; 查询语法：用 S-expression 模式匹配语法树节点
; @capture_name 将匹配的节点标记为特定高亮组

; 关键字
["if" "then" "else" "elseif"
 "end" "for" "while" "repeat"
 "until" "do" "return" "local"] @keyword

; 函数声明 — 捕获函数名
(function_declaration
  name: (identifier) @function)

; 方法调用 — 捕获方法名
(method_index_expression
  method: (identifier) @function.method)

; 字符串字面量
(string) @string

; 注释
(comment) @comment

; 数字字面量
(number) @number

; 布尔值和 nil
["true" "false" "nil"] @boolean

; 运算符
["and" "or" "not" "in"] @keyword.operator

; 表构造器
(table_constructor
  (field
    name: (identifier) @property))

-- 在 Neovim 配置中自定义 Tree-sitter 高亮组
-- 这些映射覆盖默认的捕获名到高亮组关系

-- 方法 1：直接设置高亮组
vim.api.nvim_set_hl(0, "@keyword", {
  fg = "#C678DD",    -- 紫色关键字
  bold = true,
})

vim.api.nvim_set_hl(0, "@function", {
  fg = "#61AFEF",    -- 蓝色函数名
  italic = true,
})

vim.api.nvim_set_hl(0, "@string", {
  fg = "#98C379",    -- 绿色字符串
})

vim.api.nvim_set_hl(0, "@comment", {
  fg = "#5C6370",    -- 灰色注释
  italic = true,
})

-- 方法 2：链接到现有高亮组
vim.api.nvim_set_hl(0, "@function.call", {
  link = "@function",  -- 继承 @function 的样式
})

-- 方法 3：语言特定的高亮覆盖
vim.api.nvim_set_hl(0, "@function.python", {
  fg = "#E5C07B",    -- Python 函数用黄色
})

; injections.scm — 定义语言注入规则

; Markdown 中的代码块 → 注入对应语言
(fenced_code_block
  (info_string
    (language) @injection.language)
  (code_fence_content) @injection.content)

; HTML 中的 <script> 标签 → 注入 JavaScript
(script_element
  (raw_text) @injection.content
  (#set! injection.language "javascript"))

; HTML 中的 <style> 标签 → 注入 CSS
(style_element
  (raw_text) @injection.content
  (#set! injection.language "css"))

; Lua 中的 vim.cmd() 调用 → 注入 Vim script
(function_call
  name: (identifier) @_vimcmd
  (arguments
    (string
      (string_content) @injection.content))
  (#match? @_vimcmd "^vim%.(cmd|api%.nvim_cmd)")
  (#set! injection.language "vim"))

; JavaScript 模板字面量 → 注入 SQL
(call_expression
  function: (identifier) @_sql_func
  arguments: (template_string) @injection.content
  (#match? @_sql_func "^(sql|query)")
  (#set! injection.language "sql"))

-- textobjects 配置
textobjects = {
  select = {
    enable = true,
    lookahead = true,  -- 向前查找
    keymaps = {
      -- af = 包含外围的整个函数
      ["af"] = "@function.outer",
      -- if = 只选函数体（不含声明行）
      ["if"] = "@function.inner",
      -- ac = 整个类/结构体
      ["ac"] = "@class.outer",
      ["ic"] = "@class.inner",
      -- 条件块
      ["ai"] = "@conditional.outer",
      ["ii"] = "@conditional.inner",
      -- 循环块
      ["al"] = "@loop.outer",
      ["il"] = "@loop.inner",
      -- 参数/参数列表
      ["aa"] = "@parameter.outer",
      ["ia"] = "@parameter.inner",
    },
  },
  move = {
    enable = true,
    set_jumps = true,  -- 添加到 jumplist
    goto_next_start = {
      ["]m"] = "@function.outer",
      ["]]"] = "@class.outer",
    },
    goto_previous_start = {
      ["[m"] = "@function.outer",
      ["[@"] = "@class.outer",
    },
  },
},

-- 启用 Tree-sitter 折叠
vim.o.foldmethod = "expr"
vim.o.foldexpr = "nvim_treesitter#foldexpr()"

-- 折叠行为配置
vim.o.foldlevel = 99  -- 打开文件时不自动折叠
vim.o.foldlevelstart = 99
vim.o.foldenable = true

-- 自定义折叠文本显示
vim.o.foldtext = "v:lua.vim.treesitter.foldtext()"

-- folds.scm 查询示例（Python）
;
; 这些节点类型可以折叠
; [
;   (function_definition)
;   (class_definition)
;   (if_statement)
;   (for_statement)
;   (with_statement)
;   (try_statement)
;   (list)
;   (dictionary)
;   (string)
; ] @fold

-- 折叠快捷键
vim.keymap.set("n", "ff", function()
  vim.cmd("normal! zc")  -- 关闭折叠
end)
vim.keymap.set("n", "fo", function()
  vim.cmd("normal! zo")  -- 打开折叠
end)
vim.keymap.set("n", "fO", function()
  vim.cmd("normal! zR")  -- 打开所有折叠
end)

-- 目录结构：
-- ~/.config/nvim/after/queries/
--   lua/highlights.scm    -- Lua 高亮覆盖
--   python/highlights.scm  -- Python 高亮覆盖
--   rust/folds.scm        -- Rust 折叠覆盖
--   typescript/textobjects.scm
--
-- 方法 1：完全覆盖（替换默认查询）
-- 文件内容即为完整的查询定义

-- 方法 2：追加查询（在默认基础上添加）
-- 文件第一行写 ; extends

-- after/queries/lua/highlights.scm
; extends

-- 额外高亮：vim 全局变量
("vim") @variable.builtin

-- 高亮 require 调用中的模块名
(function_call
  name: (identifier) @_require
  (arguments
    (string
      (string_content) @string.special))
  (#eq? @_require "require"))

-- 高亮 local function 中的函数名
(variable_declaration
  (declaration
    (variable_declarator
      name: (identifier) @function)
    (function_expression)))

-- 自定义 Textobjects 查询
-- after/queries/lua/textobjects.scm
; extends

-- 将 table constructor 视为类对象
(table_constructor) @class.outer

-- 选择 do...end 块的内容
(do_statement) @loop.outer

-- 调试工具配置

-- 1. Playground 语法树可视化
{
  "nvim-treesitter/playground",
  config = function()
    local pg = require("nvim-treesitter.configs")
    pg.setup({
      playground = {
        enable = true,
        updatetime = 25,  -- 刷新延迟（ms）
        persist_queries = true,  -- 保存查询历史
      },
    })
  end,
}

-- 2. 常用调试命令
-- :TSPlaygroundToggle   打开/关闭语法树面板
-- :TSHighlightCaptures  显示当前高亮捕获
-- :Inspect              显示光标位置的高亮信息
-- :TSInstallInfo        查看已安装的解析器
-- :TSUpdate all         更新所有解析器

-- 3. 查看解析耗时
vim.treesitter.query.set("lua", "highlights", "")
-- 清空查询 → 重新设置 → 对比性能差异

-- 4. 诊断解析问题
local parser = vim.treesitter.get_parser(0, "lua")
local tree = parser:parse()[1]
local root = tree:root()
-- 检查是否有错误节点
local has_errors = false
for node in root:iter_children() do
  if node:type() == "ERROR" then
    has_errors = true
    local sr, sc = node:start()
    print("Parse error at line " .. (sr+1))
  end
end

// grammar.js — 定义一个简单配置语言的语法
// 这是一个演示用的最小化示例

module.exports = grammar({
  name: "myconfig",

  rules: {
    // 顶层规则：配置文件 = 多个语句
    source_file: $ => repeat(
      choice(
        $.section_header,
        $.key_value_pair,
        $.comment,
      )
    ),

    // 段标题：[section_name]
    section_header: $ => seq(
      "[",
      field("name", $.identifier),
      "]",
      $.newline,
    ),

    // 键值对：key = value
    key_value_pair: $ => seq(
      field("key", $.identifier),
      "=",
      field("value", $.value),
      $.newline,
    ),

    // 值：字符串 | 数字 | 布尔
    value: $ => choice(
      $.string,
      $.number,
      $.boolean,
    ),

    // 注释：# 开头
    comment: $ => seq(
      "#",
      token(prec(-1, /.*/)),
      $.newline,
    ),

    // 基础 token
    identifier: $ => /[a-zA-Z_]\w*/,
    string: $ => /"[^"]*"/,
    number: $ => /\d+(\.\d+)?/,
    boolean: $ => choice("true", "false"),
    newline: $ => /\n/,
  },
});

; after/queries/lua/highlights.scm
; 请补充完整实现

; TODO 1: 高亮 vim 全局变量
; 提示：匹配 (identifier) 节点，
;       值为 "vim" 时标记为 @variable.builtin

; TODO 2: 区分函数声明和调用
; 提示：function_declaration 的 name 用 @function
;       function_call 的函数名用 @function.call

; TODO 3: 高亮 TODO/FIXME 注释
; 提示：在 (comment) 中使用 (#match? @comment.todo "TODO|FIXME")

; TODO 4: 高亮 require 模块名
; 提示：匹配 function_call name 为 "require" 的参数