feat: add color to the output using Chalk

threeal · Jul 17, 2023 · 4b98890 · 4b98890
1 parent aa304f2
commit 4b98890
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 12 deletions.
diff --git a/lib/google-rank.js b/lib/google-rank.js
diff --git a/lib/google-rank.js.map b/lib/google-rank.js.map
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -36,6 +36,7 @@
     "test": "dev test"
   },
   "dependencies": {
+    "chalk": "^4.1.2",
     "commander": "^11.0.0",
     "googlethis": "^1.7.1"
   },

diff --git a/src/google-rank.ts b/src/google-rank.ts
@@ -1,5 +1,6 @@
 #!/usr/bin/env node
 
+import chalk from "chalk";
 import { program } from "commander";
 import { getWebsiteRank, WebsiteRank } from "./utils";
 
@@ -22,13 +23,23 @@ async function run() {
     rankByKeywords.push([keyword, prom]);
   }
 
-  process.stdout.write(`Ranks for ${website} website:\n`);
+  process.stdout.write(`Ranks for ${chalk.blueBright(website)} website:\n`);
   for (const [keyword, prom] of rankByKeywords) {
     const rank = await prom;
     if (rank === undefined) {
       process.stdout.write(`page ?  rank ?`);
     } else {
-      process.stdout.write(`page ${rank.page + 1}  rank ${rank.rank + 1}`);
+      if (rank.page <= 0) {
+        process.stdout.write(`page ${chalk.greenBright(rank.page + 1)}`);
+      } else {
+        process.stdout.write(`page ${chalk.redBright(rank.page + 1)}`);
+      }
+
+      if (rank.page <= 0 && rank.rank <= 2) {
+        process.stdout.write(`  rank ${chalk.greenBright(rank.rank + 1)}`);
+      } else {
+        process.stdout.write(`  rank ${chalk.redBright(rank.rank + 1)}`);
+      }
     }
     process.stdout.write(`  ${keyword}\n`);
   }