package command

import (
	"context"
	"gitee.com/go-genie/sqlx"
	contextx "gitee.com/go-genie/xx/context"
	"io"
)

type contextKeyGeneratorOutput struct{}

func GeneratorOutputFromContext(ctx context.Context) io.Writer {
	if opts, ok := ctx.Value(contextKeyGeneratorOutput{}).(io.Writer); ok {
		if opts != nil {
			return opts
		}
	}
	return nil
}

func MustGenerator(db sqlx.DBExecutor, w io.Writer) {
	if err := Generator(db, w); err != nil {
		panic(err)
	}
}

func Generator(db sqlx.DBExecutor, output io.Writer) error {
	ctx := contextx.WithValue(db.Context(), contextKeyGeneratorOutput{}, output)

	if err := db.(sqlx.Generator).Generate(ctx, db); err != nil {
		return err
	}

	return nil
}