manage_plugin.go 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188
  1. package service
  2. import (
  3. "time"
  4. "github.com/langgenius/dify-plugin-daemon/internal/db"
  5. "github.com/langgenius/dify-plugin-daemon/internal/types/entities"
  6. "github.com/langgenius/dify-plugin-daemon/internal/types/entities/plugin_entities"
  7. "github.com/langgenius/dify-plugin-daemon/internal/types/models"
  8. "github.com/langgenius/dify-plugin-daemon/internal/utils/cache/helper"
  9. "github.com/langgenius/dify-plugin-daemon/internal/utils/strings"
  10. )
  11. func ListPlugins(tenant_id string, page int, page_size int) *entities.Response {
  12. type installation struct {
  13. ID string `json:"id"`
  14. Name string `json:"name"`
  15. PluginID string `json:"plugin_id"`
  16. TenantID string `json:"tenant_id"`
  17. PluginUniqueIdentifier string `json:"plugin_unique_identifier"`
  18. EndpointsActive int `json:"endpoints_active"`
  19. EndpointsSetups int `json:"endpoints_setups"`
  20. InstallationID string `json:"installation_id"`
  21. Declaration *plugin_entities.PluginDeclaration `json:"declaration"`
  22. RuntimeType plugin_entities.PluginRuntimeType `json:"runtime_type"`
  23. Version string `json:"version"`
  24. CreatedAt time.Time `json:"created_at"`
  25. UpdatedAt time.Time `json:"updated_at"`
  26. Source string `json:"source"`
  27. Meta map[string]any `json:"meta"`
  28. }
  29. pluginInstallations, err := db.GetAll[models.PluginInstallation](
  30. db.Equal("tenant_id", tenant_id),
  31. db.Page(page, page_size),
  32. )
  33. if err != nil {
  34. return entities.NewErrorResponse(-500, err.Error())
  35. }
  36. data := make([]installation, 0, len(pluginInstallations))
  37. for _, plugin_installation := range pluginInstallations {
  38. pluginUniqueIdentifier, err := plugin_entities.NewPluginUniqueIdentifier(
  39. plugin_installation.PluginUniqueIdentifier,
  40. )
  41. if err != nil {
  42. return entities.NewErrorResponse(-500, err.Error())
  43. }
  44. pluginDeclaration, err := helper.CombinedGetPluginDeclaration(
  45. pluginUniqueIdentifier,
  46. tenant_id,
  47. plugin_entities.PluginRuntimeType(plugin_installation.RuntimeType),
  48. )
  49. if err != nil {
  50. return entities.NewErrorResponse(-500, err.Error())
  51. }
  52. data = append(data, installation{
  53. ID: plugin_installation.ID,
  54. Name: pluginDeclaration.Name,
  55. TenantID: plugin_installation.TenantID,
  56. PluginID: pluginUniqueIdentifier.PluginID(),
  57. PluginUniqueIdentifier: pluginUniqueIdentifier.String(),
  58. InstallationID: plugin_installation.ID,
  59. Declaration: pluginDeclaration,
  60. EndpointsSetups: plugin_installation.EndpointsSetups,
  61. EndpointsActive: plugin_installation.EndpointsActive,
  62. RuntimeType: plugin_entities.PluginRuntimeType(plugin_installation.RuntimeType),
  63. Version: pluginDeclaration.Version,
  64. CreatedAt: plugin_installation.CreatedAt,
  65. UpdatedAt: plugin_installation.UpdatedAt,
  66. Source: plugin_installation.Source,
  67. Meta: plugin_installation.Meta,
  68. })
  69. }
  70. return entities.NewSuccessResponse(data)
  71. }
  72. // Using plugin_ids to fetch plugin installations
  73. func BatchFetchPluginInstallationByIDs(tenant_id string, plugin_ids []string) *entities.Response {
  74. if len(plugin_ids) == 0 {
  75. return entities.NewSuccessResponse([]models.PluginInstallation{})
  76. }
  77. pluginInstallations, err := db.GetAll[models.PluginInstallation](
  78. db.Equal("tenant_id", tenant_id),
  79. db.InArray("plugin_id", strings.Map(plugin_ids, func(id string) any { return id })),
  80. db.Page(1, 256), // TODO: pagination
  81. )
  82. if err != nil {
  83. return entities.NewErrorResponse(-500, err.Error())
  84. }
  85. return entities.NewSuccessResponse(pluginInstallations)
  86. }
  87. // check which plugin is missing
  88. func FetchMissingPluginInstallations(tenant_id string, plugin_unique_identifiers []plugin_entities.PluginUniqueIdentifier) *entities.Response {
  89. result := make([]plugin_entities.PluginUniqueIdentifier, 0, len(plugin_unique_identifiers))
  90. if len(plugin_unique_identifiers) == 0 {
  91. return entities.NewSuccessResponse(result)
  92. }
  93. installed, err := db.GetAll[models.PluginInstallation](
  94. db.Equal("tenant_id", tenant_id),
  95. db.InArray(
  96. "plugin_unique_identifier",
  97. strings.Map(
  98. plugin_unique_identifiers,
  99. func(id plugin_entities.PluginUniqueIdentifier) any {
  100. return id.String()
  101. },
  102. ),
  103. ),
  104. )
  105. if err != nil {
  106. return entities.NewErrorResponse(-500, err.Error())
  107. }
  108. // check which plugin is missing
  109. for _, pluginUniqueIdentifier := range plugin_unique_identifiers {
  110. found := false
  111. for _, installed_plugin := range installed {
  112. if installed_plugin.PluginUniqueIdentifier == pluginUniqueIdentifier.String() {
  113. found = true
  114. break
  115. }
  116. }
  117. if !found {
  118. result = append(result, pluginUniqueIdentifier)
  119. }
  120. }
  121. return entities.NewSuccessResponse(result)
  122. }
  123. func ListTools(tenant_id string, page int, page_size int) *entities.Response {
  124. providers, err := db.GetAll[models.ToolInstallation](
  125. db.Equal("tenant_id", tenant_id),
  126. db.Page(page, page_size),
  127. )
  128. if err != nil {
  129. return entities.NewErrorResponse(-500, err.Error())
  130. }
  131. return entities.NewSuccessResponse(providers)
  132. }
  133. func ListModels(tenant_id string, page int, page_size int) *entities.Response {
  134. providers, err := db.GetAll[models.AIModelInstallation](
  135. db.Equal("tenant_id", tenant_id),
  136. db.Page(page, page_size),
  137. )
  138. if err != nil {
  139. return entities.NewErrorResponse(-500, err.Error())
  140. }
  141. return entities.NewSuccessResponse(providers)
  142. }
  143. func GetTool(tenant_id string, plugin_id string, provider string) *entities.Response {
  144. // try get tool
  145. tool, err := db.GetOne[models.ToolInstallation](
  146. db.Equal("tenant_id", tenant_id),
  147. db.Equal("plugin_id", plugin_id),
  148. )
  149. if err != nil {
  150. return entities.NewErrorResponse(-500, err.Error())
  151. }
  152. if tool.Provider != provider {
  153. return entities.NewErrorResponse(-404, "tool not found")
  154. }
  155. return entities.NewSuccessResponse(tool)
  156. }