secret_storage_repository.go

package repository

import (
	"errors"

	"github.com/jinzhu/gorm"

	apperror "github.com/caraml-dev/mlp/api/pkg/errors"

	"github.com/caraml-dev/mlp/api/models"
)

// SecretStorageRepository is an interface for interacting with "secret_storages" table in DB
type SecretStorageRepository interface {
	// Get returns a Secret Storage with given ID
	Get(id models.ID) (*models.SecretStorage, error)
	// List lists all Secret Storage within a project
	List(projectID models.ID) ([]*models.SecretStorage, error)
	// Save creates or updates a Secret Storage
	Save(secretStorage *models.SecretStorage) (*models.SecretStorage, error)
	// Delete deletes a Secret Storage
	Delete(id models.ID) error
	// ListAll lists all Secret Storage
	ListAll() ([]*models.SecretStorage, error)
	// GetGlobal return a global Secret Storage with a name
	GetGlobal(name string) (*models.SecretStorage, error)
	// ListGlobal lists all global Secret Storage
	ListGlobal() ([]*models.SecretStorage, error)
}

type secretStorageRepository struct {
	db *gorm.DB
}

// NewSecretStorageRepository creates a new Secret Storage Repository
func NewSecretStorageRepository(db *gorm.DB) SecretStorageRepository {
	return &secretStorageRepository{
		db: db,
	}
}

// Get returns a Secret Storage with a name within a project
func (r *secretStorageRepository) Get(id models.ID) (*models.SecretStorage, error) {
	var ss models.SecretStorage

	err := r.db.Preload("Project").
		Where("id = ?", id).
		First(&ss).Error

	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, apperror.NewNotFoundErrorf("secret storage with ID %d not found", id)
		}
		return nil, err
	}

	return &ss, err
}

// List lists all Secret Storage within a project
func (r *secretStorageRepository) List(projectID models.ID) ([]*models.SecretStorage, error) {
	var ss []*models.SecretStorage

	err := r.db.Preload("Project").
		Where("project_id = ?", projectID).
		Find(&ss).Error

	return ss, err
}

// Save creates or updates a Secret Storage
func (r *secretStorageRepository) Save(secretStorage *models.SecretStorage) (*models.SecretStorage, error) {
	err := r.db.Save(secretStorage).Error
	return secretStorage, err
}

// Delete deletes a Secret Storage
func (r *secretStorageRepository) Delete(id models.ID) error {
	return r.db.Where("id = ?", id).Delete(models.SecretStorage{}).Error
}

// ListAll lists all Secret Storage
func (r *secretStorageRepository) ListAll() ([]*models.SecretStorage, error) {
	var ss []*models.SecretStorage

	err := r.db.Preload("Project").Find(&ss).Error
	return ss, err
}

// GetGlobal return a global Secret Storage with a name
func (r *secretStorageRepository) GetGlobal(name string) (*models.SecretStorage, error) {
	var ss models.SecretStorage

	err := r.db.Where("name = ? AND scope = ?", name, models.GlobalSecretStorageScope).Find(&ss).Error
	return &ss, err
}

// ListGlobal lists all global Secret Storage
func (r *secretStorageRepository) ListGlobal() ([]*models.SecretStorage, error) {
	var ss []*models.SecretStorage

	err := r.db.Where("scope = ?", models.GlobalSecretStorageScope).Find(&ss).Error
	return ss, err
}