function LiveGrepWithExtension(extension)
    require('telescope.builtin').live_grep({
        additional_args = function(opts)
            return { "--glob", "*." .. extension }
        end
    })
end

vim.api.nvim_create_user_command(
    "LgExtension",
    function(opts)
        LiveGrepWithExtension(opts.args)
    end,
    { nargs = 1 }
)


function LiveGrepUp(n)
    local bufnr = vim.api.nvim_get_current_buf()
    local full_path = vim.api.nvim_buf_get_name(bufnr)
    local file_path = vim.fs.dirname(full_path)
    -- 用系统的路径分隔符分割路径
    local path_sep = package.config:sub(1, 1)
    local components = {}

    -- 分割路径到各个组件
    for component in string.gmatch(file_path, "[^" .. path_sep .. "]+") do
        table.insert(components, component)
    end

    -- 计算向上n级的路径
    local up_n = #components - n
    if up_n < 1 then
        up_n = 1
    end -- 确保不会超出根目录
    local new_cwd = path_sep
    for i = 1, up_n do
        new_cwd = new_cwd .. components[i] .. (i < up_n and path_sep or "")
    end

    print("search under:" .. new_cwd)
    -- 使用新的目录执行Telescope live_grep
    require("telescope.builtin").live_grep({ cwd = new_cwd })
end

vim.api.nvim_create_user_command(
    "LgUp",
    function(opts)
        local n = tonumber(opts.args) or 0
        LiveGrepUp(n)
    end,
    { nargs = 1 } -- 命令接受一个参数
)