Price/telescope-orgmode.nvim
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: support tag searching, highlights

Browse Source
This commit is contained in:
Price Hiller 2024-03-31 03:10:54 -05:00
parent d70432192d
commit 47b485c57c
Signed by: Price
GPG Key ID: C3FADDE7A8534BEB
1 changed files with 74 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

134
lua/telescope-orgmode/utils.lua
View File

@ -1,79 +1,93 @@
local entry_display = require("telescope.pickers.entry_display")
local entry_display = require('telescope.pickers.entry_display')
local orgmode = require("orgmode.api")
local orgmode = require('orgmode.api')
local utils = {}
utils.get_entries = function(opts)
local file_results = vim.tbl_map(function(file)
return { file = file, filename = file.filename }
end, orgmode.load())
local file_results = vim.tbl_map(function(file)
return { file = file, filename = file.filename }
end, orgmode.load())
if not opts.archived then
file_results = vim.tbl_filter(function(entry)
return not entry.file.is_archive_file
end, file_results)
end
if not opts.archived then
file_results = vim.tbl_filter(function(entry)
return not entry.file.is_archive_file
end, file_results)
end
if opts.max_depth == 0 then
return file_results
end
if opts.max_depth == 0 then
return file_results
end
local results = {}
for _, file_entry in ipairs(file_results) do
for _, headline in ipairs(file_entry.file.headlines) do
local allowed_depth = opts.max_depth == nil or headline.level <= opts.max_depth
local allowed_archive = opts.archived or not headline.is_archived
if allowed_depth and allowed_archive then
local entry = {
file = file_entry.file,
filename = file_entry.filename,
headline = headline,
}
table.insert(results, entry)
end
end
end
local results = {}
for _, file_entry in ipairs(file_results) do
for _, headline in ipairs(file_entry.file.headlines) do
local allowed_depth = opts.max_depth == nil or headline.level <= opts.max_depth
local allowed_archive = opts.archived or not headline.is_archived
if allowed_depth and allowed_archive then
local entry = {
file = file_entry.file,
filename = file_entry.filename,
headline = headline,
}
table.insert(results, entry)
end
end
end
return results
return results
end
utils.make_entry = function(opts)
local displayer = entry_display.create({
separator = " ",
items = {
{ width = vim.F.if_nil(opts.location_width, 20) },
{ remaining = true },
},
})
local function make_display(entry)
local displayer = entry_display.create({
separator = ' ',
items = {
{ width = 0.2 },
{ width = (#entry.tags > 0 and 0.5 or 0.8) },
{ remaining = true },
},
})
local function make_display(entry)
return displayer({ entry.location, entry.line })
end
local location = entry.location .. ':' .. entry.lnum
local headline = entry.headline
local headline_level = (headline.level % 8 == 0 and 8 or headline.level)
local headline_hl = '@org.headline.level' .. headline_level
return displayer({
location,
{ entry.line, headline_hl },
{ ' ' .. entry.tags, '@org.tag' },
})
end
return function(entry)
local headline = entry.headline
return function(entry)
local headline = entry.headline
local lnum = nil
local location = vim.fn.fnamemodify(entry.filename, ':t')
local line = ''
local tags = ''
local lnum = nil
local location = vim.fn.fnamemodify(entry.filename, ":t")
local line = ""
if headline then
lnum = headline.position.start_line
line = headline.title
line = string.format('%s %s', string.rep('*', headline.level), headline.title)
if #headline.all_tags > 0 then
tags = ':' .. table.concat(headline.all_tags, ':') .. ':'
end
end
if headline then
lnum = headline.position.start_line
location = string.format("%s:%i", location, lnum)
line = string.format("%s %s", string.rep("*", headline.level), headline.title)
end
return {
value = entry,
ordinal = location .. " " .. line,
filename = entry.filename,
lnum = lnum,
display = make_display,
location = location,
line = line,
}
end
return {
value = entry,
ordinal = location .. ' ' .. headline.title .. ' ' .. tags,
filename = entry.filename,
lnum = lnum,
tags = tags,
headline = headline,
display = make_display,
location = location,
line = line,
}
end
end
return utils