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();
+}