feat: toggle to show headlines of current file only

Author: Sebastian Flügge

README.md

@@ -77,6 +77,12 @@ require('telescope').extensions.orgmode.insert_link
 By pressing `<C-Space>` the picker state can be toggled between two modes.
 Every mode is available in every function.
 
+### Current file only mode
+
+In headline mode, you can press `<C-f>` to toggle between showing all headlines
+vs only headlines from the current file. This is useful when you want to focus
+on the current file's structure.
+
 ### Search headlines
 
 This is the first and default mode. It shows all the headlines, initially
@@ -116,15 +122,45 @@ For a particular command you can pass it directly in your key mapping to the fun
 require('telescope').extension.orgmode.search_headings({ max_depth = 3 })
 ```
 
-You can also create a key mapping, that allows you to search directly for org files:
+### Custom keymaps
+
+You can customize the telescope picker keymaps by passing a `mappings` table:
+
+```lua
+require('telescope').extensions.orgmode.search_headings({
+  mappings = {
+    i = {
+      ['<C-l>'] = require('telescope-orgmode.actions').toggle_current_file_only,
+      ['<C-s>'] = require('telescope-orgmode.actions').toggle_headlines_orgfiles,
+    },
+    n = {
+      ['<C-l>'] = require('telescope-orgmode.actions').toggle_current_file_only,
+      ['<C-s>'] = require('telescope-orgmode.actions').toggle_headlines_orgfiles,
+    }
+  }
+})
+```
+
+You can also create key mappings for specific modes:
 
 ```lua
-vim.set.keymap(
+-- Search only org files
+vim.keymap.set(
   "n",
-  "<Leader>off", 
+  "<Leader>off",
   function()
-    require('telescope').extension.orgmode.search_headings({ mode = "orgfiles" })
+    require('telescope').extensions.orgmode.search_headings({ mode = "orgfiles" })
   end,
   { desc = "Find org files"}
 )
+
+-- Search headlines in current file only
+vim.keymap.set(
+  "n",
+  "<Leader>ofc",
+  function()
+    require('telescope').extensions.orgmode.search_headings({ only_current_file = true })
+  end,
+  { desc = "Find headlines in current file"}
+)
 ```

lua/telescope-orgmode/actions.lua

@@ -33,6 +33,18 @@ function M.search_orgfiles(opts)
   end
 end
 
+function M.toggle_current_file_only(opts)
+  return function(prompt_bufnr)
+    -- Only toggle if we're in headlines mode
+    if opts.state.current ~= 'headlines' then
+      return
+    end
+
+    opts.only_current_file = not opts.only_current_file
+    M._find_headlines(opts, prompt_bufnr)
+  end
+end
+
 function M.refile(closest_headline)
   return function(prompt_bufnr)
     local entry = action_state.get_selected_entry()

lua/telescope-orgmode/mappings.lua

@@ -5,6 +5,8 @@ local M = {}
 function M.attach_mappings(map, opts)
   map('i', '<c-space>', to_actions.toggle_headlines_orgfiles(opts), { desc = 'Toggle headline/orgfile' })
   map('n', '<c-space>', to_actions.toggle_headlines_orgfiles(opts), { desc = 'Toggle headline/orgfile' })
+  map('i', '<c-f>', to_actions.toggle_current_file_only(opts), { desc = 'Toggle current file only' })
+  map('n', '<c-f>', to_actions.toggle_current_file_only(opts), { desc = 'Toggle current file only' })
   M.attach_custom(map, opts)
 end
 

lua/telescope-orgmode/org.lua

@@ -28,6 +28,17 @@ end
 function M.load_headlines(opts)
   -- Get files sorted by modification time (most recent first)
   local files = require('orgmode').files:all()
+
+  if opts.only_current_file then
+    local current_file = opts.original_file or vim.api.nvim_buf_get_name(0)
+    if current_file == '' then
+      current_file = vim.fn.expand('%:p')
+    end
+    files = vim.tbl_filter(function(file)
+      return file.filename == current_file
+    end, files)
+  end
+
   if not opts.archived then
     files = vim.tbl_filter(function(file)
       return not (vim.fn.fnamemodify(file.filename, ':e') == 'org_archive')

lua/telescope-orgmode/picker/search_headings.lua

@@ -11,6 +11,10 @@ return function(opts)
     orgfiles = 'Search org files',
   }, "headlines")
 
+  -- Capture the current buffer before opening telescope
+  opts.original_buffer = vim.api.nvim_get_current_buf()
+  opts.original_file = vim.api.nvim_buf_get_name(opts.original_buffer)
+
   pickers
     .new(opts, {
       prompt_title = opts.prompt_titles[opts.state.current],