Moved internal/ to pkg/ to make building external plugins possible

pkg/util/params.go

@@ -0,0 +1,57 @@
+package util
+
+import (
+	"slices"
+
+	"golang.org/x/exp/maps"
+)
+
+// Params are accessible in generator.Generate().
+type Params map[string]any
+type Option func(Params)
+
+// Extract all parameters from the options passed as map[string]any.
+func GetParams(opts ...Option) Params {
+	params := Params{}
+	for _, opt := range opts {
+		opt(params)
+	}
+	return params
+}
+
+// Test if an option is present in params
+func (p *Params) OptionExists(params Params, opt string) bool {
+	var k []string = maps.Keys(params)
+	return slices.Contains(k, opt)
+}
+
+// Assert that the options exists within the params map
+func AssertOptionsExist(params Params, opts ...string) []string {
+	foundKeys := []string{}
+	for k := range params {
+		index := slices.IndexFunc(opts, func(s string) bool {
+			return s == k
+		})
+		if index >= 0 {
+			foundKeys = append(foundKeys, k)
+		}
+	}
+	return foundKeys
+}
+
+func WithDefault[T any](v T) Option {
+	return func(p Params) {
+		p["default"] = v
+	}
+}
+
+// Syntactic sugar generic function to get parameter from util.Params.
+func Get[T any](params Params, key string, opts ...Option) *T {
+	if v, ok := params[key].(T); ok {
+		return &v
+	}
+	if defaultValue, ok := params["default"].(T); ok {
+		return &defaultValue
+	}
+	return nil
+}