From e3f5b494cd352fb477339edc918f8877e55e78a4 Mon Sep 17 00:00:00 2001
From: Jesse Wierzbinski <contact@cpluspatch.com>
Date: Sat, 2 Dec 2023 18:55:42 -1000
Subject: [PATCH] feat: Better indexing CLI

---
 cli.ts | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/cli.ts b/cli.ts
index 1d619948..ba0b7a72 100644
--- a/cli.ts
+++ b/cli.ts
@@ -4,6 +4,7 @@ import { client } from "~database/datasource";
 import { createNewLocalUser } from "~database/entities/User";
 import Table from "cli-table";
 import { rebuildSearchIndexes, MeiliIndexType } from "@meilisearch";
+import { getConfig } from "@config";
 
 const args = process.argv;
 
@@ -110,6 +111,8 @@ if (args.length < 3) {
 
 const command = args[2];
 
+const config = getConfig();
+
 switch (command) {
 	case "help":
 		console.log(help);
@@ -521,6 +524,12 @@ switch (command) {
 		break;
 	}
 	case "index": {
+		if (!config.meilisearch.enabled) {
+			console.log(
+				`${chalk.red(`✗`)} Meilisearch is not enabled in the config`
+			);
+			process.exit(1);
+		}
 		switch (args[3]) {
 			case "rebuild": {
 				const statuses = args.includes("--statuses");
@@ -541,6 +550,8 @@ switch (command) {
 						)}`
 					);
 
+					const timeBefore = performance.now();
+
 					await rebuildSearchIndexes(
 						[MeiliIndexType.Statuses],
 						batchSize
@@ -548,7 +559,9 @@ switch (command) {
 
 					console.log(
 						`${chalk.green(`✓`)} ${chalk.bold(
-							`Meilisearch index for statuses rebuilt`
+							`Meilisearch index for statuses rebuilt in ${chalk.bgGreen(
+								(performance.now() - timeBefore).toFixed(2)
+							)}ms`
 						)}`
 					);
 				}
@@ -560,6 +573,8 @@ switch (command) {
 						)}`
 					);
 
+					const timeBefore = performance.now();
+
 					await rebuildSearchIndexes(
 						[MeiliIndexType.Accounts],
 						batchSize
@@ -567,7 +582,9 @@ switch (command) {
 
 					console.log(
 						`${chalk.green(`✓`)} ${chalk.bold(
-							`Meilisearch index for users rebuilt`
+							`Meilisearch index for users rebuilt in ${chalk.bgGreen(
+								(performance.now() - timeBefore).toFixed(2)
+							)}ms`
 						)}`
 					);
 				}