main.c

#include <dolphinfs.h>

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>

#define CLI_DISK_DEV "disk"

// Function declarations
int cmd_mkfs(void);
int cmd_put(const char* src_path, const char* dest_path);
int cmd_get(const char* src_path, const char* dest_path);
int cmd_list(const char* dir_path);
int cmd_delete_file(const char* file_path);
int cmd_copy(const char* src_path, const char* dest_path);
int cmd_move(const char* src_path, const char* dest_path);
int cmd_mkdir(const char* dir_path);
int cmd_test(void);

// Function implementations
int cmd_mkfs(void) {
    // Implement file system creation operation
    printf("Creating file system\n");

    /* init disk */
    init_blkdev();
    list_blkdev();

    struct super_block dolphin_sb;

    /* 挂载文件系统 */
    if (!dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("Filesystem exist, can not format!\n");
        dolphin_unmount(&dolphin_sb);
        return -1;
    }

    /* 创建文件系统 */
    dolphin_mkfs(CLI_DISK_DEV, &dolphin_sb);
    /* 再次挂载 */
    if (dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("mount dolphinfs failed!\n");
        return -1;
    }

    /* 挂载成功 */
    dolphin_unmount(&dolphin_sb);

    exit_blkdev();

    printf("Create filesystem success.\n");

    return 0; // Return 0 to indicate success
}

int cmd_put(const char* src_path, const char* dest_path) {
    // Implement the operation to copy a file from the current directory to the file system
    printf("Copying %s to %s in the file system\n", src_path, dest_path);
    int ret = 0;

    /* init disk */
    init_blkdev();
    list_blkdev();

    struct super_block dolphin_sb;

    /* 挂载文件系统 */
    if (dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("Mount filesystem failed!\n");
        return -1;
    }

    FILE *src_file;
    int dest_file;
    
    char buffer[1024];
    size_t bytes_read;

    // Open the source file in the current directory
    src_file = fopen(src_path, "rb");
    if (src_file == NULL) {
        perror("Error opening source file");
        ret = 1;
        goto end;
    }

    // Open the destination file in the file system
    // NOTE: This example uses fopen, but you should use your file system's API
    dest_file = open_file(&dolphin_sb, dest_path, FF_WRITE | FF_CRATE);
    if (dest_file == -1) {
        perror("Error opening destination file");
        fclose(src_file);
        ret = 1;
        goto end;
    }

    // Copy the file contents
    while ((bytes_read = fread(buffer, 1, sizeof(buffer), src_file)) > 0) {
        if (write_file(dest_file, buffer, bytes_read) != bytes_read) {
            perror("Error writing to destination file");
            fclose(src_file);
            close_file(dest_file);
            ret = 1;
            goto end;
        }
    }

    if (ferror(src_file)) {
        perror("Error reading source file");
        fclose(src_file);
        close_file(dest_file);
        ret = 1;
        goto end;
    }

    // Close both files
    fclose(src_file);
    close_file(dest_file);
end:
    /* 挂载成功 */
    dolphin_unmount(&dolphin_sb);

    exit_blkdev();

    printf("File %s copied to %s %s\n", src_path, dest_path, !ret ? "successfully" : "failed");
    return ret; // Return 0 to indicate success
}

int cmd_get(const char* src_path, const char* dest_path) {
    // Implement the operation to copy a file from the file system to the current directory
    printf("Copying %s from the file system to %s\n", src_path, dest_path);

    /* init disk */
    init_blkdev();
    list_blkdev();
    int ret = 0;

    struct super_block dolphin_sb;

    /* 挂载文件系统 */
    if (dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("Mount filesystem failed!\n");
        return -1;
    }

    int src_file;
    FILE *dest_file;
    char buffer[1024];
    size_t bytes_read;

    // Open the source file in the file system
    // NOTE: This example uses fopen, but you should use your file system's API
    src_file = open_file(&dolphin_sb, src_path, FF_READ);
    if (src_file == -1) {
        perror("Error opening source file");
        ret = 1;
        goto end;
    }

    // Open the destination file in the current directory
    dest_file = fopen(dest_path, "wb");
    if (dest_file == NULL) {
        perror("Error opening destination file");
        close_file(src_file);
        ret = 1;
        goto end;
    }

    // Copy the file contents
    while ((bytes_read = read_file(src_file, buffer, sizeof(buffer))) > 0) {
        if (fwrite(buffer, 1, bytes_read, dest_file) != bytes_read) {
            perror("Error writing to destination file");
            close_file(src_file);
            fclose(dest_file);
            ret = 1;
            goto end;
        }
    }

    if (tell_file(src_file) != seek_file(src_file, 0, SEEK_END)) {
        perror("Error reading source file");
        close_file(src_file);
        fclose(dest_file);
        ret = 1;
        goto end;
    }

    // Close both files
    close_file(src_file);
    fclose(dest_file);
end:
    /* 挂载成功 */
    dolphin_unmount(&dolphin_sb);

    exit_blkdev();

    printf("File %s copied to %s %s\n", src_path, dest_path, !ret ? "successfully" : "failed");
    return 0; // Return 0 to indicate success
}

int cmd_list(const char* dir_path) {
    // Implement the operation to list files in a directory of the file system
    printf("Listing %s in the file system\n", dir_path);

    /* init disk */
    init_blkdev();
    list_blkdev();

    struct super_block dolphin_sb;

    /* 挂载文件系统 */
    if (dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("Mount filesystem failed!\n");
        return -1;
    }

    list_files(&dolphin_sb);

    /* 挂载成功 */
    dolphin_unmount(&dolphin_sb);

    exit_blkdev();

    printf("List filesystem success.\n");

    return 0; // Return 0 to indicate success
}

int cmd_delete_file(const char* file_path) {
    // Implement the operation to delete a file in the file system
    printf("Deleting %s in the file system\n", file_path);

    /* init disk */
    init_blkdev();
    list_blkdev();

    struct super_block dolphin_sb;

    /* 挂载文件系统 */
    if (dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("Mount filesystem failed!\n");
            
        /* 挂载成功 */
        dolphin_unmount(&dolphin_sb);

        exit_blkdev();
        return -1;
    }

    if (!delete_file(&dolphin_sb, file_path)) {    
        printf("Delete file %s success.\n", file_path);
    } else {
        printf("Delete file %s failed!\n", file_path);
    }

    /* 挂载成功 */
    dolphin_unmount(&dolphin_sb);

    exit_blkdev();

    return 0; // Return 0 to indicate success
}

int cmd_copy(const char* src_path, const char* dest_path) {
    // Implement the operation to copy a file within the file system
    printf("Copying %s to %s in the file system\n", src_path, dest_path);

    /* init disk */
    init_blkdev();
    list_blkdev();
    int ret = 0;

    struct super_block dolphin_sb;

    /* 挂载文件系统 */
    if (dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("Mount filesystem failed!\n");
        return -1;
    }

    int src_file, dest_file;
    char buffer[1024];
    size_t bytes_read;

    // Open the source file in the file system
    src_file = open_file(&dolphin_sb, src_path, FF_READ);
    if (src_file == -1) {
        perror("Error opening source file");
        ret = 1;
        goto end;
    }

    // Open the destination file in the file system
    dest_file = open_file(&dolphin_sb, dest_path, FF_WRITE | FF_CRATE);
    if (dest_file == -1) {
        perror("Error opening destination file");
        close_file(src_file);
        ret = 1;
        goto end;
    }

    // Copy the file contents
    while ((bytes_read = read_file(src_file, buffer, sizeof(buffer))) > 0) {
        if (write_file(dest_file, buffer,bytes_read) != bytes_read) {
            perror("Error writing to destination file");
            close_file(src_file);
            close_file(dest_file);
            ret = 1;
        goto end;
        }
    }

    if (tell_file(src_file) != seek_file(src_file, 0, SEEK_END)) {
        perror("Error reading source file");
        close_file(src_file);
        close_file(dest_file);
        ret = 1;
        goto end;
    }

    // Close both files
    close_file(src_file);
    close_file(dest_file);
end:
    /* 挂载成功 */
    dolphin_unmount(&dolphin_sb);

    exit_blkdev();

    printf("File %s copied to %s %s\n", src_path, dest_path, !ret ? "successfully" : "failed");
    return 0; // Return 0 to indicate success
}

int cmd_move(const char* src_path, const char* dest_path) {
    // Implement the operation to move a file within the file system
    printf("Moving %s to %s in the file system\n", src_path, dest_path);

    /* init disk */
    init_blkdev();
    list_blkdev();

    struct super_block dolphin_sb;

    /* 挂载文件系统 */
    if (dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("Mount filesystem failed!\n");
        return -1;
    }

    if (!rename_file(&dolphin_sb, src_path, dest_path)) {    
        printf("Move file %s to %s success.\n", src_path, dest_path);
    } else {
        printf("Move file%s to %s failed!\n", src_path, dest_path);
    }

    /* 挂载成功 */
    dolphin_unmount(&dolphin_sb);

    exit_blkdev();

    return 0; // Return 0 to indicate success
}

int cmd_mkdir(const char* dir_path) {
    // Implement the operation to create a directory in the file system
    printf("Creating directory %s not support!\n", dir_path);

    
    /* init disk */
    init_blkdev();
    list_blkdev();

    struct super_block dolphin_sb;

    /* 挂载文件系统 */
    if (dolphin_mount(CLI_DISK_DEV, &dolphin_sb)) {
        printf("Mount filesystem failed!\n");
        return -1;
    }

    int fd = open_file(&dolphin_sb, dir_path, FF_WRITE | FF_CRATE);
    if (fd == -1) {
        perror("Error create dir file\n");
            
        /* 挂载成功 */
        dolphin_unmount(&dolphin_sb);

        exit_blkdev();

        return 1;
    }

    close_file(fd);

    /* 挂载成功 */
    dolphin_unmount(&dolphin_sb);

    exit_blkdev();

    printf("Creating directory %s success.\n", dir_path);

    return 0; // Return 0 to indicate success
}

int main(int argc, char* argv[]) {
    // Check if the number of arguments is less than 2
    if (argc < 2) {
        printf("Usage: %s <command> [<args>]\n", argv[0]);
        printf("Commands:\n");
        printf("  mkfs\n");
        printf("  put <src> <dest>\n");
        printf("  get <src> <dest>\n");
        printf("  list <dir>\n");
        printf("  delete <file>\n");
        printf("  copy <src> <dest>\n");
        printf("  move <src> <dest>\n");
        printf("  mkdir <dir>\n");
        printf("  test\n");
        return 1;
    }

    // Process commands using if-else statements
    if (strcmp(argv[1], "mkfs") == 0) {
        return cmd_mkfs();
    } else if (strcmp(argv[1], "put") == 0) {
        if (argc != 4) {
            printf("Usage: %s put <src> <dest>\n", argv[0]);
            return 1;
        }
        return cmd_put(argv[2], argv[3]);
    } else if (strcmp(argv[1], "get") == 0) {
        if (argc != 4) {
            printf("Usage: %s get <src> <dest>\n", argv[0]);
            return 1;
        }
        return cmd_get(argv[2], argv[3]);
    } else if (strcmp(argv[1], "list") == 0) {
        if (argc != 3) {
            printf("Usage: %s list <dir>\n", argv[0]);
            return 1;
        }
        return cmd_list(argv[2]);
    } else if (strcmp(argv[1], "delete") == 0) {
        if (argc != 3) {
            printf("Usage: %s delete <file>\n", argv[0]);
            return 1;
        }
        return cmd_delete_file(argv[2]);
    } else if (strcmp(argv[1], "copy") == 0) {
        if (argc != 4) {
            printf("Usage: %s copy <src> <dest>\n", argv[0]);
            return 1;
        }
        return cmd_copy(argv[2], argv[3]);
    } else if (strcmp(argv[1], "move") == 0) {
        if (argc != 4) {
            printf("Usage: %s move <src> <dest>\n", argv[0]);
            return 1;
        }
        return cmd_move(argv[2], argv[3]);
    } else if (strcmp(argv[1], "mkdir") == 0) {
        if (argc != 3) {
            printf("Usage: %s mkdir <dir>\n", argv[0]);
            return 1;
        }
        return cmd_mkdir(argv[2]);
    } else if (strcmp(argv[1], "test") == 0) {
        if (argc != 3) {
            printf("Usage: %s test\n", argv[0]);
            return 1;
        }
        return cmd_test();
    } else {
        printf("Unknown command: %s\n", argv[1]);
        return 1;
    }

    return 0;
}