package db import ( "blackforestbytes.com/simplecloudnotifier/models" "database/sql" "gogs.mikescher.com/BlackForestBytes/goext/sq" "time" ) func (db *Database) CreateSubscription(ctx TxContext, subscriberUID models.UserID, channel models.Channel, confirmed bool) (models.Subscription, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return models.Subscription{}, err } now := time.Now().UTC() res, err := tx.Exec(ctx, "INSERT INTO subscriptions (subscriber_user_id, channel_owner_user_id, channel_name, channel_id, timestamp_created, confirmed) VALUES (:suid, :ouid, :cnam, :cid, :ts, :conf)", sq.PP{ "suid": subscriberUID, "ouid": channel.OwnerUserID, "cnam": channel.Name, "cid": channel.ChannelID, "ts": time2DB(now), "conf": confirmed, }) if err != nil { return models.Subscription{}, err } liid, err := res.LastInsertId() if err != nil { return models.Subscription{}, err } return models.Subscription{ SubscriptionID: models.SubscriptionID(liid), SubscriberUserID: subscriberUID, ChannelOwnerUserID: channel.OwnerUserID, ChannelID: channel.ChannelID, ChannelName: channel.Name, TimestampCreated: now, Confirmed: confirmed, }, nil } func (db *Database) ListSubscriptionsByChannel(ctx TxContext, 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 channel_id = :cid", sq.PP{"cid": channelID}) if err != nil { return nil, err } data, err := models.DecodeSubscriptions(rows) if err != nil { return nil, err } return data, nil } func (db *Database) ListSubscriptionsByOwner(ctx TxContext, ownerUserID models.UserID) ([]models.Subscription, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return nil, err } rows, err := tx.Query(ctx, "SELECT * FROM subscriptions WHERE channel_owner_user_id = :ouid", sq.PP{"ouid": ownerUserID}) 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 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 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 err == sql.ErrNoRows { return nil, nil } if err != nil { return nil, err } return &user, nil } func (db *Database) DeleteSubscription(ctx 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 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 }