From ab6c05ad9275dd8be47cda804f751c4c19e889af Mon Sep 17 00:00:00 2001
From: Lawliet-Chan <1576710154@qq.com>
Date: Thu, 18 Apr 2024 11:43:59 +0800
Subject: [PATCH] pretty log init

---
 core/startup/init.go    |  3 +++
 utils/log/pretty_log.go | 23 ++++++++++++++++++-----
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/core/startup/init.go b/core/startup/init.go
index aac2c8f9..45a0f978 100644
--- a/core/startup/init.go
+++ b/core/startup/init.go
@@ -3,7 +3,9 @@ package startup
 import (
 	"github.com/sirupsen/logrus"
 	"github.com/yu-org/yu/config"
+	"github.com/yu-org/yu/utils/log"
 	"os"
+	"path/filepath"
 )
 
 func InitKernelConfigFromPath(cfgPath string) {
@@ -52,6 +54,7 @@ func initLog(cfg *config.KernelConf) {
 		}
 	}
 
+	log.InitPrettyLog(filepath.Dir(cfg.LogOutput))
 	logrus.SetOutput(logfile)
 	lvl, err := logrus.ParseLevel(cfg.LogLevel)
 	if err != nil {
diff --git a/utils/log/pretty_log.go b/utils/log/pretty_log.go
index 5d8a9a36..a75db1a0 100644
--- a/utils/log/pretty_log.go
+++ b/utils/log/pretty_log.go
@@ -12,13 +12,26 @@ var (
 	PlusConsole       *console.Console
 )
 
-func init() {
+func InitPrettyLog(output string) {
 	doubleLine := text.WithDoubleLine()
-	DoubleLineConsole = console.NewConsole(console.WithTextOption(doubleLine))
+	DoubleLineConsole = console.NewConsole(console.WithOutputDir(output), console.WithTextOption(doubleLine))
+	doLog(DoubleLineConsole)
+
 	star := text.WithStarStyle()
-	StarConsole = console.NewConsole(console.WithTextOption(star))
+	StarConsole = console.NewConsole(console.WithOutputDir(output), console.WithTextOption(star))
+	doLog(StarConsole)
+
 	plus := text.WithPlusStyle()
-	PlusConsole = console.NewConsole(console.WithTextOption(plus))
+	PlusConsole = console.NewConsole(console.WithOutputDir(output), console.WithTextOption(plus))
+	doLog(PlusConsole)
+
 	dot := text.WithDotStyle()
-	DotConsole = console.NewConsole(console.WithTextOption(dot))
+	DotConsole = console.NewConsole(console.WithOutputDir(output), console.WithTextOption(dot))
+	doLog(DotConsole)
+}
+
+func doLog(c *console.Console) {
+	c.DisableColor()
+	c.DisableBorder()
+	c.Close()
 }