package db import ( "blackforestbytes.com/simplecloudnotifier/models" "database/sql" "time" ) func (db *Database) GetChannelByName(ctx TxContext, userid models.UserID, chanName string) (*models.Channel, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return nil, err } rows, err := tx.QueryContext(ctx, "SELECT * FROM channels WHERE owner_user_id = ? OR name = ? LIMIT 1", userid, chanName) if err != nil { return nil, err } channel, err := models.DecodeChannel(rows) if err == sql.ErrNoRows { return nil, nil } if err != nil { return nil, err } return &channel, nil } func (db *Database) CreateChannel(ctx TxContext, userid models.UserID, name string, subscribeKey string, sendKey string) (models.Channel, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return models.Channel{}, err } now := time.Now().UTC() res, err := tx.ExecContext(ctx, "INSERT INTO channels (owner_user_id, name, subscribe_key, send_key, timestamp_created) VALUES (?, ?, ?, ?, ?)", userid, name, subscribeKey, sendKey, time2DB(now)) if err != nil { return models.Channel{}, err } liid, err := res.LastInsertId() if err != nil { return models.Channel{}, err } return models.Channel{ ChannelID: models.ChannelID(liid), OwnerUserID: userid, Name: name, SubscribeKey: subscribeKey, SendKey: sendKey, TimestampCreated: now, TimestampLastSent: nil, MessagesSent: 0, }, nil } func (db *Database) ListChannelsByOwner(ctx TxContext, userid models.UserID) ([]models.Channel, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return nil, err } rows, err := tx.QueryContext(ctx, "SELECT * FROM channels WHERE owner_user_id = ?", userid) if err != nil { return nil, err } data, err := models.DecodeChannels(rows) if err != nil { return nil, err } return data, nil } func (db *Database) ListChannelsBySubscriber(ctx TxContext, userid models.UserID, confirmed bool) ([]models.Channel, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return nil, err } confCond := "" if confirmed { confCond = " AND sub.confirmed = 1" } rows, err := tx.QueryContext(ctx, "SELECT * FROM channels LEFT JOIN subscriptions sub on channels.channel_id = sub.channel_id WHERE sub.subscriber_user_id = ? "+confCond, userid) if err != nil { return nil, err } data, err := models.DecodeChannels(rows) if err != nil { return nil, err } return data, nil } func (db *Database) ListChannelsByAccess(ctx TxContext, userid models.UserID, confirmed bool) ([]models.Channel, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return nil, err } confCond := "OR sub.subscriber_user_id = ?" if confirmed { confCond = "OR (sub.subscriber_user_id = ? AND sub.confirmed = 1)" } rows, err := tx.QueryContext(ctx, "SELECT * FROM channels LEFT JOIN subscriptions sub on channels.channel_id = sub.channel_id WHERE owner_user_id = ? "+confCond, userid) if err != nil { return nil, err } data, err := models.DecodeChannels(rows) if err != nil { return nil, err } return data, nil } func (db *Database) GetChannel(ctx TxContext, userid models.UserID, channelid models.ChannelID) (models.Channel, error) { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return models.Channel{}, err } rows, err := tx.QueryContext(ctx, "SELECT * FROM channels WHERE owner_user_id = ? AND channel_id = ? LIMIT 1", userid, channelid) if err != nil { return models.Channel{}, err } client, err := models.DecodeChannel(rows) if err != nil { return models.Channel{}, err } return client, nil } func (db *Database) IncChannelMessageCounter(ctx TxContext, channel models.Channel) error { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return err } _, err = tx.ExecContext(ctx, "UPDATE channels SET messages_sent = ?, timestamp_lastsent = ? WHERE channel_id = ?", channel.MessagesSent+1, time2DB(time.Now()), channel.ChannelID) if err != nil { return err } return nil } func (db *Database) UpdateChannelSendKey(ctx TxContext, channelid models.ChannelID, newkey string) error { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return err } _, err = tx.ExecContext(ctx, "UPDATE channels SET send_key = ? WHERE channel_id = ?", newkey, channelid) if err != nil { return err } return nil } func (db *Database) UpdateChannelSubscribeKey(ctx TxContext, channelid models.ChannelID, newkey string) error { tx, err := ctx.GetOrCreateTransaction(db) if err != nil { return err } _, err = tx.ExecContext(ctx, "UPDATE channels SET subscribe_key = ? WHERE channel_id = ?", newkey, channelid) if err != nil { return err } return nil }