pit/src/db.c

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <pwd.h>
#include <sys/types.h>
#include "pit.h"

#define PITFILE "~/.pit"

char *mem2str(char *mem, int len) {
    char *str = malloc(len + 1);
    memcpy(str, mem, len);
    str[len] = '\0';

    return str;
}

static char *home_dir(char *username, int len) {
    char *str = mem2str(username, len);
    struct passwd *pw = getpwnam(str);
    free(str);

    return (pw ? pw->pw_dir : NULL);
}

static char *expand_path(char *path, char *expanded) {
    if (!path || *path != '~') {
        return path;
    } else {
        char *next = path + 1;
        if (*next == '/') { /* Path without the username, i.e. ~/file */
            strcpy(expanded, getenv("HOME"));
            strcat(expanded, next);
        } else {            /* Path with the username, i.e. ~username/file */
            char *slash = strchr(next, '/');
            if (!slash) {
                slash = next + strlen(next);
            }
            char *home = home_dir(next, slash - next);
            if (!home) {    /* Ex. non-existent username. */
                perish(path);
            } else {
                strcpy(expanded, home);
                strcat(expanded, slash);
            }
        }
    }

    return expanded;
}

static char *pit_file_name()
{
    static char file_name[128];

    if (!*file_name) {
        strcpy(file_name, expand_path(PITFILE, file_name));
    }

    return file_name;
}

int pit_init(char *argv[]) {
    char **arg = &argv[1];
    char *file_name = pit_file_name();
    int reply = 'Y';

    if (!access(file_name, F_OK)) {
        if (!*arg || strcmp(*arg, "-f")) { /* Do not prompt user if forced init (-f). */
            printf("%s already exist, do you want to override it [y/N]: ", file_name);
            reply = getchar();
        }
    }

    if (reply == 'y' || reply == 'Y') {
        pit_db_initialize();
        printf("Created empty %s\n", file_name);
    }
    return 1;
}

void pit_db_load() {
    char *file_name = pit_file_name();
    FILE *file = fopen(file_name, "r");

    if (!file) {
        perish(file_name);
    } else {
        projects   = pit_table_load(file);
        tasks      = pit_table_load(file);
        notes      = pit_table_load(file);
        activities = pit_table_load(file);
        users      = pit_table_load(file);
        fclose(file);
    }
}

void pit_db_initialize() {
    projects   = pit_table_initialize(sizeof(Project),  TABLE_HAS_ID | TABLE_HAS_TIMESTAMPS);
    tasks      = pit_table_initialize(sizeof(Task),     TABLE_HAS_ID | TABLE_HAS_TIMESTAMPS);
    notes      = pit_table_initialize(sizeof(Note),     TABLE_HAS_ID | TABLE_HAS_TIMESTAMPS);
    activities = pit_table_initialize(sizeof(Activity), TABLE_HAS_CREATED_AT);
    users      = pit_table_initialize(sizeof(User),     TABLE_HAS_ID);

    pit_table_insert(users,      (uchar *)pit_current_user());
    pit_table_insert(activities, (uchar *)pit_add_activity(0, "", "Initialized pit", 1));

    pit_db_save();
}

void pit_db_save() {
    char *file_name = pit_file_name();
    FILE *file = fopen(file_name, "wb");

    if (!file) {
        perish(file_name);
    } else {
        pit_table_save(file, projects);
        pit_table_save(file, tasks);
        pit_table_save(file, notes);
        pit_table_save(file, activities);
        pit_table_save(file, users);
        fclose(file);
    }
}