feat: start local database from schema files (#4774)

* feat: start local database from schema files

* fix: do not fail on missing declarative schema

Author: sweatybridge

internal/db/diff/diff.go

@@ -157,14 +157,14 @@ func DiffDatabase(ctx context.Context, schema []string, config pgconn.Config, w
 		Database: "postgres",
 	}
 	if utils.IsLocalDatabase(config) {
-		if declared, err := loadDeclaredSchemas(fsys); err != nil {
-			return "", err
-		} else if len(declared) > 0 {
+		if declared, err := loadDeclaredSchemas(fsys); len(declared) > 0 {
 			config = shadowConfig
 			config.Database = "contrib_regression"
 			if err := migrateBaseDatabase(ctx, config, declared, fsys, options...); err != nil {
 				return "", err
 			}
+		} else if err != nil {
+			return "", err
 		}
 	}
 	// Load all user defined schemas

internal/migration/apply/apply.go

@@ -2,16 +2,23 @@ package apply
 
 import (
 	"context"
+	"fmt"
+	"io/fs"
 
 	"github.com/jackc/pgx/v4"
 	"github.com/spf13/afero"
+	"github.com/spf13/viper"
 	"github.com/supabase/cli/internal/migration/list"
 	"github.com/supabase/cli/internal/utils"
 	"github.com/supabase/cli/pkg/migration"
 )
 
 func MigrateAndSeed(ctx context.Context, version string, conn *pgx.Conn, fsys afero.Fs) error {
-	if err := applyMigrationFiles(ctx, version, conn, fsys); err != nil {
+	if viper.GetBool("EXPERIMENTAL") && len(version) == 0 {
+		if err := applySchemaFiles(ctx, conn, afero.NewIOFS(fsys)); err != nil {
+			return err
+		}
+	} else if err := applyMigrationFiles(ctx, version, conn, fsys); err != nil {
 		return err
 	}
 	return applySeedFiles(ctx, conn, fsys)
@@ -38,3 +45,22 @@ func applySeedFiles(ctx context.Context, conn *pgx.Conn, fsys afero.Fs) error {
 	}
 	return migration.SeedData(ctx, seeds, conn, afero.NewIOFS(fsys))
 }
+
+func applySchemaFiles(ctx context.Context, conn *pgx.Conn, fsys fs.FS) error {
+	declared, err := utils.Config.Db.Migrations.SchemaPaths.Files(fsys)
+	if len(declared) == 0 {
+		return err
+	}
+	for _, fp := range declared {
+		schema, err := migration.NewMigrationFromFile(fp, fsys)
+		if err != nil {
+			return err
+		}
+		schema.Version = ""
+		if err := schema.ExecBatch(ctx, conn); err != nil {
+			utils.CmdSuggestion = fmt.Sprintf("See schema file: %s", utils.Bold(fp))
+			return err
+		}
+	}
+	return nil
+}