2019-12-22 02:45:19 +01:00
|
|
|
from AbstractModule import AbstractModule
|
2019-12-16 12:08:22 +01:00
|
|
|
from subprocess import call
|
2019-12-22 03:59:38 +01:00
|
|
|
from PyInquirer import prompt
|
2019-12-16 12:08:22 +01:00
|
|
|
import os
|
2019-12-22 04:23:03 +01:00
|
|
|
import json
|
2019-12-14 04:48:57 +01:00
|
|
|
|
2019-12-22 02:45:19 +01:00
|
|
|
|
|
|
|
class VimModule(AbstractModule):
|
2019-12-14 04:22:34 +01:00
|
|
|
def __init__(self):
|
2019-12-14 05:47:33 +01:00
|
|
|
super().__init__()
|
2019-12-22 04:23:03 +01:00
|
|
|
self.plugins = {}
|
|
|
|
with open("modules/vim/vim_plugins.json") as plugins:
|
|
|
|
self.plugins = json.load(plugins)['plugins']
|
2019-12-22 03:59:38 +01:00
|
|
|
|
2019-12-22 07:49:51 +01:00
|
|
|
def do_plugin_selection(self):
|
2019-12-22 03:59:38 +01:00
|
|
|
mapped_list = []
|
|
|
|
for choice in self.plugins.keys():
|
|
|
|
mapped_list.append({
|
|
|
|
"name": choice
|
|
|
|
})
|
|
|
|
|
|
|
|
print("Available Plugins:")
|
|
|
|
module_selection = [
|
|
|
|
{
|
|
|
|
'type': 'checkbox',
|
|
|
|
'message': 'Select plugins',
|
|
|
|
'name': 'modules',
|
|
|
|
'choices': mapped_list
|
|
|
|
}
|
2019-12-14 04:22:34 +01:00
|
|
|
]
|
2019-12-22 04:23:03 +01:00
|
|
|
return prompt(module_selection)['modules']
|
2019-12-14 04:22:34 +01:00
|
|
|
|
2019-12-22 03:59:38 +01:00
|
|
|
def run(self):
|
2019-12-14 04:22:34 +01:00
|
|
|
|
2019-12-22 07:49:51 +01:00
|
|
|
selected_plugins = self.do_plugin_selection()
|
2019-12-15 07:30:20 +01:00
|
|
|
|
|
|
|
# print out selected plugins
|
2019-12-22 07:49:51 +01:00
|
|
|
[print(x) for x in selected_plugins]
|
2019-12-15 07:30:20 +01:00
|
|
|
|
|
|
|
# Create vimrc file
|
2019-12-22 07:49:51 +01:00
|
|
|
try:
|
|
|
|
with open("vim_testfile", "r") as vim_file:
|
|
|
|
vimrc_content = vim_file.readlines()
|
|
|
|
except FileNotFoundError:
|
|
|
|
vimrc_content = []
|
2019-12-15 07:30:20 +01:00
|
|
|
|
2019-12-22 07:49:51 +01:00
|
|
|
try:
|
|
|
|
index = vimrc_content.index("call plug#begin('~/.vim/plugged')\n") + 1
|
|
|
|
print("Index of plugman " + str(index))
|
|
|
|
for plugin in selected_plugins:
|
|
|
|
vimrc_content.insert(index, self.plugins[plugin] + "\n")
|
|
|
|
except ValueError:
|
|
|
|
vimrc_content.append("call plug#begin('~/.vim/plugged')\n")
|
|
|
|
for element in selected_plugins:
|
|
|
|
vimrc_content.append(self.plugins[element] + "\n")
|
|
|
|
vimrc_content.append("call plug#end()\n")
|
2019-12-19 14:03:31 +01:00
|
|
|
|
2019-12-22 07:49:51 +01:00
|
|
|
seen = set()
|
|
|
|
seen_add = seen.add
|
|
|
|
vimrc_content = [x1 for x1 in vimrc_content if not (x1 in seen or seen_add(x1))]
|
2019-12-15 07:30:20 +01:00
|
|
|
|
2019-12-22 07:49:51 +01:00
|
|
|
with open("vim_testfile", "w") as vimrc_file:
|
|
|
|
for line in vimrc_content:
|
|
|
|
vimrc_file.write(line)
|
2019-12-17 23:36:42 +01:00
|
|
|
|
2019-12-19 14:03:31 +01:00
|
|
|
# Create necessary folder structure for Plugin Manager
|
2019-12-22 07:49:51 +01:00
|
|
|
if not os.path.isdir("/home/clay/Gitlab/linux-tools/testdirectory/autoload/"):
|
|
|
|
if os.path.isdir("/home/clay/Gitlab/linux-tools/testdirectory/"):
|
2019-12-22 04:23:03 +01:00
|
|
|
install_vimplug = "cd ~/Gitlab/linux-tools/testdirectory/ " \
|
|
|
|
"&& mkdir autoload " \
|
|
|
|
"&& cd autoload " \
|
|
|
|
"&& touch test.txt"
|
2019-12-19 14:03:31 +01:00
|
|
|
call(install_vimplug, shell=True)
|
2019-12-17 23:36:42 +01:00
|
|
|
|
2019-12-19 14:03:31 +01:00
|
|
|
else:
|
2019-12-22 04:23:03 +01:00
|
|
|
make_vimdir = "cd ~/Gitlab/linux-tools/ " \
|
|
|
|
"&& mkdir testdirectory " \
|
|
|
|
"&& cd testdirectory " \
|
|
|
|
"&& mkdir autoload " \
|
|
|
|
"&& cd autoload " \
|
|
|
|
"&& touch test.txt"
|
2019-12-22 02:45:19 +01:00
|
|
|
call(make_vimdir, shell=True)
|
2019-12-17 23:36:42 +01:00
|
|
|
else:
|
2019-12-22 02:45:19 +01:00
|
|
|
print("Vim-Plug already installed!")
|