def create_database(driver, db_name) -> bool:
    print("Creating a new database", end="...")
    driver.databases.create(db_name)
    print("OK")
    with driver.session(db_name, SessionType.SCHEMA) as session:
        db_schema_setup(session)
    with driver.session(db_name, SessionType.DATA) as session:
        db_dataset_setup(session)
    return True

def replace_database(driver, db_name) -> bool:
    print("Deleting an existing database", end="...")
    driver.databases.get(db_name).delete()  # Delete the database if it exists already
    print("OK")
    if create_database(driver, db_name):
        return True
    else:
        print("Failed to create a new database. Terminating...")
        return False

def db_schema_setup(schema_session, schema_file='iam-schema.tql'):
    with open(schema_file, 'r') as data:
        define_query = data.read()
    with schema_session.transaction(TransactionType.WRITE) as tx:
        print("Defining schema", end="...")
        tx.query.define(define_query)
        tx.commit()
        print("OK")

def db_dataset_setup(data_session, data_file='iam-data-single-query.tql'):
    with open(data_file, 'r') as data:
        insert_query = data.read()
    with data_session.transaction(TransactionType.WRITE) as tx:
        print("Loading data", end="...")
        tx.query.insert(insert_query)
        tx.commit()
        print("OK")

def db_check(data_session) -> bool:
    with data_session.transaction(TransactionType.READ) as tx:
        test_query = "match $u isa user; get $u; count;"
        print("Testing the database", end="...")
        response = tx.query.get_aggregate(test_query)
        result = response.resolve().as_value().as_long()
        if result == 3:
            print("Passed")
            return True
        else:
            print("Failed the test with the result:", result, "\n Expected result: 3.")
            return False

def fetch_all_users(driver, db_name) -> list:
    with driver.session(db_name, SessionType.DATA) as data_session:
        with data_session.transaction(TransactionType.READ) as tx:
            users = list(
                tx.query.fetch("match $u isa user; fetch $u: full-name, email;")
            )
            for i, JSON in enumerate(users, start=0):
                print(f"User #{i + 1} — Full-name:", JSON['u']['full-name'][0]['value'],
                      "E-mail:", JSON['u']['email'][0]['value'])
            return users

for i, JSON in enumerate(users, start=0):
    print(f"User #{i + 1} —", end="")
    for result in JSON['u']['full-name']:
        print(f" Full-name:", result['value'], end="")
    for result in JSON['u']['email']:
        print(f" E-mail:", result['value'], end="")
    print()

def insert_new_user(driver, db_name, name, email) -> list:
    with driver.session(db_name, SessionType.DATA) as data_session:
        with data_session.transaction(TransactionType.WRITE) as tx:
            response = list(
                tx.query.insert(
                    f"insert $p isa person, has full-name $fn, has email $e; $fn == '{name}'; $e == '{email}';")
            )
            tx.commit()
            for i, concept_map in enumerate(response, start=1):
                name = concept_map.get("fn").as_attribute().get_value()
                email = concept_map.get("e").as_attribute().get_value()
                print("Added new user. Name: " + name + ", E-mail:" + email)
            return response

def get_files_by_user(driver, db_name, name, inference=False):
    options = TypeDBOptions(infer=inference)
    with driver.session(db_name, SessionType.DATA) as data_session:
        with data_session.transaction(TransactionType.READ, options) as tx:
            users = list(
                tx.query.get(f"match $u isa user, has full-name '{name}'; get;")
            )
            if len(users) > 1:
                print("Error: Found more than one user with that name.")
                return None
            elif len(users) == 1:
                response = list(
                    tx.query.get(f"""
                                    match
                                    $fn == '{name}';
                                    $u isa user, has full-name $fn;
                                    $p($u, $pa) isa permission;
                                    $o isa object, has path $fp;
                                    $pa($o, $va) isa access;
                                    $va isa action, has name 'view_file';
                                    get $fp; sort $fp asc;
                                    """)
                )
                for i, file in enumerate(response, start=1):
                    print(f"File #{i}:", file.get("fp").as_attribute().get_value())
                if len(response) == 0:
                    print("No files found. Try enabling inference.")
                return response
            else:
                print("Error: No users found with that name.")
                return None

def update_filepath(driver, db_name, old, new):
    with driver.session(db_name, SessionType.DATA) as data_session:
        with data_session.transaction(TransactionType.WRITE) as tx:
            response = list(
                tx.query.update(f"""
                                match
                                $f isa file, has path $old_path;
                                $old_path = '{old}';
                                delete
                                $f has $old_path;
                                insert
                                $f has path $new_path;
                                $new_path = '{new}';
                                """)
            )
            if len(response) > 0:
                tx.commit()
                print(f"Total number of paths updated: {len(response)}.")
                return response
            else:
                print("No matched paths: nothing to update.")
                return None

def delete_file(driver, db_name, path):
    with driver.session(db_name, SessionType.DATA) as data_session:
        with data_session.transaction(TransactionType.WRITE) as tx:
            response = list(
                tx.query.get(f"""
                                match
                                $f isa file, has path '{path}';
                                get;
                                """)
            )
            if len(response) == 1:
                tx.query.delete(f"""
                                match
                                $f isa file, has path '{path}';
                                delete
                                $f isa file;
                                """).resolve()
                tx.commit()
                print("The file has been deleted.")
                return True
            elif len(response) > 1:
                print("Matched more than one file with the same path.")
                print("No files were deleted.")
                return False
            else:
                print("No files matched in the database.")
                print("No files were deleted.")
                return False