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()