feat: add banner printing functionality to server startup

2026-05-24 20:13:18 +08:00
parent 8bbd6c9414
commit c9ad978da6
3 changed files with 86 additions and 0 deletions
@@ -0,0 +1,43 @@
+package bootstrap
+
+import (
+	"fmt"
+	"io"
+	"os"
+
+	"cs-agent/internal/pkg/config"
+)
+
+func printBanner() {
+	printBannerTo(os.Stdout, config.Current())
+}
+
+func printBannerTo(w io.Writer, cfg config.Config) {
+	_, _ = fmt.Fprint(w, renderBanner(cfg))
+}
+
+func renderBanner(cfg config.Config) string {
+	port := cfg.Server.Port
+	if port <= 0 {
+		port = 8080
+	}
+	dbType := cfg.DB.Type
+	if dbType == "" {
+		dbType = "unknown"
+	}
+
+	return fmt.Sprintf(`
+ ____  _          _ _        _    ___
+/ ___|| |__   ___| | |      / \  |_ _|
+\___ \| '_ \ / _ \ | |     / _ \  | |
+ ___) | | | |  __/ | |    / ___ \ | |
+|____/|_| |_|\___|_|_|   /_/   \_\___|
+
+:: Shell AI ::
+
+Port     : %d
+DB       : %s
+Address  : http://127.0.0.1:%d
+
+`, port, dbType, port)
+}