kristoferssolo/telescope-frecency.nvim
1
0
Fork 0
mirror of https://github.com/kristoferssolo/telescope-frecency.nvim.git synced 2025-10-21 20:10:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: make database location configurable

Browse Source
This commit is contained in:
Senghan Bright 2021-08-04 10:52:02 +02:00
parent 8b6f4b8981
commit 65a7d89eb8
3 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lua/telescope/_extensions/frecency.lua
View File

@ -117,7 +117,6 @@ local frecency = function(opts)
local make_display = function(entry) local make_display = function(entry)
bufnr = vim.fn.bufnr bufnr = vim.fn.bufnr
buf_is_loaded = vim.api.nvim_buf_is_loaded buf_is_loaded = vim.api.nvim_buf_is_loaded
filename = entry.name filename = entry.name
hl_filename = buf_is_loaded(bufnr(filename)) and "TelescopeBufferLoaded" or "" hl_filename = buf_is_loaded(bufnr(filename)) and "TelescopeBufferLoaded" or ""
filename = format_filepath(filename, opts) filename = format_filepath(filename, opts)
@ -256,13 +255,14 @@ end
return telescope.register_extension { return telescope.register_extension {
setup = function(ext_config) setup = function(ext_config)
set_config_state('db_root', ext_config.db_root, nil)
set_config_state('show_scores', ext_config.show_scores, false) set_config_state('show_scores', ext_config.show_scores, false)
set_config_state('show_unindexed', ext_config.show_unindexed, true) set_config_state('show_unindexed', ext_config.show_unindexed, true)
set_config_state('show_filter_column', ext_config.show_filter_column, true) set_config_state('show_filter_column', ext_config.show_filter_column, true)
set_config_state('user_workspaces', ext_config.workspaces, {}) set_config_state('user_workspaces', ext_config.workspaces, {})
-- start the database client -- start the database client
db_client.init(ext_config.ignore_patterns, ext_config.db_safe_mode or true, ext_config.auto_validate or true) db_client.init(ext_config.db_root, ext_config.ignore_patterns, ext_config.db_safe_mode or true, ext_config.auto_validate or true)
end, end,
exports = { exports = {
frecency = frecency, frecency = frecency,

5
lua/telescope/_extensions/frecency/db_client.lua
View File

@ -85,10 +85,11 @@ local function validate_db(safe_mode)
end end
end end
local function init(config_ignore_patterns, safe_mode, auto_validate) -- TODO: make init params a keyed table
local function init(db_root, config_ignore_patterns, safe_mode, auto_validate)
if sql_wrapper then return end if sql_wrapper then return end
sql_wrapper = sqlwrap:new() sql_wrapper = sqlwrap:new()
local first_run = sql_wrapper:bootstrap() local first_run = sql_wrapper:bootstrap(db_root)
ignore_patterns = config_ignore_patterns or default_ignore_patterns ignore_patterns = config_ignore_patterns or default_ignore_patterns
if auto_validate then if auto_validate then

8
lua/telescope/_extensions/frecency/sql_wrapper.lua
View File

@ -34,14 +34,14 @@ function M:new()
return o return o
end end
function M:bootstrap(opts) function M:bootstrap(db_root)
if self.db then return end if self.db then return end
opts = opts or {} -- opts = opts or {}
self.max_entries = opts.max_entries or 2000 -- self.max_entries = opts.max_entries or 2000
-- create the db if it doesn't exist -- create the db if it doesn't exist
local db_root = opts.docs_root or vim.fn.stdpath('data') db_root = db_root or vim.fn.stdpath('data')
local db_filename = db_root .. "/file_frecency.sqlite3" local db_filename = db_root .. "/file_frecency.sqlite3"
self.db = sql:open(db_filename) self.db = sql:open(db_filename)
if not self.db then if not self.db then