diff options
author | 2023-10-25 16:04:53 +0200 | |
---|---|---|
committer | 2023-10-25 15:04:53 +0100 | |
commit | c7b6cd7770cad9bfdc23decffa7c4068752dbbbd (patch) | |
tree | 0f039fd34fb0287860fce06ff1c30dedd1882136 /internal/db/bundb/thread.go | |
parent | [bugfix] allow store smaller PNG image than 261 bytes (#2263) (#2298) (diff) | |
download | gotosocial-c7b6cd7770cad9bfdc23decffa7c4068752dbbbd.tar.xz |
[feature] Status thread mute/unmute functionality (#2278)
* add db models + functions for keeping track of threads
* give em the old linty testy
* create, remove, check mutes
* swagger
* testerino
* test mute/unmute via api
* add info log about new index creation
* thread + allow muting of any remote statuses that mention a local account
* IsStatusThreadMutedBy -> IsThreadMutedByAccount
* use common processing functions in status processor
* set = NULL
* favee!
* get rekt darlings, darlings get rekt
* testrig please, have mercy muy liege
Diffstat (limited to 'internal/db/bundb/thread.go')
-rw-r--r-- | internal/db/bundb/thread.go | 117 |
1 files changed, 117 insertions, 0 deletions
diff --git a/internal/db/bundb/thread.go b/internal/db/bundb/thread.go new file mode 100644 index 000000000..e6d6154d4 --- /dev/null +++ b/internal/db/bundb/thread.go @@ -0,0 +1,117 @@ +// GoToSocial +// Copyright (C) GoToSocial Authors admin@gotosocial.org +// SPDX-License-Identifier: AGPL-3.0-or-later +// +// 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" + "errors" + + "github.com/superseriousbusiness/gotosocial/internal/db" + "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" + "github.com/superseriousbusiness/gotosocial/internal/state" + "github.com/uptrace/bun" +) + +type threadDB struct { + db *DB + state *state.State +} + +func (t *threadDB) PutThread(ctx context.Context, thread *gtsmodel.Thread) error { + _, err := t.db. + NewInsert(). + Model(thread). + Exec(ctx) + + return err +} + +func (t *threadDB) GetThreadMute(ctx context.Context, id string) (*gtsmodel.ThreadMute, error) { + return t.state.Caches.GTS.ThreadMute().Load("ID", func() (*gtsmodel.ThreadMute, error) { + var threadMute gtsmodel.ThreadMute + + q := t.db. + NewSelect(). + Model(&threadMute). + Where("? = ?", bun.Ident("thread_mute.id"), id) + + if err := q.Scan(ctx); err != nil { + return nil, err + } + + return &threadMute, nil + }, id) +} + +func (t *threadDB) GetThreadMutedByAccount( + ctx context.Context, + threadID string, + accountID string, +) (*gtsmodel.ThreadMute, error) { + return t.state.Caches.GTS.ThreadMute().Load("ThreadID.AccountID", func() (*gtsmodel.ThreadMute, error) { + var threadMute gtsmodel.ThreadMute + + q := t.db. + NewSelect(). + Model(&threadMute). + Where("? = ?", bun.Ident("thread_mute.thread_id"), threadID). + Where("? = ?", bun.Ident("thread_mute.account_id"), accountID) + + if err := q.Scan(ctx); err != nil { + return nil, err + } + + return &threadMute, nil + }, threadID, accountID) +} + +func (t *threadDB) IsThreadMutedByAccount( + ctx context.Context, + threadID string, + accountID string, +) (bool, error) { + if threadID == "" { + return false, nil + } + + mute, err := t.GetThreadMutedByAccount(ctx, threadID, accountID) + if err != nil && !errors.Is(err, db.ErrNoEntries) { + return false, err + } + + return (mute != nil), nil +} + +func (t *threadDB) PutThreadMute(ctx context.Context, threadMute *gtsmodel.ThreadMute) error { + return t.state.Caches.GTS.ThreadMute().Store(threadMute, func() error { + _, err := t.db.NewInsert().Model(threadMute).Exec(ctx) + return err + }) +} + +func (t *threadDB) DeleteThreadMute(ctx context.Context, id string) error { + if _, err := t.db. + NewDelete(). + TableExpr("? AS ?", bun.Ident("thread_mutes"), bun.Ident("thread_mute")). + Where("? = ?", bun.Ident("thread_mute.id"), id).Exec(ctx); err != nil { + return err + } + + t.state.Caches.GTS.ThreadMute().Invalidate("ID", id) + return nil +} |