package restfulapi import ( "net/http" "strings" "github.com/go-chi/chi/v5" "github.com/go-chi/chi/v5/middleware" "github.com/go-chi/render" "github.com/go-playground/validator/v10" core "github.com/v2fly/v2ray-core/v5" "github.com/v2fly/v2ray-core/v5/common/net" "github.com/v2fly/v2ray-core/v5/transport/internet" ) var validate *validator.Validate type StatsBound struct { // Better name? Uplink int64 `json:"uplink"` Downlink int64 `json:"downlink"` } func (rs *restfulService) tagStats(w http.ResponseWriter, r *http.Request) { boundType := chi.URLParam(r, "bound_type") tag := chi.URLParam(r, "tag") if validate.Var(boundType, "required,oneof=inbounds outbounds") != nil || validate.Var(tag, "required,min=1,max=255") != nil { render.Status(r, http.StatusUnprocessableEntity) render.JSON(w, r, render.M{}) return } bound := boundType[:len(boundType)-1] upCounter := rs.stats.GetCounter(bound + ">>>" + tag + ">>>traffic>>>uplink") downCounter := rs.stats.GetCounter(bound + ">>>" + tag + ">>>traffic>>>downlink") if upCounter == nil || downCounter == nil { render.Status(r, http.StatusNotFound) render.JSON(w, r, render.M{}) return } render.JSON(w, r, &StatsBound{ Uplink: upCounter.Value(), Downlink: downCounter.Value(), }) } func (rs *restfulService) version(w http.ResponseWriter, r *http.Request) { render.JSON(w, r, render.M{"version": core.Version()}) } func (rs *restfulService) TokenAuthMiddleware(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { header := r.Header.Get("Authorization") text := strings.SplitN(header, " ", 2) hasInvalidHeader := text[0] != "Bearer" hasInvalidSecret := len(text) != 2 || text[1] != rs.config.AuthToken if hasInvalidHeader || hasInvalidSecret { render.Status(r, http.StatusUnauthorized) render.JSON(w, r, render.M{}) return } next.ServeHTTP(w, r) }) } func (rs *restfulService) start() error { r := chi.NewRouter() r.Use(middleware.Heartbeat("/ping")) validate = validator.New() r.Route("/v1", func(r chi.Router) { r.Get("/{bound_type}/{tag}/stats", rs.tagStats) }) r.Get("/version", rs.version) var listener net.Listener var err error address := net.ParseAddress(rs.config.ListenAddr) switch { case address.Family().IsIP(): listener, err = internet.ListenSystem(rs.ctx, &net.TCPAddr{IP: address.IP(), Port: int(rs.config.ListenPort)}, nil) case strings.EqualFold(address.Domain(), "localhost"): listener, err = internet.ListenSystem(rs.ctx, &net.TCPAddr{IP: net.IP{127, 0, 0, 1}, Port: int(rs.config.ListenPort)}, nil) default: return newError("restful api cannot listen on the address: ", address) } if err != nil { return newError("restful api cannot listen on the port ", rs.config.ListenPort).Base(err) } go func() { err := http.Serve(listener, r) if err != nil { newError("unable to serve restful api").WriteToLog() } }() return nil }