59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"git.dragse.it/anthrove/otter-space-sdk/pkg/models"
|
|
"git.dragse.it/anthrove/otter-space-sdk/pkg/models/pgModels"
|
|
log "github.com/sirupsen/logrus"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func EstablishAnthrovePostToSourceLink(ctx context.Context, db *gorm.DB, anthrovePostID models.AnthrovePostID, anthroveSourceDomain string) error {
|
|
var post pgModels.Post
|
|
var source pgModels.Source
|
|
|
|
// Find the post
|
|
err := db.WithContext(ctx).Where("id = ?", anthrovePostID).First(&post).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Find the source
|
|
err = db.WithContext(ctx).Where("domain = ?", anthroveSourceDomain).First(&source).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Establish the relationship
|
|
err = db.WithContext(ctx).Model(&post).Association("Sources").Append(&source)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"anthrove_post_id": anthrovePostID,
|
|
"anthrove_source_domain": anthroveSourceDomain,
|
|
}).Trace("database: created anthrove post to source link")
|
|
|
|
return nil
|
|
}
|
|
|
|
func EstablishUserToPostLink(ctx context.Context, db *gorm.DB, anthroveUserID models.AnthroveUserID, anthrovePostID models.AnthrovePostID) error {
|
|
userFavorite := pgModels.UserFavorite{
|
|
UserID: string(anthroveUserID),
|
|
PostID: string(anthrovePostID),
|
|
}
|
|
|
|
err := db.WithContext(ctx).Create(&userFavorite).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
log.WithFields(log.Fields{
|
|
"anthrove_user_id": anthroveUserID,
|
|
"anthrove_post_id": anthrovePostID,
|
|
}).Trace("database: created user to post link")
|
|
|
|
return nil
|
|
}
|