add data struct

data/data.go

@@ -0,0 +1,62 @@
+package data
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/tyler-sommer/stick"
+	"github.com/tyler-sommer/stick/twig"
+	"gitnet.fr/deblan/database-anonymizer/logger"
+	"strconv"
+	"strings"
+)
+
+type Data struct {
+	Value        string
+	Faker        string
+	IsVirtual    bool
+	IsPrimaryKey bool
+}
+
+func (d *Data) FromByte(v []byte) *Data {
+	d.Value = string(v)
+
+	return d
+}
+
+func (d *Data) FromInt64(v int64) *Data {
+	d.Value = strconv.FormatInt(v, 10)
+
+	return d
+}
+
+func (d *Data) FromString(v string) *Data {
+	d.Value = v
+
+	return d
+}
+
+func (d *Data) IsTwigExpression() bool {
+	return strings.Contains(d.Faker, "{{") || strings.Contains(d.Faker, "}}")
+}
+
+func (d *Data) Update(row map[string]Data) {
+	if !d.IsPrimaryKey && d.Faker != "" {
+		if d.IsTwigExpression() {
+			env := twig.New(nil)
+			params := map[string]stick.Value{}
+
+			for key, value := range row {
+				params[key] = value.Value
+			}
+
+			var buf bytes.Buffer
+			err := env.Execute(d.Faker, &buf, params)
+
+			logger.LogFatalExitIf(err)
+
+			d.Value = buf.String()
+		} else {
+			d.Value = fmt.Sprintf("%s__UPDATE", d.Value)
+		}
+	}
+}