mihomo/hub/configs.go

85 lines
1.7 KiB
Go
Raw Normal View History

2018-06-20 22:41:02 +08:00
package hub
import (
2018-07-15 22:23:20 +08:00
"fmt"
2018-06-20 22:41:02 +08:00
"net/http"
"github.com/Dreamacro/clash/config"
2018-07-15 22:23:20 +08:00
C "github.com/Dreamacro/clash/constant"
"github.com/Dreamacro/clash/proxy"
2018-06-20 22:41:02 +08:00
"github.com/go-chi/chi"
"github.com/go-chi/render"
)
func configRouter() http.Handler {
r := chi.NewRouter()
2018-07-15 22:23:20 +08:00
r.Get("/", getConfigs)
r.Put("/", updateConfigs)
2018-06-20 22:41:02 +08:00
return r
}
type configSchema struct {
Port int `json:"port"`
SocketPort int `json:"socket-port"`
AllowLan bool `json:"allow-lan"`
Mode string `json:"mode"`
LogLevel string `json:"log-level"`
}
2018-06-20 22:41:02 +08:00
2018-07-15 22:23:20 +08:00
func getConfigs(w http.ResponseWriter, r *http.Request) {
general := cfg.General()
render.JSON(w, r, configSchema{
Port: *general.Port,
SocketPort: *general.SocketPort,
AllowLan: *general.AllowLan,
Mode: general.Mode.String(),
LogLevel: general.LogLevel.String(),
})
2018-06-20 22:41:02 +08:00
}
2018-07-15 22:23:20 +08:00
func updateConfigs(w http.ResponseWriter, r *http.Request) {
general := &C.General{}
err := render.DecodeJSON(r.Body, general)
2018-06-20 22:41:02 +08:00
if err != nil {
w.WriteHeader(http.StatusBadRequest)
render.JSON(w, r, Error{
Error: "Format error",
})
return
}
2018-07-15 22:23:20 +08:00
// update errors
var proxyErr, modeErr error
// update proxy
listener := proxy.Instance()
proxyErr = listener.Update(&config.Base{
AllowLan: general.AllowLan,
Port: general.Port,
SocketPort: general.SocksPort,
})
2018-07-15 22:23:20 +08:00
// update mode
if general.Mode != nil {
mode, ok := config.ModeMapping[*general.Mode]
2018-07-15 22:23:20 +08:00
if !ok {
modeErr = fmt.Errorf("Mode error")
} else {
cfg.SetMode(mode)
2018-07-15 22:23:20 +08:00
}
}
hasError, errors := formatErrors(map[string]error{
"proxy": proxyErr,
"mode": modeErr,
})
if hasError {
w.WriteHeader(http.StatusBadRequest)
2018-07-15 22:23:20 +08:00
render.JSON(w, r, errors)
2018-06-20 22:41:02 +08:00
return
}
w.WriteHeader(http.StatusNoContent)
}