From b063f1cbcccd4c10d5dfad8f14ca9dcfeed9c641 Mon Sep 17 00:00:00 2001
From: JustSong <songquanpeng@foxmail.com>
Date: Mon, 17 Apr 2023 21:25:58 +0800
Subject: [PATCH] chore: update text prompt

---
 common/init.go | 26 +++++++++++++++++++++++---
 main.go        |  2 +-
 2 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/common/init.go b/common/init.go
index 8c2c9d82b..81c351bfb 100644
--- a/common/init.go
+++ b/common/init.go
@@ -6,14 +6,30 @@ import (
 	"log"
 	"os"
 	"path/filepath"
+	"strings"
 )
 
 var (
-	Port         = flag.Int("port", 3000, "the listening port")
-	PrintVersion = flag.Bool("version", false, "print version and exit")
-	LogDir       = flag.String("log-dir", "", "specify the log directory")
+	PrintVersion = flag.Bool("version", false, "Print the version of the program and exits.")
+	PrintHelp    = flag.Bool("help", false, "Print the help message and exits.")
+	Port         = flag.Int("port", 3000, "Specify the listening port. Default is 3000.")
+	LogDir       = flag.String("log-dir", "", "Specify the directory for log files.")
 )
 
+func printHelp() {
+	fmt.Println(fmt.Sprintf("Message Pusher %s - Your all in one message push system.", Version))
+	fmt.Println("Copyright (C) 2023 JustSong. All rights reserved.")
+	fmt.Println("GitHub: https://github.com/songquanpeng/message-pusher")
+	fmt.Println("Usage: message-pusher [options]")
+	fmt.Println("Options:")
+	flag.CommandLine.VisitAll(func(f *flag.Flag) {
+		name := fmt.Sprintf("-%s", f.Name)
+		usage := strings.Replace(f.Usage, "\n", "\n    ", -1)
+		fmt.Printf("        -%-14s%s\n", name, usage)
+	})
+	os.Exit(0)
+}
+
 func init() {
 	flag.Parse()
 
@@ -22,6 +38,10 @@ func init() {
 		os.Exit(0)
 	}
 
+	if *PrintHelp {
+		printHelp()
+	}
+
 	if os.Getenv("SESSION_SECRET") != "" {
 		SessionSecret = os.Getenv("SESSION_SECRET")
 	}
diff --git a/main.go b/main.go
index 81b617245..7907ac6af 100644
--- a/main.go
+++ b/main.go
@@ -24,7 +24,7 @@ var indexPage []byte
 
 func main() {
 	common.SetupGinLog()
-	common.SysLog("system started")
+	common.SysLog("Message Pusher " + common.Version + " started")
 	if os.Getenv("GIN_MODE") != "debug" {
 		gin.SetMode(gin.ReleaseMode)
 	}