--=============================================================================

-- projectmanager.lua --- The lua version of projectmanager..vim

-- Copyright (c) 2016-2019 Wang Shidong & Contributors

-- Author: Wang Shidong < wsdjeg@outlook.com >

-- URL: https://spacevim.org

-- License: GPLv3

--=============================================================================

local logger = require('spacevim.logger').derive('project')

local sp = require('spacevim')

local sp_file = require('spacevim.api.file')

local sp_json = require('spacevim.api.data.json')

local sp_opt = require('spacevim.opt')

local fn = sp.fn

local layer = require('spacevim.layer')

local project_paths = {}

local project_cache_path = sp_file.unify_path(sp_opt.data_dir, ':p') .. 'SpaceVim/projects.json'

local spacevim_project_rooter_patterns = {}

local project_rooter_patterns = {}

local project_rooter_ignores = {}

local project_callback = {}

local cd = 'cd'

if fn.exists(':tcd') then

cd = 'tcd'

elseif fn.exists(':lcd') then

cd = 'lcd'

end

local function update_rooter_patterns()

project_rooter_patterns = {}

project_rooter_ignores = {}

for _,v in pairs(sp_opt.project_rooter_patterns) do

if string.match(v, '^!') == nil then

table.insert(project_rooter_patterns, v)

else

table.insert(project_rooter_ignores, string.sub(v, 2, -1))

end

end

end

local function is_ignored_dir(dir)

for _,v in pairs(project_rooter_ignores) do

if string.match(dir, v) ~= nil then

logger.debug('this is an ignored dir:' .. dir)

return true

end

end

return false

end

local function cache()

local path = sp_file.unify_path(project_cache_path, ':p')

local file = io.open(path, 'w')

if file then

if file:write(sp_json.json_encode(project_paths)) == nil then

logger.debug('failed to write to file:' .. path)

end

io.close(file)

else

logger.debug('failed to open file:' .. path)

end

end

local function readfile(path)

local file = io.open(path, "r")

if file then

local content = file:read("*a")

io.close(file)

return content

end

return nil

end

local function filereadable(fpath)

local f = io.open(fpath, 'r')

if f ~= nil then io.close(f) return true else return false end

end

local function isdirectory(fpath)

local f, err, code = io.open(fpath, "r")

if f ~= nil then

f:close()

return false

end

return code == 13

end

local function filter_invalid(projects)

for key, value in pairs(projects) do

if fn.isdirectory(value.path) == 0 then

projects[key] = nil

end

end

return projects

end

local function load_cache()

if filereadable(project_cache_path) then

logger.info('Load projects cache from: ' .. project_cache_path)

local cache_context = readfile(project_cache_path)

if cache_context ~= nil then

local cache_object = sp_json.json_decode(cache_context)

if type(cache_object) == 'table' then

project_paths = filter_invalid(cache_object)

end

end

else

logger.info('projects cache file does not exists!')

end

end

local function sort_by_opened_time()

local paths = {}

for k,v in pairs(project_paths) do table.insert(paths, k) end

table.sort(paths, compare_time)

if sp_opt.projects_cache_num > 0 and #paths >= sp_opt.projects_cache_num then

for i = sp_opt.projects_cache_num, #paths, 1 do

project_paths[paths[sp_opt.projects_cache_num]] = nil

table.remove(paths, sp_opt.projects_cache_num)

end

end

return paths

end

local function compare_time(d1, d2)

local proj1 = project_paths[d1] or {}

local proj1time = proj1['opened_time'] or 0

local proj2 = project_paths[d2] or {}

local proj2time = proj2['opened_time'] or 0

return proj2time - proj1time

end

local function change_dir(dir)

if dir == sp_file.unify_path(fn.getcwd()) then

logger.debug('same as current directory, no need to change.')

else

if dir ~= nil then

logger.info('change to root: ' .. dir)

sp.cmd(cd .. ' ' .. sp.fn.fnameescape(sp.fn.fnamemodify(dir, ':p')))

end

end

end

local function sort_dirs(dirs)

table.sort(dirs, compare)

local dir = dirs[1]

local bufdir = fn.getbufvar('%', 'rootDir', '')

if bufdir == dir then

return ''

else

return dir

end

end

local function compare(d1, d2)

local _, al = string.gsub(d1, "/", "")

local _, bl = string.gsub(d2, "/", "")

if sp_opt.project_rooter_outermost == 0 then

return bl - al

else

return al - bl

end

end

local function find_root_directory()

local fd = fn.bufname('%')

if fn == '' then

logger.debug('bufname is empty')

fd = fn.getcwd()

end

logger.debug('start to find root for: ' .. fd)

local dirs = {}

for _,pattern in pairs(project_rooter_patterns) do

logger.debug('searching rooter_patterns:' .. pattern)

local find_path = ''

if string.sub(pattern, -1) == '/' then

if sp_opt.project_rooter_outermost == 1 then

find_path = sp_file.finddir(pattern, fd, -1)

else

find_path = sp_file.finddir(pattern, fd)

end

else

if sp_opt.project_rooter_outermost == 1 then

find_path = sp_file.findfile(pattern, fd, -1)

else

find_path = sp_file.findfile(pattern, fd)

end

end

logger.debug('find_path is:' .. find_path)

local path_type = fn.getftype(find_path)

logger.debug('path_type is:' .. path_type)

if ( path_type == 'dir' or path_type == 'file' )

and not(is_ignored_dir(find_path)) then

find_path = sp_file.unify_path(find_path, ':p')

if path_type == 'dir' then

find_path = sp_file.unify_path(find_path, ':h:h')

else

find_path = sp_file.unify_path(find_path, ':h')

end

logger.debug('find_path is:' .. find_path)

if find_path ~= sp_file.unify_path(fn.expand('$HOME')) then

logger.info(' (' .. pattern .. '):' .. find_path)

table.insert(dirs, find_path)

else

logger.info('ignore $HOME directory:' .. find_path)

end

end

end

return sort_dirs(dirs)

end

local function cache_project(prj)

project_paths[prj.path] = prj

sp.cmd('let g:unite_source_menu_menus.Projects.command_candidates = []')

for _, key in pairs(sort_by_opened_time()) do

local desc = '[' .. project_paths[key].name .. '] ' .. project_paths[key].path .. ' <' .. fn.strftime('%Y-%m-%d %T', project_paths[key].opened_time) .. '>'

local cmd = "call SpaceVim#plugins#projectmanager#open('" .. project_paths[key].path .. "')"

sp.cmd('call add(g:unite_source_menu_menus.Projects.command_candidates, ["'

.. desc

.. '", "'

.. cmd

.. '"])')

end

if sp_opt.enable_projects_cache then

cache()

end

end

-- call add(g:spacevim_project_rooter_patterns, '.SpaceVim.d/')

-- let s:spacevim_project_rooter_patterns = copy(g:spacevim_project_rooter_patterns)

update_rooter_patterns()

if sp_opt.enable_projects_cache == 1 then

load_cache()

end

sp.cmd([[

let g:unite_source_menu_menus = get(g:,'unite_source_menu_menus',{})

let g:unite_source_menu_menus.Projects = {'description': 'Custom mapped keyboard shortcuts [SPC] p p'}

let g:unite_source_menu_menus.Projects.command_candidates = get(g:unite_source_menu_menus.Projects,'command_candidates', [])

]])

if sp_opt.project_rooter_automatically == 1 then

sp.cmd("augroup spacevim_project_rooter")

sp.cmd("autocmd!")

sp.cmd("autocmd VimEnter,BufEnter * call SpaceVim#plugins#projectmanager#current_root()")

sp.cmd("autocmd BufWritePost * :call setbufvar('%', 'rootDir', '') | call SpaceVim#plugins#projectmanager#current_root()")

sp.cmd("augroup END")

end

local M = {}

function M.list()

if layer.isLoaded('unite') then

sp.cmd('Unite menu:Projects')

elseif layer.isLoaded('denite') then

sp.cmd('Denite menu:Projects')

elseif layer.isLoaded('fzf') then

sp.cmd('FzfMenu Projects')

elseif layer.isLoaded('leaderf') then

sp.cmd("call SpaceVim#layers#leaderf#run_menu('Projects')")

else

logger.warn('fuzzy find layer is needed to find project!')

end

end

function M.open(project)

local path = project_paths[project]['path']

local name = project_paths[project]['name']

sp.cmd('tabnew')

-- I am not sure we should set the project name here.

-- sp.cmd('let t:_spacevim_tab_name = "[' .. name .. ']"')

sp.cmd(cd .. ' ' .. path)

if sp_opt.filemanager == 'vimfiler' then

sp.cmd('Startify | VimFiler')

elseif sp_opt.filemanager == 'nerdtree' then

sp.cmd('Startify | NERDTree')

elseif sp_opt.filemanager == 'defx' then

sp.cmd('Startify | Defx -new')

end

end

function M.current_name()

return sp.eval('b:_spacevim_project_name')

end

function M.RootchandgeCallback()

local path = sp_file.unify_path(fn.getcwd(), ':p')

local name = fn.fnamemodify(path, ':h:t')

logger.debug('project name is:' .. name)

logger.debug('project path is:' .. path)

local project = {

['path'] = path,

['name'] = name,

['opened_time'] = os.time()

}

if project.path == '' then

return

end

cache_project(project)

-- let g:_spacevim_project_name = project.name

-- let b:_spacevim_project_name = g:_spacevim_project_name

fn.setbufvar('%', '_spacevim_project_name', project.name)

for _, Callback in pairs(project_callback) do

logger.debug('run callback:' .. Callback)

fn.call(Callback, {})

end

end

function M.reg_callback(func)

if type(func) == 'string' then

if string.match(func, '^function%(') ~= nil then

table.insert(project_callback, string.sub(func, 11, -3))

else

table.insert(project_callback, func)

end

else

logger.warn('type of func is:' .. type(func))

logger.warn('can not register the project callback: ' .. fn.string(func))

end

end

function M.kill_project()

local name = sp.eval('b:_spacevim_project_name')

if name ~= '' then

sp_buffer.filter_do(

{

['expr'] = {

'buflisted(v:val)',

'getbufvar(v:val, "_spacevim_project_name") == "' .. name .. '"',

},

['do'] = 'bd %d'

}

)

end

end

function M.complete_project(arglead, cmdline, cursorpos)

local dir = '~'

local result = fn.split(fn.globpath(dir, '*'), "\n")

local ps = {}

for p in result do

if fn.isdirectory(p) == 1 and fn.isdirectory(p .. sp_file.separator .. '.git') == 1 then

table.insert(ps, fn.fnamemodify(p, ':t'))

end

end

return fn.join(ps, "\n")

end

function M.OpenProject(p)

sp.cmd('CtrlP '.. dir)

end

function M.current_root()

local bufname = fn.bufname('%')

if bufname == '[denite]'

or bufname == 'denite-filter'

or bufname == '[defx]' then

return

end

if table.concat(sp_opt.project_rooter_patterns, ':') ~= table.concat(spacevim_project_rooter_patterns, ':') then

logger.info('project_rooter_patterns option has been change, clear b:rootDir')

fn.setbufvar('%', 'rootDir', '')

spacevim_project_rooter_patterns = sp_opt.project_rooter_patterns

update_rooter_patterns()

end

local rootdir = fn.getbufvar('%', 'rootDir', '')

-- @bug fn.getbufvar('%', 'rootDir', '') return nil

if rootdir == nil or rootdir == '' then

rootdir = find_root_directory()

if rootdir == nil or rootdir == '' then

rootdir = sp_file.unify_path(fn.getcwd())

end

fn.setbufvar('%', 'rootDir', rootdir)

end

change_dir(rootdir)

M.RootchandgeCallback()

return rootdir

end

return M