gen.go

package process

import (
	"bytes"
	"context"
	"errors"
	"fmt"
	"os"
	"os/exec"

	"google.golang.org/protobuf/proto"

	"github.com/sqlc-dev/sqlc/internal/plugin"
)

type Runner struct {
	Cmd string
	Env []string
}

// TODO: Update the gen func signature to take a ctx
func (r Runner) Generate(ctx context.Context, req *plugin.CodeGenRequest) (*plugin.CodeGenResponse, error) {
	stdin, err := proto.Marshal(req)
	if err != nil {
		return nil, fmt.Errorf("failed to encode codegen request: %s", err)
	}

	// Check if the output plugin exists
	path, err := exec.LookPath(r.Cmd)
	if err != nil {
		return nil, fmt.Errorf("process: %s not found", r.Cmd)
	}

	cmd := exec.CommandContext(ctx, path)
	cmd.Stdin = bytes.NewReader(stdin)
	cmd.Env = []string{
		fmt.Sprintf("SQLC_VERSION=%s", req.SqlcVersion),
	}
	for _, key := range r.Env {
		if key == "SQLC_AUTH_TOKEN" {
			continue
		}
		cmd.Env = append(cmd.Env, fmt.Sprintf("%s=%s", key, os.Getenv(key)))
	}

	out, err := cmd.Output()
	if err != nil {
		stderr := err.Error()
		var exit *exec.ExitError
		if errors.As(err, &exit) {
			stderr = string(exit.Stderr)
		}
		return nil, fmt.Errorf("process: error running command %s", stderr)
	}

	var resp plugin.CodeGenResponse
	if err := proto.Unmarshal(out, &resp); err != nil {
		return nil, fmt.Errorf("process: failed to read codegen resp: %s", err)
	}

	return &resp, nil
}