86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
otterError "git.anthrove.art/anthrove/otter-space-sdk/v2/pkg/error"
|
|
"git.anthrove.art/anthrove/otter-space-sdk/v2/pkg/models"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// CreateSource creates a pgModels.Source
|
|
func CreateSource(ctx context.Context, db *gorm.DB, anthroveSource *models.Source) error {
|
|
|
|
if anthroveSource.Domain == "" {
|
|
return &otterError.EntityValidationFailed{Reason: "Domain is required"}
|
|
}
|
|
|
|
result := db.WithContext(ctx).Where(models.Source{Domain: anthroveSource.Domain}).FirstOrCreate(anthroveSource)
|
|
|
|
if result.Error != nil {
|
|
if errors.Is(result.Error, gorm.ErrDuplicatedKey) {
|
|
return &otterError.EntityAlreadyExists{}
|
|
}
|
|
return result.Error
|
|
}
|
|
|
|
if result.RowsAffected == 0 {
|
|
return &otterError.NoDataWritten{}
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"node_source_url": anthroveSource.Domain,
|
|
"node_source_displayName": anthroveSource.DisplayName,
|
|
"node_source_icon": anthroveSource.Icon,
|
|
}).Trace("database: created source node")
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetAllSource returns a list of all pgModels.Source
|
|
func GetAllSource(ctx context.Context, db *gorm.DB) ([]models.Source, error) {
|
|
var sources []models.Source
|
|
|
|
result := db.WithContext(ctx).Find(&sources)
|
|
|
|
if result.Error != nil {
|
|
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
|
return nil, &otterError.NoDataFound{}
|
|
}
|
|
return nil, result.Error
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"tag_amount": result.RowsAffected,
|
|
}).Trace("database: get all source nodes")
|
|
|
|
return sources, nil
|
|
}
|
|
|
|
// GetSourceByDomain returns the first source it finds based on the domain
|
|
func GetSourceByDomain(ctx context.Context, db *gorm.DB, sourceDomain models.AnthroveSourceDomain) (*models.Source, error) {
|
|
var sources models.Source
|
|
|
|
if sourceDomain == "" {
|
|
return nil, &otterError.EntityValidationFailed{Reason: "AnthroveSourceDomain is not set"}
|
|
}
|
|
|
|
result := db.WithContext(ctx).Where("domain = ?", sourceDomain).First(&sources)
|
|
|
|
if result.Error != nil {
|
|
if errors.Is(result.Error, gorm.ErrRecordNotFound) {
|
|
return nil, &otterError.NoDataFound{}
|
|
}
|
|
return nil, result.Error
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"tag_amount": result.RowsAffected,
|
|
}).Trace("database: get all source nodes")
|
|
|
|
return &sources, nil
|
|
}
|