135 lines
3.4 KiB
Go
135 lines
3.4 KiB
Go
package primary
|
|
|
|
import (
|
|
"blackforestbytes.com/simplecloudnotifier/db"
|
|
"blackforestbytes.com/simplecloudnotifier/models"
|
|
"database/sql"
|
|
"errors"
|
|
"gogs.mikescher.com/BlackForestBytes/goext/sq"
|
|
"time"
|
|
)
|
|
|
|
func (db *Database) CreateSubscription(ctx db.TxContext, subscriberUID models.UserID, channel models.Channel, confirmed bool) (models.Subscription, error) {
|
|
tx, err := ctx.GetOrCreateTransaction(db)
|
|
if err != nil {
|
|
return models.Subscription{}, err
|
|
}
|
|
|
|
entity := models.SubscriptionDB{
|
|
SubscriptionID: models.NewSubscriptionID(),
|
|
SubscriberUserID: subscriberUID,
|
|
ChannelOwnerUserID: channel.OwnerUserID,
|
|
ChannelID: channel.ChannelID,
|
|
ChannelInternalName: channel.InternalName,
|
|
TimestampCreated: time2DB(time.Now()),
|
|
Confirmed: bool2DB(confirmed),
|
|
}
|
|
|
|
_, err = sq.InsertSingle(ctx, tx, "subscriptions", entity)
|
|
if err != nil {
|
|
return models.Subscription{}, err
|
|
}
|
|
|
|
return entity.Model(), nil
|
|
}
|
|
|
|
func (db *Database) ListSubscriptions(ctx db.TxContext, filter models.SubscriptionFilter) ([]models.Subscription, error) {
|
|
tx, err := ctx.GetOrCreateTransaction(db)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
filterCond, filterJoin, prepParams, err := filter.SQL()
|
|
|
|
orderClause := " ORDER BY subscriptions.timestamp_created DESC, subscriptions.subscription_id DESC "
|
|
|
|
sqlQuery := "SELECT " + "subscriptions.*" + " FROM subscriptions " + filterJoin + " WHERE ( " + filterCond + " ) " + orderClause
|
|
|
|
rows, err := tx.Query(ctx, sqlQuery, prepParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
data, err := models.DecodeSubscriptions(rows)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return data, nil
|
|
}
|
|
|
|
func (db *Database) GetSubscription(ctx db.TxContext, subid models.SubscriptionID) (models.Subscription, error) {
|
|
tx, err := ctx.GetOrCreateTransaction(db)
|
|
if err != nil {
|
|
return models.Subscription{}, err
|
|
}
|
|
|
|
rows, err := tx.Query(ctx, "SELECT * FROM subscriptions WHERE subscription_id = :sid LIMIT 1", sq.PP{"sid": subid})
|
|
if err != nil {
|
|
return models.Subscription{}, err
|
|
}
|
|
|
|
sub, err := models.DecodeSubscription(rows)
|
|
if err != nil {
|
|
return models.Subscription{}, err
|
|
}
|
|
|
|
return sub, nil
|
|
}
|
|
|
|
func (db *Database) GetSubscriptionBySubscriber(ctx db.TxContext, subscriberId models.UserID, channelId models.ChannelID) (*models.Subscription, error) {
|
|
tx, err := ctx.GetOrCreateTransaction(db)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
rows, err := tx.Query(ctx, "SELECT * FROM subscriptions WHERE subscriber_user_id = :suid AND channel_id = :cid LIMIT 1", sq.PP{
|
|
"suid": subscriberId,
|
|
"cid": channelId,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
user, err := models.DecodeSubscription(rows)
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil, nil
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (db *Database) DeleteSubscription(ctx db.TxContext, subid models.SubscriptionID) error {
|
|
tx, err := ctx.GetOrCreateTransaction(db)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = tx.Exec(ctx, "DELETE FROM subscriptions WHERE subscription_id = :sid", sq.PP{"sid": subid})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (db *Database) UpdateSubscriptionConfirmed(ctx db.TxContext, subscriptionID models.SubscriptionID, confirmed bool) error {
|
|
tx, err := ctx.GetOrCreateTransaction(db)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = tx.Exec(ctx, "UPDATE subscriptions SET confirmed = :conf WHERE subscription_id = :sid", sq.PP{
|
|
"conf": confirmed,
|
|
"sid": subscriptionID,
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|