From e7ce7d53985057426630d8bd4b709865237c6061 Mon Sep 17 00:00:00 2001
From: paladine <todd@voidpointer.org>
Date: Thu, 11 Jan 2024 12:56:43 -0700
Subject: [PATCH] Adds a script to create a user (#617)

---
 scripts/create_user.py | 46 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100755 scripts/create_user.py

diff --git a/scripts/create_user.py b/scripts/create_user.py
new file mode 100755
index 00000000..081190c7
--- /dev/null
+++ b/scripts/create_user.py
@@ -0,0 +1,46 @@
+#!/usr/bin/python3
+
+import argparse
+import base64
+import hashlib
+import os
+import sqlite3
+import sys
+
+def _create_parser():
+  parser = argparse.ArgumentParser(description='Create MBBSEmu user accounts.')
+  parser.add_argument('--username', help='Username to create', required=True)
+  parser.add_argument('--password', help='Password to use', required=True)
+  parser.add_argument('--keys', help='Account keys to add to the new account', action='append', default=['NORMAL','PAYING'])
+  parser.add_argument('--email', help='Email address to use', default='test@test.bbs')
+
+  return parser.parse_args()
+
+def _make_password_hash(password, salt_bytes):
+ m = hashlib.sha512()
+ m.update(password.encode(encoding = 'UTF-8', errors = 'strict'))
+ m.update(salt_bytes)
+ return m.digest()
+
+def _main():
+  args = _create_parser()
+
+  conn = sqlite3.connect('mbbs.db')
+
+  passwordSaltBytes=os.urandom(32)
+  passwordHashBytes=_make_password_hash(args.password, passwordSaltBytes)
+
+  cur = conn.cursor()
+  t = (args.username, str(base64.b64encode(passwordHashBytes), encoding='utf-8'), str(base64.b64encode(passwordSaltBytes), encoding='utf-8'), args.email)
+  cur.execute('INSERT INTO Accounts (userName, passwordHash, passwordSalt, email, createDate, updateDate) VALUES (?,?,?,?, datetime(\'now\'), datetime(\'now\'))', t)
+  conn.commit()
+
+  account_id = cur.lastrowid
+
+  for user_key in args.keys:
+    t = (account_id, user_key)
+    cur.execute('INSERT INTO AccountKeys (accountId, accountKey, createDate, updateDate) VALUES (?,?,datetime(\'now\'), datetime(\'now\'))', t)
+    conn.commit()
+
+if __name__ == '__main__':
+  _main()