summary refs log tree commit diff
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--litterbox.c407
1 files changed, 181 insertions, 226 deletions
diff --git a/litterbox.c b/litterbox.c
index c1938cd..1e49bf5 100644
--- a/litterbox.c
+++ b/litterbox.c
@@ -27,190 +27,6 @@
 
 #include "database.h"
 
-static struct {
-	sqlite3_stmt *motd;
-	sqlite3_stmt *context;
-	sqlite3_stmt *topic;
-	sqlite3_stmt *event;
-	sqlite3_stmt *events;
-} insert;
-
-static void prepare(void) {
-	const char *InsertMOTD = SQL(
-		INSERT OR IGNORE INTO motds (time, network, motd)
-		VALUES (coalesce(datetime(:time), datetime('now')), :network, :motd);
-	);
-	dbPersist(&insert.motd, InsertMOTD);
-
-	const char *InsertContext = SQL(
-		INSERT OR IGNORE INTO contexts (network, name, query)
-		VALUES (:network, :context, :query);
-	);
-	dbPersist(&insert.context, InsertContext);
-
-	const char *InsertTopic = SQL(
-		INSERT OR IGNORE INTO topics (time, context, topic)
-		SELECT coalesce(datetime(:time), datetime('now')), context, :topic
-		FROM contexts WHERE network = :network AND name = :context;
-	);
-	dbPersist(&insert.topic, InsertTopic);
-
-	const char *InsertEvent = SQL(
-		INSERT INTO events (time, type, context, name, target, message)
-		SELECT
-			coalesce(datetime(:time), datetime('now')),
-			:type, context, names.name, :target, :message
-		FROM contexts, names
-		WHERE contexts.network = :network
-			AND contexts.name = :context
-			AND names.nick = :nick
-			AND names.user = :user
-			AND names.host = :host;
-	);
-	dbPersist(&insert.event, InsertEvent);
-
-	const char *CreateJoins = SQL(
-		CREATE TEMPORARY TABLE joins (
-			nick TEXT NOT NULL,
-			channel TEXT NOT NULL,
-			UNIQUE (nick, channel)
-		);
-	);
-	dbExec(CreateJoins);
-
-	const char *InsertEvents = SQL(
-		INSERT INTO events (time, type, context, name, target, message)
-		SELECT
-			coalesce(datetime(:time), datetime('now')),
-			:type, context, names.name, :target, :message
-		FROM joins, contexts, names
-		WHERE joins.nick = :nick
-			AND contexts.name = joins.channel
-			AND contexts.network = :network
-			AND names.nick = :nick
-			AND names.user = :user
-			AND names.host = :host;
-	);
-	dbPersist(&insert.events, InsertEvents);
-}
-
-static void bindNetwork(const char *network) {
-	dbBindTextCopy(insert.motd, ":network", network);
-	dbBindTextCopy(insert.context, ":network", network);
-	dbBindTextCopy(insert.topic, ":network", network);
-	dbBindTextCopy(insert.event, ":network", network);
-	dbBindTextCopy(insert.events, ":network", network);
-}
-
-static void insertMOTD(const char *time, const char *motd) {
-	dbBindText(insert.motd, ":time", time);
-	dbBindText(insert.motd, ":motd", motd);
-	dbRun(insert.motd);
-}
-
-static void insertContext(const char *context, bool query) {
-	dbBindText(insert.context, ":context", context);
-	dbBindInt(insert.context, ":query", query);
-	dbRun(insert.context);
-}
-
-static void insertTopic(
-	const char *time, const char *context, const char *topic
-) {
-	dbBindText(insert.topic, ":time", time);
-	dbBindText(insert.topic, ":context", context);
-	dbBindText(insert.topic, ":topic", topic);
-	dbRun(insert.topic);
-}
-
-static void insertEvent(
-	const char *time, enum Type type, const char *context,
-	const char *nick, const char *user, const char *host,
-	const char *target, const char *message
-) {
-	dbBindText(insert.event, ":time", time);
-	dbBindInt(insert.event, ":type", type);
-	dbBindText(insert.event, ":context", context);
-	dbBindText(insert.event, ":nick", nick);
-	dbBindText(insert.event, ":user", user);
-	dbBindText(insert.event, ":host", host);
-	dbBindText(insert.event, ":target", target);
-	dbBindText(insert.event, ":message", message);
-	dbRun(insert.event);
-}
-
-static void insertEvents(
-	const char *time, enum Type type,
-	const char *nick, const char *user, const char *host,
-	const char *target, const char *message
-) {
-	dbBindText(insert.events, ":time", time);
-	dbBindInt(insert.events, ":type", type);
-	dbBindText(insert.events, ":nick", nick);
-	dbBindText(insert.events, ":user", user);
-	dbBindText(insert.events, ":host", host);
-	dbBindText(insert.events, ":target", target);
-	dbBindText(insert.events, ":message", message);
-	dbRun(insert.events);
-}
-
-static void insertName(const char *nick, const char *user, const char *host) {
-	static sqlite3_stmt *stmt;
-	const char *sql = SQL(
-		INSERT OR IGNORE INTO names (nick, user, host)
-		VALUES (:nick, :user, :host);
-	);
-	dbPersist(&stmt, sql);
-	dbBindText(stmt, ":nick", nick);
-	dbBindText(stmt, ":user", user);
-	dbBindText(stmt, ":host", host);
-	dbRun(stmt);
-}
-
-static void insertJoin(const char *nick, const char *channel) {
-	static sqlite3_stmt *stmt;
-	const char *sql = SQL(
-		INSERT OR IGNORE INTO joins (nick, channel) VALUES (:nick, :channel);
-	);
-	dbPersist(&stmt, sql);
-	dbBindText(stmt, ":nick", nick);
-	dbBindText(stmt, ":channel", channel);
-	dbRun(stmt);
-}
-
-static void deleteJoin(const char *nick, const char *channel) {
-	static sqlite3_stmt *stmt;
-	const char *sql = SQL(
-		DELETE FROM joins WHERE nick = :nick AND channel = :channel;
-	);
-	dbPersist(&stmt, sql);
-	dbBindText(stmt, ":nick", nick);
-	dbBindText(stmt, ":channel", channel);
-	dbRun(stmt);
-}
-
-static void updateJoin(const char *old, const char *new) {
-	static sqlite3_stmt *stmt;
-	const char *sql = SQL(
-		UPDATE joins SET nick = :new WHERE nick = :old;
-	);
-	dbPersist(&stmt, sql);
-	dbBindText(stmt, ":old", old);
-	dbBindText(stmt, ":new", new);
-	dbRun(stmt);
-}
-
-static void clearJoins(const char *nick, const char *channel) {
-	static sqlite3_stmt *stmt;
-	const char *sql = SQL(
-		DELETE FROM joins WHERE nick = :nick OR channel = :channel;
-	);
-	dbPersist(&stmt, sql);
-	dbBindText(stmt, ":nick", nick);
-	dbBindText(stmt, ":channel", channel);
-	dbRun(stmt);
-}
-
 static struct tls *client;
 
 static void clientWrite(const char *ptr, size_t len) {
@@ -285,6 +101,7 @@ static void require(const struct Message *msg, bool nick, size_t len) {
 static const char *join;
 
 static char *self;
+static char *network;
 static char *chanTypes;
 static char *prefixes;
 
@@ -313,7 +130,7 @@ static void handleReplyISupport(struct Message *msg) {
 		char *key = strsep(&msg->params[i], "=");
 		if (!msg->params[i]) continue;
 		if (!strcmp(key, "NETWORK")) {
-			bindNetwork(msg->params[i]);
+			set(&network, msg->params[i]);
 		} else if (!strcmp(key, "CHANTYPES")) {
 			set(&chanTypes, msg->params[i]);
 		} else if (!strcmp(key, "PREFIX")) {
@@ -353,12 +170,76 @@ static void handleReplyMOTD(struct Message *msg) {
 }
 
 static void handleReplyEndOfMOTD(struct Message *msg) {
-	motd.buf[motd.len - 1] = '\0';
-	insertMOTD(msg->time, motd.buf);
+	const char *sql = SQL(
+		INSERT OR IGNORE INTO motds (time, network, motd)
+		VALUES (coalesce(datetime(:time), datetime('now')), :network, :motd);
+	);
+	sqlite3_stmt *stmt = dbPrepare(sql);
+	dbBindText(stmt, ":time", msg->time);
+	dbBindText(stmt, ":network", network);
+	dbBindTextLen(stmt, ":motd", motd.buf, motd.len);
+	dbRun(stmt);
+	sqlite3_finalize(stmt);
 	free(motd.buf);
 	memset(&motd, 0, sizeof(motd));
 }
 
+static void insertContext(const char *context, bool query) {
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		INSERT OR IGNORE INTO contexts (network, name, query)
+		VALUES (:network, :context, :query);
+	);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":network", network);
+	dbBindText(stmt, ":context", context);
+	dbBindInt(stmt, ":query", query);
+	dbRun(stmt);
+}
+
+static void insertName(const struct Message *msg) {
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		INSERT OR IGNORE INTO names (nick, user, host)
+		VALUES (:nick, :user, :host);
+	);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":nick", msg->nick);
+	dbBindText(stmt, ":user", msg->user);
+	dbBindText(stmt, ":host", msg->host);
+	dbRun(stmt);
+}
+
+static void insertEvent(
+	const struct Message *msg, enum Type type, const char *context,
+	const char *target, const char *message
+) {
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		INSERT INTO events (time, type, context, name, target, message)
+		SELECT
+			coalesce(datetime(:time), datetime('now')),
+			:type, context, names.name, :target, :message
+		FROM contexts, names
+		WHERE contexts.network = :network
+			AND contexts.name = :context
+			AND names.nick = :nick
+			AND names.user = :user
+			AND names.host = :host;
+	);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":time", msg->time);
+	dbBindInt(stmt, ":type", type);
+	dbBindText(stmt, ":network", network);
+	dbBindText(stmt, ":context", context);
+	dbBindText(stmt, ":nick", msg->nick);
+	dbBindText(stmt, ":user", msg->user);
+	dbBindText(stmt, ":host", msg->host);
+	dbBindText(stmt, ":target", target);
+	dbBindText(stmt, ":message", message);
+	dbRun(stmt);
+}
+
 static void handlePrivmsg(struct Message *msg) {
 	require(msg, true, 2);
 
@@ -376,11 +257,25 @@ static void handlePrivmsg(struct Message *msg) {
 	}
 
 	insertContext(context, query);
-	insertName(msg->nick, msg->user, msg->host);
-	insertEvent(
-		msg->time, type, context,
-		msg->nick, msg->user, msg->host, NULL, message
+	insertName(msg);
+	insertEvent(msg, type, context, NULL, message);
+}
+
+static void insertTopic(
+	const char *time, const char *context, const char *topic
+) {
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		INSERT OR IGNORE INTO topics (time, context, topic)
+		SELECT coalesce(datetime(:time), datetime('now')), context, :topic
+		FROM contexts WHERE network = :network AND name = :context;
 	);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":time", time);
+	dbBindText(stmt, ":network", network);
+	dbBindText(stmt, ":context", context);
+	dbBindText(stmt, ":topic", topic);
+	dbRun(stmt);
 }
 
 static void handleReplyTopic(struct Message *msg) {
@@ -389,6 +284,50 @@ static void handleReplyTopic(struct Message *msg) {
 	insertTopic(msg->time, msg->params[1], msg->params[2]);
 }
 
+static void createJoins(void) {
+	const char *sql = SQL(
+		CREATE TEMPORARY TABLE joins (
+			nick TEXT NOT NULL,
+			channel TEXT NOT NULL,
+			UNIQUE (nick, channel)
+		);
+	);
+	dbExec(sql);
+}
+
+static void insertJoin(const char *nick, const char *channel) {
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		INSERT OR IGNORE INTO joins (nick, channel) VALUES (:nick, :channel);
+	);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":nick", nick);
+	dbBindText(stmt, ":channel", channel);
+	dbRun(stmt);
+}
+
+static void deleteJoin(const char *nick, const char *channel) {
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		DELETE FROM joins WHERE nick = :nick AND channel = :channel;
+	);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":nick", nick);
+	dbBindText(stmt, ":channel", channel);
+	dbRun(stmt);
+}
+
+static void clearJoins(const char *nick, const char *channel) {
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		DELETE FROM joins WHERE nick = :nick OR channel = :channel;
+	);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":nick", nick);
+	dbBindText(stmt, ":channel", channel);
+	dbRun(stmt);
+}
+
 static void handleReplyNames(struct Message *msg) {
 	require(msg, false, 3);
 	char *names = msg->params[3];
@@ -402,22 +341,16 @@ static void handleReplyNames(struct Message *msg) {
 static void handleJoin(struct Message *msg) {
 	require(msg, true, 1);
 	insertContext(msg->params[0], false);
-	insertName(msg->nick, msg->user, msg->host);
-	insertEvent(
-		msg->time, Join, msg->params[0],
-		msg->nick, msg->user, msg->host, NULL, NULL
-	);
+	insertName(msg);
+	insertEvent(msg, Join, msg->params[0], NULL, NULL);
 	insertJoin(msg->nick, msg->params[0]);
 }
 
 static void handlePart(struct Message *msg) {
 	require(msg, true, 1);
 	insertContext(msg->params[0], false);
-	insertName(msg->nick, msg->user, msg->host);
-	insertEvent(
-		msg->time, Part, msg->params[0],
-		msg->nick, msg->user, msg->host, NULL, msg->params[1]
-	);
+	insertName(msg);
+	insertEvent(msg, Part, msg->params[0], NULL, msg->params[1]);
 	if (!strcmp(msg->nick, self)) {
 		clearJoins(NULL, msg->params[0]);
 	} else {
@@ -428,12 +361,8 @@ static void handlePart(struct Message *msg) {
 static void handleKick(struct Message *msg) {
 	require(msg, true, 2);
 	insertContext(msg->params[0], false);
-	insertName(msg->nick, msg->user, msg->host);
-	insertEvent(
-		msg->time, Kick, msg->params[0],
-		msg->nick, msg->user, msg->host,
-		msg->params[1], msg->params[2]
-	);
+	insertName(msg);
+	insertEvent(msg, Kick, msg->params[0], msg->params[1], msg->params[2]);
 	if (!strcmp(msg->params[1], self)) {
 		clearJoins(NULL, msg->params[0]);
 	} else {
@@ -441,36 +370,64 @@ static void handleKick(struct Message *msg) {
 	}
 }
 
+static void insertEvents(
+	const struct Message *msg, enum Type type,
+	const char *target, const char *message
+) {
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		INSERT INTO events (time, type, context, name, target, message)
+		SELECT
+			coalesce(datetime(:time), datetime('now')),
+			:type, context, names.name, :target, :message
+		FROM joins, contexts, names
+		WHERE joins.nick = :nick
+			AND contexts.name = joins.channel
+			AND contexts.network = :network
+			AND names.nick = :nick
+			AND names.user = :user
+			AND names.host = :host;
+	);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":time", msg->time);
+	dbBindInt(stmt, ":type", type);
+	dbBindText(stmt, ":network", network);
+	dbBindText(stmt, ":nick", msg->nick);
+	dbBindText(stmt, ":user", msg->user);
+	dbBindText(stmt, ":host", msg->host);
+	dbBindText(stmt, ":target", target);
+	dbBindText(stmt, ":message", message);
+	dbRun(stmt);
+}
+
 static void handleNick(struct Message *msg) {
 	require(msg, true, 1);
 	if (!strcmp(msg->nick, self)) set(&self, msg->params[0]);
-	insertName(msg->nick, msg->user, msg->host);
-	insertEvents(
-		msg->time, Nick,
-		msg->nick, msg->user, msg->host, msg->params[0], NULL
+	insertName(msg);
+	insertEvents(msg, Nick, msg->params[0], NULL);
+	static sqlite3_stmt *stmt;
+	const char *sql = SQL(
+		UPDATE joins SET nick = :new WHERE nick = :old;
 	);
-	updateJoin(msg->nick, msg->params[0]);
+	dbPersist(&stmt, sql);
+	dbBindText(stmt, ":old", msg->nick);
+	dbBindText(stmt, ":new", msg->params[0]);
+	dbRun(stmt);
 }
 
 static void handleQuit(struct Message *msg) {
 	require(msg, true, 0);
-	insertName(msg->nick, msg->user, msg->host);
-	insertEvents(
-		msg->time, Quit,
-		msg->nick, msg->user, msg->host, NULL, msg->params[0]
-	);
+	insertName(msg);
+	insertEvents(msg, Quit, NULL, msg->params[0]);
 	clearJoins(msg->nick, NULL);
 }
 
 static void handleTopic(struct Message *msg) {
 	require(msg, true, 1);
 	insertContext(msg->params[0], false);
-	insertTopic(msg->params[0], msg->params[1]);
-	insertName(msg->nick, msg->user, msg->host);
-	insertEvent(
-		msg->time, Topic, msg->params[0],
-		msg->nick, msg->user, msg->host, NULL, msg->params[1]
-	);
+	insertTopic(msg->time, msg->params[0], msg->params[1]);
+	insertName(msg);
+	insertEvent(msg, Topic, msg->params[0], NULL, msg->params[1]);
 }
 
 static void handlePing(struct Message *msg) {
@@ -573,15 +530,13 @@ int main(int argc, char *argv[]) {
 	if (dbVersion() != DatabaseVersion) {
 		errx(EX_CONFIG, "database out of date; migrate with -m");
 	}
+	createJoins();
 
 	if (!host) errx(EX_USAGE, "host required");
 	set(&self, "*");
 	set(&chanTypes, "#&");
 	set(&prefixes, "@+");
 
-	prepare();
-	bindNetwork(host);
-
 	client = tls_client();
 	if (!client) errx(EX_SOFTWARE, "tls_client");