Fix irregular output format

tdh8316 · Dec 30, 2019 · 9433e2f · 9433e2f
1 parent 9a7b40c
commit 9433e2f
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/investigo.go b/investigo.go
@@ -198,7 +198,7 @@ func initializeSiteData(forceUpdate bool) {
 				fmt.Fprintf(
 					color.Output,
 					"[%s] Cannot open database \"%s\"\n",
-					color.HiRedString("!"), color.HiBlueString(dataFileName),
+					color.HiRedString("!"), (dataFileName),
 				)
 			}
 		}
@@ -211,8 +211,8 @@ func initializeSiteData(forceUpdate bool) {
 		} else {
 			fmt.Fprintf(
 				color.Output,
-				"%s Update database: %s",
-				color.HiBlueString("[!]"),
+				"[%s] Update database: %s",
+				color.HiBlueString("!"),
 				color.HiYellowString("Downloading..."),
 			)
 		}
@@ -250,7 +250,11 @@ func initializeSiteData(forceUpdate bool) {
 		_updateFile.Close()
 		jsonFile, _ = os.Open(dataFileName)
 
-		fmt.Println(" [Done]")
+		if options.noColor {
+			fmt.Println(" [Done]")
+		} else {
+			fmt.Fprintf(color.Output, " [%s]\n", color.GreenString("Done"))
+		}
 	}
 
 	defer jsonFile.Close()