add feature CLICOLOR

lixiaoqity · Oct 9, 2020 · 22e6937 · 22e6937
1 parent cac3f27
commit 22e6937
Show file tree

Hide file tree

Showing 6 changed files with 54 additions and 12 deletions.
diff --git a/config/keys.env b/config/keys.env
@@ -0,0 +1 @@
+CLICOLOR=1
diff --git a/index.js b/index.js
@@ -1,9 +1,10 @@
 #!/usr/bin/env node
 
-const process = require("process");
 const fs = require('fs');
 const pjson = require('./package.json');
 const testLink = require("./src/test-link");
+const testLinkColor = require('./src/test-link-color');
+require('dotenv').config({path:"./config/keys.env"});
 
 const regex = /(https?)(:\/\/)([-A-Za-z0-9+&@#/%?=~_|!:,.;]+[-A-Za-z0-9+&@#/%=~_|]+)/gi;
 
@@ -40,14 +41,22 @@ else {
                     if (urls) {
                         for (let u of urls) {
                             try {
-
-                                if (flag && !u.startsWith("https")) {
-                                    testLink(u.replace(/^http/, "https"));
+                                if(process.env.CLICOLOR==1){
+                                    if (flag && !u.startsWith("https")) {
+                                        testLinkColor(u.replace(/^http/, "https"));
+                                    }
+                                    else {
+                                        testLinkColor(u);
+                                    }
                                 }
-                                else {
-                                    testLink(u);
+                                if(process.env.CLICOLOR==0){
+                                    if (flag && !u.startsWith("https")) {
+                                        testLink(u.replace(/^http/, "https"));
+                                    }
+                                    else {
+                                        testLink(u);
+                                    }
                                 }
-
                             }
                             catch (error) {
                                 console.log(error);

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -13,6 +13,7 @@
   },
   "dependencies": {
     "chalk": "^4.1.0",
+    "dotenv": "^8.2.0",
     "node-fetch": "^2.6.1",
     "request": "^2.88.2"
   },

diff --git a/src/test-link-color.js b/src/test-link-color.js
@@ -0,0 +1,26 @@
+const chalk = require('chalk');
+const request = require('request');
+
+const testLinkColor = (u)=> {
+    request({url: u, method: 'HEAD'},(error,response)=>{
+        if(!error){
+            if (response.statusCode == 200) {
+                console.log(chalk.green(u + ' is good.'));
+            }
+            else if (response.statusCode == 400 || response.statusCode == 404) {
+                console.log(chalk.red(u + ' is bad.'));  
+            }
+            else if (response.statusCode == 301 || response.statusCode == 307 || response.statusCode == 308){
+                console.log(chalk.yellow(u + ' is redirect.')); 
+            }
+            else {
+                console.log(chalk.gray(u + ' is unknow.'));
+            }
+        }
+        else{
+            console.log(chalk.gray(u + ' is unknow.'));
+        }
+    })
+}
+
+module.exports = testLinkColor;
diff --git a/src/test-link.js b/src/test-link.js
@@ -5,20 +5,20 @@ const testLink = (u)=> {
     request({url: u, method: 'HEAD'},(error,response)=>{
         if(!error){
             if (response.statusCode == 200) {
-                console.log(chalk.green(u + ' is good.'));
+                console.log(u + ' is good.');
             }
             else if (response.statusCode == 400 || response.statusCode == 404) {
-                console.log(chalk.red(u + ' is bad.'));  
+                console.log(u + ' is bad.');  
             }
             else if (response.statusCode == 301 || response.statusCode == 307 || response.statusCode == 308){
-                console.log(chalk.yellow(u + ' is redirect.')); 
+                console.log(u + ' is redirect.'); 
             }
             else {
-                console.log(chalk.gray(u + ' is unknow.'));
+                console.log(u + ' is unknow.');
             }
         }
         else{
-            console.log(chalk.gray(u + ' is unknow.'));
+            console.log(u + ' is unknow.');
         }
     })
 }