plugins.go 3.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495
  1. package controllers
  2. import (
  3. "net/http"
  4. "github.com/gin-gonic/gin"
  5. "github.com/langgenius/dify-plugin-daemon/internal/core/plugin_manager"
  6. "github.com/langgenius/dify-plugin-daemon/internal/service"
  7. "github.com/langgenius/dify-plugin-daemon/internal/types/app"
  8. "github.com/langgenius/dify-plugin-daemon/internal/types/entities"
  9. "github.com/langgenius/dify-plugin-daemon/internal/types/entities/plugin_entities"
  10. )
  11. func GetAsset(c *gin.Context) {
  12. plugin_manager := plugin_manager.Manager()
  13. asset, err := plugin_manager.GetAsset(c.Param("id"))
  14. if err != nil {
  15. c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
  16. return
  17. }
  18. c.Data(http.StatusOK, "application/octet-stream", asset)
  19. }
  20. func InstallPluginFromPkg(app *app.Config) gin.HandlerFunc {
  21. return func(c *gin.Context) {
  22. dify_pkg_file_header, err := c.FormFile("dify_pkg")
  23. if err != nil {
  24. c.JSON(http.StatusOK, entities.NewErrorResponse(-400, err.Error()))
  25. return
  26. }
  27. tenant_id := c.Param("tenant_id")
  28. if tenant_id == "" {
  29. c.JSON(http.StatusOK, entities.NewErrorResponse(-400, "Tenant ID is required"))
  30. return
  31. }
  32. if dify_pkg_file_header.Size > app.MaxPluginPackageSize {
  33. c.JSON(http.StatusOK, entities.NewErrorResponse(-413, "File size exceeds the maximum limit"))
  34. return
  35. }
  36. verify_signature := c.PostForm("verify_signature") == "true"
  37. dify_pkg_file, err := dify_pkg_file_header.Open()
  38. if err != nil {
  39. c.JSON(http.StatusOK, entities.NewErrorResponse(-500, err.Error()))
  40. return
  41. }
  42. defer dify_pkg_file.Close()
  43. service.InstallPluginFromPkg(app, c, tenant_id, dify_pkg_file, verify_signature)
  44. }
  45. }
  46. func InstallPluginFromIdentifier(app *app.Config) gin.HandlerFunc {
  47. return func(c *gin.Context) {
  48. BindRequest(c, func(request struct {
  49. TenantID string `uri:"tenant_id" validate:"required"`
  50. PluginUniqueIdentifier plugin_entities.PluginUniqueIdentifier `json:"plugin_unique_identifier" validate:"required,plugin_unique_identifier"`
  51. }) {
  52. c.JSON(http.StatusOK, service.InstallPluginFromIdentifier(request.TenantID, request.PluginUniqueIdentifier))
  53. })
  54. }
  55. }
  56. func UninstallPlugin(c *gin.Context) {
  57. BindRequest(c, func(request struct {
  58. TenantID string `uri:"tenant_id" validate:"required"`
  59. PluginInstallationID string `json:"plugin_installation_id" validate:"required"`
  60. }) {
  61. c.JSON(http.StatusOK, service.UninstallPlugin(request.TenantID, request.PluginInstallationID))
  62. })
  63. }
  64. func FetchPluginFromIdentifier(c *gin.Context) {
  65. BindRequest(c, func(request struct {
  66. PluginUniqueIdentifier plugin_entities.PluginUniqueIdentifier `form:"plugin_unique_identifier" validate:"required,plugin_unique_identifier"`
  67. }) {
  68. c.JSON(http.StatusOK, service.FetchPluginFromIdentifier(request.PluginUniqueIdentifier))
  69. })
  70. }
  71. func ListPlugins(c *gin.Context) {
  72. BindRequest(c, func(request struct {
  73. TenantID string `uri:"tenant_id" validate:"required"`
  74. Page int `form:"page" validate:"required,min=1"`
  75. PageSize int `form:"page_size" validate:"required,min=1,max=256"`
  76. }) {
  77. c.JSON(http.StatusOK, service.ListPlugins(request.TenantID, request.Page, request.PageSize))
  78. })
  79. }