refactor: move major parts of the code

Move the actual code from telescope/_extension folder into the actual
plugin folder.

Author: Sebastian Flügge

lua/telescope-orgmode/init.lua

@@ -0,0 +1,4 @@
+return {
+  refile_heading = require('telescope-orgmode.refile_heading'),
+  search_headings = require('telescope-orgmode.search_headings'),
+}

lua/telescope-orgmode/refile_heading.lua

@@ -0,0 +1,79 @@
+local pickers = require('telescope.pickers')
+local finders = require('telescope.finders')
+local conf = require('telescope.config').values
+local action_set = require('telescope.actions.set')
+local actions = require('telescope.actions')
+local action_state = require('telescope.actions.state')
+local state = require('telescope.state')
+
+local utils = require('telescope-orgmode.utils')
+
+local api = require('orgmode.api')
+
+return function(opts)
+  opts = opts or {}
+
+  local closest_headline = api.current():get_closest_headline()
+
+  local function refile(prompt_bufnr)
+    local entry = action_state.get_selected_entry()
+    actions.close(prompt_bufnr)
+
+    -- Refile to the file by default
+    local destination = entry.value.file
+
+    -- Refile to a specific heading if is set
+    if entry.value.headline then
+      destination = entry.value.headline
+    end
+
+    return api.refile({
+      source = closest_headline,
+      destination = destination,
+    })
+  end
+
+  local function gen_depth_toggle(opts, prompt_bufnr)
+    local status = state.get_status(prompt_bufnr)
+    status._ot_current_depth = opts.max_depth
+    status._ot_next_depth = nil
+    if status._ot_current_depth ~= 0 then
+      status._ot_next_depth = 0
+    end
+
+    return function()
+      local current_picker = action_state.get_current_picker(prompt_bufnr)
+
+      local aux = status._ot_current_depth
+      status._ot_current_depth = status._ot_next_depth
+      status._ot_next_depth = aux
+
+      opts.max_depth = status._ot_current_depth
+      local new_finder = finders.new_table({
+        results = utils.get_entries(opts),
+        entry_maker = opts.entry_maker or utils.make_entry(opts),
+      })
+
+      current_picker:refresh(new_finder, opts)
+    end
+  end
+
+  pickers
+    .new(opts, {
+      -- TODO: alter prompt title when depth is 0: Refile under file, Refile
+      -- under Headline
+      prompt_title = 'Refile Destination',
+      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),
+      attach_mappings = function(prompt_bufnr, map)
+        action_set.select:replace(refile)
+        map('i', '<c-space>', gen_depth_toggle(opts, prompt_bufnr))
+        return true
+      end,
+    })
+    :find()
+end

lua/telescope/_extensions/orgmode/search_headings.lua renamed to lua/telescope-orgmode/search_headings.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-orgmode').search_headings,
+    refile_heading = require('telescope-orgmode').refile_heading,
+  },
 })