gtsocial-umbx

commit 99ddaba57e998fd990a3a900ea7867d7b438573c
parent c7702c47bb887a220233fd92d80a1084a25f2a27
Author: tobi <31960611+tsmethurst@users.noreply.github.com>
Date:   Sun, 21 May 2023 13:45:24 +0200

[bugfix] Fix incorrect default for empty emoji domain (#1803)


Diffstat:
A
internal/db/bundb/migrations/20230521105850_emoji_empty_domain_fix.go
 | 
154
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

1 file changed, 154 insertions(+), 0 deletions(-)
diff --git a/internal/db/bundb/migrations/20230521105850_emoji_empty_domain_fix.go b/internal/db/bundb/migrations/20230521105850_emoji_empty_domain_fix.go
@@ -0,0 +1,154 @@
+// 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 {
+			// SQLite doesn't mind creating multiple constraints with
+			// the same name, but Postgres balks at it, so remove
+			// constraints before we go editing the emoji tables.
+			if tx.Dialect().Name() == dialect.PG {
+				for _, constraint := range []string{
+					"shortcodedomain", // initial constraint name
+					"domainshortcode", // later constraint name
+				} {
+					if _, err := tx.ExecContext(
+						ctx,
+						"ALTER TABLE ? DROP CONSTRAINT IF EXISTS ?",
+						bun.Ident("emojis"),
+						bun.Safe(constraint),
+					); err != nil {
+						return err
+					}
+				}
+			}
+
+			// Set all existing emoji domains to null
+			// where domain is '', to fix the bug!
+			if _, err := tx.
+				NewUpdate().
+				Table("emojis").
+				Set("? = NULL", bun.Ident("domain")).
+				Where("? = ?", bun.Ident("domain"), "").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			// Create the new emojis table.
+			if _, err := tx.
+				NewCreateTable().
+				ModelTableExpr("new_emojis").
+				Model(&gtsmodel.Emoji{}).
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			// Specify columns explicitly to
+			// avoid any Postgres shenanigans.
+			columns := []string{
+				"id",
+				"created_at",
+				"updated_at",
+				"shortcode",
+				"domain",
+				"image_remote_url",
+				"image_static_remote_url",
+				"image_url",
+				"image_static_url",
+				"image_path",
+				"image_static_path",
+				"image_content_type",
+				"image_static_content_type",
+				"image_file_size",
+				"image_static_file_size",
+				"image_updated_at",
+				"disabled",
+				"uri",
+				"visible_in_picker",
+				"category_id",
+			}
+
+			// Copy existing emojis to the new table.
+			if _, err := tx.
+				NewInsert().
+				Table("new_emojis").
+				Table("emojis").
+				Column(columns...).
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			// Drop the old table.
+			if _, err := tx.
+				NewDropTable().
+				Table("emojis").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			// Rename new table to old table.
+			if _, err := tx.
+				ExecContext(
+					ctx,
+					"ALTER TABLE ? RENAME TO ?",
+					bun.Ident("new_emojis"),
+					bun.Ident("emojis"),
+				); err != nil {
+				return err
+			}
+
+			// Add indexes to the new table.
+			for index, columns := range map[string][]string{
+				"emojis_id_idx":               {"id"},
+				"emojis_available_custom_idx": {"visible_in_picker", "disabled", "shortcode"},
+				"emojis_image_static_url_idx": {"image_static_url"},
+				"emojis_uri_idx":              {"uri"},
+			} {
+				if _, err := tx.
+					NewCreateIndex().
+					Table("emojis").
+					Index(index).
+					Column(columns...).
+					Exec(ctx); 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)
+	}
+}