From 6b2d76153fa0d3b65f6c7b1b28dd2554979aaeee Mon Sep 17 00:00:00 2001 From: Stephan Seitz Date: Thu, 30 Apr 2020 22:11:44 +0200 Subject: Add function 'nvim-treesitter/install'.ensure_installed --- lua/nvim-treesitter/install.lua | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'lua') diff --git a/lua/nvim-treesitter/install.lua b/lua/nvim-treesitter/install.lua index ca84ac036..298a0e9f7 100644 --- a/lua/nvim-treesitter/install.lua +++ b/lua/nvim-treesitter/install.lua @@ -3,6 +3,7 @@ local fn = vim.fn local luv = vim.loop local configs = require'nvim-treesitter/configs' local parsers = configs.get_parser_configs() +local has_parser = require'nvim-treesitter/parsers'.has_parser local M = {} @@ -143,6 +144,24 @@ local function install(ft) run_install(cache_folder, package_path, ft, install_info) end + +M.ensure_installed = function(languages) + if type(languages) == 'string' then + if languages == 'all' then + languages = configs.available_parsers() + else + languages = {languages} + end + end + + for _, ft in ipairs(languages) do + if not has_parser(ft) then + install(ft) + end + end +end + + M.commands = { TSInstall = { run = install, -- cgit v1.2.3-70-g09d2