feat: webhook

internal/server/controllers/webhook.go

@@ -0,0 +1,23 @@
+package controllers
+
+import (
+	"github.com/gin-gonic/gin"
+)
+
+func WebhookHead(c *gin.Context) {
+}
+
+func WebhookGet(c *gin.Context) {
+}
+
+func WebhookPost(c *gin.Context) {
+}
+
+func WebhookPut(c *gin.Context) {
+}
+
+func WebhookDelete(c *gin.Context) {
+}
+
+func WebhookOptions(c *gin.Context) {
+}

internal/server/http.go

@@ -16,61 +16,61 @@ func (app *App) server(config *app.Config) {
 	engine.POST(
 		"/plugin/tool/invoke",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.InvokeTool,
 	)
 	engine.POST(
 		"/plugin/tool/validate_credentials",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.ValidateToolCredentials,
 	)
 	engine.POST(
 		"/plugin/llm/invoke",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.InvokeLLM,
 	)
 	engine.POST(
 		"/plugin/text_embedding/invoke",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.InvokeTextEmbedding,
 	)
 	engine.POST(
 		"/plugin/rerank/invoke",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.InvokeRerank,
 	)
 	engine.POST(
 		"/plugin/tts/invoke",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.InvokeTTS,
 	)
 	engine.POST(
 		"/plugin/speech2text/invoke",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.InvokeSpeech2Text,
 	)
 	engine.POST(
 		"/plugin/moderation/invoke",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.InvokeModeration,
 	)
 	engine.POST(
 		"/plugin/model/validate_provider_credentials",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.ValidateProviderCredentials,
 	)
 	engine.POST(
 		"/plugin/model/validate_model_credentials",
 		CheckingKey(config.PluginInnerApiKey),
-		app.Redirect(),
+		app.RedirectPluginInvoke(),
 		controllers.ValidateModelCredentials,
 	)
 
@@ -78,10 +78,19 @@ func (app *App) server(config *app.Config) {
 		engine.POST(
 			"/plugin/debugging/key",
 			CheckingKey(config.PluginInnerApiKey),
-			app.Redirect(),
+			app.RedirectPluginInvoke(),
 			controllers.GetRemoteDebuggingKey,
 		)
 	}
 
+	if config.PluginWebhookEnabled {
+		engine.HEAD("/webhook/:hook_id/*path", controllers.WebhookHead)
+		engine.POST("/webhook/:hook_id/*path", controllers.WebhookPost)
+		engine.GET("/webhook/:hook_id/*path", controllers.WebhookGet)
+		engine.PUT("/webhook/:hook_id/*path", controllers.WebhookPut)
+		engine.DELETE("/webhook/:hook_id/*path", controllers.WebhookDelete)
+		engine.OPTIONS("/webhook/:hook_id/*path", controllers.WebhookOptions)
+	}
+
 	engine.Run(fmt.Sprintf(":%d", config.ServerPort))
 }

internal/server/middleware.go

@@ -35,8 +35,8 @@ func (g *ginContextReader) Close() error {
 	return nil
 }
 
-// Redirect redirects the request to the correct cluster node
-func (app *App) Redirect() gin.HandlerFunc {
+// RedirectPluginInvoke redirects the request to the correct cluster node
+func (app *App) RedirectPluginInvoke() gin.HandlerFunc {
 	return func(ctx *gin.Context) {
 		// get plugin identity
 		raw, err := ctx.GetRawData()
@@ -107,6 +107,9 @@ func (app *App) Redirect() gin.HandlerFunc {
 					ctx.Writer.Write(buf[:n])
 				}
 			}
+
+			ctx.Abort()
+			return
 		} else {
 			ctx.Next()
 		}

internal/types/app/config.go

@@ -18,6 +18,8 @@ type Config struct {
 	PluginRemoteInstallingMaxConn          int    `envconfig:"PLUGIN_REMOTE_INSTALLING_MAX_CONN"`
 	PluginRemoteInstallServerEventLoopNums int    `envconfig:"PLUGIN_REMOTE_INSTALL_SERVER_EVENT_LOOP_NUMS"`
 
+	PluginWebhookEnabled bool `envconfig:"PLUGIN_WEBHOOK_ENABLED"`
+
 	StoragePath        string `envconfig:"STORAGE_PATH"  validate:"required"`
 	ProcessCachingPath string `envconfig:"PROCESS_CACHING_PATH"  validate:"required"`