Skip to content

Commit

Permalink
fix: etc. etc.
Browse files Browse the repository at this point in the history
  • Loading branch information
soedirgo committed Dec 9, 2021
1 parent 1185d8f commit 3a3f20b
Show file tree
Hide file tree
Showing 6 changed files with 321 additions and 215 deletions.
49 changes: 48 additions & 1 deletion internal/db/changes/changes.go
Original file line number Diff line number Diff line change
Expand Up @@ -163,12 +163,59 @@ func run(p *tea.Program) error {
return errors.New("Error creating shadow database: " + errBuf.String())
}

{
out, err := utils.DockerExec(ctx, utils.DbId, []string{
"sh", "-c", "psql --username postgres --host localhost --dbname '" + utils.ShadowDbName + `' <<'EOSQL'
BEGIN;
` + utils.InitialSchemaSql + `
COMMIT;
EOSQL
`,
})
if err != nil {
return err
}
if err := utils.ProcessPsqlOutput(out, p); err != nil {
return err
}
}

{
extensionsSql, err := os.ReadFile("supabase/extensions.sql")
if errors.Is(err, os.ErrNotExist) {
// skip
} else if err != nil {
return err
} else {
out, err := utils.DockerExec(ctx, utils.DbId, []string{
"sh", "-c", "psql --username postgres --host localhost --dbname '" + utils.ShadowDbName + `' <<'EOSQL'
BEGIN;
` + string(extensionsSql) + `
COMMIT;
EOSQL
`,
})
if err != nil {
return err
}
if err := utils.ProcessPsqlOutput(out, p); err != nil {
return err
}
}
}

migrations, err := os.ReadDir("supabase/migrations")
if err != nil {
return err
}

for _, migration := range migrations {
for i, migration := range migrations {
// NOTE: To handle backward-compatibility. `<timestamp>_init.sql` as
// the first migration (prev versions of the CLI) is deprecated.
if i == 0 && strings.HasSuffix(migration.Name(), "_init.sql") {
continue
}

p.Send(utils.StatusMsg("Applying migration " + utils.Bold(migration.Name()) + "..."))

content, err := os.ReadFile("supabase/migrations/" + migration.Name())
Expand Down
49 changes: 48 additions & 1 deletion internal/db/commit/commit.go
Original file line number Diff line number Diff line change
Expand Up @@ -164,12 +164,59 @@ func run(p *tea.Program, name string) error {
return errors.New("Error creating shadow database: " + errBuf.String())
}

{
out, err := utils.DockerExec(ctx, utils.DbId, []string{
"sh", "-c", "psql --username postgres --host localhost --dbname '" + utils.ShadowDbName + `' <<'EOSQL'
BEGIN;
` + utils.InitialSchemaSql + `
COMMIT;
EOSQL
`,
})
if err != nil {
return err
}
if err := utils.ProcessPsqlOutput(out, p); err != nil {
return err
}
}

{
extensionsSql, err := os.ReadFile("supabase/extensions.sql")
if errors.Is(err, os.ErrNotExist) {
// skip
} else if err != nil {
return err
} else {
out, err := utils.DockerExec(ctx, utils.DbId, []string{
"sh", "-c", "psql --username postgres --host localhost --dbname '" + utils.ShadowDbName + `' <<'EOSQL'
BEGIN;
` + string(extensionsSql) + `
COMMIT;
EOSQL
`,
})
if err != nil {
return err
}
if err := utils.ProcessPsqlOutput(out, p); err != nil {
return err
}
}
}

migrations, err := os.ReadDir("supabase/migrations")
if err != nil {
return err
}

for _, migration := range migrations {
for i, migration := range migrations {
// NOTE: To handle backward-compatibility. `<timestamp>_init.sql` as
// the first migration (prev versions of the CLI) is deprecated.
if i == 0 && strings.HasSuffix(migration.Name(), "_init.sql") {
continue
}

p.Send(utils.StatusMsg("Applying migration " + utils.Bold(migration.Name()) + "..."))

content, err := os.ReadFile("supabase/migrations/" + migration.Name())
Expand Down
82 changes: 46 additions & 36 deletions internal/db/remote/commit/commit.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ import (
"io"
"os"
"regexp"
"strconv"
"strings"

"github.com/charmbracelet/bubbles/progress"
Expand Down Expand Up @@ -201,31 +202,31 @@ func run(p *tea.Program, url string) error {
if rows, err := conn.Query(ctx, "SELECT version FROM supabase_migrations.schema_migrations ORDER BY version"); err != nil {
return err
} else {
versions := []string{}
remoteMigrations := []string{}
for rows.Next() {
var version string
if err := rows.Scan(&version); err != nil {
return err
}
versions = append(versions, version)
remoteMigrations = append(remoteMigrations, version)
}

migrations, err := os.ReadDir("supabase/migrations")
localMigrations, err := os.ReadDir("supabase/migrations")
if err != nil {
return err
}

conflictErr := errors.New("supabase_migrations.schema_migrations table is not in sync with the contents of " + utils.Bold("supabase/migrations") + ".")

if len(versions) != len(migrations) {
if len(remoteMigrations) != len(localMigrations) {
return conflictErr
}

re := regexp.MustCompile(`([0-9]+)_.*\.sql`)
for i, version := range versions {
migrationTimestamp := re.FindStringSubmatch(migrations[i].Name())[1]
for i, remoteTimestamp := range remoteMigrations {
localTimestamp := re.FindStringSubmatch(localMigrations[i].Name())[1]

if version == migrationTimestamp {
if localTimestamp == remoteTimestamp {
continue
}

Expand All @@ -234,27 +235,37 @@ func run(p *tea.Program, url string) error {
}

// 2. Create shadow db and run migrations.
p.Send(utils.StatusMsg("Creating shadow database..."))
{
p.Send(utils.StatusMsg("Creating shadow database..."))
cmd := []string{}
if dbVersion, err := strconv.ParseUint(utils.DbVersion, 10, 64); err != nil {
return err
} else if dbVersion >= 140000 {
cmd = []string{"postgres", "-c", "config_file=/etc/postgresql/postgresql.conf"}
}

if _, err := utils.DockerRun(
ctx,
dbId,
&container.Config{
Image: utils.DbImage,
Env: []string{"POSTGRES_PASSWORD=postgres"},
Cmd: []string{"postgres", "-c", "wal_level=logical"},
},
&container.Config{Image: utils.DbImage, Env: []string{"POSTGRES_PASSWORD=postgres"}, Cmd: cmd},
&container.HostConfig{NetworkMode: netId},
); err != nil {
return err
}

out, err := utils.DockerExec(ctx, dbId, []string{
"sh", "-c", "until pg_isready --host $(hostname --ip-address); do sleep 0.1; done",
"sh", "-c", "until pg_isready --host $(hostname --ip-address); do sleep 0.1; done " +
`&& psql --username postgres --host localhost <<'EOSQL'
BEGIN;
` + utils.GlobalsSql + `
COMMIT;
EOSQL
`,
})
if err != nil {
return err
}

var errBuf bytes.Buffer
if _, err := stdcopy.StdCopy(io.Discard, &errBuf, out); err != nil {
return err
Expand All @@ -264,17 +275,10 @@ func run(p *tea.Program, url string) error {
}

{
globalsSql, err := os.ReadFile("supabase/globals.sql")
if errors.Is(err, os.ErrNotExist) {
return errors.New("Cannot find " + utils.Bold("supabase/globals.sql") + ".")
} else if err != nil {
return err
}

out, err := utils.DockerExec(ctx, dbId, []string{
"sh", "-c", `psql --username postgres --host localhost --dbname postgres <<'EOSQL'
"sh", "-c", `psql --username postgres --host localhost <<'EOSQL'
BEGIN;
` + string(globalsSql) + `
` + utils.InitialSchemaSql + `
COMMIT;
EOSQL
`,
Expand All @@ -290,24 +294,24 @@ EOSQL
{
extensionsSql, err := os.ReadFile("supabase/extensions.sql")
if errors.Is(err, os.ErrNotExist) {
return errors.New("Cannot find " + utils.Aqua("supabase/extensions.sql") + ".")
// skip
} else if err != nil {
return err
}

out, err := utils.DockerExec(ctx, dbId, []string{
"sh", "-c", `psql --username postgres --host localhost --dbname postgres <<'EOSQL'
} else {
out, err := utils.DockerExec(ctx, dbId, []string{
"sh", "-c", `psql --username postgres --host localhost <<'EOSQL'
BEGIN;
` + string(extensionsSql) + `
COMMIT;
EOSQL
`,
})
if err != nil {
return err
}
if err := utils.ProcessPsqlOutput(out, p); err != nil {
return err
})
if err != nil {
return err
}
if err := utils.ProcessPsqlOutput(out, p); err != nil {
return err
}
}
}

Expand All @@ -316,7 +320,13 @@ EOSQL
return err
}

for _, migration := range migrations {
for i, migration := range migrations {
// NOTE: To handle backward-compatibility. `<timestamp>_init.sql` as
// the first migration (prev versions of the CLI) is deprecated.
if i == 0 && strings.HasSuffix(migration.Name(), "_init.sql") {
continue
}

p.Send(utils.StatusMsg("Applying migration " + utils.Bold(migration.Name()) + "..."))

content, err := os.ReadFile("supabase/migrations/" + migration.Name())
Expand All @@ -325,7 +335,7 @@ EOSQL
}

out, err := utils.DockerExec(ctx, dbId, []string{
"sh", "-c", `psql --username postgres --host localhost --dbname postgres <<'EOSQL'
"sh", "-c", `psql --username postgres --host localhost <<'EOSQL'
BEGIN;
` + string(content) + `
COMMIT;
Expand Down
47 changes: 11 additions & 36 deletions internal/db/remote/set/set.go
Original file line number Diff line number Diff line change
Expand Up @@ -35,74 +35,49 @@ func Run(url string) error {

// 2. Setup & validate `schema_migrations`.

// If `schema_migrations` doesn't exist on the remote database, create it
// and insert the timestamp for the init migration.
// If `schema_migrations` doesn't exist on the remote database, create it.
if _, err := conn.Exec(ctx, "SELECT 1 FROM supabase_migrations.schema_migrations"); err != nil {
tx, err := conn.BeginTx(ctx, pgx.TxOptions{})
if err != nil {
return err
}
defer tx.Rollback(context.Background()) //nolint:errcheck

if _, err := tx.Exec(
if _, err := conn.Exec(
ctx,
`CREATE SCHEMA IF NOT EXISTS supabase_migrations;
CREATE TABLE supabase_migrations.schema_migrations (version text NOT NULL PRIMARY KEY);
`,
); err != nil {
return err
}

migrations, err := os.ReadDir("supabase/migrations")
if err != nil {
return err
}

re := regexp.MustCompile(`([0-9]+)_.*\.sql`)
migrationTimestamp := re.FindStringSubmatch(migrations[0].Name())[1]
if _, err := tx.Exec(
ctx,
"INSERT INTO supabase_migrations.schema_migrations(version) VALUES($1);",
migrationTimestamp,
); err != nil {
return err
}

if err := tx.Commit(ctx); err != nil {
return err
}
}
// If `migrations` is not a "prefix" of list of migrations in repo, fail &

// If `schema_migrations` is not a "prefix" of list of migrations in repo, fail &
// warn user.
rows, err := conn.Query(ctx, "SELECT version FROM supabase_migrations.schema_migrations ORDER BY version")
if err != nil {
return err
} else {
var versions []string
var remoteMigrations []string
for rows.Next() {
var version string
if err := rows.Scan(&version); err != nil {
return err
}
versions = append(versions, version)
remoteMigrations = append(remoteMigrations, version)
}

migrations, err := os.ReadDir("supabase/migrations")
localMigrations, err := os.ReadDir("supabase/migrations")
if err != nil {
return err
}

conflictErr := errors.New("supabase_migrations.schema_migrations table conflicts with the contents of " + utils.Bold("supabase/migrations") + ".")

if len(versions) > len(migrations) {
if len(remoteMigrations) > len(localMigrations) {
return conflictErr
}

re := regexp.MustCompile(`([0-9]+)_.*\.sql`)
for i, version := range versions {
migrationTimestamp := re.FindStringSubmatch(migrations[i].Name())[1]
for i, remoteTimestamp := range remoteMigrations {
localTimestamp := re.FindStringSubmatch(localMigrations[i].Name())[1]

if version == migrationTimestamp {
if localTimestamp == remoteTimestamp {
continue
}

Expand Down
Loading

0 comments on commit 3a3f20b

Please sign in to comment.