Mercurial > sci
diff scictl.c @ 0:f1de39079243
misc: initial import
author | David Demelier <markand@malikania.fr> |
---|---|
date | Mon, 07 Jun 2021 09:41:37 +0200 |
parents | |
children | 5afdb14df924 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/scictl.c Mon Jun 07 09:41:37 2021 +0200 @@ -0,0 +1,180 @@ +#define _BSD_SOURCE +#include <err.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <stdnoreturn.h> +#include <unistd.h> + +#include "config.h" +#include "project.h" +#include "req.h" +#include "util.h" +#include "worker.h" + +noreturn static void +usage(void) +{ + fprintf(stderr, "usage: %s [-s sock] command [args...]\n", getprogname()); + exit(1); +} + +noreturn static void +help(void) +{ + fprintf(stderr, "usage: job-queue project tag\n"); + fprintf(stderr, " project-add name desc url script\n"); + fprintf(stderr, " project-list\n"); + fprintf(stderr, " worker-add name desc\n"); + fprintf(stderr, " worker-list\n"); + exit(0); +} + +static struct req +cmd_job_queue(int argc, char **argv) +{ + if (argc < 2) + usage(); + + return req_job_queue(argv[0], argv[1]); +} + +static struct req +cmd_project_add(int argc, char **argv) +{ + struct project pc; + + if (argc < 4) + usage(); + + memset(&pc, 0, sizeof (pc)); + strlcpy(pc.name, argv[0], sizeof (pc.name)); + strlcpy(pc.desc, argv[1], sizeof (pc.desc)); + strlcpy(pc.url, argv[2], sizeof (pc.url)); + strlcpy(pc.script, argv[3], sizeof (pc.script)); + + return req_project_add(&pc); +} + +static struct req +cmd_project_list(int argc, char **argv) +{ + (void)argc; + (void)argv; + + struct project pc[SCI_PROJECT_MAX]; + struct req req; + size_t pcsz = UTIL_SIZE(pc); + + memset(pc, 0, sizeof (pc)); + + if ((req = req_project_list(pc, &pcsz)).status) + return req; + + for (size_t i = 0; i < pcsz; ++i) { + printf("%-16s%s\n", "name:", pc[i].name); + printf("%-16s%s\n", "description:", pc[i].desc); + printf("%-16s%s\n", "url:", pc[i].url); + printf("%-16s%s\n", "script:", pc[i].script); + + if (i + 1 < pcsz) + printf("\n"); + } + + return req; +} + +static struct req +cmd_worker_add(int argc, char **argv) +{ + struct worker wk; + + if (argc < 2) + usage(); + + memset(&wk, 0, sizeof (wk)); + strlcpy(wk.name, argv[0], sizeof (wk.name)); + strlcpy(wk.desc, argv[1], sizeof (wk.desc)); + + return req_worker_add(&wk); +} + +static struct req +cmd_worker_list(int argc, char **argv) +{ + (void)argc; + (void)argv; + + struct worker wk[SCI_WORKER_MAX]; + struct req req; + size_t wksz = UTIL_SIZE(wk); + + if ((req = req_worker_list(wk, &wksz)).status) + return req; + + for (size_t i = 0; i < wksz; ++i) { + printf("%-16s%s\n", "name:", wk[i].name); + printf("%-16s%s\n", "description:", wk[i].desc); + + if (i + 1 < wksz) + printf("\n"); + } + + return req; +} + +static struct { + const char *name; + struct req (*exec)(int, char **); +} commands[] = { + { "job-queue", cmd_job_queue }, + { "project-add", cmd_project_add }, + { "project-list", cmd_project_list }, + { "worker-add", cmd_worker_add }, + { "worker-list", cmd_worker_list }, + { NULL, NULL } +}; + +int +main(int argc, char **argv) +{ + const char *sock = VARDIR "/run/sci.sock"; + int ch; + + setprogname("scictl"); + + while ((ch = getopt(argc, argv, "s:")) != -1) { + switch (ch) { + case 's': + sock = optarg; + break; + default: + break; + } + } + + argc -= optind; + argv += optind; + + if (argc <= 0) + usage(); + if (strcmp(argv[0], "help") == 0) + help(); + if (req_connect(sock) < 0) + err(1, "%s", sock); + + for (size_t i = 0; commands[i].name; ++i) { + struct req res; + + if (strcmp(commands[i].name, argv[0]) == 0) { + res = commands[i].exec(--argc, ++argv); + + if (res.status) + warnx("%s", res.msg); + + break; + } + } + + req_finish(); +}