feat(vet): Support managed databases

Author: kyleconroy

internal/cmd/vet.go

@@ -26,6 +26,8 @@ import (
 	"github.com/sqlc-dev/sqlc/internal/debug"
 	"github.com/sqlc-dev/sqlc/internal/opts"
 	"github.com/sqlc-dev/sqlc/internal/plugin"
+	"github.com/sqlc-dev/sqlc/internal/quickdb"
+	pb "github.com/sqlc-dev/sqlc/internal/quickdb/v1"
 	"github.com/sqlc-dev/sqlc/internal/shfmt"
 	"github.com/sqlc-dev/sqlc/internal/vet"
 )
@@ -376,6 +378,62 @@ type checker struct {
 	Envmap     map[string]string
 	Stderr     io.Writer
 	NoDatabase bool
+	Client     pb.QuickClient
+}
+
+func (c *checker) fetchDatabaseUri(ctx context.Context, s config.SQL) (string, func() error, error) {
+	cleanup := func() error {
+		return nil
+	}
+
+	if s.Database == nil {
+		panic("fetch database URI called with nil database")
+	}
+	if !s.Database.Managed {
+		uri, err := c.DSN(s.Database.URI)
+		return uri, cleanup, err
+	}
+	if s.Engine != config.EnginePostgreSQL {
+		return "", cleanup, fmt.Errorf("managed: only PostgreSQL currently")
+	}
+
+	if c.Client == nil {
+		// FIXME: Eventual race condition
+		client, err := quickdb.NewClientFromConfig(c.Conf.Cloud)
+		if err != nil {
+			return "", cleanup, fmt.Errorf("managed: client: %w", err)
+		}
+		c.Client = client
+	}
+
+	var migrations []string
+	for _, query := range s.Schema {
+		contents, err := os.ReadFile(query)
+		if err != nil {
+			return "", cleanup, fmt.Errorf("read file: %w", err)
+		}
+		migrations = append(migrations, string(contents))
+	}
+
+	start := time.Now()
+	resp, err := c.Client.CreateEphemeralDatabase(ctx, &pb.CreateEphemeralDatabaseRequest{
+		Engine:     "postgresql",
+		Region:     "sjc",
+		Migrations: migrations,
+	})
+	if err != nil {
+		return "", cleanup, fmt.Errorf("managed: create database: %w", err)
+	}
+
+	cleanup = func() error {
+		_, err := c.Client.DropEphemeralDatabase(ctx, &pb.DropEphemeralDatabaseRequest{
+			DatabaseId: resp.DatabaseId,
+		})
+		return err
+	}
+
+	fmt.Println("createdb+template", time.Since(start))
+	return resp.Uri, cleanup, nil
 }
 
 func (c *checker) DSN(dsn string) (string, error) {
@@ -422,10 +480,16 @@ func (c *checker) checkSQL(ctx context.Context, s config.SQL) error {
 		if c.NoDatabase {
 			return fmt.Errorf("database: connections disabled via command line flag")
 		}
-		dburl, err := c.DSN(s.Database.URI)
+		dburl, cleanup, err := c.fetchDatabaseUri(ctx, s)
 		if err != nil {
 			return err
 		}
+		defer func() {
+			if err := cleanup(); err != nil {
+				fmt.Fprintf(c.Stderr, "error cleaning up: %s\n", err)
+			}
+		}()
+
 		switch s.Engine {
 		case config.EnginePostgreSQL:
 			conn, err := pgx.Connect(ctx, dburl)

internal/config/config.go

@@ -69,7 +69,8 @@ type Project struct {
 }
 
 type Database struct {
-	URI string `json:"uri" yaml:"uri"`
+	URI     string `json:"uri" yaml:"uri"`
+	Managed bool   `json:"managed" yaml:"managed"`
 }
 
 type Cloud struct {

internal/config/validate.go

@@ -12,8 +12,8 @@ func Validate(c *Config) error {
 			return fmt.Errorf("invalid config: emit_methods_with_db_argument and emit_prepared_queries settings are mutually exclusive")
 		}
 		if sql.Database != nil {
-			if sql.Database.URI == "" {
-				return fmt.Errorf("invalid config: database must have a non-empty URI")
+			if sql.Database.URI == "" && !sql.Database.Managed {
+				return fmt.Errorf("invalid config: database must be managed or have a non-empty URI")
 			}
 		}
 	}