diff --git a/main.py b/main.py index 916b400..120e94a 100644 --- a/main.py +++ b/main.py @@ -8,6 +8,7 @@ from modules.vim.VimModule import VimModule from modules.teamspeak.Teamspeak import Teamspeak from modules.hacker.Hacker import Hacker import modules.fail2ban.Fail2BanModule as fail2ban +from modules.bashrc.BashrcModule import BashrcModule modules = { "helloworld": HelloWorld().run, @@ -17,7 +18,8 @@ modules = { "vim": VimModule().run, "teamspeak": Teamspeak, "hacker": Hacker, - "fail2ban": fail2ban.run() + "fail2ban": fail2ban.run(), + "bashrc": BashrcModule().run } fire.Fire(modules) diff --git a/modules/bashrc/BashrcModule.py b/modules/bashrc/BashrcModule.py new file mode 100644 index 0000000..9c3a30e --- /dev/null +++ b/modules/bashrc/BashrcModule.py @@ -0,0 +1,26 @@ +from PyInquirer import prompt +import os +from shutil import copyfile +from print_helpers import print_gr, print_warn, print_fail + + +class BashrcModule: + + def run(self): + bashrc_path = os.path.join(os.path.expanduser('~'), ".bashrc") + bashrc_exists = os.path.exists(bashrc_path) + + if bashrc_exists: + print_gr(f"Found .bashrc in {bashrc_path}") + ask_to_keep = { + 'type': 'confirm', + 'message': 'Do you want to overwrite the existing .bashrc?', + 'name': 'overwrite', + 'default': False, + } + if not prompt(ask_to_keep)['overwrite']: + print_fail("Stopping") + return + + copyfile("./modules/bashrc/bashrc_default", bashrc_path) + print_gr("Overwritten") if bashrc_exists else print_gr("Created")