mirror of
https://gitclone.com/github.com/MetaCubeX/Clash.Meta
synced 2024-11-14 05:11:17 +08:00
63 lines
1.4 KiB
Go
63 lines
1.4 KiB
Go
package route
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/metacubex/mihomo/component/updater"
|
|
"github.com/metacubex/mihomo/log"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/render"
|
|
)
|
|
|
|
func upgradeRouter() http.Handler {
|
|
r := chi.NewRouter()
|
|
r.Post("/", upgradeCore)
|
|
r.Post("/ui", updateUI)
|
|
r.Post("/geo", updateGeoDatabases)
|
|
return r
|
|
}
|
|
|
|
func upgradeCore(w http.ResponseWriter, r *http.Request) {
|
|
// modify from https://github.com/AdguardTeam/AdGuardHome/blob/595484e0b3fb4c457f9bb727a6b94faa78a66c5f/internal/home/controlupdate.go#L108
|
|
log.Infoln("start update")
|
|
execPath, err := os.Executable()
|
|
if err != nil {
|
|
render.Status(r, http.StatusInternalServerError)
|
|
render.JSON(w, r, newError(fmt.Sprintf("getting path: %s", err)))
|
|
return
|
|
}
|
|
|
|
err = updater.UpdateCore(execPath)
|
|
if err != nil {
|
|
log.Warnln("%s", err)
|
|
render.Status(r, http.StatusInternalServerError)
|
|
render.JSON(w, r, newError(fmt.Sprintf("%s", err)))
|
|
return
|
|
}
|
|
|
|
render.JSON(w, r, render.M{"status": "ok"})
|
|
if f, ok := w.(http.Flusher); ok {
|
|
f.Flush()
|
|
}
|
|
|
|
go restartExecutable(execPath)
|
|
}
|
|
|
|
func updateUI(w http.ResponseWriter, r *http.Request) {
|
|
err := updater.DefaultUiUpdater.DownloadUI()
|
|
if err != nil {
|
|
log.Warnln("%s", err)
|
|
render.Status(r, http.StatusInternalServerError)
|
|
render.JSON(w, r, newError(fmt.Sprintf("%s", err)))
|
|
return
|
|
}
|
|
|
|
render.JSON(w, r, render.M{"status": "ok"})
|
|
if f, ok := w.(http.Flusher); ok {
|
|
f.Flush()
|
|
}
|
|
}
|