chore: allow upgrade ui in embed mode (#1692)

hingbong · web-flow · commit d6b496d3c0eb · 2024-12-04T08:54:01.000+08:00
diff --git 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
@@ -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
 }
 
