From f5d1a606b38fcadfcbe38c2caa30eb37f5a2ebe3 Mon Sep 17 00:00:00 2001
From: tsosunchia <59512455+tsosunchia@users.noreply.github.com>
Date: Thu, 30 Nov 2023 17:11:14 +0800
Subject: [PATCH] add param: support disable color output

---
 cmd/cmd.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/cmd/cmd.go b/cmd/cmd.go
index f78c951..4c1ec87 100644
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -3,6 +3,7 @@ package cmd
 import (
 	"encoding/json"
 	"fmt"
+	"github.com/fatih/color"
 	"log"
 	"net"
 	"os"
@@ -70,6 +71,7 @@ func Excute() {
 	lang := parser.Selector("g", "language", []string{"en", "cn"}, &argparse.Options{Default: "cn",
 		Help: "Choose the language for displaying [en, cn]"})
 	file := parser.String("", "file", &argparse.Options{Help: "Read IP Address or domain name from file"})
+	nocolor := parser.Flag("C", "nocolor", &argparse.Options{Help: "Disable Colorful Output"})
 
 	err := parser.Parse(os.Args)
 	if err != nil {
@@ -86,6 +88,12 @@ func Excute() {
 		os.Exit(0)
 	}
 
+	if *nocolor {
+		color.NoColor = true
+	} else {
+		color.NoColor = false
+	}
+
 	domain := *str
 
 	if *port == 0 {