chore: logic seperation/refactor

main
Scott McKendry 2024-06-21 18:53:26 +12:00
parent 979272e167
commit 63ad46275d
3 changed files with 80 additions and 60 deletions

View File

@ -1,62 +1,5 @@
local actions = require("telescope.actions")
local action_state = require("telescope.actions.state")
local telescope = require("telescope")
local themes = require("telescope.themes")
local sorters = require("telescope.sorters")
local pickers = require("telescope.pickers")
local ok, _ = pcall(require, "telescope")
if not ok then
error("This plugin requires telescope.nvim")
end
ok, _ = pcall(require, "resession")
if not ok then
error("This plugin requires resession.nvim")
end
local function encode_session(session_str)
return session_str:gsub(":\\", "__"):gsub("\\", "_")
end
local function decode_sessions(sessions)
for i, session in ipairs(sessions) do
sessions[i] = session:gsub("__", ":\\"):gsub("_", "\\")
end
return sessions
end
local function get_sessions()
local sessions = require("resession").list({ dir = "dirsession" })
return sessions
end
local function load_session(prompt_bufnr)
local session = action_state.get_selected_entry()
local encoded = encode_session(session[1])
actions.close(prompt_bufnr)
require("resession").load(encoded, { dir = "dirsession" })
end
local dropdown = themes.get_dropdown({})
local resession_opts = {
prompt_title = "Find session",
finder = require("telescope.finders").new_table({
results = decode_sessions(get_sessions()),
}),
attach_mappings = function(_, map)
map("i", "<CR>", load_session)
map("n", "<CR>", load_session)
return true
end,
sorter = sorters.get_fzy_sorter(),
}
local function resession_picker()
pickers.new(dropdown, resession_opts):find()
end
return telescope.register_extension({ exports = { resession = resession_picker } })
return telescope.register_extension({
exports = { resession = require("telescope._extensions.resession.picker").resession_picker },
})

View File

@ -0,0 +1,42 @@
local action_state = require("telescope.actions.state")
local actions = require("telescope.actions")
local pickers = require("telescope.pickers")
local sorters = require("telescope.sorters")
local themes = require("telescope.themes")
local utils = require("telescope._extensions.resession.utils")
local M = {}
-- TODO: add mapping to delete sessions
--- Load the selected session
---@param prompt_bufnr number
function M.load_session(prompt_bufnr)
local session = action_state.get_selected_entry()
local encoded = utils.encode_session(session[1])
actions.close(prompt_bufnr)
require("resession").load(encoded, { dir = "dirsession" })
end
--- Render the session picker
function M.resession_picker()
local dropdown = themes.get_dropdown({})
local resession_opts = {
prompt_title = "Find session",
finder = require("telescope.finders").new_table({
results = utils.get_results(),
}),
attach_mappings = function(_, map)
map("i", "<CR>", M.load_session)
map("n", "<CR>", M.load_session)
return true
end,
sorter = sorters.get_fzy_sorter(),
}
return pickers.new(dropdown, resession_opts):find()
end
return M

View File

@ -0,0 +1,35 @@
local M = {}
--- Encode a session string in the format used by resession
--- @param session_str string The session string to encode
--- @return string The encoded session string
M.encode_session = function(session_str)
session_str = session_str:gsub(":\\", "__"):gsub("\\", "_")
return session_str
end
--- Decode a list of session strings in the format used by resession to a friendlier format
--- TODO: Test in linux with different paths
--- TODO: Support path formatting options (e.g. relative paths) ✨
--- @param sessions string[] The list of session strings to decode. Usually the ouotput of resession.list()
--- @return string[] The decoded session strings
M.decode_sessions = function(sessions)
for i, session in ipairs(sessions) do
sessions[i] = session:gsub("__", ":\\"):gsub("_", "\\")
end
return sessions
end
--- Get a list of sessions from resession
--- @return string[] The list of sessions
M.get_sessions = function()
return require("resession").list({ dir = "dirsession" })
end
--- Get a list of sessions from resession and decode them
--- @return string[] The list of decoded sessions
M.get_results = function()
return M.decode_sessions(M.get_sessions())
end
return M