提交

cool-tools/internal/utility/allyes/allyes.go

@@ -0,0 +1,22 @@
+package allyes
+
+import (
+	"github.com/gogf/gf/v2/os/gcmd"
+	"github.com/gogf/gf/v2/os/genv"
+)
+
+const (
+	EnvName = "GF_CLI_ALL_YES"
+)
+
+// Init initializes the package manually.
+func Init() {
+	if gcmd.GetOpt("y") != nil {
+		genv.MustSet(EnvName, "1")
+	}
+}
+
+// Check checks whether option allow all yes for command.
+func Check() bool {
+	return genv.Get(EnvName).String() == "1"
+}

cool-tools/internal/utility/mlog/mlog.go

@@ -0,0 +1,66 @@
+package mlog
+
+import (
+	"context"
+
+	"github.com/gogf/gf/v2/os/gcmd"
+	"github.com/gogf/gf/v2/os/genv"
+	"github.com/gogf/gf/v2/os/glog"
+)
+
+const (
+	headerPrintEnvName = "GF_CLI_MLOG_HEADER"
+)
+
+var (
+	ctx    = context.TODO()
+	logger = glog.New()
+)
+
+func init() {
+	logger.SetStack(false)
+	if genv.Get(headerPrintEnvName).String() == "1" {
+		logger.SetHeaderPrint(true)
+	} else {
+		logger.SetHeaderPrint(false)
+	}
+	if gcmd.GetOpt("debug") != nil || gcmd.GetOpt("gf.debug") != nil {
+		logger.SetDebug(true)
+	} else {
+		logger.SetDebug(false)
+	}
+}
+
+// SetHeaderPrint enables/disables header printing to stdout.
+func SetHeaderPrint(enabled bool) {
+	logger.SetHeaderPrint(enabled)
+	if enabled {
+		_ = genv.Set(headerPrintEnvName, "1")
+	} else {
+		_ = genv.Set(headerPrintEnvName, "0")
+	}
+}
+
+func Print(v ...interface{}) {
+	logger.Print(ctx, v...)
+}
+
+func Printf(format string, v ...interface{}) {
+	logger.Printf(ctx, format, v...)
+}
+
+func Fatal(v ...interface{}) {
+	logger.Fatal(ctx, v...)
+}
+
+func Fatalf(format string, v ...interface{}) {
+	logger.Fatalf(ctx, format, v...)
+}
+
+func Debug(v ...interface{}) {
+	logger.Debug(ctx, v...)
+}
+
+func Debugf(format string, v ...interface{}) {
+	logger.Debugf(ctx, format, v...)
+}