Merge pull request #143 from enrichman/inject

Added `inject` function

Author: ugol

pkg/functions/functions.go

@@ -24,11 +24,6 @@ import (
 	"bufio"
 	"encoding/csv"
 	"fmt"
-	"github.com/google/uuid"
-	"github.com/ugol/jr/pkg/constants"
-	"github.com/ugol/jr/pkg/ctx"
-	"golang.org/x/text/cases"
-	"golang.org/x/text/language"
 	"log"
 	"math"
 	"math/rand"
@@ -37,6 +32,12 @@ import (
 	"strconv"
 	"strings"
 	"text/template"
+
+	"github.com/google/uuid"
+	"github.com/ugol/jr/pkg/constants"
+	"github.com/ugol/jr/pkg/ctx"
+	"golang.org/x/text/cases"
+	"golang.org/x/text/language"
 )
 
 func FunctionsMap() template.FuncMap {
@@ -47,6 +48,8 @@ var Random = rand.New(rand.NewSource(0))
 var data = map[string][]string{}
 var fmap = map[string]interface{}{
 
+	"inject": Inject,
+
 	// text utilities
 	"atoi":                     Atoi,
 	"itoa":                     strconv.Itoa,
@@ -523,3 +526,10 @@ func InitCSV(csvpath string) {
 	}
 
 }
+
+func Inject(perc float64, injected, orig string) string {
+	if rand.Float64() < perc {
+		return injected
+	}
+	return orig
+}