feat: added json and bcrypt generators

Author: radim

go.mod

@@ -9,6 +9,8 @@ require (
 	github.com/mndrix/tap-go v0.0.0-20171203230836-629fa407e90b
 	github.com/pmezard/go-difflib v1.0.0
 	github.com/spf13/cobra v1.10.1
+	golang.org/x/crypto v0.31.0
+	golang.org/x/term v0.39.0
 	golang.org/x/text v0.28.0
 	gopkg.in/yaml.v3 v3.0.1
 )
@@ -21,9 +23,7 @@ require (
 	github.com/kr/pretty v0.3.1 // indirect
 	github.com/spf13/pflag v1.0.10 // indirect
 	github.com/stretchr/testify v1.11.1 // indirect
-	golang.org/x/crypto v0.31.0 // indirect
 	golang.org/x/sync v0.16.0 // indirect
 	golang.org/x/sys v0.40.0 // indirect
-	golang.org/x/term v0.39.0 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 )

regresql/fixture_manager.go

@@ -50,6 +50,7 @@ func (fm *FixtureManager) registerBuiltinGenerators() error {
 		NewDecimalGenerator(),
 		NewBoolGenerator(),
 		NewRangeGenerator(),
+		NewBcryptHashGenerator(),
 	}
 
 	for _, gen := range basicGens {
@@ -62,6 +63,7 @@ func (fm *FixtureManager) registerBuiltinGenerators() error {
 		NewForeignKeyGenerator(fm.db),
 		NewPatternGenerator(fm.generators),
 		NewTemplateGenerator(fm.generators),
+		NewJSONGenerator(fm.generators),
 	}
 
 	for _, gen := range advancedGens {
@@ -549,7 +551,11 @@ func (fm *FixtureManager) autoDetectForeignKeys(genSpec *GenerateSpec, tableInfo
 // Helper functions
 
 func joinColumns(columns []string) string {
-	return joinStrings(columns, ", ")
+	quoted := make([]string, len(columns))
+	for i, col := range columns {
+		quoted[i] = `"` + col + `"`
+	}
+	return joinStrings(quoted, ", ")
 }
 
 func joinStrings(strs []string, sep string) string {

regresql/generators_basic.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/google/uuid"
+	"golang.org/x/crypto/bcrypt"
 )
 
 const defaultCharset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
@@ -96,6 +97,11 @@ type (
 	BoolGenerator struct {
 		BaseGenerator
 	}
+
+	// BcryptHashGenerator generates bcrypt password hashes
+	BcryptHashGenerator struct {
+		BaseGenerator
+	}
 )
 
 func NewSequenceGenerator() *SequenceGenerator {
@@ -517,3 +523,35 @@ func (g *BoolGenerator) Validate(params map[string]any, column *ColumnInfo) erro
 	}
 	return nil
 }
+
+func NewBcryptHashGenerator() *BcryptHashGenerator {
+	return &BcryptHashGenerator{
+		BaseGenerator: BaseGenerator{name: "bcrypt_hash"},
+	}
+}
+
+func (g *BcryptHashGenerator) Generate(params map[string]any, column *ColumnInfo) (any, error) {
+	password, err := getRequiredParam[string](params, "password")
+	if err != nil {
+		return nil, err
+	}
+	cost := getParam(params, "cost", 10)
+
+	hash, err := bcrypt.GenerateFromPassword([]byte(password), cost)
+	if err != nil {
+		return nil, fmt.Errorf("bcrypt hash generation failed: %w", err)
+	}
+	return string(hash), nil
+}
+
+func (g *BcryptHashGenerator) Validate(params map[string]any, column *ColumnInfo) error {
+	if _, err := getRequiredParam[string](params, "password"); err != nil {
+		return fmt.Errorf("bcrypt_hash requires 'password' parameter")
+	}
+
+	cost := getParam(params, "cost", 10)
+	if cost < bcrypt.MinCost || cost > bcrypt.MaxCost {
+		return fmt.Errorf("bcrypt cost must be between %d and %d, got %d", bcrypt.MinCost, bcrypt.MaxCost, cost)
+	}
+	return nil
+}

regresql/generators_json.go

@@ -0,0 +1,96 @@
+package regresql
+
+import "fmt"
+
+type JSONGenerator struct {
+	registry *GeneratorRegistry
+}
+
+func NewJSONGenerator(registry *GeneratorRegistry) *JSONGenerator {
+	return &JSONGenerator{registry: registry}
+}
+
+func (g *JSONGenerator) Name() string { return "json" }
+
+func (g *JSONGenerator) Validate(params map[string]any, column *ColumnInfo) error {
+	_, hasSchema := params["schema"]
+	_, hasValue := params["value"]
+
+	if !hasSchema && !hasValue {
+		return fmt.Errorf("json generator requires either 'schema' or 'value' parameter")
+	}
+	if hasSchema && hasValue {
+		return fmt.Errorf("json generator cannot have both 'schema' and 'value'")
+	}
+
+	if hasSchema {
+		schema, ok := params["schema"].(map[string]any)
+		if !ok {
+			return fmt.Errorf("'schema' must be an object")
+		}
+		return g.validateSchema(schema)
+	}
+	return nil
+}
+
+func (g *JSONGenerator) validateSchema(schema map[string]any) error {
+	for field, spec := range schema {
+		specMap, ok := spec.(map[string]any)
+		if !ok {
+			continue // static value
+		}
+
+		genName, ok := specMap["generator"].(string)
+		if !ok {
+			continue // static value (object without generator key)
+		}
+
+		gen, err := g.registry.Get(genName)
+		if err != nil {
+			return fmt.Errorf("field '%s': %w", field, err)
+		}
+
+		if err := gen.Validate(specMap, nil); err != nil {
+			return fmt.Errorf("field '%s': %w", field, err)
+		}
+	}
+	return nil
+}
+
+func (g *JSONGenerator) Generate(params map[string]any, column *ColumnInfo) (any, error) {
+	// Static value mode
+	if value, ok := params["value"]; ok {
+		return value, nil
+	}
+
+	// Schema mode
+	schema := params["schema"].(map[string]any)
+	result := make(map[string]any)
+
+	for field, spec := range schema {
+		specMap, ok := spec.(map[string]any)
+		if !ok {
+			result[field] = spec
+			continue
+		}
+
+		genName, ok := specMap["generator"].(string)
+		if !ok {
+			result[field] = spec
+			continue
+		}
+
+		gen, err := g.registry.Get(genName)
+		if err != nil {
+			return nil, fmt.Errorf("field '%s': %w", field, err)
+		}
+
+		value, err := gen.Generate(specMap, nil)
+		if err != nil {
+			return nil, fmt.Errorf("field '%s': %w", field, err)
+		}
+		result[field] = value
+	}
+
+	return result, nil
+}