config.go

package python

import (
	"bytes"
	json "encoding/json"
	"fmt"
	"strings"
)

type OverrideColumn struct {
	Column   string `json:"column"`
	DbType   string `json:"db_type"`
	PyType   string `json:"py_type"`
	PyImport string `json:"py_import"`
}

type Config struct {
	EmitExactTableNames         bool             `json:"emit_exact_table_names"`
	EmitSyncQuerier             bool             `json:"emit_sync_querier"`
	EmitAsyncQuerier            bool             `json:"emit_async_querier"`
	Package                     string           `json:"package"`
	Out                         string           `json:"out"`
	EmitPydanticModels          bool             `json:"emit_pydantic_models"`
	EmitStrEnum                 bool             `json:"emit_str_enum"`
	EmitSchemaNamePrefix        bool             `json:"emit_schema_name_prefix"`
	QueryParameterLimit         *int32           `json:"query_parameter_limit"`
	InflectionExcludeTableNames []string         `json:"inflection_exclude_table_names"`
	Overrides                   []OverrideColumn `json:"overrides"`
}

func parseConfig(raw []byte) (Config, error) {
	var conf Config
	if len(raw) == 0 {
		return conf, nil
	}

	dec := json.NewDecoder(bytes.NewReader(raw))
	dec.DisallowUnknownFields()
	if err := dec.Decode(&conf); err != nil {
		msg := strings.TrimPrefix(err.Error(), "json: ")
		return Config{}, fmt.Errorf("invalid plugin options: %s", msg)
	}

	return conf, nil
}