From d6b496d3c0eb99cf966e0f93ed00434797e5216c Mon Sep 17 00:00:00 2001
From: hingbong <hingbonglo@gmail.com>
Date: Wed, 4 Dec 2024 00:54:01 +0000
Subject: [PATCH] chore: allow upgrade ui in embed mode (#1692)

---
 hub/route/server.go  | 4 ++--
 hub/route/upgrade.go | 6 ++++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/hub/route/server.go b/hub/route/server.go
index 21102cc3..3c3e5ca4 100644
--- a/hub/route/server.go
+++ b/hub/route/server.go
@@ -127,10 +127,10 @@ func router(isDebug bool, secret string, dohServer string, cors Cors) *chi.Mux {
 		r.Mount("/providers/rules", ruleProviderRouter())
 		r.Mount("/cache", cacheRouter())
 		r.Mount("/dns", dnsRouter())
-		if !embedMode { // disallow restart and upgrade in embed mode
+		if !embedMode { // disallow restart in embed mode
 			r.Mount("/restart", restartRouter())
-			r.Mount("/upgrade", upgradeRouter())
 		}
+		r.Mount("/upgrade", upgradeRouter())
 		addExternalRouters(r)
 
 	})
diff --git a/hub/route/upgrade.go b/hub/route/upgrade.go
index 2fed3f67..beee77a2 100644
--- a/hub/route/upgrade.go
+++ b/hub/route/upgrade.go
@@ -14,9 +14,11 @@ import (
 
 func upgradeRouter() http.Handler {
 	r := chi.NewRouter()
-	r.Post("/", upgradeCore)
 	r.Post("/ui", updateUI)
-	r.Post("/geo", updateGeoDatabases)
+	if !embedMode { // disallow upgrade core/geo in embed mode
+		r.Post("/", upgradeCore)
+		r.Post("/geo", updateGeoDatabases)
+	}
 	return r
 }