chore: use wildcard pattern and escape like operator

Author: sweatybridge

internal/db/diff/migra.go

@@ -106,10 +106,8 @@ func LoadUserSchemas(ctx context.Context, conn *pgx.Conn, exclude ...string) ([]
 			// Exclude functions because Webhooks support is early alpha
 			"supabase_functions",
 			"supabase_migrations",
-			// Exclude postgres temporary schemas
-			"pg_temp_%",
-			"pg_toast_temp_%",
 		}, utils.SystemSchemas...)
+		exclude = likeEscapeSchema(exclude)
 	}
 	rows, err := conn.Query(ctx, LIST_SCHEMAS, exclude)
 	if err != nil {
@@ -126,6 +124,15 @@ func LoadUserSchemas(ctx context.Context, conn *pgx.Conn, exclude ...string) ([]
 	return schemas, nil
 }
 
+func likeEscapeSchema(schemas []string) (result []string) {
+	// Treat _ as literal, * as any character
+	replacer := strings.NewReplacer("_", `\_`, "*", "%")
+	for _, sch := range schemas {
+		result = append(result, replacer.Replace(sch))
+	}
+	return result
+}
+
 func CreateShadowDatabase(ctx context.Context) (string, error) {
 	config := container.Config{
 		Image: utils.DbImage,

internal/db/diff/migra_test.go

@@ -108,7 +108,7 @@ func TestRunMigra(t *testing.T) {
 		// Setup mock postgres
 		conn := pgtest.NewConn()
 		defer conn.Close(t)
-		conn.Query("SELECT schema_name FROM information_schema.schemata WHERE NOT schema_name LIKE ANY('{pgbouncer,realtime,_realtime,supabase_functions,supabase_migrations,pg_temp_%,pg_toast_temp_%,information_schema,pg_catalog,pg_toast,cron,graphql,graphql_public,net,pgsodium,pgsodium_masks,vault}') ORDER BY schema_name").
+		conn.Query(`SELECT schema_name FROM information_schema.schemata WHERE NOT schema_name LIKE ANY('{pgbouncer,realtime,"\\_realtime","supabase\\_functions","supabase\\_migrations","information\\_schema","pg\\_%",cron,graphql,"graphql\\_public",net,pgsodium,"pgsodium\\_masks",vault}') ORDER BY schema_name`).
 			ReplyError(pgerrcode.DuplicateTable, `relation "test" already exists`)
 		// Run test
 		err := RunMigra(context.Background(), []string{}, "", "password", fsys, conn.Intercept)

internal/utils/misc.go

@@ -109,11 +109,10 @@ var (
 	FuncSlugPattern    = regexp.MustCompile(`^[A-Za-z][A-Za-z0-9_-]*$`)
 	ImageNamePattern   = regexp.MustCompile(`\/(.*):`)
 
-	// These schemas are ignored from schema diffs
+	// These schemas are ignored from db diff and db dump
 	SystemSchemas = []string{
 		"information_schema",
-		"pg_catalog",
-		"pg_toast",
+		"pg_*", // Wildcard pattern follows pg_dump
 		// Owned by extensions
 		"cron",
 		"graphql",