62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
from IModule import IModule
|
|
from subprocess import call
|
|
import os
|
|
import shutil
|
|
|
|
|
|
class VimModule(IModule):
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def get_command(self):
|
|
|
|
plugins = [
|
|
("1. Colorschemes", "Plug 'https://github.com/rafi/awesome-vim-colorschemes.git'"),
|
|
("2. Unix Operations", "Plug 'https://github.com/tpope/vim-eunuch.git'"),
|
|
("3. Emmet", "Plug 'https://github.com/mattn/emmet-vim.git'"),
|
|
("4. Meta5 Theme", "Plug 'https://github.com/christophermca/meta5.git'"),
|
|
("5. Nerdtree", "Plug 'https://github.com/scrooloose/nerdtree'"),
|
|
("6. Lightline", "Plug 'https://github.com/itchyny/lightline.vim'")
|
|
]
|
|
|
|
print("Available Plugins:\n")
|
|
|
|
for key in plugins:
|
|
print(key[0])
|
|
|
|
print("")
|
|
|
|
# Select Plugins
|
|
user_input = [int(userInput) for userInput in input("Select modules: ").split()]
|
|
clean_plugin_list = set(user_input)
|
|
|
|
# print out selected plugins
|
|
for element in clean_plugin_list:
|
|
selection = element - 1
|
|
try:
|
|
print(plugins[selection][0])
|
|
except Exception:
|
|
print("Fail")
|
|
|
|
# Create vimrc file
|
|
vimrc_exists = os.path.isfile("testfile")
|
|
|
|
if vimrc_exists:
|
|
with open("testfile", "r") as vimrc_in_file:
|
|
buf = vimrc_in_file.readlines()
|
|
|
|
with open("testfile", "w") as vimrc_out_file:
|
|
for line in buf:
|
|
if line == "call plug#begin('~/.vim/plugged')\n":
|
|
line = line + plugins[selection][1] + "\n"
|
|
vimrc_out_file.write(line)
|
|
|
|
else:
|
|
vimrc_file = open("testfile", "w+")
|
|
vimrc_file.write("\n" + "call plug#begin('~/.vim/plugged')\n")
|
|
for element in clean_plugin_list:
|
|
selection = element -1
|
|
vimrc_file.write(plugins[selection][1] + "\n")
|
|
vimrc_file.write("call plug#end()\n")
|
|
vimrc_file.close()
|