catboy/command.c

/* Copyright (C) 2020  C. McEnroe <june@causal.agency>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#include <ctype.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "chat.h"

typedef void Command(size_t id, char *params);

static void commandDebug(size_t id, char *params) {
	(void)id;
	(void)params;
	self.debug ^= true;
	uiFormat(
		Debug, Warm, NULL,
		"\3%dDebug is %s", Gray, (self.debug ? "on" : "off")
	);
}

static void commandQuote(size_t id, char *params) {
	(void)id;
	if (params) ircFormat("%s\r\n", params);
}

static void commandPrivmsg(size_t id, char *params) {
	if (!params || !params[0]) return;
	ircFormat("PRIVMSG %s :%s\r\n", idNames[id], params);
	struct Message msg = {
		.nick = self.nick,
		.user = self.user,
		.cmd = "PRIVMSG",
		.params[0] = idNames[id],
		.params[1] = params,
	};
	handle(msg);
}

static void commandNotice(size_t id, char *params) {
	if (!params || !params[0]) return;
	ircFormat("NOTICE %s :%s\r\n", idNames[id], params);
	struct Message msg = {
		.nick = self.nick,
		.user = self.user,
		.cmd = "NOTICE",
		.params[0] = idNames[id],
		.params[1] = params,
	};
	handle(msg);
}

static void commandMe(size_t id, char *params) {
	char buf[512];
	snprintf(buf, sizeof(buf), "\1ACTION %s\1", (params ? params : ""));
	commandPrivmsg(id, buf);
}

static void commandMsg(size_t id, char *params) {
	(void)id;
	char *nick = strsep(&params, " ");
	if (!params) return;
	commandPrivmsg(idFor(nick), params);
}

static void commandJoin(size_t id, char *params) {
	size_t count = 1;
	if (params) {
		for (char *ch = params; *ch && *ch != ' '; ++ch) {
			if (*ch == ',') count++;
		}
	}
	ircFormat("JOIN %s\r\n", (params ? params : idNames[id]));
	replies.join += count;
	replies.topic += count;
	replies.names += count;
}

static void commandPart(size_t id, char *params) {
	if (params) {
		ircFormat("PART %s :%s\r\n", idNames[id], params);
	} else {
		ircFormat("PART %s\r\n", idNames[id]);
	}
}

static void commandQuit(size_t id, char *params) {
	(void)id;
	set(&self.quit, (params ? params : "Goodbye"));
}

static void commandNick(size_t id, char *params) {
	(void)id;
	if (!params) return;
	ircFormat("NICK :%s\r\n", params);
}

static void commandTopic(size_t id, char *params) {
	if (params) {
		ircFormat("TOPIC %s :%s\r\n", idNames[id], params);
	} else {
		ircFormat("TOPIC %s\r\n", idNames[id]);
		replies.topic++;
	}
}

static void commandNames(size_t id, char *params) {
	(void)params;
	ircFormat("NAMES :%s\r\n", idNames[id]);
	replies.names++;
}

static void commandList(size_t id, char *params) {
	(void)id;
	if (params) {
		ircFormat("LIST :%s\r\n", params);
	} else {
		ircFormat("LIST\r\n");
	}
	replies.list++;
}

static void commandWhois(size_t id, char *params) {
	(void)id;
	if (!params) return;
	ircFormat("WHOIS :%s\r\n", params);
	replies.whois++;
}

static void commandQuery(size_t id, char *params) {
	if (!params) return;
	size_t query = idFor(params);
	idColors[query] = completeColor(id, params);
	uiShowID(query);
}

static void commandWindow(size_t id, char *params) {
	if (!params) return;
	if (isdigit(params[0])) {
		uiShowNum(strtoul(params, NULL, 10));
	} else {
		id = idFind(params);
		if (id) uiShowID(id);
	}
}

static void commandMove(size_t id, char *params) {
	if (!params) return;
	char *name = strsep(&params, " ");
	if (params) {
		id = idFind(name);
		if (id) uiMoveID(id, strtoul(params, NULL, 10));
	} else {
		uiMoveID(id, strtoul(name, NULL, 10));
	}
}

static void commandClose(size_t id, char *params) {
	if (!params) {
		uiCloseID(id);
	} else if (isdigit(params[0])) {
		uiCloseNum(strtoul(params, NULL, 10));
	} else {
		id = idFind(params);
		if (id) uiCloseID(id);
	}
}

static void commandOpen(size_t id, char *params) {
	if (!params) {
		urlOpenCount(id, 1);
	} else if (isdigit(params[0])) {
		urlOpenCount(id, strtoul(params, NULL, 10));
	} else {
		urlOpenMatch(id, params);
	}
}

static void commandCopy(size_t id, char *params) {
	urlCopyMatch(id, params);
}

static void commandExec(size_t id, char *params) {
	execID = id;

	pid_t pid = fork();
	if (pid < 0) err(EX_OSERR, "fork");
	if (pid) return;

	const char *shell = getenv("SHELL");
	if (!shell) shell = "/bin/sh";

	close(STDIN_FILENO);
	dup2(execPipe[1], STDOUT_FILENO);
	dup2(utilPipe[1], STDERR_FILENO);
	execlp(shell, shell, "-c", params, NULL);
	warn("%s", shell);
	_exit(EX_UNAVAILABLE);
}

static void commandHelp(size_t id, char *params) {
	(void)id;
	uiHide();

	pid_t pid = fork();
	if (pid < 0) err(EX_OSERR, "fork");
	if (pid) return;

	char buf[256];
	snprintf(buf, sizeof(buf), "ip%s$", (params ? params : "COMMANDS"));
	setenv("LESS", buf, 1);
	execlp("man", "man", "1", "catgirl", NULL);
	dup2(utilPipe[1], STDERR_FILENO);
	warn("man");
	_exit(EX_UNAVAILABLE);
}

static const struct Handler {
	const char *cmd;
	Command *fn;
	bool restricted;
} Commands[] = {
	{ "/close", .fn = commandClose },
	{ "/copy", .fn = commandCopy, .restricted = true },
	{ "/debug", .fn = commandDebug, .restricted = true },
	{ "/exec", .fn = commandExec, .restricted = true },
	{ "/help", .fn = commandHelp },
	{ "/join", .fn = commandJoin, .restricted = true },
	{ "/list", .fn = commandList },
	{ "/me", .fn = commandMe },
	{ "/move", .fn = commandMove },
	{ "/msg", .fn = commandMsg, .restricted = true },
	{ "/names", .fn = commandNames },
	{ "/nick", .fn = commandNick },
	{ "/notice", .fn = commandNotice },
	{ "/open", .fn = commandOpen, .restricted = true },
	{ "/part", .fn = commandPart },
	{ "/query", .fn = commandQuery, .restricted = true },
	{ "/quit", .fn = commandQuit },
	{ "/quote", .fn = commandQuote, .restricted = true },
	{ "/topic", .fn = commandTopic },
	{ "/whois", .fn = commandWhois },
	{ "/window", .fn = commandWindow },
};

static int compar(const void *cmd, const void *_handler) {
	const struct Handler *handler = _handler;
	return strcmp(cmd, handler->cmd);
}

const char *commandIsPrivmsg(size_t id, const char *input) {
	if (id == Network || id == Debug) return NULL;
	if (input[0] != '/') return input;
	const char *space = strchr(&input[1], ' ');
	const char *slash = strchr(&input[1], '/');
	if (slash && (!space || slash < space)) return input;
	return NULL;
}

const char *commandIsNotice(size_t id, const char *input) {
	if (id == Network || id == Debug) return NULL;
	if (strncmp(input, "/notice ", 8)) return NULL;
	return &input[8];
}

const char *commandIsAction(size_t id, const char *input) {
	if (id == Network || id == Debug) return NULL;
	if (strncmp(input, "/me ", 4)) return NULL;
	return &input[4];
}

void command(size_t id, char *input) {
	if (id == Debug && input[0] != '/') {
		commandQuote(id, input);
	} else if (commandIsPrivmsg(id, input)) {
		commandPrivmsg(id, input);
	} else if (input[0] == '/' && isdigit(input[1])) {
		commandWindow(id, &input[1]);
	} else {
		const char *cmd = strsep(&input, " ");
		const char *unique = complete(None, cmd);
		if (unique && !complete(None, cmd)) {
			cmd = unique;
			completeReject();
		}
		const struct Handler *handler = bsearch(
			cmd, Commands, ARRAY_LEN(Commands), sizeof(*handler), compar
		);
		if (self.restricted && handler && handler->restricted) {
			handler = NULL;
		}
		if (handler) {
			if (input) {
				input += strspn(input, " ");
				size_t len = strlen(input);
				while (input[len - 1] == ' ') input[--len] = '\0';
				if (!input[0]) input = NULL;
			}
			if (input && !input[0]) input = NULL;
			handler->fn(id, input);
		} else {
			uiFormat(id, Hot, NULL, "No such command %s", cmd);
		}
	}
}

void commandComplete(void) {
	for (size_t i = 0; i < ARRAY_LEN(Commands); ++i) {
		completeAdd(None, Commands[i].cmd, Default);
	}
}