450 lines
13 KiB
Go
450 lines
13 KiB
Go
/*
|
|
GoToSocial
|
|
Copyright (C) 2021-2022 GoToSocial Authors admin@gotosocial.org
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package bundb
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"time"
|
|
|
|
"codeberg.org/gruf/go-cache/v3/result"
|
|
"github.com/superseriousbusiness/gotosocial/internal/db"
|
|
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
|
"github.com/superseriousbusiness/gotosocial/internal/log"
|
|
"github.com/uptrace/bun"
|
|
"github.com/uptrace/bun/dialect"
|
|
)
|
|
|
|
type emojiDB struct {
|
|
conn *DBConn
|
|
emojiCache *result.Cache[*gtsmodel.Emoji]
|
|
categoryCache *result.Cache[*gtsmodel.EmojiCategory]
|
|
}
|
|
|
|
func (e *emojiDB) init() {
|
|
// Initialize emoji result cache
|
|
e.emojiCache = result.NewSized([]result.Lookup{
|
|
{Name: "ID"},
|
|
{Name: "URI"},
|
|
{Name: "Shortcode.Domain"},
|
|
{Name: "ImageStaticURL"},
|
|
}, func(e1 *gtsmodel.Emoji) *gtsmodel.Emoji {
|
|
e2 := new(gtsmodel.Emoji)
|
|
*e2 = *e1
|
|
return e2
|
|
}, 1000)
|
|
|
|
// Set cache TTL and start sweep routine
|
|
e.emojiCache.SetTTL(time.Minute*5, false)
|
|
e.emojiCache.Start(time.Second * 10)
|
|
|
|
// Initialize category result cache
|
|
e.categoryCache = result.NewSized([]result.Lookup{
|
|
{Name: "ID"},
|
|
{Name: "Name"},
|
|
}, func(c1 *gtsmodel.EmojiCategory) *gtsmodel.EmojiCategory {
|
|
c2 := new(gtsmodel.EmojiCategory)
|
|
*c2 = *c1
|
|
return c2
|
|
}, 1000)
|
|
|
|
// Set cache TTL and start sweep routine
|
|
e.categoryCache.SetTTL(time.Minute*5, false)
|
|
e.categoryCache.Start(time.Second * 10)
|
|
}
|
|
|
|
func (e *emojiDB) newEmojiQ(emoji *gtsmodel.Emoji) *bun.SelectQuery {
|
|
return e.conn.
|
|
NewSelect().
|
|
Model(emoji).
|
|
Relation("Category")
|
|
}
|
|
|
|
func (e *emojiDB) newEmojiCategoryQ(emojiCategory *gtsmodel.EmojiCategory) *bun.SelectQuery {
|
|
return e.conn.
|
|
NewSelect().
|
|
Model(emojiCategory)
|
|
}
|
|
|
|
func (e *emojiDB) PutEmoji(ctx context.Context, emoji *gtsmodel.Emoji) db.Error {
|
|
return e.emojiCache.Store(emoji, func() error {
|
|
_, err := e.conn.NewInsert().Model(emoji).Exec(ctx)
|
|
return e.conn.ProcessError(err)
|
|
})
|
|
}
|
|
|
|
func (e *emojiDB) UpdateEmoji(ctx context.Context, emoji *gtsmodel.Emoji, columns ...string) (*gtsmodel.Emoji, db.Error) {
|
|
// Update the emoji's last-updated
|
|
emoji.UpdatedAt = time.Now()
|
|
|
|
if _, err := e.conn.
|
|
NewUpdate().
|
|
Model(emoji).
|
|
Where("? = ?", bun.Ident("emoji.id"), emoji.ID).
|
|
Column(columns...).
|
|
Exec(ctx); err != nil {
|
|
return nil, e.conn.ProcessError(err)
|
|
}
|
|
|
|
e.emojiCache.Invalidate("ID", emoji.ID)
|
|
return emoji, nil
|
|
}
|
|
|
|
func (e *emojiDB) DeleteEmojiByID(ctx context.Context, id string) db.Error {
|
|
if err := e.conn.RunInTx(ctx, func(tx bun.Tx) error {
|
|
// delete links between this emoji and any statuses that use it
|
|
if _, err := tx.
|
|
NewDelete().
|
|
TableExpr("? AS ?", bun.Ident("status_to_emojis"), bun.Ident("status_to_emoji")).
|
|
Where("? = ?", bun.Ident("status_to_emoji.emoji_id"), id).
|
|
Exec(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
// delete links between this emoji and any accounts that use it
|
|
if _, err := tx.
|
|
NewDelete().
|
|
TableExpr("? AS ?", bun.Ident("account_to_emojis"), bun.Ident("account_to_emoji")).
|
|
Where("? = ?", bun.Ident("account_to_emoji.emoji_id"), id).
|
|
Exec(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err := tx.
|
|
NewDelete().
|
|
TableExpr("? AS ?", bun.Ident("emojis"), bun.Ident("emoji")).
|
|
Where("? = ?", bun.Ident("emoji.id"), id).
|
|
Exec(ctx); err != nil {
|
|
return e.conn.ProcessError(err)
|
|
}
|
|
|
|
return nil
|
|
}); err != nil {
|
|
return err
|
|
}
|
|
|
|
e.emojiCache.Invalidate("ID", id)
|
|
return nil
|
|
}
|
|
|
|
func (e *emojiDB) GetEmojis(ctx context.Context, domain string, includeDisabled bool, includeEnabled bool, shortcode string, maxShortcodeDomain string, minShortcodeDomain string, limit int) ([]*gtsmodel.Emoji, db.Error) {
|
|
emojiIDs := []string{}
|
|
|
|
subQuery := e.conn.
|
|
NewSelect().
|
|
ColumnExpr("? AS ?", bun.Ident("emoji.id"), bun.Ident("emoji_ids"))
|
|
|
|
// To ensure consistent ordering and make paging possible, we sort not by shortcode
|
|
// but by [shortcode]@[domain]. Because sqlite and postgres have different syntax
|
|
// for concatenation, that means we need to switch here. Depending on which driver
|
|
// is in use, query will look something like this (sqlite):
|
|
//
|
|
// SELECT
|
|
// "emoji"."id" AS "emoji_ids",
|
|
// lower("emoji"."shortcode" || '@' || COALESCE("emoji"."domain", '')) AS "shortcode_domain"
|
|
// FROM
|
|
// "emojis" AS "emoji"
|
|
// ORDER BY
|
|
// "shortcode_domain" ASC
|
|
//
|
|
// Or like this (postgres):
|
|
//
|
|
// SELECT
|
|
// "emoji"."id" AS "emoji_ids",
|
|
// LOWER(CONCAT("emoji"."shortcode", '@', COALESCE("emoji"."domain", ''))) AS "shortcode_domain"
|
|
// FROM
|
|
// "emojis" AS "emoji"
|
|
// ORDER BY
|
|
// "shortcode_domain" ASC
|
|
switch e.conn.Dialect().Name() {
|
|
case dialect.SQLite:
|
|
subQuery = subQuery.ColumnExpr("LOWER(? || ? || COALESCE(?, ?)) AS ?", bun.Ident("emoji.shortcode"), "@", bun.Ident("emoji.domain"), "", bun.Ident("shortcode_domain"))
|
|
case dialect.PG:
|
|
subQuery = subQuery.ColumnExpr("LOWER(CONCAT(?, ?, COALESCE(?, ?))) AS ?", bun.Ident("emoji.shortcode"), "@", bun.Ident("emoji.domain"), "", bun.Ident("shortcode_domain"))
|
|
default:
|
|
panic("db conn was neither pg not sqlite")
|
|
}
|
|
|
|
subQuery = subQuery.TableExpr("? AS ?", bun.Ident("emojis"), bun.Ident("emoji"))
|
|
|
|
if domain == "" {
|
|
subQuery = subQuery.Where("? IS NULL", bun.Ident("emoji.domain"))
|
|
} else if domain != db.EmojiAllDomains {
|
|
subQuery = subQuery.Where("? = ?", bun.Ident("emoji.domain"), domain)
|
|
}
|
|
|
|
switch {
|
|
case includeDisabled && !includeEnabled:
|
|
// show only disabled emojis
|
|
subQuery = subQuery.Where("? = ?", bun.Ident("emoji.disabled"), true)
|
|
case includeEnabled && !includeDisabled:
|
|
// show only enabled emojis
|
|
subQuery = subQuery.Where("? = ?", bun.Ident("emoji.disabled"), false)
|
|
default:
|
|
// show emojis regardless of emoji.disabled value
|
|
}
|
|
|
|
if shortcode != "" {
|
|
subQuery = subQuery.Where("LOWER(?) = LOWER(?)", bun.Ident("emoji.shortcode"), shortcode)
|
|
}
|
|
|
|
// assume we want to sort ASC (a-z) unless informed otherwise
|
|
order := "ASC"
|
|
|
|
if maxShortcodeDomain != "" {
|
|
subQuery = subQuery.Where("? > LOWER(?)", bun.Ident("shortcode_domain"), maxShortcodeDomain)
|
|
}
|
|
|
|
if minShortcodeDomain != "" {
|
|
subQuery = subQuery.Where("? < LOWER(?)", bun.Ident("shortcode_domain"), minShortcodeDomain)
|
|
// if we have a minShortcodeDomain we're paging upwards/backwards
|
|
order = "DESC"
|
|
}
|
|
|
|
subQuery = subQuery.Order("shortcode_domain " + order)
|
|
|
|
if limit > 0 {
|
|
subQuery = subQuery.Limit(limit)
|
|
}
|
|
|
|
// Wrap the subQuery in a query, since we don't need to select the shortcode_domain column.
|
|
//
|
|
// The final query will come out looking something like...
|
|
//
|
|
// SELECT
|
|
// "subquery"."emoji_ids"
|
|
// FROM (
|
|
// SELECT
|
|
// "emoji"."id" AS "emoji_ids",
|
|
// LOWER("emoji"."shortcode" || '@' || COALESCE("emoji"."domain", '')) AS "shortcode_domain"
|
|
// FROM
|
|
// "emojis" AS "emoji"
|
|
// ORDER BY
|
|
// "shortcode_domain" ASC
|
|
// ) AS "subquery"
|
|
if err := e.conn.
|
|
NewSelect().
|
|
Column("subquery.emoji_ids").
|
|
TableExpr("(?) AS ?", subQuery, bun.Ident("subquery")).
|
|
Scan(ctx, &emojiIDs); err != nil {
|
|
return nil, e.conn.ProcessError(err)
|
|
}
|
|
|
|
if order == "DESC" {
|
|
// Reverse the slice order so the caller still
|
|
// gets emojis in expected a-z alphabetical order.
|
|
//
|
|
// See https://github.com/golang/go/wiki/SliceTricks#reversing
|
|
for i := len(emojiIDs)/2 - 1; i >= 0; i-- {
|
|
opp := len(emojiIDs) - 1 - i
|
|
emojiIDs[i], emojiIDs[opp] = emojiIDs[opp], emojiIDs[i]
|
|
}
|
|
}
|
|
|
|
return e.emojisFromIDs(ctx, emojiIDs)
|
|
}
|
|
|
|
func (e *emojiDB) GetUseableEmojis(ctx context.Context) ([]*gtsmodel.Emoji, db.Error) {
|
|
emojiIDs := []string{}
|
|
|
|
q := e.conn.
|
|
NewSelect().
|
|
TableExpr("? AS ?", bun.Ident("emojis"), bun.Ident("emoji")).
|
|
Column("emoji.id").
|
|
Where("? = ?", bun.Ident("emoji.visible_in_picker"), true).
|
|
Where("? = ?", bun.Ident("emoji.disabled"), false).
|
|
Where("? IS NULL", bun.Ident("emoji.domain")).
|
|
Order("emoji.shortcode ASC")
|
|
|
|
if err := q.Scan(ctx, &emojiIDs); err != nil {
|
|
return nil, e.conn.ProcessError(err)
|
|
}
|
|
|
|
return e.emojisFromIDs(ctx, emojiIDs)
|
|
}
|
|
|
|
func (e *emojiDB) GetEmojiByID(ctx context.Context, id string) (*gtsmodel.Emoji, db.Error) {
|
|
return e.getEmoji(
|
|
ctx,
|
|
"ID",
|
|
func(emoji *gtsmodel.Emoji) error {
|
|
return e.newEmojiQ(emoji).Where("? = ?", bun.Ident("emoji.id"), id).Scan(ctx)
|
|
},
|
|
id,
|
|
)
|
|
}
|
|
|
|
func (e *emojiDB) GetEmojiByURI(ctx context.Context, uri string) (*gtsmodel.Emoji, db.Error) {
|
|
return e.getEmoji(
|
|
ctx,
|
|
"URI",
|
|
func(emoji *gtsmodel.Emoji) error {
|
|
return e.newEmojiQ(emoji).Where("? = ?", bun.Ident("emoji.uri"), uri).Scan(ctx)
|
|
},
|
|
uri,
|
|
)
|
|
}
|
|
|
|
func (e *emojiDB) GetEmojiByShortcodeDomain(ctx context.Context, shortcode string, domain string) (*gtsmodel.Emoji, db.Error) {
|
|
return e.getEmoji(
|
|
ctx,
|
|
"Shortcode.Domain",
|
|
func(emoji *gtsmodel.Emoji) error {
|
|
q := e.newEmojiQ(emoji)
|
|
|
|
if domain != "" {
|
|
q = q.Where("? = ?", bun.Ident("emoji.shortcode"), shortcode)
|
|
q = q.Where("? = ?", bun.Ident("emoji.domain"), domain)
|
|
} else {
|
|
q = q.Where("? = ?", bun.Ident("emoji.shortcode"), strings.ToLower(shortcode))
|
|
q = q.Where("? IS NULL", bun.Ident("emoji.domain"))
|
|
}
|
|
|
|
return q.Scan(ctx)
|
|
},
|
|
shortcode,
|
|
domain,
|
|
)
|
|
}
|
|
|
|
func (e *emojiDB) GetEmojiByStaticURL(ctx context.Context, imageStaticURL string) (*gtsmodel.Emoji, db.Error) {
|
|
return e.getEmoji(
|
|
ctx,
|
|
"ImageStaticURL",
|
|
func(emoji *gtsmodel.Emoji) error {
|
|
return e.
|
|
newEmojiQ(emoji).
|
|
Where("? = ?", bun.Ident("emoji.image_static_url"), imageStaticURL).
|
|
Scan(ctx)
|
|
},
|
|
imageStaticURL,
|
|
)
|
|
}
|
|
|
|
func (e *emojiDB) PutEmojiCategory(ctx context.Context, emojiCategory *gtsmodel.EmojiCategory) db.Error {
|
|
return e.categoryCache.Store(emojiCategory, func() error {
|
|
_, err := e.conn.NewInsert().Model(emojiCategory).Exec(ctx)
|
|
return e.conn.ProcessError(err)
|
|
})
|
|
}
|
|
|
|
func (e *emojiDB) GetEmojiCategories(ctx context.Context) ([]*gtsmodel.EmojiCategory, db.Error) {
|
|
emojiCategoryIDs := []string{}
|
|
|
|
q := e.conn.
|
|
NewSelect().
|
|
TableExpr("? AS ?", bun.Ident("emoji_categories"), bun.Ident("emoji_category")).
|
|
Column("emoji_category.id").
|
|
Order("emoji_category.name ASC")
|
|
|
|
if err := q.Scan(ctx, &emojiCategoryIDs); err != nil {
|
|
return nil, e.conn.ProcessError(err)
|
|
}
|
|
|
|
return e.emojiCategoriesFromIDs(ctx, emojiCategoryIDs)
|
|
}
|
|
|
|
func (e *emojiDB) GetEmojiCategory(ctx context.Context, id string) (*gtsmodel.EmojiCategory, db.Error) {
|
|
return e.getEmojiCategory(
|
|
ctx,
|
|
"ID",
|
|
func(emojiCategory *gtsmodel.EmojiCategory) error {
|
|
return e.newEmojiCategoryQ(emojiCategory).Where("? = ?", bun.Ident("emoji_category.id"), id).Scan(ctx)
|
|
},
|
|
id,
|
|
)
|
|
}
|
|
|
|
func (e *emojiDB) GetEmojiCategoryByName(ctx context.Context, name string) (*gtsmodel.EmojiCategory, db.Error) {
|
|
return e.getEmojiCategory(
|
|
ctx,
|
|
"Name",
|
|
func(emojiCategory *gtsmodel.EmojiCategory) error {
|
|
return e.newEmojiCategoryQ(emojiCategory).Where("LOWER(?) = ?", bun.Ident("emoji_category.name"), strings.ToLower(name)).Scan(ctx)
|
|
},
|
|
name,
|
|
)
|
|
}
|
|
|
|
func (e *emojiDB) getEmoji(ctx context.Context, lookup string, dbQuery func(*gtsmodel.Emoji) error, keyParts ...any) (*gtsmodel.Emoji, db.Error) {
|
|
return e.emojiCache.Load(lookup, func() (*gtsmodel.Emoji, error) {
|
|
var emoji gtsmodel.Emoji
|
|
|
|
// Not cached! Perform database query
|
|
if err := dbQuery(&emoji); err != nil {
|
|
return nil, e.conn.ProcessError(err)
|
|
}
|
|
|
|
return &emoji, nil
|
|
}, keyParts...)
|
|
}
|
|
|
|
func (e *emojiDB) emojisFromIDs(ctx context.Context, emojiIDs []string) ([]*gtsmodel.Emoji, db.Error) {
|
|
// Catch case of no emojis early
|
|
if len(emojiIDs) == 0 {
|
|
return nil, db.ErrNoEntries
|
|
}
|
|
|
|
emojis := make([]*gtsmodel.Emoji, 0, len(emojiIDs))
|
|
|
|
for _, id := range emojiIDs {
|
|
emoji, err := e.GetEmojiByID(ctx, id)
|
|
if err != nil {
|
|
log.Errorf("emojisFromIDs: error getting emoji %q: %v", id, err)
|
|
}
|
|
|
|
emojis = append(emojis, emoji)
|
|
}
|
|
|
|
return emojis, nil
|
|
}
|
|
|
|
func (e *emojiDB) getEmojiCategory(ctx context.Context, lookup string, dbQuery func(*gtsmodel.EmojiCategory) error, keyParts ...any) (*gtsmodel.EmojiCategory, db.Error) {
|
|
return e.categoryCache.Load(lookup, func() (*gtsmodel.EmojiCategory, error) {
|
|
var category gtsmodel.EmojiCategory
|
|
|
|
// Not cached! Perform database query
|
|
if err := dbQuery(&category); err != nil {
|
|
return nil, e.conn.ProcessError(err)
|
|
}
|
|
|
|
return &category, nil
|
|
}, keyParts...)
|
|
}
|
|
|
|
func (e *emojiDB) emojiCategoriesFromIDs(ctx context.Context, emojiCategoryIDs []string) ([]*gtsmodel.EmojiCategory, db.Error) {
|
|
// Catch case of no emoji categories early
|
|
if len(emojiCategoryIDs) == 0 {
|
|
return nil, db.ErrNoEntries
|
|
}
|
|
|
|
emojiCategories := make([]*gtsmodel.EmojiCategory, 0, len(emojiCategoryIDs))
|
|
|
|
for _, id := range emojiCategoryIDs {
|
|
emojiCategory, err := e.GetEmojiCategory(ctx, id)
|
|
if err != nil {
|
|
log.Errorf("emojiCategoriesFromIDs: error getting emoji category %q: %v", id, err)
|
|
}
|
|
|
|
emojiCategories = append(emojiCategories, emojiCategory)
|
|
}
|
|
|
|
return emojiCategories, nil
|
|
}
|