tanabata/cli/cli.c

605 lines
21 KiB
C
Raw Blame History

This file contains invisible Unicode characters

This file contains invisible Unicode characters that are indistinguishable to humans but may be processed differently by a computer. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#include <stdlib.h>
#include <getopt.h>
#include <string.h>
#include <sys/stat.h>
#include <time.h>
#include "../include/tanabata.h"
// Stylization macros
#define TABLE_HEADER(s) ""s""
#define HIGHLIGHT(s) ""s""
#define SUCCESS(s) ""s""
#define ERROR(s) ""s""
#define DT_FORMAT "%F %T"
static Tanabata tanabata;
// Print the list of all sasa
void print_sasa_all() {
printf(TABLE_HEADER(" Sasa ID\tFile path")"\n");
for (uint64_t i = 0; i < tanabata.sasahyou.size; i++) {
if (tanabata.sasahyou.database[i].id != HOLE_ID) {
printf("%16lx\t%s\n", tanabata.sasahyou.database[i].id, tanabata.sasahyou.database[i].path);
}
}
}
// Print the list of all tanzaku
void print_tanzaku_all() {
printf(TABLE_HEADER(" Tanzaku ID\tName")"\n");
for (uint64_t i = 0; i < tanabata.sappyou.size; i++) {
if (tanabata.sappyou.database[i].id != HOLE_ID) {
printf("%16lx\t%s\n", tanabata.sappyou.database[i].id, tanabata.sappyou.database[i].name);
}
}
}
// Sasa view menu handler
int menu_view_sasa(const char *arg) {
if (arg == NULL) {
return 1;
}
if (strcmp(arg, ".") == 0) {
print_sasa_all();
return 0;
}
char *endptr;
uint64_t sasa_id = strtoull(arg, &endptr, 16);
if (*endptr == 0) {
Sasa current_sasa = tanabata_sasa_get_by_id(&tanabata, sasa_id);
if (current_sasa.id != HOLE_ID) {
char datetime[20];
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &current_sasa.created_ts));
printf(HIGHLIGHT("Sasa ID")" %lx\n"
HIGHLIGHT("File path")" %s\n"
HIGHLIGHT("Added datetime")" %s\n\n",
sasa_id, current_sasa.path, datetime);
Tanzaku *related_tanzaku = tanabata_tanzaku_get_by_sasa(&tanabata, current_sasa.id);
if (related_tanzaku != NULL) {
printf(HIGHLIGHT("↓ Related tanzaku ↓")"\n");
for (Tanzaku *current_tanzaku = related_tanzaku;
current_tanzaku->id != HOLE_ID; current_tanzaku++) {
printf("'%s'\n", current_tanzaku->name);
}
printf(HIGHLIGHT("↑ Related tanzaku ↑")"\n");
} else {
printf(HIGHLIGHT("No related tanzaku")"\n");
}
return 0;
}
fprintf(stderr, ERROR("No sasa with this ID")"\n");
return 1;
}
fprintf(stderr, ERROR("Invalid ID")"\n");
return 1;
}
// Tanzaku view menu handler
int menu_view_tanzaku(const char *arg) {
if (arg == NULL) {
return 1;
}
if (strcmp(arg, ".") == 0) {
print_tanzaku_all();
return 0;
}
char *endptr;
uint64_t tanzaku_id = strtoull(arg, &endptr, 16);
if (*endptr == 0) {
Tanzaku current_tanzaku = tanabata_tanzaku_get_by_id(&tanabata, tanzaku_id);
if (current_tanzaku.id != HOLE_ID) {
char datetime[20];
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &current_tanzaku.created_ts));
printf(HIGHLIGHT("Tanzaku ID")" %lx\n"
HIGHLIGHT("Name")" %s\n"
HIGHLIGHT("Created datetime")" %s\n",
tanzaku_id, current_tanzaku.name, datetime);
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &current_tanzaku.modified_ts));
printf(HIGHLIGHT("Modified datetime")" %s\n\n", datetime);
if (*current_tanzaku.description != 0) {
printf(HIGHLIGHT("↓ Description ↓")"\n"
"%s\n"
HIGHLIGHT("↑ Description ↑")"\n\n", current_tanzaku.description);
} else {
printf(HIGHLIGHT("No description")"\n\n");
}
Sasa *related_sasa = tanabata_sasa_get_by_tanzaku(&tanabata, tanzaku_id);
if (related_sasa != NULL) {
printf(HIGHLIGHT("↓ Related sasa ↓")"\n");
for (Sasa *current_sasa = related_sasa;
current_sasa->id != HOLE_ID; current_sasa++) {
printf("'%s'\n", current_sasa->path);
}
printf(HIGHLIGHT("↑ Related sasa ↑")"\n");
} else {
printf(HIGHLIGHT("No related sasa")"\n");
}
return 0;
}
fprintf(stderr, ERROR("No tanzaku with this ID")"\n");
return 1;
}
fprintf(stderr, ERROR("Invalid ID")"\n");
return 1;
}
// Sasa add menu handler
int menu_add_sasa(const char *arg) {
if (arg == NULL) {
return 1;
}
if (tanabata.sasahyou.size == -1 && tanabata.sasahyou.hole_cnt == 0) {
fprintf(stderr, ERROR("Failed to add file to database: sasahyou is full")"\n");
return 1;
}
if (tanabata_sasa_add(&tanabata, arg) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully added file to database")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to add file to database")"\n");
return 1;
}
// Tanzaku add menu handler
int menu_add_tanzaku(const char *arg) {
if (arg == NULL) {
return 1;
}
if (tanabata.sappyou.size == -1 && tanabata.sappyou.hole_cnt == 0) {
fprintf(stderr, ERROR("Failed to add tanzaku: sappyou is full")"\n");
return 1;
}
if (*arg != 0) {
char description[4096];
printf(HIGHLIGHT("Enter tanzaku description:")"\n");
fgets(description, 4096, stdin);
description[strlen(description) - 1] = 0;
if (tanabata_tanzaku_add(&tanabata, arg, description) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully added tanzaku to database")"\n");
return 0;
}
}
fprintf(stderr, ERROR("Failed to add tanzaku to database")"\n");
return 1;
}
// Kazari add menu handler
int menu_add_kazari(char *arg) {
if (arg == NULL) {
return 1;
}
if (tanabata.shoppyou.size == -1 && tanabata.shoppyou.hole_cnt == 0) {
fprintf(stderr, ERROR("Failed to add kazari: shoppyou is full")"\n");
return 1;
}
char *left = arg, *right = "\0", *endptr;
for (size_t i = 0; i < strlen(arg); i++) {
if (arg[i] == '-') {
arg[i] = 0;
right = arg + i + 1;
break;
}
}
if (*left == 0 || *right == 0) {
fprintf(stderr, ERROR("Failed to add kazari: invalid argument")"\n");
return 1;
}
uint64_t sasa_id = strtoull(left, &endptr, 16);
if (*endptr != 0) {
fprintf(stderr, ERROR("Failed to add kazari: invalid sasa ID")"\n");
return 1;
}
uint64_t tanzaku_id = strtoull(right, &endptr, 16);
if (*endptr != 0) {
fprintf(stderr, ERROR("Failed to add kazari: invalid tanzaku ID")"\n");
return 1;
}
if (tanabata_kazari_add(&tanabata, sasa_id, tanzaku_id) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully added kazari")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to add kazari")"\n");
return 1;
}
// Sasa remove menu handler
int menu_rem_sasa(const char *arg) {
if (arg == NULL) {
return 1;
}
char *endptr;
uint64_t sasa_id = strtoull(arg, &endptr, 16);
if (*endptr == 0) {
if (tanabata_sasa_rem_by_id(&tanabata, sasa_id) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully removed sasa")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to remove sasa")"\n");
return 1;
}
fprintf(stderr, ERROR("Invalid ID")"\n");
return 1;
}
// Tanzaku remove menu handler
int menu_rem_tanzaku(const char *arg) {
if (arg == NULL) {
return 1;
}
char *endptr;
uint64_t tanzaku_id = strtoull(arg, &endptr, 16);
if (*endptr == 0) {
if (tanabata_tanzaku_rem_by_id(&tanabata, tanzaku_id) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully removed tanzaku")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to remove tanzaku")"\n");
return 1;
}
fprintf(stderr, ERROR("Invalid ID")"\n");
return 1;
}
// Kazari remove menu handler
int menu_rem_kazari(char *arg) {
if (arg == NULL) {
return 1;
}
char *left = arg, *right = "\0", *endptr;
for (size_t i = 0; i < strlen(arg); i++) {
if (arg[i] == '-') {
arg[i] = 0;
right = arg + i + 1;
break;
}
}
if (*left == 0 || *right == 0) {
fprintf(stderr, ERROR("Failed to remove kazari: invalid argument")"\n");
return 1;
}
uint64_t sasa_id = strtoull(left, &endptr, 16);
if (*endptr != 0) {
fprintf(stderr, ERROR("Failed to remove kazari: invalid sasa ID")"\n");
return 1;
}
uint64_t tanzaku_id = strtoull(right, &endptr, 16);
if (*endptr != 0) {
fprintf(stderr, ERROR("Failed to remove kazari: invalid tanzaku ID")"\n");
return 1;
}
if (tanabata_kazari_rem(&tanabata, sasa_id, tanzaku_id) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully removed kazari")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to remove kazari")"\n");
return 1;
}
// Sasa update menu handler
int menu_upd_sasa(const char *arg) {
if (arg == NULL) {
return 1;
}
char *endptr;
uint64_t sasa_id = strtoull(arg, &endptr, 16);
if (*endptr == 0) {
char *path = malloc(4096);
printf(HIGHLIGHT("Enter the new file path (leave blank to keep current):")"\n");
fgets(path, 4096, stdin);
if (*path == '\n') {
free(path);
path = NULL;
} else {
path[strlen(path) - 1] = 0;
}
if (tanabata_sasa_upd(&tanabata, sasa_id, path) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully updated sasa")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to update sasa")"\n");
return 1;
}
fprintf(stderr, ERROR("Invalid ID")"\n");
return 1;
}
// Tanzaku update menu handler
int menu_upd_tanzaku(const char *arg) {
if (arg == NULL) {
return 1;
}
char *endptr;
uint64_t tanzaku_id = strtoull(arg, &endptr, 16);
if (*endptr == 0) {
char *name = malloc(4096), *description = malloc(4096);
printf(HIGHLIGHT("Enter the new name of tanzaku (leave blank to keep current):")"\n");
fgets(name, 4096, stdin);
if (*name == '\n') {
free(name);
name = NULL;
} else {
name[strlen(name) - 1] = 0;
}
printf(HIGHLIGHT("Enter the new description of tanzaku (leave blank to keep current):")"\n");
fgets(description, 4096, stdin);
if (*description == '\n') {
free(description);
description = NULL;
} else {
description[strlen(description) - 1] = 0;
}
if (tanabata_tanzaku_upd(&tanabata, tanzaku_id, name, description) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully updated tanzaku")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to update tanzaku")"\n");
return 1;
}
fprintf(stderr, ERROR("Invalid ID")"\n");
return 1;
}
int main(int argc, char **argv) {
if (argc == 1) {
fprintf(stderr, ERROR("No options provided")"\n");
return 1;
}
char *tanabata_path;
FILE *config = fopen("/etc/tfm/config", "r");
if (config == NULL) {
tanabata_path = NULL;
struct stat st;
if (stat("/etc/tfm", &st) == -1) {
if (mkdir("/etc/tfm", 0755) != 0) {
fprintf(stderr, ERROR("Failed to create '/etc/tfm' directory. "
"Try again with 'sudo' or check your permissions")"\n");
return 1;
}
}
config = fopen("/etc/tfm/config", "w");
if (config == NULL) {
fprintf(stderr, ERROR("Failed to create config file. "
"Try again with 'sudo' or check your permissions")"\n");
return 1;
}
} else {
fseek(config, 0L, SEEK_END);
long fsize = ftell(config);
rewind(config);
if (fsize == 0) {
tanabata_path = NULL;
} else {
tanabata_path = malloc(fsize + 1);
if (fgets(tanabata_path, INT32_MAX, config) == NULL) {
fprintf(stderr, ERROR("Failed to read config file")"\n");
return 1;
}
}
}
const char *shortopts = "hI:O:isuef:t:c:wV";
char *abspath = NULL;
int opt;
_Bool opt_i = 0;
_Bool opt_s = 0;
_Bool opt_u = 0;
_Bool opt_e = 0;
_Bool opt_f = 0;
_Bool opt_t = 0;
_Bool opt_c = 0;
_Bool opt_w = 0;
char *opt_f_arg;
char *opt_t_arg;
char *opt_c_arg;
while ((opt = getopt(argc, argv, shortopts)) != -1) {
switch (opt) {
case 'h':
printf(
HIGHLIGHT("(C) Masahiko AMANO aka H1K0, 2022—present")"\n"
HIGHLIGHT("(https://github.com/H1K0/tanabata)")"\n\n"
HIGHLIGHT("Usage:")"\n"
"tfm <options>\n\n"
HIGHLIGHT("Options:")"\n"
HIGHLIGHT("-h")" Print this help and exit\n"
HIGHLIGHT("-I <dir>")" Initialize new Tanabata database in directory <dir>\n"
HIGHLIGHT("-O <dir>")" Open existing Tanabata database from directory <dir>\n"
HIGHLIGHT("-i")" View database info\n"
HIGHLIGHT("-s")" Set or add\n"
HIGHLIGHT("-u")" Unset or remove\n"
HIGHLIGHT("-e")" Edit or update\n"
HIGHLIGHT("-f <sasa_id or path>")" File-sasa menu\n"
HIGHLIGHT("-t <tanzaku_id or name>")" Tanzaku menu\n"
HIGHLIGHT("-c <sasa_id>-<tanzaku_id>")" Kazari menu "
"(can only be used with the '-s' or '-u' option)\n"
HIGHLIGHT("-w")" Weed (defragment) database\n"
HIGHLIGHT("-V")" Print version and exit\n\n"
);
if (tanabata_path != NULL) {
printf(HIGHLIGHT("Current database location: %s")"\n", tanabata_path);
} else {
printf(HIGHLIGHT("No database connected")"\n");
}
return 0;
case 'V':
printf("0.1.3-dev\n");
return 0;
case 'I':
abspath = realpath(optarg, abspath);
if (abspath == NULL) {
fprintf(stderr, ERROR("Invalid path")"\n");
return 1;
}
if (tanabata_init(&tanabata) == 0 &&
tanabata_dump(&tanabata, abspath) == 0) {
config = freopen(NULL, "w", config);
if (config == NULL) {
fprintf(stderr, ERROR("Failed to update config file. "
"Try again with 'sudo' or check your permissions")"\n");
return 1;
}
fputs(abspath, config);
fclose(config);
printf(SUCCESS("Successfully initialized Tanabata database")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to initialize Tanabata database")"\n");
return 1;
case 'O':
abspath = realpath(optarg, abspath);
if (abspath == NULL) {
fprintf(stderr, ERROR("Invalid path")"\n");
return 1;
}
if (tanabata_open(&tanabata, abspath) == 0) {
config = freopen(NULL, "w", config);
if (config == NULL) {
fprintf(stderr, ERROR("Failed to update config file. "
"Try again with 'sudo' or check your permissions")"\n");
return 1;
}
fputs(abspath, config);
fclose(config);
printf(SUCCESS("Successfully opened Tanabata database")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to open Tanabata database")"\n");
return 1;
case 'i':
opt_i = 1;
break;
case 's':
opt_s = 1;
break;
case 'u':
opt_u = 1;
break;
case 'e':
opt_e = 1;
break;
case 'f':
opt_f = 1;
opt_f_arg = optarg;
break;
case 't':
opt_t = 1;
opt_t_arg = optarg;
break;
case 'c':
opt_c = 1;
opt_c_arg = optarg;
break;
case 'w':
opt_w = 1;
break;
case '?':
return 1;
default:
break;
}
}
if (tanabata_path == NULL) {
fprintf(stderr, ERROR("No connected database")"\n");
return 1;
}
if (tanabata_open(&tanabata, tanabata_path) != 0) {
fprintf(stderr, ERROR("Failed to load database")"\n");
return 1;
}
fclose(config);
if (opt_i) {
char datetime[20];
printf(HIGHLIGHT("Current database location: %s")"\n\n"
HIGHLIGHT("SASAHYOU")"\n", tanabata_path);
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &tanabata.sasahyou.created_ts));
printf(" "HIGHLIGHT("Created")" %s\n", datetime);
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &tanabata.sasahyou.modified_ts));
printf(" "HIGHLIGHT("Last modified")" %s\n"
" "HIGHLIGHT("Number of sasa")" %lu\n"
" "HIGHLIGHT("Number of holes")" %lu\n\n"
HIGHLIGHT("SAPPYOU")"\n", datetime, tanabata.sasahyou.size, tanabata.sasahyou.hole_cnt);
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &tanabata.sappyou.created_ts));
printf(" "HIGHLIGHT("Created")" %s\n", datetime);
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &tanabata.sappyou.modified_ts));
printf(" "HIGHLIGHT("Last modified")" %s\n"
" "HIGHLIGHT("Number of tanzaku")" %lu\n"
" "HIGHLIGHT("Number of holes")" %lu\n\n"
HIGHLIGHT("SHOPPYOU")"\n", datetime, tanabata.sappyou.size, tanabata.sappyou.hole_cnt);
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &tanabata.shoppyou.created_ts));
printf(" "HIGHLIGHT("Created")" %s\n", datetime);
strftime(datetime, 20, DT_FORMAT,
localtime((const time_t *) &tanabata.shoppyou.modified_ts));
printf(" "HIGHLIGHT("Last modified")" %s\n"
" "HIGHLIGHT("Number of kazari")" %lu\n"
" "HIGHLIGHT("Number of holes")" %lu\n",
datetime, tanabata.shoppyou.size, tanabata.shoppyou.hole_cnt);
return 0;
}
free(tanabata_path);
if (opt_w) {
if (tanabata_weed(&tanabata) == 0 &&
tanabata_save(&tanabata) == 0) {
printf(SUCCESS("Successfully weeded database")"\n");
return 0;
}
fprintf(stderr, ERROR("Failed to weed database")"\n");
return 1;
}
if (opt_s && opt_u) {
opt_s = 0;
opt_u = 0;
}
if (opt_s) {
if (opt_f) {
return menu_add_sasa(opt_f_arg);
}
if (opt_t) {
return menu_add_tanzaku(opt_t_arg);
}
if (opt_c) {
return menu_add_kazari(opt_c_arg);
}
} else if (opt_u) {
if (opt_f) {
return menu_rem_sasa(opt_f_arg);
}
if (opt_t) {
return menu_rem_tanzaku(opt_t_arg);
}
if (opt_c) {
return menu_rem_kazari(opt_c_arg);
}
} else if (opt_e) {
if (opt_f) {
return menu_upd_sasa(opt_f_arg);
}
if (opt_t) {
return menu_upd_tanzaku(opt_t_arg);
}
} else {
if (opt_f) {
return menu_view_sasa(opt_f_arg);
}
if (opt_t) {
return menu_view_tanzaku(opt_t_arg);
}
}
return 0;
}