From 777cca967a19245341e0f377366be6d167c2bec0 Mon Sep 17 00:00:00 2001
From: Joao Sa <me@joaomsa.com>
Date: Sat, 29 Oct 2022 18:32:59 +0200
Subject: [PATCH] Initial version of search_headings

---
 lua/telescope/_extensions/orgmode/init.lua | 87 ++++++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 lua/telescope/_extensions/orgmode/init.lua

diff --git a/lua/telescope/_extensions/orgmode/init.lua b/lua/telescope/_extensions/orgmode/init.lua
new file mode 100644
index 0000000..3739e61
--- /dev/null
+++ b/lua/telescope/_extensions/orgmode/init.lua
@@ -0,0 +1,87 @@
+local orgmode = require('orgmode.api')
+
+local pickers = require("telescope.pickers")
+local finders = require("telescope.finders")
+local conf = require("telescope.config").values
+
+local function get_entries(opts)
+
+    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 opts.max_depth == 0 then
+        return file_results
+    end
+
+    local results = {}
+    for _, file_entry in ipairs(file_results) do
+        local agenda_file = orgmode.load(file_entry.filename)
+        for _, headline in ipairs(agenda_file.headlines) do
+
+            local allowed_depth = opts.max_depth == nil or headline._section.level <= opts.max_depth
+            local allowed_archive = opts.archived or not headline._section:is_archived()
+            if allowed_depth and allowed_archive then
+                local entry = {
+                    file = file_entry.file,
+                    filename = file_entry.filename,
+                    headline = headline
+                }
+
+                assert(headline.position.start_line == headline._section.line_number)
+                table.insert(results, entry)
+            end
+        end
+    end
+
+    return results
+end
+
+local function search_headings(opts)
+    opts = opts or {}
+
+    pickers.new(opts, {
+        prompt_title = "Search Headings",
+        finder = finders.new_table {
+            results = get_entries(opts),
+            entry_maker = opts.entry_maker or function (entry)
+
+                local headline = entry.headline
+
+                local display = entry.filename
+                local lnum = nil
+
+                if headline then
+                    lnum = headline.position.start_line
+
+                    local text = string.format('%s %s', string.rep('*', headline._section.level), headline.title)
+                    local line =  string.format('%s:%i:%s', entry.filename, lnum, text)
+                    display = line
+
+                end
+
+                return {
+                    value = display,
+                    ordinal = display,
+                    display = display,
+                    filename = entry.filename,
+                    lnum = lnum
+                }
+            end,
+        },
+        sorter = conf.generic_sorter(opts),
+        previewer = conf.grep_previewer(opts),
+    }):find()
+end
+
+return require("telescope").register_extension {
+  exports = {
+    search_headings = search_headings
+  },
+}