From d90f41269b299639a17bb21a7b09ab6eaadafb6a Mon Sep 17 00:00:00 2001 From: Lyz Date: Thu, 8 Feb 2024 01:14:13 +0100 Subject: [PATCH] style: format code --- lua/telescope-orgmode/utils.lua | 119 +++++++++--------- lua/telescope/_extensions/orgmode/init.lua | 12 +- .../_extensions/orgmode/search_headings.lua | 24 ++-- 3 files changed, 77 insertions(+), 78 deletions(-) diff --git a/lua/telescope-orgmode/utils.lua b/lua/telescope-orgmode/utils.lua index e014db4..43af337 100644 --- a/lua/telescope-orgmode/utils.lua +++ b/lua/telescope-orgmode/utils.lua @@ -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 diff --git a/lua/telescope/_extensions/orgmode/init.lua b/lua/telescope/_extensions/orgmode/init.lua index 9ae10f1..1526e2b 100644 --- a/lua/telescope/_extensions/orgmode/init.lua +++ b/lua/telescope/_extensions/orgmode/init.lua @@ -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"), + }, +}) diff --git a/lua/telescope/_extensions/orgmode/search_headings.lua b/lua/telescope/_extensions/orgmode/search_headings.lua index a2783c5..c946fe7 100644 --- a/lua/telescope/_extensions/orgmode/search_headings.lua +++ b/lua/telescope/_extensions/orgmode/search_headings.lua @@ -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