state.go 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. package install_task
  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/models"
  7. "github.com/langgenius/dify-plugin-daemon/internal/types/models/curd"
  8. "github.com/langgenius/dify-plugin-daemon/internal/utils/strings"
  9. )
  10. func InstallPlugin(
  11. tenant_id string,
  12. user_id string,
  13. runtime entities.PluginRuntimeInterface,
  14. configuration map[string]any,
  15. ) error {
  16. identity, err := runtime.Identity()
  17. if err != nil {
  18. return err
  19. }
  20. plugin := &models.Plugin{
  21. PluginID: identity,
  22. Refers: 0,
  23. Checksum: runtime.Checksum(),
  24. InstallType: runtime.Type(),
  25. ManifestType: runtime.Configuration().Type,
  26. }
  27. plugin, installation, err := curd.CreatePlugin(tenant_id, user_id, plugin, configuration)
  28. if err != nil {
  29. return err
  30. }
  31. // check if there is a webhook for the plugin
  32. if runtime.Configuration().Resource.Permission.AllowRegistryWebhook() {
  33. _, err := InstallWebhook(plugin.PluginID, installation.ID, tenant_id, user_id)
  34. if err != nil {
  35. return err
  36. }
  37. }
  38. return nil
  39. }
  40. // installs a plugin to db,
  41. // returns the webhook id
  42. func InstallWebhook(plugin_id string, installation_id string, tenant_id string, user_id string) (string, error) {
  43. installation := &models.Webhook{
  44. HookID: strings.RandomString(64),
  45. PluginID: plugin_id,
  46. TenantID: tenant_id,
  47. UserID: user_id,
  48. ExpiredAt: time.Now().Add(time.Hour * 24 * 365 * 10),
  49. PluginInstallationId: installation_id,
  50. }
  51. if err := db.Create(&installation); err != nil {
  52. return "", err
  53. }
  54. return installation.HookID, nil
  55. }
  56. // uninstalls a plugin from db
  57. func UninstallWebhook(webhook *models.Webhook) error {
  58. return db.Delete(webhook)
  59. }