shithub: ircd

ref: 7a1a90f46e6c7de5ce85bac73ecc8c4931900d7c
dir: /cmd.c/

#include <u.h>
#include <libc.h>
#include "dat.h"
#include "fns.h"
#include "cmd.h"
#include "version.h"

static void
cversion(Client *c, Request *r)
{
	/* (/lib/rfc/rfc2812:/^3.4.3) */
	if (r->args[0]) {
		fprint(2, "get version of '%s' (not implemented yet!)\n", r->args[0]);
		reply(c, Enosuchserver, r->args[0]);
		return;
	}
	reply(c, Rversion, getversion());
}

static void
cuser(Client *c, Request *r)
{
	/* (/lib/rfc/rfc2812:/^3.1.3) */
	User *u;
	if (!r->args[3]) {
		reply(c, Eneedmoreparams, r->cmd->name);
		return;
	}
	u = finduser(r->args[0]);
	if (u) {
		reply(c, Ealreadyregistered);
		return;
	}
	u = adduser(r->args[0]);
	u->realname = strdup(r->args[3]);
	c->user = u;
	
	if (c->nick && !u->greeted) {
		u->greeted = 1;
		reply(c, Rwelcome, welcome, c->nick, u->name, sysnameb);
	}
}

static void
cnick(Client *c, Request *r)
{
	/* (/lib/rfc/rfc2812:/^3.1.2) */
	if (!r->args[0]) {
		reply(c, Enonicknamegiven);
		return;
	}
	
	if (findnick(r->args[0])) {
		reply(c, Enicknameinuse, r->args[0]);
		return;
	}
	
	c->nick = strdup(r->args[0]);
	
	if (c->user && !c->user->greeted) {
		c->user->greeted = 1;
		reply(c, Rwelcome, welcome, c->nick, c->user->name, sysnameb);
	}
}

static void
cwhois(Client *c, Request *r)
{
	/* (/lib/rfc/rfc2812:/^3.6.2) */
	// TODO: implement remaining replies: (/lib/rfc/rfc2812:/^3.6.2)
	int start = 0;
	char *server = nil;
	User *u;
	Client *cl;
	
	if (!r->args[0]) {
		reply(c, Enonicknamegiven);
		return;
	}
	if (r->args[1] && strcmp(r->args[1], ",")) {
		server = r->args[0];
		start = 2;
	}
	for (int i = start; i < 15; i += 2) {
		/* args[i] is nick, args[i+1] is ',' */
		if (!r->args[i])
			break;
		if (!server) {
			cl = findnick(r->args[i]);
			if (cl) {
				u = cl->user;
				if (!u)
					goto Next;
				reply(c, Rwhoisuser, cl->nick, u->name, sysnameb, u->realname);
			} else
				reply(c, Enosuchnick, r->args[i]);
		} else {
			// TODO: forward to server: send message to server
		}
Next:
		reply(c, Rendofwhois, r->args[i]);
	}
}

static void
cprivmsg(Client *c, Request *r)
{
	/* (/lib/rfc/rfc2812:/^3.3.1) */
	Client *tgt;
	if (!r->args[0]) {
		reply(c, Enorecipient, "PRIVMSG");
		return;
	}
	if (!r->args[1]) {
		reply(c, Enotexttosend);
		return;
	}
	tgt = findnick(r->args[0]);
	if (!tgt) {
		reply(c, Enosuchnick, r->args[0]);
		return;
	}
	if (tgt->away) {
		reply(c, Raway, r->args[0], tgt->away);
		return;
	}
	ircsend(tgt, c, Sprivmsg, r->args[0], r->args[1]);
}

static void
caway(Client *c, Request *r)
{
	/* (/lib/rfc/rfc2812:/^4.1) */
	if (r->args[0]) {
		if (c->away)
			free(c->away);
		c->away = strdup(r->args[0]);
		reply(c, Rnowaway);
		return;
	}
	if (c->away)
		free(c->away);
	c->away = nil;
	reply(c, Runaway);
}

static Command commands[] = {
	{ "whois", cwhois },
	{ "version", cversion },
	{ "user", cuser },
	{ "nick", cnick },
	{ "privmsg", cprivmsg },
	{ "away", caway },
};
int ncommands = sizeof(commands) / sizeof(Command);

Command*
findcommand(char *s)
{
	for (int i = 0; i < ncommands; i++) {
		if (cistrcmp(commands[i].name, s) == 0)
			return &commands[i];
	}
	return nil;
}

Command*
findcommandn(int n)
{
	assert(0);
	return nil;
}

void
execrequest(Client *c, Request r)
{
	if (!(r.cmd && r.cmd->func)) {
		fprint(2, "cannot execute request: no command\n");
		return;
	}
	if (debug)
		fprint(2, "run command '%s'\n", r.cmd->name);
	r.cmd->func(c, &r);
}