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

style: format code

Browse Source
This commit is contained in:
Lyz 2024-02-08 01:14:13 +01:00
parent ed1c31acb4
commit d90f41269b
No known key found for this signature in database
GPG Key ID: 6C7D7C1612CDE02F
3 changed files with 77 additions and 78 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,82 +1,79 @@
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 displayer = entry_display.create({
separator = ' ',
items = {
{ width = vim.F.if_nil(opts.location_width, 20) },
{ remaining = true }
}
})
local function make_display(entry)
return displayer({ entry.location, entry.line })
end
local function make_display(entry)
return displayer({ entry.location, entry.line })
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 lnum = nil
local location = vim.fn.fnamemodify(entry.filename, ':t')
local line = ""
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
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 .. " " .. line,
filename = entry.filename,
lnum = lnum,
display = make_display,
location = location,
line = line,
}
end
end
return utils

12
lua/telescope/_extensions/orgmode/init.lua
View File

@ -2,9 +2,9 @@
-- public orgmode api
-- TODO: add highlight groups
return require("telescope").register_extension {
exports = {
search_headings = require("telescope._extensions.orgmode.search_headings"),
refile_heading = require("telescope._extensions.orgmode.refile_heading")
},
}
return require("telescope").register_extension({
exports = {
search_headings = require("telescope._extensions.orgmode.search_headings"),
refile_heading = require("telescope._extensions.orgmode.refile_heading"),
},
})

24
lua/telescope/_extensions/orgmode/search_headings.lua
View File

@ -2,18 +2,20 @@ local pickers = require("telescope.pickers")
local finders = require("telescope.finders")
local conf = require("telescope.config").values
local utils = require('telescope-orgmode.utils')
local utils = require("telescope-orgmode.utils")
return function(opts)
opts = opts or {}
opts = opts or {}
pickers.new(opts, {
prompt_title = "Search Headings",
finder = finders.new_table {
results = utils.get_entries(opts),
entry_maker = opts.entry_maker or utils.make_entry(opts),
},
sorter = conf.generic_sorter(opts),
previewer = conf.grep_previewer(opts),
}):find()
pickers
.new(opts, {
prompt_title = "Search Headings",
finder = finders.new_table({
results = utils.get_entries(opts),
entry_maker = opts.entry_maker or utils.make_entry(opts),
}),
sorter = conf.generic_sorter(opts),
previewer = conf.grep_previewer(opts),
})
:find()
end