private static boolean createDatabase(TypeDBDriver driver, String dbName) {
    System.out.print("Creating a new database...");
    driver.databases().create(dbName);
    System.out.println("OK");
    try (TypeDBSession session = driver.session(dbName, TypeDBSession.Type.SCHEMA)) {
        dbSchemaSetup(session);
    }
    try (TypeDBSession session = driver.session(dbName, TypeDBSession.Type.DATA)) {
        dbDatasetSetup(session);
    }
    return true;
}

private static boolean replaceDatabase(TypeDBDriver driver, String dbName) {
    System.out.print("Deleting an existing database...");
    driver.databases().get(dbName).delete();  // Delete the database if it exists already
    System.out.println("OK");
    if (createDatabase(driver,dbName)) {
        return true;
    } else {
        System.out.println("Failed to create a new database. Terminating...");
        return false;
    }
}

private static void dbSchemaSetup(TypeDBSession session) {
    String schemaFile = "iam-schema.tql";
    try (TypeDBTransaction tx = session.transaction(TypeDBTransaction.Type.WRITE)) {
        String defineQuery = new String(Files.readAllBytes(Paths.get(schemaFile)));
        System.out.print("Defining schema...");
        tx.query().define(defineQuery).resolve();
        tx.commit();
        System.out.println("OK");
    } catch (IOException e) {
        throw new RuntimeException("Failed to read the schema file.", e);
    }
}

private static void dbDatasetSetup(TypeDBSession session) {
    String dataFile = "iam-data-single-query.tql";
    try (TypeDBTransaction tx = session.transaction(TypeDBTransaction.Type.WRITE)) {
        String insertQuery = new String(Files.readAllBytes(Paths.get(dataFile)));
        System.out.print("Loading data...");
        tx.query().insert(insertQuery).toList();
        tx.commit();
        System.out.println("OK");
    } catch (IOException e) {
        throw new RuntimeException("Failed to read the data file.", e);
    }
}

private static boolean dbCheck(TypeDBSession session) {
    try (TypeDBTransaction transaction = session.transaction(TypeDBTransaction.Type.READ)) {
        String testQuery = "match $u isa user; get $u; count;";
        System.out.print("Testing the database...");
        long result = transaction.query().getAggregate(testQuery).resolve().get().asLong();
        if (result == 3) {
            System.out.println("Passed");
            return true;
        } else {
            System.out.println("Failed the test with the result: " + result + "\n Expected result: 3.");
            return false;
        }
    }
}

private static List<JSON> fetchAllUsers(TypeDBDriver driver, String dbName) {
    try (TypeDBSession session = driver.session(dbName, TypeDBSession.Type.DATA)) {
        try (TypeDBTransaction tx = session.transaction(TypeDBTransaction.Type.READ)) {
            String query = "match $u isa user; fetch $u: full-name, email;";
            List<JSON> answers = tx.query().fetch(query).toList();
            answers.forEach(json -> System.out.println("JSON: " + json.toString()));
            return answers;
        }
    }
}

public static List<ConceptMap> insertNewUser(TypeDBDriver driver, String dbName, String name, String email) {
    try (TypeDBSession session = driver.session(dbName, TypeDBSession.Type.DATA)) {
        try (TypeDBTransaction tx = session.transaction(TypeDBTransaction.Type.WRITE)) {
            String query = String.format(
                    "insert $p isa person, has full-name $fn, has email $e; $fn \"%s\"; $e \"%s\";", name, email);
            List<ConceptMap> response = tx.query().insert(query).toList();
            tx.commit();
            for (ConceptMap conceptMap : response) {
                String fullName = conceptMap.get("fn").asAttribute().getValue().asString();
                String emailAddress = conceptMap.get("e").asAttribute().getValue().asString();
                System.out.println("Added new user. Name: " + fullName + ", E-mail: " + emailAddress);
            }
            return response;
        }
    }
}

public static List<ConceptMap> getFilesByUser(TypeDBDriver driver, String dbName, String name, boolean inference) {
    List<ConceptMap> filePaths = new ArrayList<>();
    TypeDBOptions options = new TypeDBOptions().infer(inference);
    try (TypeDBSession session = driver.session(dbName, TypeDBSession.Type.DATA);
         TypeDBTransaction tx = session.transaction(TypeDBTransaction.Type.READ, options)) {

        String userQuery = String.format("match $u isa user, has full-name '%s'; get;", name);
        List<ConceptMap> users = tx.query().get(userQuery).toList();

        if (users.size() > 1) {
            System.out.println("Error: Found more than one user with that name.");
            return null;
        } else if (users.size() == 1) {
            String fileQuery = String.format("""
                                            match
                                            $fn '%s';
                                            $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;""", name);
            tx.query().get(fileQuery).forEach(filePaths::add);
            filePaths.forEach(path -> System.out.println("File: " + path.get("fp").asAttribute().getValue().toString()));
            if (filePaths.isEmpty()) {
                System.out.println("No files found. Try enabling inference.");
            }
            return filePaths;
        } else {
            System.out.println("Warning: No users found with that name.");
            return null;
        }
    } catch (TypeDBDriverException e) {
        e.printStackTrace();
        return null;
    }
}

public static List<ConceptMap> updateFilePath(TypeDBDriver driver, String dbName, String oldPath, String newPath) {
    List<ConceptMap> response = new ArrayList<>();
    try (TypeDBSession session = driver.session(dbName, TypeDBSession.Type.DATA);
         TypeDBTransaction tx = session.transaction(TypeDBTransaction.Type.WRITE)) {
        String query = String.format("""
                                    match
                                    $f isa file, has path $old_path;
                                    $old_path = '%s';
                                    delete
                                    $f has $old_path;
                                    insert
                                    $f has path '%s';""", oldPath, newPath);
        response = tx.query().update(query).toList();
        if (!response.isEmpty()) {
            tx.commit();
            System.out.println(String.format("Total number of paths updated: %s", response.size()));
            return response;
        } else {
            System.out.println("No matched paths: nothing to update");
        }

    } catch (TypeDBDriverException e) {
        e.printStackTrace();
    }
    return response;
}

public static boolean deleteFile(TypeDBDriver driver, String dbName, String path) {
    try (TypeDBSession session = driver.session(dbName, TypeDBSession.Type.DATA);
         TypeDBTransaction tx = session.transaction(TypeDBTransaction.Type.WRITE)) {

        String query = String.format("match $f isa file, has path '%s'; get;", path);
        List<ConceptMap> response = tx.query().get(query).toList();

        if (response.size() == 1) {
            tx.query().delete(String.format("match $f isa file, has path '%s'; delete $f isa file;", path)).resolve();
            tx.commit();
            System.out.println("The file has been deleted.");
            return true;
        } else if (response.size() > 1) {
            System.out.println("Matched more than one file with the same path. No files were deleted.");
            return false;
        } else {
            System.out.println("No files matched in the database. No files were deleted.");
            return false;
        }
    } catch (TypeDBDriverException e) {
        e.printStackTrace();
        return false;
    }
}