dify-plugin-daemon/internal/server/http_server.go

package server

import (
	"context"
	"fmt"
	"net/http"

	"github.com/gin-gonic/gin"
	"github.com/langgenius/dify-plugin-daemon/internal/core/plugin_daemon/backwards_invocation/transaction"
	"github.com/langgenius/dify-plugin-daemon/internal/server/controllers"
	"github.com/langgenius/dify-plugin-daemon/internal/service"
	"github.com/langgenius/dify-plugin-daemon/internal/types/app"
	"github.com/langgenius/dify-plugin-daemon/internal/utils/log"
)

func (app *App) server(config *app.Config) func() {
	engine := gin.Default()
	engine.GET("/health/check", controllers.HealthCheck)

	app.pluginInvokeGroup(engine.Group("/plugin"), config)
	app.remoteDebuggingGroup(engine.Group("/plugin/debugging"), config)
	app.webhookGroup(engine.Group("/webhook"), config)
	app.awsLambdaTransactionGroup(engine.Group("/aws-lambda"), config)

	srv := &http.Server{
		Addr:    fmt.Sprintf(":%d", config.ServerPort),
		Handler: engine,
	}

	go func() {
		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Panic("listen: %s\n", err)
		}
	}()

	return func() {
		if err := srv.Shutdown(context.Background()); err != nil {
			log.Panic("Server Shutdown: %s\n", err)
		}
	}
}

func (app *App) pluginInvokeGroup(group *gin.RouterGroup, config *app.Config) {
	group.Use(CheckingKey(config.PluginInnerApiKey))
	group.Use(app.RedirectPluginInvoke())
	group.Use(app.InitClusterID())

	group.POST("/tool/invoke", controllers.InvokeTool)
	group.POST("/tool/validate_credentials", controllers.ValidateToolCredentials)
	group.POST("/llm/invoke", controllers.InvokeLLM)
	group.POST("/text_embedding/invoke", controllers.InvokeTextEmbedding)
	group.POST("/rerank/invoke", controllers.InvokeRerank)
	group.POST("/tts/invoke", controllers.InvokeTTS)
	group.POST("/speech2text/invoke", controllers.InvokeSpeech2Text)
	group.POST("/moderation/invoke", controllers.InvokeModeration)
	group.POST("/model/validate_provider_credentials", controllers.ValidateProviderCredentials)
	group.POST("/model/validate_model_credentials", controllers.ValidateModelCredentials)
}

func (app *App) remoteDebuggingGroup(group *gin.RouterGroup, config *app.Config) {
	if config.PluginRemoteInstallingEnabled {
		group.POST("/key", CheckingKey(config.PluginInnerApiKey), controllers.GetRemoteDebuggingKey)
	}
}

func (app *App) webhookGroup(group *gin.RouterGroup, config *app.Config) {
	if config.PluginWebhookEnabled {
		group.HEAD("/:hook_id/*path", app.Webhook())
		group.POST("/:hook_id/*path", app.Webhook())
		group.GET("/:hook_id/*path", app.Webhook())
		group.PUT("/:hook_id/*path", app.Webhook())
		group.DELETE("/:hook_id/*path", app.Webhook())
		group.OPTIONS("/:hook_id/*path", app.Webhook())
	}
}

func (appRef *App) awsLambdaTransactionGroup(group *gin.RouterGroup, config *app.Config) {
	if config.Platform == app.PLATFORM_AWS_LAMBDA {
		appRef.aws_transaction_handler = transaction.NewAWSTransactionHandler(config.MaxAWSLambdaTransactionTimeout)
		group.POST(
			"/transaction",
			appRef.RedirectAWSLambdaTransaction,
			service.HandleAWSPluginTransaction(appRef.aws_transaction_handler),
		)
	}
}