fix: add credential_type to credentials

opengovern · Nov 13, 2024 · 2fd2d70 · 2fd2d70
1 parent 64d387b
commit 2fd2d70
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/services/integration/api/integrations/api.go b/services/integration/api/integrations/api.go
@@ -179,6 +179,7 @@ func (h API) DiscoverIntegrations(c echo.Context) error {
 		err = h.database.CreateCredential(&models2.Credential{
 			ID:              credentialID,
 			IntegrationType: req.IntegrationType,
+			CredentialType:  req.CredentialType,
 			Secret:          secret,
 			Metadata:        credentialMetadataJsonb,
 		})

diff --git a/services/integration/api/models/credential.go b/services/integration/api/models/credential.go
@@ -9,6 +9,7 @@ type Credential struct {
 	ID              string            `json:"id"`
 	Secret          string            `json:"secret"`
 	IntegrationType integration.Type  `json:"integration_type"`
+	CredentialType  string            `json:"credential_type"`
 	Metadata        map[string]string `json:"metadata"`
 	CreatedAt       time.Time         `json:"created_at"`
 	UpdatedAt       time.Time         `json:"updated_at"`

diff --git a/services/integration/api/models/integration.go b/services/integration/api/models/integration.go
@@ -15,6 +15,7 @@ const (
 
 type DiscoverIntegrationRequest struct {
 	IntegrationType integration.Type `json:"integration_type"`
+	CredentialType  string           `json:"credential_type"`
 	CredentialID    *string          `json:"credential_id"`
 	Credentials     map[string]any   `json:"credentials"`
 }

diff --git a/services/integration/models/credential.go b/services/integration/models/credential.go
@@ -14,6 +14,7 @@ import (
 type Credential struct {
 	ID              uuid.UUID `gorm:"primaryKey;type:uuid;default:uuid_generate_v4()"`
 	IntegrationType integration.Type
+	CredentialType  string
 	Secret          string
 	Metadata        pgtype.JSONB
 
@@ -33,6 +34,7 @@ func (c *Credential) ToApi(returnSecret bool) (*models.Credential, error) {
 	credential := &models.Credential{
 		ID:              c.ID.String(),
 		IntegrationType: c.IntegrationType,
+		CredentialType:  c.CredentialType,
 		Metadata:        metadata,
 		CreatedAt:       c.CreatedAt,
 		UpdatedAt:       c.UpdatedAt,