config.py

import os
import sys
import string
import random
import hashlib
import sys
from getpass import getpass

from utils.dbconfig import dbconfig

from rich import print as printc
from rich.console import Console

console = Console()

def checkConfig():
	db = dbconfig()
	cursor = db.cursor()
	query = "SELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA  WHERE SCHEMA_NAME = 'pm'"
	cursor.execute(query)
	results = cursor.fetchall()
	db.close()
	if len(results)!=0:
		return True
	return False


def generateDeviceSecret(length=10):
	return ''.join(random.choices(string.ascii_uppercase + string.digits, k = length))


def make():
	if checkConfig():
		printc("[red][!] Already Configured! [/red]")
		return

	printc("[green][+] Creating new config [/green]")

	# Create database
	db = dbconfig()
	cursor = db.cursor()
	try:
		cursor.execute("CREATE DATABASE pm")
	except Exception as e:
		printc("[red][!] An error occurred while trying to create db. Check if database with name 'pm' already exists - if it does, delete it and try again.")
		console.print_exception(show_locals=True)
		sys.exit(0)

	printc("[green][+][/green] Database 'pm' created")

	# Create tables
	query = "CREATE TABLE pm.secrets (masterkey_hash TEXT NOT NULL, device_secret TEXT NOT NULL)"
	res = cursor.execute(query)
	printc("[green][+][/green] Table 'secrets' created ")

	query = "CREATE TABLE pm.entries (sitename TEXT NOT NULL, siteurl TEXT NOT NULL, email TEXT, username TEXT, password TEXT NOT NULL)"
	res = cursor.execute(query)
	printc("[green][+][/green] Table 'entries' created ")


	mp = ""
	printc("[green][+] A [bold]MASTER PASSWORD[/bold] is the only password you will need to remember in-order to access all your other passwords. Choosing a strong [bold]MASTER PASSWORD[/bold] is essential because all your other passwords will be [bold]encrypted[/bold] with a key that is derived from your [bold]MASTER PASSWORD[/bold]. Therefore, please choose a strong one that has upper and lower case characters, numbers and also special characters. Remember your [bold]MASTER PASSWORD[/bold] because it won't be stored anywhere by this program, and you also cannot change it once chosen. [/green]\n")

	while 1:
		mp = getpass("Choose a MASTER PASSWORD: ")
		if mp == getpass("Re-type: ") and mp!="":
			break
		printc("[yellow][-] Please try again.[/yellow]")

	# Hash the MASTER PASSWORD
	hashed_mp = hashlib.sha256(mp.encode()).hexdigest()
	printc("[green][+][/green] Generated hash of MASTER PASSWORD")


	# Generate a device secret
	ds = generateDeviceSecret()
	printc("[green][+][/green] Device Secret generated")

	# Add them to db
	query = "INSERT INTO pm.secrets (masterkey_hash, device_secret) values (%s, %s)"
	val = (hashed_mp, ds)
	cursor.execute(query, val)
	db.commit()

	printc("[green][+][/green] Added to the database")

	printc("[green][+] Configuration done![/green]")

	db.close()


def delete():
	printc("[red][-] Deleting a config clears the device secret and all your entries from the database. This means you will loose access to all your passwords that you have added into the password manager until now. Only do this if you truly want to 'destroy' all your entries. This action cannot be undone. [/red]")

	while 1:
		op = input("So are you sure you want to continue? (y/N): ")
		if op.upper() == "Y":
			break
		if op.upper() == "N" or op.upper == "":
			sys.exit(0)
		else:
			continue

	printc("[green][-][/green] Deleting config")


	if not checkConfig():
		printc("[yellow][-][/yellow] No configuration exists to delete!")
		return

	db = dbconfig()
	cursor = db.cursor()
	query = "DROP DATABASE pm"
	cursor.execute(query)
	db.commit()
	db.close()
	printc("[green][+] Config deleted![/green]")

def remake():
	printc("[green][+][/green] Remaking config")
	delete()
	make()


if __name__ == "__main__":

	if len(sys.argv)!=2:
		print("Usage: python config.py <make/delete/remake>")
		sys.exit(0)

	if sys.argv[1] == "make":
		make()
	elif sys.argv[1] == "delete":
		delete()
	elif sys.argv[1] == "remake":
		remake()
	else:
		print("Usage: python config.py <make/delete/remake>")