1 files changed, 167 insertions, 0 deletions
diff --git a/internal/db/bundb/migrations/20230328105630_chore_refactoring.go b/internal/db/bundb/migrations/20230328105630_chore_refactoring.go
new file mode 100644
index 000000000..3bf9d59ef
--- /dev/null
+++ b/internal/db/bundb/migrations/20230328105630_chore_refactoring.go
@@ -0,0 +1,167 @@
+// 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"
+)
+
+func init() {
+	up := func(ctx context.Context, db *bun.DB) error {
+		// To update unique constraint on public key, we need to migrate accounts into a new table.
+		// See section 7 here: https://www.sqlite.org/lang_altertable.html
+
+		return db.RunInTx(ctx, nil, func(ctx context.Context, tx bun.Tx) error {
+			// Create the new accounts table.
+			if _, err := tx.
+				NewCreateTable().
+				ModelTableExpr("new_accounts").
+				Model(&gtsmodel.Account{}).
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			// If we don't specify columns explicitly,
+			// Postgres gives the following error when
+			// transferring accounts to new_accounts:
+			//
+			//	ERROR:  column "fetched_at" is of type timestamp with time zone but expression is of type character varying at character 35
+			//	HINT:  You will need to rewrite or cast the expression.
+			//
+			// Rather than do funky casting to fix this,
+			// it's simpler to just specify all columns.
+			columns := []string{
+				"id",
+				"created_at",
+				"updated_at",
+				"fetched_at",
+				"username",
+				"domain",
+				"avatar_media_attachment_id",
+				"avatar_remote_url",
+				"header_media_attachment_id",
+				"header_remote_url",
+				"display_name",
+				"emojis",
+				"fields",
+				"note",
+				"note_raw",
+				"memorial",
+				"also_known_as",
+				"moved_to_account_id",
+				"bot",
+				"reason",
+				"locked",
+				"discoverable",
+				"privacy",
+				"sensitive",
+				"language",
+				"status_content_type",
+				"custom_css",
+				"uri",
+				"url",
+				"inbox_uri",
+				"shared_inbox_uri",
+				"outbox_uri",
+				"following_uri",
+				"followers_uri",
+				"featured_collection_uri",
+				"actor_type",
+				"private_key",
+				"public_key",
+				"public_key_uri",
+				"sensitized_at",
+				"silenced_at",
+				"suspended_at",
+				"hide_collections",
+				"suspension_origin",
+				"enable_rss",
+			}
+
+			// Copy all accounts to the new table.
+			if _, err := tx.
+				NewInsert().
+				Table("new_accounts").
+				Table("accounts").
+				Column(columns...).
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			// Drop the old table.
+			if _, err := tx.
+				NewDropTable().
+				Table("accounts").
+				Exec(ctx); err != nil {
+				return err
+			}
+
+			// Rename new table to old table.
+			if _, err := tx.
+				ExecContext(
+					ctx,
+					"ALTER TABLE ? RENAME TO ?",
+					bun.Ident("new_accounts"),
+					bun.Ident("accounts"),
+				); err != nil {
+				return err
+			}
+
+			// Add all account indexes to the new table.
+			for index, columns := range map[string][]string{
+				// Standard indices.
+				"accounts_id_idx":              {"id"},
+				"accounts_suspended_at_idx":    {"suspended_at"},
+				"accounts_domain_idx":          {"domain"},
+				"accounts_username_domain_idx": {"username", "domain"},
+				// URI indices.
+				"accounts_uri_idx":            {"uri"},
+				"accounts_url_idx":            {"url"},
+				"accounts_inbox_uri_idx":      {"inbox_uri"},
+				"accounts_outbox_uri_idx":     {"outbox_uri"},
+				"accounts_followers_uri_idx":  {"followers_uri"},
+				"accounts_following_uri_idx":  {"following_uri"},
+				"accounts_public_key_uri_idx": {"public_key_uri"},
+			} {
+				if _, err := tx.
+					NewCreateIndex().
+					Table("accounts").
+					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)
+	}
+}