From 34f62a091926e3dd3a2cdcff8eae4c8a0021a011 Mon Sep 17 00:00:00 2001
From: xishang0128 <xishang02@gmail.com>
Date: Sat, 23 Sep 2023 17:54:20 +0800
Subject: [PATCH] feat: add provider proxies api

---
 hub/route/provider.go | 38 ++++++++++++++++++++++++++++++++++++--
 1 file changed, 36 insertions(+), 2 deletions(-)

diff --git a/hub/route/provider.go b/hub/route/provider.go
index 1ba0d32c..c050a9f1 100644
--- a/hub/route/provider.go
+++ b/hub/route/provider.go
@@ -4,22 +4,35 @@ import (
 	"context"
 	"net/http"
 
+	C "github.com/Dreamacro/clash/constant"
 	"github.com/Dreamacro/clash/constant/provider"
 	"github.com/Dreamacro/clash/tunnel"
 
 	"github.com/go-chi/chi/v5"
 	"github.com/go-chi/render"
+	"github.com/samber/lo"
 )
 
 func proxyProviderRouter() http.Handler {
 	r := chi.NewRouter()
 	r.Get("/", getProviders)
 
-	r.Route("/{name}", func(r chi.Router) {
+	r.Route("/{providerName}", func(r chi.Router) {
 		r.Use(parseProviderName, findProviderByName)
 		r.Get("/", getProvider)
 		r.Put("/", updateProvider)
 		r.Get("/healthcheck", healthCheckProvider)
+		r.Mount("/", proxyProviderProxyRouter())
+	})
+	return r
+}
+
+func proxyProviderProxyRouter() http.Handler {
+	r := chi.NewRouter()
+	r.Route("/{name}", func(r chi.Router) {
+		r.Use(parseProxyName, findProviderProxyByName)
+		r.Get("/", getProxy)
+		r.Get("/healthcheck", getProxyDelay)
 	})
 	return r
 }
@@ -54,7 +67,7 @@ func healthCheckProvider(w http.ResponseWriter, r *http.Request) {
 
 func parseProviderName(next http.Handler) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		name := getEscapeParam(r, "name")
+		name := getEscapeParam(r, "providerName")
 		ctx := context.WithValue(r.Context(), CtxKeyProviderName, name)
 		next.ServeHTTP(w, r.WithContext(ctx))
 	})
@@ -76,6 +89,27 @@ func findProviderByName(next http.Handler) http.Handler {
 	})
 }
 
+func findProviderProxyByName(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		var (
+			name = r.Context().Value(CtxKeyProxyName).(string)
+			pd   = r.Context().Value(CtxKeyProvider).(provider.ProxyProvider)
+		)
+		proxy, exist := lo.Find(pd.Proxies(), func(proxy C.Proxy) bool {
+			return proxy.Name() == name
+		})
+
+		if !exist {
+			render.Status(r, http.StatusNotFound)
+			render.JSON(w, r, ErrNotFound)
+			return
+		}
+
+		ctx := context.WithValue(r.Context(), CtxKeyProxy, proxy)
+		next.ServeHTTP(w, r.WithContext(ctx))
+	})
+}
+
 func ruleProviderRouter() http.Handler {
 	r := chi.NewRouter()
 	r.Get("/", getRuleProviders)