From 1d4e6a3f8d09c46a088b4942f2fe2e3f45150eaa Mon Sep 17 00:00:00 2001
From: Lucas Noki <lucas-noki@gmx.de>
Date: Sat, 25 Jul 2020 04:50:30 +0200
Subject: [PATCH] Implement bashrc module code

---
 main.py                        |  4 +++-
 modules/bashrc/BashrcModule.py | 26 ++++++++++++++++++++++++++
 2 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 modules/bashrc/BashrcModule.py

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