From 84431bda1d30dd0e2412268cbacbe750a7bec7e1 Mon Sep 17 00:00:00 2001
From: Darksider3 <github@darksider3.de>
Date: Sun, 20 Oct 2019 15:25:14 +0200
Subject: [PATCH] Edit: Sceleton, Interface and first argument

The sceleton for the interface and it's first possible argument are
finished, so a users pbukey can be changed now without touching the DB
itself directly.

Checking for users existence and the presence of options/arguments is
done beforehand and will throw out when nothing useful got delivered to
the script.
---
 private/editUsers.py | 83 +++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 79 insertions(+), 4 deletions(-)

diff --git a/private/editUsers.py b/private/editUsers.py
index b38341f..3137a51 100644
--- a/private/editUsers.py
+++ b/private/editUsers.py
@@ -1,7 +1,82 @@
-import lib.uis.default
+import configparser
+import lib.uis.config_ui  # only follow -c flag
+import lib.validator
+import sqlite3
 
-argparser = lib.uis.default.argparser
+lib.uis.config_ui.argparser.description += " - Edit Tilde Users"
+ArgParser = lib.uis.config_ui.argparser
+ArgParser.add_argument('--user', type=str,
+                       help='Tilde users name to edit', required=True)
 
-args = argparser.parse_args()
-config = argparser.ConfigParser()
+Mutually = ArgParser.add_mutually_exclusive_group()
+Mutually.add_argument('-r', '--remove', default=False, action="store_true",
+                      help='Remove an approved/unapproved User from the system. Effectively purges him.',
+                      required=False)
+Mutually.add_argument('-a', '--approve', default=False, action="store_true",
+                      help="Approve the given user", required=False)
+Mutually.add_argument("--verify", default=True, action="store_false",
+                      help="Turns off value checks",
+                      required=False)
+
+ArgParser.add_argument('--sshpubkey', type=str, default=None,
+                       help="Stores the new given SSH-Key in given user", required=False)
+ArgParser.add_argument('--name', type=str, default=None,
+                       help="Sets the stored name of the given user")
+ArgParser.add_argument('--username', type=str, default=None,
+                       help="Rename given User")
+ArgParser.add_argument('--email', type=str, default=None,
+                       help="Set new email address for given user")
+ArgParser.add_argument('--status', type=int, default=None,
+                       help="Set status of given user")
+args = ArgParser.parse_args()
+config = configparser.ConfigParser()
 config.read(args.config)
+
+
+if __name__ == "__main__":
+    try:
+        db = config['DEFAULT']['applications_db']
+        if not args.sshpubkey and not args.name and not args.username and not args.email and not args.status \
+                and not args.approve and not args.remove:
+            print(f"Well, SOMETHING must be done with {args.user} ;-)")
+            exit(1)
+        if not lib.validator.checkUserInDB(args.user, db):
+            print(f"User {args.user} doesn't exist in the database.")
+            exit(1)
+        import lib.sqlitedb
+        import lib.System
+        import lib.UserExceptions
+        DB = lib.sqlitedb.SQLitedb(db)
+        Sysctl = lib.System.System()
+        if not DB:
+            print("Couldn't establish connection to database")
+            exit(1)
+        if args.sshpubkey:
+            if not lib.validator.checkSSHKey(args.sshpubkey):
+                print(f"Pubkey {args.sshpubkey} isn't valid.")
+                exit(1)
+            try:
+                DB.safequery("UPDATE `applications` SET `pubkey`=? WHERE `username`=?",
+                             tuple([args.sshpubkey, args.user]))
+            except sqlite3.Error as e:
+                print(f"Something unexpected happened! {e}")
+                exit(1)
+            fetch = DB.safequery("SELECT * FROM `applications` WHERE `username` = ? ", tuple([args.user]))
+            if int(fetch[0]["status"]) == 1:
+                try:
+                    Sysctl.make_ssh_usable(args.user, args.sshpubkey)
+                except lib.UserExceptions.ModifyFilesystem as e:
+                    print(f"One action failed during writing the ssh key back into the authorization file")
+            print(f"{args.user} updated successfully.")
+
+        if args.name:
+            pass
+        if args.username:
+            print(f"{args.username}")
+        if args.email:
+            print(f"{args.email}")
+        if args.status:
+            print(f"{args.status}")
+        exit(0)
+    except KeyboardInterrupt as e:
+        pass