From 51235aae70a03ee3deeeaa3336bf80d1863c71e4 Mon Sep 17 00:00:00 2001
From: zijiren233 <pyh1670605849@gmail.com>
Date: Fri, 13 Oct 2023 19:31:18 +0800
Subject: [PATCH] Feat: add disable-update-check flag

---
 cmd/flags/config.go |  2 ++
 cmd/server.go       | 11 ++++++++---
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/cmd/flags/config.go b/cmd/flags/config.go
index c7f14e0..ad788c4 100644
--- a/cmd/flags/config.go
+++ b/cmd/flags/config.go
@@ -13,4 +13,6 @@ var (
 	SkipConfig bool
 
 	EnvNoPrefix bool
+
+	DisableUpdateCheck bool
 )
diff --git a/cmd/server.go b/cmd/server.go
index 0b9353c..a825926 100644
--- a/cmd/server.go
+++ b/cmd/server.go
@@ -9,6 +9,7 @@ import (
 	log "github.com/sirupsen/logrus"
 	"github.com/soheilhy/cmux"
 	"github.com/spf13/cobra"
+	"github.com/synctv-org/synctv/cmd/flags"
 	"github.com/synctv-org/synctv/internal/bootstrap"
 	"github.com/synctv-org/synctv/internal/conf"
 	"github.com/synctv-org/synctv/server"
@@ -19,12 +20,15 @@ var ServerCmd = &cobra.Command{
 	Short: "Start synctv-server",
 	Long:  `Start synctv-server`,
 	PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
-		return bootstrap.New(bootstrap.WithContext(cmd.Context())).Add(
+		boot := bootstrap.New(bootstrap.WithContext(cmd.Context())).Add(
 			bootstrap.InitSysNotify,
 			bootstrap.InitConfig,
 			bootstrap.InitLog,
-			bootstrap.InitCheckUpdate,
-		).Run()
+		)
+		if !flags.DisableUpdateCheck {
+			boot.Add(bootstrap.InitCheckUpdate)
+		}
+		return boot.Run()
 	},
 	PreRunE: func(cmd *cobra.Command, args []string) error { return InitGinMode() },
 	Run:     Server,
@@ -121,4 +125,5 @@ func Server(cmd *cobra.Command, args []string) {
 
 func init() {
 	RootCmd.AddCommand(ServerCmd)
+	ServerCmd.Flags().BoolVar(&flags.DisableUpdateCheck, "disable-update-check", false, "disable update check")
 }