user.py

from db_operations import insert_user_values, fetch_user_data, delete_user_rows, check_user_password, insert_session_values, if_logged_in, logout_session
from time import sleep
import socket
import getpass
import secrets
import subprocess
import sys
from client import main as client
from server import main as server

def generate_session_id():
    return secrets.token_hex(16)

def expire_session_id():
	pass

def login():
    print("Type EXIT at username to quit the app")
    username = input("Enter Username-> ")
    if username == 'EXIT':
        print("Exiting.... ")
        sleep(1)
        sys.exit()
    password = getpass.getpass("Enter your password: ")

    print("Authenticating......")
    sleep(2)

    if check_user_password(username, password) == True:
        print("Authenticated!")
        print(f"Welcome {username.split(' ')[0]}!!")
        print(":)")

        insert_session_values(generate_session_id(), username)
            
    else:
        print("Invalid Credentials")
        print("Please Try Again")

    return username

def dashboard(username):
    while True:
        try:
            number = int(input("Enter 0 to chat with a user, print 1 to logout-> "))
            if number == 0:
                chat(username)
            elif number == 1:
                logout(username)
            else:
                print("Invalid Request")
        except ValueError:
            print("Invalid Input! Enter only numbers")

def isServerRunning(host, portno):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        return s.connect((host, portno)) == 0

def chat(username):
    client(username)

def logout(username):
    sleep(1)
    print("Logging out.......")
    logout_session(username)
    login()

def register():
    print("Please enter your credentials-> ")
    while True:
        username = input("Enter Username-> ")
        password = getpass.getpass("Enter your password: ")
        password_confirm = getpass.getpass("Confirm your password: ")

        if password != password_confirm:
            print("The password is not the same. Please try again")
        else:
            break

    insert_user_values(username, password)

    print("Succesfully made a new account!")
    print("Please Log in")