go.charczuk.com@v0.0.0-20240327042549-bc490516bd1a/projects/chirp/pkg/model/migrations.go

/*

Copyright (c) 2023 - Present. Will Charczuk. All rights reserved.
Use of this source code is governed by a MIT license that can be found in the LICENSE file at the root of the repository.

*/

package model

import (
	"context"
	"database/sql"

	"go.charczuk.com/sdk/apputil"
	"go.charczuk.com/sdk/db"
	"go.charczuk.com/sdk/db/dbgen"
	"go.charczuk.com/sdk/db/migration"
	"go.charczuk.com/sdk/uuid"
)

// Migrations returns the migration suite to bootstrap the database.
func Migrations(opts ...migration.SuiteOption) *migration.Suite {
	return migration.New(
		append(opts,
			migration.OptGroups(
				append(
					apputil.MigrationGroups(), // this includes e.g. `User` and `Session`
					migration.NewGroupWithAction(
						dbgen.TableFrom(UserInfo{},
							dbgen.ForeignKey(UserInfo{}, "user_id", apputil.User{}, "id"),
						),
					),
					migration.NewGroupWithAction(
						dbgen.TableFrom(Chirp{},
							dbgen.ForeignKey(Chirp{}, "user_id", apputil.User{}, "id"),
							dbgen.ForeignKey(Chirp{}, "reply_id", Chirp{}, "id"),
							dbgen.ForeignKey(Chirp{}, "quoted_id", Chirp{}, "id"),
						),
					),
					migration.NewGroupWithAction(
						dbgen.TableFrom(ChirpStats{},
							dbgen.ForeignKey(ChirpStats{}, "chirp_id", Chirp{}, "id"),
							dbgen.ForeignKey(ChirpStats{}, "user_id", apputil.User{}, "id"),
						),
					),
					migration.NewGroupWithAction(
						dbgen.TableFrom(ChirpLike{},
							dbgen.ForeignKey(ChirpLike{}, "chirp_id", Chirp{}, "id"),
							dbgen.ForeignKey(ChirpLike{}, "user_id", apputil.User{}, "id"),
						),
					),
					migration.NewGroupWithAction(
						dbgen.TableFrom(Graph{},
							dbgen.ForeignKey(Graph{}, "user_id", apputil.User{}, "id"),
							dbgen.ForeignKey(Graph{}, "target_id", apputil.User{}, "id"),
						),
					),
					migration.NewGroupWithAction(
						dbgen.TableFrom(Feed{},
							dbgen.ForeignKey(Feed{}, "user_id", apputil.User{}, "id"),
							dbgen.ForeignKey(Feed{}, "chirp_id", Chirp{}, "id"),
						),
					),
					migration.NewGroupWithAction(
						dbgen.TableFrom(Notification{},
							dbgen.ForeignKey(Notification{}, "user_id", apputil.User{}, "id"),
							dbgen.ForeignKey(Notification{}, "acting_user_id", apputil.User{}, "id"),
							dbgen.ForeignKey(Notification{}, "chirp_id", Chirp{}, "id"),
						),
					),
					migration.NewGroupWithAction(
						dbgen.TableFrom(Audit{},
							dbgen.ForeignKey(Audit{}, "user_id", apputil.User{}, "id"),
						),
					),
					migration.NewGroupWithStep(
						migration.IfNotExists(`select 1 from users where email = $1`, "will.charczuk@gmail.com"),
						migration.ActionFunc(func(ctx context.Context, conn *db.Connection, tx *sql.Tx) error {
							user := &apputil.User{
								ID:         uuid.MustParse("476a78aa-8c46-45f6-aca4-dd52384d60bd"),
								GivenName:  "Will",
								FamilyName: "Charczuk",
								Email:      "will.charczuk@gmail.com",
								PictureURL: "https://lh3.googleusercontent.com/a/AAcHTte1Yp7G57Gs5c_BGXZGXVQVuXQBvvq-Uvyo573Awi9S5Q=s96-c",
								Locale:     "en",
							}
							if err := conn.Invoke(db.OptContext(ctx), db.OptTx(tx)).Create(user); err != nil {
								return err
							}
							if err := conn.Invoke(db.OptContext(ctx), db.OptTx(tx)).Create(&UserInfo{
								UserID:  user.ID,
								IsAdmin: true,
							}); err != nil {
								return err
							}
							return nil
						}),
					),
					migration.NewGroupWithStep(
						migration.ConstraintExists("chirp_like", "pk_chirp_like_chirp_id"),
						migration.Statements(
							`ALTER TABLE chirp_like DROP CONSTRAINT pk_chirp_like_chirp_id`,
							`ALTER TABLE chirp_like ADD CONSTRAINT pk_chirp_like_chirp_id_user_id PRIMARY KEY (chirp_id, user_id)`,
						),
					),
				)...,
			),
		)...,
	)
}