about summary refs log tree commit diff
diff options
context:
space:
mode:
authorJune McEnroe <june@causal.agency>2019-12-30 23:15:42 -0500
committerJune McEnroe <june@causal.agency>2019-12-30 23:15:42 -0500
commit942a8466dca6b493228195eba4aa992bf8732e30 (patch)
tree1d8c8e99bbf52c341fcf7037be23e57042648ada
parentJoin with USING wherever possible (diff)
downloadlitterbox-942a8466dca6b493228195eba4aa992bf8732e30.tar.gz
litterbox-942a8466dca6b493228195eba4aa992bf8732e30.zip
Add initial rough version of scoop
-rw-r--r--.gitignore1
-rw-r--r--Makefile2
-rw-r--r--scoop.c208
3 files changed, 210 insertions, 1 deletions
diff --git a/.gitignore b/.gitignore
index c4295bb..272bd1a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,4 +1,5 @@
 *.o
 litterbox
+scoop
 tags
 unscoop
diff --git a/Makefile b/Makefile
index 73df59d..a7ba555 100644
--- a/Makefile
+++ b/Makefile
@@ -6,7 +6,7 @@ LDFLAGS += ${LIBS_PREFIX:%=-L%/lib}
 LDLIBS = -lsqlite3
 LDLIBS_litterbox = -ltls
 
-BINS = litterbox unscoop
+BINS = litterbox scoop unscoop
 
 -include config.mk
 
diff --git a/scoop.c b/scoop.c
new file mode 100644
index 0000000..48f291c
--- /dev/null
+++ b/scoop.c
@@ -0,0 +1,208 @@
+/* Copyright (C) 2019  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 <err.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sysexits.h>
+#include <unistd.h>
+
+#include "database.h"
+
+static const char *Inner = SQL(
+	SELECT
+		contexts.network,
+		contexts.name AS context,
+		date(events.time) || 'T' || time(events.time) || 'Z' AS time,
+		events.type,
+		names.nick,
+		CASE WHEN names.user = '*'
+			THEN names.nick
+			ELSE names.user
+		END,
+		events.target,
+		highlight(search, 6, :open, :close),
+		events.event
+	FROM events
+	JOIN contexts USING (context)
+	JOIN names USING (name)
+	JOIN search ON search.rowid = events.event
+	WHERE coalesce(contexts.network = :network, true)
+		AND coalesce(contexts.name = :context, true)
+		AND coalesce(contexts.query = :query, true)
+		AND coalesce(date(events.time) = :date, true)
+		AND coalesce(events.type = :type, true)
+		AND coalesce(names.nick = :nick, true)
+		AND coalesce(names.user = :user, true)
+		AND coalesce(names.host = :host, true)
+		AND coalesce(events.target = :target, true)
+);
+
+static const char *Search = SQL(search MATCH :search);
+
+static const char *Limit = SQL(
+	ORDER BY time DESC, event DESC
+	LIMIT :limit
+);
+
+static const char *Outer = SQL(
+	SELECT * FROM results
+	ORDER BY time, event
+);
+
+static const char *TypeNames[] = {
+#define X(id, name) [id] = name,
+	ENUM_TYPE
+#undef X
+};
+
+static const enum Type *parseType(const char *input) {
+	static enum Type type;
+	for (type = 0; type < ARRAY_LEN(TypeNames); ++type) {
+		if (!strcmp(input, TypeNames[type])) return &type;
+	}
+	errx(EX_USAGE, "no such type %s", input);
+}
+
+int main(int argc, char *argv[]) {
+	char *path = NULL;
+	bool shell = false;
+
+	bool public = false;
+	bool query = false;
+	const char *network = NULL;
+	const char *context = NULL;
+	const char *date = NULL;
+	const enum Type *type = NULL;
+	const char *nick = NULL;
+	const char *user = NULL;
+	const char *host = NULL;
+	const char *target = NULL;
+	const char *search = NULL;
+	int limit = -1;
+
+	int opt;
+	while (0 < (opt = getopt(argc, argv, "D:N:T:c:d:h:l:n:pqst:u:v"))) {
+		switch (opt) {
+			break; case 'D': date = optarg;
+			break; case 'N': network = optarg;
+			break; case 'T': target = optarg;
+			break; case 'c': context = optarg;
+			break; case 'd': path = optarg;
+			break; case 'h': host = optarg;
+			break; case 'l': limit = strtol(optarg, NULL, 0);
+			break; case 'n': nick = optarg;
+			break; case 'p': public = true;
+			break; case 'q': query = true;
+			break; case 's': shell = true;
+			break; case 't': type = parseType(optarg);
+			break; case 'u': user = optarg;
+			break; case 'v': verbose = true;
+			break; default:  return EX_USAGE;
+		}
+	}
+	if (optind < argc) search = argv[optind];
+
+	dbFind(path, SQLITE_OPEN_READONLY);
+	if (dbVersion() != DatabaseVersion) {
+		errx(EX_CONFIG, "database out of date; migrate with litterbox -m");
+	}
+
+	if (shell) {
+		path = strdup(sqlite3_db_filename(db, "main"));
+		if (!path) err(EX_OSERR, "strdup");
+		dbClose();
+		execlp("sqlite3", "sqlite3", path, NULL);
+		err(EX_UNAVAILABLE, "sqlite3");
+	}
+
+	// TODO: Set up pipe to $PAGER.
+
+	char sql[4096];
+	if (search) {
+		snprintf(
+			sql, sizeof(sql),
+			"WITH results AS (%s AND %s %s) %s;",
+			Inner, Search, Limit, Outer
+		);
+	} else {
+		snprintf(
+			sql, sizeof(sql),
+			"WITH results AS (%s %s) %s;",
+			Inner, Limit, Outer
+		);
+	}
+
+	sqlite3_stmt *stmt = dbPrepare(sql);
+	dbBindText(stmt, ":network", network);
+	dbBindText(stmt, ":context", context);
+	if (public) dbBindInt(stmt, ":query", false);
+	if (query) dbBindInt(stmt, ":query", true);
+	dbBindText(stmt, ":date", date);
+	if (type) dbBindInt(stmt, ":type", *type);
+	dbBindText(stmt, ":nick", nick);
+	dbBindText(stmt, ":user", user);
+	dbBindText(stmt, ":host", host);
+	dbBindText(stmt, ":target", target);
+	if (search) dbBindText(stmt, ":search", search);
+	dbBindInt(stmt, ":limit", limit);
+
+	// FIXME: Conditional on terminal.
+	dbBindText(stmt, ":open", "\33[33m");
+	dbBindText(stmt, ":close", "\33[m");
+
+	if (verbose) {
+		char *expand = sqlite3_expanded_sql(stmt);
+		if (!expand) errx(EX_SOFTWARE, "sqlite3_expanded_sql");
+		fprintf(stderr, "%s\n", expand);
+		sqlite3_free(expand);
+	}
+
+	int result;
+	while (SQLITE_ROW == (result = sqlite3_step(stmt))) {
+		int i = 0;
+		const char *network = (const char *)sqlite3_column_text(stmt, i++);
+		const char *context = (const char *)sqlite3_column_text(stmt, i++);
+		const char *time = (const char *)sqlite3_column_text(stmt, i++);
+		enum Type type = sqlite3_column_int(stmt, i++);
+		const char *nick = (const char *)sqlite3_column_text(stmt, i++);
+		const char *user = (const char *)sqlite3_column_text(stmt, i++);
+		const char *target = (const char *)sqlite3_column_text(stmt, i++);
+		const char *message = (const char *)sqlite3_column_text(stmt, i++);
+		if (!target) target = "";
+		if (!message) message = "";
+
+		// TODO: Nick coloring.
+		printf("%s/%s: [%s] ", network, context, time);
+		switch (type) {
+			break; case Privmsg: printf("<%s> %s\n", nick, message);
+			break; case Notice: printf("-%s- %s\n", nick, message);
+			break; case Action: printf("* %s %s\n", nick, message);
+			break; case Join: printf("%s joined\n", nick);
+			break; case Part: printf("%s parted: %s\n", nick, message);
+			break; case Quit: printf("%s quit: %s\n", nick, message);
+			break; case Kick: printf("%s kicked %s: %s\n", nick, target, message);
+			break; case Nick: printf("%s changed nick to %s\n", nick, target);
+			break; case Topic: printf("%s set the topic: %s\n", nick, message);
+		}
+	}
+	if (result != SQLITE_DONE) warnx("%s", sqlite3_errmsg(db));
+
+	sqlite3_finalize(stmt);
+	dbClose();
+}