import json import os import subprocess from pathlib import Path from PyInquirer import prompt from print_helpers import print_gr, print_bold vim_root = Path().home().joinpath(".vim") vimrc_path = str(Path().home().joinpath(".vimrc")) json_config = {} def create_plugin_section(): plugins = json_config['plugins'] print("Available Plugins:") plugin_selection = [ { 'type': 'checkbox', 'message': 'Select plugins', 'name': 'modules', 'choices': list(map(lambda x: {"name": x}, plugins.keys())) } ] selected_plugins = prompt(plugin_selection)['modules'] print_bold("Your selection:") [print(x) for x in selected_plugins] # Install vimplug if plugins selected vimplug_path = str(vim_root.joinpath("autoload", "plug.vim")) if selected_plugins and not os.path.isfile(vimplug_path): curl_request = f"curl {json_config['pluginmanager_url']} -o {vimplug_path}" subprocess.call(curl_request, shell=True) print_gr("Pluginmanager was set up") vimrc_content = ["call plug#begin('~/.vim/plugged')\n"] for element in selected_plugins: vimrc_content.append(plugins[element] + "\n") vimrc_content.append("call plug#end()\n") return vimrc_content def create_setting_section(): default_settings = list(json_config['settings'].values()) print_bold("Default settings:") [print(i) for i in default_settings] vimrc_content = list(map(lambda x: x + "\n", default_settings)) return vimrc_content def write_vimfile(vimrc_content): seen = set() seen_add = seen.add vimrc_content = [x1 for x1 in vimrc_content if not (x1 in seen or seen_add(x1))] with open(vimrc_path, "w") as vimrc_file: for line in vimrc_content: vimrc_file.write(line) # Unused def _get_vimfile_working_copy(): try: with open(vimrc_path, "r") as vimrc_file: vimrc_content = vimrc_file.readlines() except FileNotFoundError: vimrc_content = [] return vimrc_content def run(): global json_config with open("modules/vim/vimrc_conf.json", "r") as config: json_config = json.load(config) # Create folders os.makedirs(str(vim_root.joinpath("autoload")), exist_ok=True) os.makedirs(str(vim_root.joinpath("plugged")), exist_ok=True) plugin_section = create_plugin_section() settings_section = create_setting_section() write_vimfile(settings_section + plugin_section) # Exec plugin manager install_plugins = "vim +PlugInstall +qall" subprocess.call(install_plugins, shell=True) print_gr("Module finished successfully!") functions = run