summary refs log tree commit diff
path: root/litterbox.c
diff options
context:
space:
mode:
authorJune McEnroe <june@causal.agency>2019-12-25 15:46:41 -0500
committerJune McEnroe <june@causal.agency>2019-12-25 15:46:41 -0500
commited99ecc2fe913fa047bc6e87cc363b60010befab (patch)
tree984c5c5912d4b1dea0046c46dffd0529b2746533 /litterbox.c
parentHandle SIGINT and SIGTERM (diff)
downloadlitterbox-ed99ecc2fe913fa047bc6e87cc363b60010befab.tar.gz
litterbox-ed99ecc2fe913fa047bc6e87cc363b60010befab.zip
Insert existing topics into the database
Not sure how to handle the 333 reply that contains the user who set the
topic and the timestamp of when it was set, since they're two separate
messages that aren't really easily correlated since there's no guarantee
that you're even going to get a 333 at all.
Diffstat (limited to '')
-rw-r--r--litterbox.c26
1 files changed, 26 insertions, 0 deletions
diff --git a/litterbox.c b/litterbox.c
index 181227a..964a6ea 100644
--- a/litterbox.c
+++ b/litterbox.c
@@ -29,6 +29,7 @@
 
 static struct {
 	sqlite3_stmt *context;
+	sqlite3_stmt *topic;
 	sqlite3_stmt *event;
 	sqlite3_stmt *events;
 } insert;
@@ -49,6 +50,13 @@ static void prepare(void) {
 	);
 	dbPersist(&insert.context, InsertContext);
 
+	const char *InsertTopic = SQL(
+		INSERT INTO topics (context, time, topic)
+		SELECT context, coalesce(datetime(:time), datetime('now')), :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
@@ -81,6 +89,7 @@ static void prepare(void) {
 
 static void bindNetwork(const char *network) {
 	dbBindTextCopy(insert.context, ":network", network);
+	dbBindTextCopy(insert.topic, ":network", network);
 	dbBindTextCopy(insert.event, ":network", network);
 	dbBindTextCopy(insert.events, ":network", network);
 }
@@ -91,6 +100,15 @@ static void insertContext(const char *context, bool query) {
 	dbRun(insert.context);
 }
 
+static void insertTopic(
+	const char *context, const char *time, const char *topic
+) {
+	dbBindText(insert.topic, ":context", context);
+	dbBindText(insert.topic, ":time", time);
+	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,
@@ -316,6 +334,12 @@ static void handlePrivmsg(struct Message *msg) {
 	);
 }
 
+static void handleReplyTopic(struct Message *msg) {
+	require(msg, false, 2);
+	if (!strcmp(msg->cmd, "331")) msg->params[2] = NULL;
+	insertTopic(msg->params[1], msg->time, msg->params[2]);
+}
+
 static void handleReplyNames(struct Message *msg) {
 	require(msg, false, 3);
 	char *names = msg->params[3];
@@ -411,6 +435,8 @@ static const struct {
 } Handlers[] = {
 	{ "001", false, handleReplyWelcome },
 	{ "005", false, handleReplyISupport },
+	{ "331", true, handleReplyTopic },
+	{ "332", true, handleReplyTopic },
 	{ "353", true, handleReplyNames },
 	{ "CAP", false, handleCap },
 	{ "JOIN", true, handleJoin },