mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-11-22 19:56:39 +00:00
Compare commits
1 commit
df74bcf7af
...
7223c63a11
Author | SHA1 | Date | |
---|---|---|---|
7223c63a11 |
|
@ -247,54 +247,6 @@ func (suite *FilterTestSuite) TestFilterCRUD() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (suite *FilterTestSuite) TestFilterTitleOverlap() {
|
|
||||||
var (
|
|
||||||
ctx = context.Background()
|
|
||||||
account1 = "01HNEJXCPRTJVJY9MV0VVHGD47"
|
|
||||||
account2 = "01JAG5BRJPJYA0FSA5HR2MMFJH"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Create an empty filter for account 1.
|
|
||||||
account1filter1 := >smodel.Filter{
|
|
||||||
ID: "01HNEJNVZZVXJTRB3FX3K2B1YF",
|
|
||||||
AccountID: account1,
|
|
||||||
Title: "my filter",
|
|
||||||
Action: gtsmodel.FilterActionWarn,
|
|
||||||
ContextHome: util.Ptr(true),
|
|
||||||
}
|
|
||||||
if err := suite.db.PutFilter(ctx, account1filter1); err != nil {
|
|
||||||
suite.FailNow("", "error putting account1filter1: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create a filter for account 2 with
|
|
||||||
// the same title, should be no issue.
|
|
||||||
account2filter1 := >smodel.Filter{
|
|
||||||
ID: "01JAG5GPXG7H5Y4ZP78GV1F2ET",
|
|
||||||
AccountID: account2,
|
|
||||||
Title: "my filter",
|
|
||||||
Action: gtsmodel.FilterActionWarn,
|
|
||||||
ContextHome: util.Ptr(true),
|
|
||||||
}
|
|
||||||
if err := suite.db.PutFilter(ctx, account2filter1); err != nil {
|
|
||||||
suite.FailNow("", "error putting account2filter1: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Try to create another filter for
|
|
||||||
// account 1 with the same name as
|
|
||||||
// an existing filter of theirs.
|
|
||||||
account1filter2 := >smodel.Filter{
|
|
||||||
ID: "01JAG5J8NYKQE2KYCD28Y4P05V",
|
|
||||||
AccountID: account1,
|
|
||||||
Title: "my filter",
|
|
||||||
Action: gtsmodel.FilterActionWarn,
|
|
||||||
ContextHome: util.Ptr(true),
|
|
||||||
}
|
|
||||||
err := suite.db.PutFilter(ctx, account1filter2)
|
|
||||||
if !errors.Is(err, db.ErrAlreadyExists) {
|
|
||||||
suite.FailNow("", "wanted ErrAlreadyExists, got %s", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFilterTestSuite(t *testing.T) {
|
func TestFilterTestSuite(t *testing.T) {
|
||||||
suite.Run(t, new(FilterTestSuite))
|
suite.Run(t, new(FilterTestSuite))
|
||||||
}
|
}
|
||||||
|
|
|
@ -20,7 +20,7 @@
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
|
||||||
gtsmodel "github.com/superseriousbusiness/gotosocial/internal/db/bundb/migrations/20240126064004_add_filters"
|
gtsmodel "github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
||||||
"github.com/uptrace/bun"
|
"github.com/uptrace/bun"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
@ -1,65 +0,0 @@
|
||||||
// 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 gtsmodel
|
|
||||||
|
|
||||||
import (
|
|
||||||
"regexp"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Filter stores a filter created by a local account.
|
|
||||||
type Filter struct {
|
|
||||||
ID string `bun:"type:CHAR(26),pk,nullzero,notnull,unique"` // id of this item in the database
|
|
||||||
CreatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item created
|
|
||||||
UpdatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item last updated
|
|
||||||
ExpiresAt time.Time `bun:"type:timestamptz,nullzero"` // Time filter should expire. If null, should not expire.
|
|
||||||
AccountID string `bun:"type:CHAR(26),notnull,nullzero"` // ID of the local account that created the filter.
|
|
||||||
Title string `bun:",nullzero,notnull,unique"` // The name of the filter.
|
|
||||||
Action string `bun:",nullzero,notnull"` // The action to take.
|
|
||||||
Keywords []*FilterKeyword `bun:"-"` // Keywords for this filter.
|
|
||||||
Statuses []*FilterStatus `bun:"-"` // Statuses for this filter.
|
|
||||||
ContextHome *bool `bun:",nullzero,notnull,default:false"` // Apply filter to home timeline and lists.
|
|
||||||
ContextNotifications *bool `bun:",nullzero,notnull,default:false"` // Apply filter to notifications.
|
|
||||||
ContextPublic *bool `bun:",nullzero,notnull,default:false"` // Apply filter to home timeline and lists.
|
|
||||||
ContextThread *bool `bun:",nullzero,notnull,default:false"` // Apply filter when viewing a status's associated thread.
|
|
||||||
ContextAccount *bool `bun:",nullzero,notnull,default:false"` // Apply filter when viewing an account profile.
|
|
||||||
}
|
|
||||||
|
|
||||||
// FilterKeyword stores a single keyword to filter statuses against.
|
|
||||||
type FilterKeyword struct {
|
|
||||||
ID string `bun:"type:CHAR(26),pk,nullzero,notnull,unique"` // id of this item in the database
|
|
||||||
CreatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item created
|
|
||||||
UpdatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item last updated
|
|
||||||
AccountID string `bun:"type:CHAR(26),notnull,nullzero"` // ID of the local account that created the filter keyword.
|
|
||||||
FilterID string `bun:"type:CHAR(26),notnull,nullzero,unique:filter_keywords_filter_id_keyword_uniq"` // ID of the filter that this keyword belongs to.
|
|
||||||
Filter *Filter `bun:"-"` // Filter corresponding to FilterID
|
|
||||||
Keyword string `bun:",nullzero,notnull,unique:filter_keywords_filter_id_keyword_uniq"` // The keyword or phrase to filter against.
|
|
||||||
WholeWord *bool `bun:",nullzero,notnull,default:false"` // Should the filter consider word boundaries?
|
|
||||||
Regexp *regexp.Regexp `bun:"-"` // pre-prepared regular expression
|
|
||||||
}
|
|
||||||
|
|
||||||
// FilterStatus stores a single status to filter.
|
|
||||||
type FilterStatus struct {
|
|
||||||
ID string `bun:"type:CHAR(26),pk,nullzero,notnull,unique"` // id of this item in the database
|
|
||||||
CreatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item created
|
|
||||||
UpdatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item last updated
|
|
||||||
AccountID string `bun:"type:CHAR(26),notnull,nullzero"` // ID of the local account that created the filter keyword.
|
|
||||||
FilterID string `bun:"type:CHAR(26),notnull,nullzero,unique:filter_statuses_filter_id_status_id_uniq"` // ID of the filter that this keyword belongs to.
|
|
||||||
Filter *Filter `bun:"-"` // Filter corresponding to FilterID
|
|
||||||
StatusID string `bun:"type:CHAR(26),notnull,nullzero,unique:filter_statuses_filter_id_status_id_uniq"` // ID of the status to filter.
|
|
||||||
}
|
|
|
@ -1,131 +0,0 @@
|
||||||
// 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 migrations
|
|
||||||
|
|
||||||
import (
|
|
||||||
"context"
|
|
||||||
|
|
||||||
"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
|
|
||||||
"github.com/uptrace/bun"
|
|
||||||
"github.com/uptrace/bun/dialect"
|
|
||||||
)
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
up := func(ctx context.Context, db *bun.DB) error {
|
|
||||||
return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
|
|
||||||
|
|
||||||
// Create the new filters table
|
|
||||||
// with the unique constraint
|
|
||||||
// set on AccountID + Title.
|
|
||||||
if _, err := tx.
|
|
||||||
NewCreateTable().
|
|
||||||
ModelTableExpr("new_filters").
|
|
||||||
Model((*gtsmodel.Filter)(nil)).
|
|
||||||
Exec(ctx); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Explicitly specify columns to bring
|
|
||||||
// from old table to new, to avoid any
|
|
||||||
// potential Postgres shenanigans.
|
|
||||||
columns := []string{
|
|
||||||
"id",
|
|
||||||
"created_at",
|
|
||||||
"updated_at",
|
|
||||||
"expires_at",
|
|
||||||
"account_id",
|
|
||||||
"title",
|
|
||||||
"action",
|
|
||||||
"context_home",
|
|
||||||
"context_notifications",
|
|
||||||
"context_public",
|
|
||||||
"context_thread",
|
|
||||||
"context_account",
|
|
||||||
}
|
|
||||||
|
|
||||||
// Copy all data for existing
|
|
||||||
// filters to the new table.
|
|
||||||
if _, err := tx.
|
|
||||||
NewInsert().
|
|
||||||
Table("new_filters").
|
|
||||||
Table("filters").
|
|
||||||
Column(columns...).
|
|
||||||
Exec(ctx); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Drop the old table.
|
|
||||||
if _, err := tx.
|
|
||||||
NewDropTable().
|
|
||||||
Table("filters").
|
|
||||||
Exec(ctx); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Rename new table to old table.
|
|
||||||
if _, err := tx.
|
|
||||||
ExecContext(
|
|
||||||
ctx,
|
|
||||||
"ALTER TABLE ? RENAME TO ?",
|
|
||||||
bun.Ident("new_filters"),
|
|
||||||
bun.Ident("filters"),
|
|
||||||
); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Index the new version
|
|
||||||
// of the filters table.
|
|
||||||
if _, err := tx.
|
|
||||||
NewCreateIndex().
|
|
||||||
Table("filters").
|
|
||||||
Index("filters_account_id_idx").
|
|
||||||
Column("account_id").
|
|
||||||
IfNotExists().
|
|
||||||
Exec(ctx); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
if db.Dialect().Name() == dialect.PG {
|
|
||||||
// Rename "new_filters_pkey" from the
|
|
||||||
// new table to just "filters_pkey".
|
|
||||||
// This is only necessary on Postgres.
|
|
||||||
if _, err := tx.ExecContext(
|
|
||||||
ctx,
|
|
||||||
"ALTER TABLE ? RENAME CONSTRAINT ? TO ?",
|
|
||||||
bun.Ident("public.filters"),
|
|
||||||
bun.Safe("new_filters_pkey"),
|
|
||||||
bun.Safe("filters_pkey"),
|
|
||||||
); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
down := func(ctx context.Context, db *bun.DB) error {
|
|
||||||
return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := Migrations.Register(up, down); err != nil {
|
|
||||||
panic(err)
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -26,20 +26,20 @@
|
||||||
|
|
||||||
// Filter stores a filter created by a local account.
|
// Filter stores a filter created by a local account.
|
||||||
type Filter struct {
|
type Filter struct {
|
||||||
ID string `bun:"type:CHAR(26),pk,nullzero,notnull,unique"` // id of this item in the database
|
ID string `bun:"type:CHAR(26),pk,nullzero,notnull,unique"` // id of this item in the database
|
||||||
CreatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item created
|
CreatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item created
|
||||||
UpdatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item last updated
|
UpdatedAt time.Time `bun:"type:timestamptz,nullzero,notnull,default:current_timestamp"` // when was item last updated
|
||||||
ExpiresAt time.Time `bun:"type:timestamptz,nullzero"` // Time filter should expire. If null, should not expire.
|
ExpiresAt time.Time `bun:"type:timestamptz,nullzero"` // Time filter should expire. If null, should not expire.
|
||||||
AccountID string `bun:"type:CHAR(26),notnull,nullzero,unique:filters_account_id_title_uniq"` // ID of the local account that created the filter.
|
AccountID string `bun:"type:CHAR(26),notnull,nullzero"` // ID of the local account that created the filter.
|
||||||
Title string `bun:",nullzero,notnull,unique:filters_account_id_title_uniq"` // The name of the filter.
|
Title string `bun:",nullzero,notnull,unique"` // The name of the filter.
|
||||||
Action FilterAction `bun:",nullzero,notnull"` // The action to take.
|
Action FilterAction `bun:",nullzero,notnull"` // The action to take.
|
||||||
Keywords []*FilterKeyword `bun:"-"` // Keywords for this filter.
|
Keywords []*FilterKeyword `bun:"-"` // Keywords for this filter.
|
||||||
Statuses []*FilterStatus `bun:"-"` // Statuses for this filter.
|
Statuses []*FilterStatus `bun:"-"` // Statuses for this filter.
|
||||||
ContextHome *bool `bun:",nullzero,notnull,default:false"` // Apply filter to home timeline and lists.
|
ContextHome *bool `bun:",nullzero,notnull,default:false"` // Apply filter to home timeline and lists.
|
||||||
ContextNotifications *bool `bun:",nullzero,notnull,default:false"` // Apply filter to notifications.
|
ContextNotifications *bool `bun:",nullzero,notnull,default:false"` // Apply filter to notifications.
|
||||||
ContextPublic *bool `bun:",nullzero,notnull,default:false"` // Apply filter to home timeline and lists.
|
ContextPublic *bool `bun:",nullzero,notnull,default:false"` // Apply filter to home timeline and lists.
|
||||||
ContextThread *bool `bun:",nullzero,notnull,default:false"` // Apply filter when viewing a status's associated thread.
|
ContextThread *bool `bun:",nullzero,notnull,default:false"` // Apply filter when viewing a status's associated thread.
|
||||||
ContextAccount *bool `bun:",nullzero,notnull,default:false"` // Apply filter when viewing an account profile.
|
ContextAccount *bool `bun:",nullzero,notnull,default:false"` // Apply filter when viewing an account profile.
|
||||||
}
|
}
|
||||||
|
|
||||||
// Expired returns whether the filter has expired at a given time.
|
// Expired returns whether the filter has expired at a given time.
|
||||||
|
|
Loading…
Reference in a new issue