import sqlite3
from hashlib import sha256


def user_data():
    login = input("Input login: ")
    password = sha256(input("Input password: ").encode()).hexdigest()
    input_task = input("You want registration or authorization?: ")

    conn = sqlite3.connect("mydatabase.db")
    cursor = conn.cursor()

    cursor.execute(
        """
            CREATE TABLE
            IF NOT EXISTS user
            (
                id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
                login,
                password
            )
        """
    )

    cursor.execute(
        """
            CREATE TABLE
            IF NOT EXISTS notes
            (
                user_Id Integer not null,
                note_title,
                note_text,
                FOREIGN KEY (user_id) REFERENCES information (id)
            )
        """
    )

    check_user = cursor.execute(
        """
            SELECT id, login, password
            FROM user
            WHERE login = ?""", (login,)
    ).fetchone()
    print(check_user)

    conn.commit()
    conn.close()


def registration(login, password, check_user = ''):
    if check_user is None:
        cursor.execute(
            """
                INSERT INTO user (login, password)
                VALUES (?, ?)""", (login, password)
            )
    else:
        print("Thy another user!")

def authorization():
    if check_user is not None:
        if password == check_user[2]:
            print(f"Welcome, {login}")

            while True:
                user_choice = input("Do you want to create a note or sign out:")
                if user_choice == "create a note":
                    note_title = input("Enter the names for the task: ")
                    note_text = input("Enter text of the task: ")
                    cursor.execute(
                        """
                            INSERT INTO notes (user_id, note_title,note_text)
                            VALUES(?, ?, ?)
                        """, (check_user[0], note_title, note_text))
                    continue
                else:
                    break

        else:
            print("Please login or password incorrect, try again!!")