From b37db99618c3cc7adb0eef0a6bd14008bd11ab83 Mon Sep 17 00:00:00 2001
From: Sven Slootweg <admin@cryto.net>
Date: Sun, 21 Mar 2021 01:18:08 +0100
Subject: [PATCH] Colorize logs

---
 src/kernel.js                | 3 ++-
 src/log-status.js            | 7 +++++++
 src/mutation-api/database.js | 4 +++-
 src/task-stream.js           | 6 ++++--
 4 files changed, 16 insertions(+), 4 deletions(-)
 create mode 100644 src/log-status.js

diff --git a/src/kernel.js b/src/kernel.js
index 302ab54..2ffaa13 100644
--- a/src/kernel.js
+++ b/src/kernel.js
@@ -12,6 +12,7 @@ const pipe = require("@promistream/pipe");
 const parallelize = require("@ppstreams/parallelize");
 
 const initialize = require("./initialize");
+const logStatus = require("./log-status");
 
 // TODO: Publish this as a separate package
 // Inverts an object of arrays, eg. {a: [x, y], b: [x, z]} becomes {x: [a, b], y: [a], z: [b]}
@@ -107,7 +108,7 @@ module.exports = function createKernel(configuration) {
 					return pipe([
 						taskStream,
 						simpleSink((completedItem) => {
-							console.log(`[completed] ${completedItem.id}`);
+							logStatus(task, chalk.bold.green, "completed", completedItem.id);
 						})
 					]).read();
 				} else {
diff --git a/src/log-status.js b/src/log-status.js
new file mode 100644
index 0000000..52f1c6e
--- /dev/null
+++ b/src/log-status.js
@@ -0,0 +1,7 @@
+"use strict";
+
+const chalk = require("chalk");
+
+module.exports = function logStatus(task, color, type, message) {
+	console.log(`${chalk.bold(`[${task}]`)} ${color(`[${type}]`)} ${message}`);
+};
diff --git a/src/mutation-api/database.js b/src/mutation-api/database.js
index a7d9da0..d7de7d5 100644
--- a/src/mutation-api/database.js
+++ b/src/mutation-api/database.js
@@ -1,6 +1,8 @@
 "use strict";
 
 const defaultValue = require("default-value");
+const chalk = require("chalk");
+const logStatus = require("../log-status");
 
 module.exports = function (state) {
 	const queries = require("../queries")(state);
@@ -8,7 +10,7 @@ module.exports = function (state) {
 	return function createDatabaseMutationAPI({ tx, item, taskVersion, task }) {
 		return {
 			createItem: function (options) {
-				console.log(`[${task}][new] ${options.id}`);
+				logStatus(task, chalk.gray, "new", options.id);
 
 				return queries.createItem(tx, {
 					...options,
diff --git a/src/task-stream.js b/src/task-stream.js
index 2e65efa..53f4961 100644
--- a/src/task-stream.js
+++ b/src/task-stream.js
@@ -11,6 +11,8 @@ const buffer = require("@promistream/buffer");
 const pipe = require("@promistream/pipe");
 
 const createMutationAPIWrapper = require("./mutation-api/wrapper");
+const logStatus = require("./log-status");
+const chalk = require("chalk");
 
 // FIXME: Revert inlining of task_states once switched to PostgreSQL 12+, which can do this automatically using NOT MATERIALIZED
 let query = `
@@ -128,7 +130,7 @@ module.exports = function (state) {
 			buffer(),
 			globalRateLimiter,
 			processTaskSafely(task, (item, tx) => {
-				console.log(`[${task}][started] ${item.id}`);
+				logStatus(task, chalk.bold.cyan, "started", item.id);
 
 				let context = { tx, item, task, taskVersion };
 
@@ -171,7 +173,7 @@ module.exports = function (state) {
 						expires_at: dateFns.add(new Date(), { seconds: ttlInSeconds })
 					});
 				}).catch((error) => {
-					console.warn(`[${task}][failed] ${item.id}`, error);
+					logStatus(task, chalk.bold.red, "failed", item.id);
 
 					return Promise.try(() => {
 						// Task failed -- note, cannot use tx here because it has failed