Compare commits
15 Commits
fd1df6e40e
...
master
Author | SHA1 | Date | |
---|---|---|---|
969ccae25c | |||
5b46ff0efd | |||
d234616406 | |||
af3e55e518 | |||
13b48229ac | |||
670f6ed7a0 | |||
bbc7c50fae | |||
779d1e0a0e | |||
54581f0216 | |||
3d01822e77 | |||
4e0ca92c77 | |||
388e54b3e3 | |||
6f2e76221a | |||
e0d3b938e3 | |||
491a030bf8 |
13
.vscode/launch.json
vendored
13
.vscode/launch.json
vendored
@@ -98,6 +98,19 @@
|
|||||||
"args": [
|
"args": [
|
||||||
"cook_tacz.yml",
|
"cook_tacz.yml",
|
||||||
]
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "Launch Package (ICARUS)",
|
||||||
|
"type": "go",
|
||||||
|
"request": "launch",
|
||||||
|
"mode": "auto",
|
||||||
|
"program": "${workspaceFolder}",
|
||||||
|
"cwd": "C:/Users/Administrator/Seafile/Games-ICARUS/Icarus/Saved/IME3/Mods",
|
||||||
|
"args": [
|
||||||
|
"-loglevel",
|
||||||
|
"trace",
|
||||||
|
"cook_processorrecipes.yml",
|
||||||
|
]
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
9
go.mod
9
go.mod
@@ -13,7 +13,6 @@ require (
|
|||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/davecgh/go-spew v1.1.1 // indirect
|
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||||
github.com/google/go-cmp v0.6.0 // indirect
|
|
||||||
github.com/hexops/valast v1.5.0 // indirect
|
github.com/hexops/valast v1.5.0 // indirect
|
||||||
github.com/jinzhu/inflection v1.0.0 // indirect
|
github.com/jinzhu/inflection v1.0.0 // indirect
|
||||||
github.com/jinzhu/now v1.1.5 // indirect
|
github.com/jinzhu/now v1.1.5 // indirect
|
||||||
@@ -21,6 +20,8 @@ require (
|
|||||||
github.com/mattn/go-sqlite3 v1.14.22 // indirect
|
github.com/mattn/go-sqlite3 v1.14.22 // indirect
|
||||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||||
github.com/rogpeppe/go-internal v1.14.1 // indirect
|
github.com/rogpeppe/go-internal v1.14.1 // indirect
|
||||||
|
github.com/tidwall/match v1.1.1 // indirect
|
||||||
|
github.com/tidwall/pretty v1.2.0 // indirect
|
||||||
golang.org/x/mod v0.21.0 // indirect
|
golang.org/x/mod v0.21.0 // indirect
|
||||||
golang.org/x/sync v0.11.0 // indirect
|
golang.org/x/sync v0.11.0 // indirect
|
||||||
golang.org/x/text v0.22.0 // indirect
|
golang.org/x/text v0.22.0 // indirect
|
||||||
@@ -29,4 +30,8 @@ require (
|
|||||||
mvdan.cc/gofumpt v0.4.0 // indirect
|
mvdan.cc/gofumpt v0.4.0 // indirect
|
||||||
)
|
)
|
||||||
|
|
||||||
require gorm.io/driver/sqlite v1.6.0
|
require (
|
||||||
|
github.com/google/go-cmp v0.6.0
|
||||||
|
github.com/tidwall/gjson v1.18.0
|
||||||
|
gorm.io/driver/sqlite v1.6.0
|
||||||
|
)
|
||||||
|
6
go.sum
6
go.sum
@@ -36,6 +36,12 @@ github.com/rogpeppe/go-internal v1.14.1 h1:UQB4HGPB6osV0SQTLymcB4TgvyWu6ZyliaW0t
|
|||||||
github.com/rogpeppe/go-internal v1.14.1/go.mod h1:MaRKkUm5W0goXpeCfT7UZI6fk/L7L7so1lCWt35ZSgc=
|
github.com/rogpeppe/go-internal v1.14.1/go.mod h1:MaRKkUm5W0goXpeCfT7UZI6fk/L7L7so1lCWt35ZSgc=
|
||||||
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
||||||
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||||
|
github.com/tidwall/gjson v1.18.0 h1:FIDeeyB800efLX89e5a8Y0BNH+LOngJyGrIWxG2FKQY=
|
||||||
|
github.com/tidwall/gjson v1.18.0/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
|
||||||
|
github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=
|
||||||
|
github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
|
||||||
|
github.com/tidwall/pretty v1.2.0 h1:RWIZEg2iJ8/g6fDDYzMpobmaoGh5OLl4AXtGUGPcqCs=
|
||||||
|
github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
|
||||||
github.com/yuin/gopher-lua v1.1.1 h1:kYKnWBjvbNP4XLT3+bPEwAXJx262OhaHDWDVOPjL46M=
|
github.com/yuin/gopher-lua v1.1.1 h1:kYKnWBjvbNP4XLT3+bPEwAXJx262OhaHDWDVOPjL46M=
|
||||||
github.com/yuin/gopher-lua v1.1.1/go.mod h1:GBR0iDaNXjAgGg9zfCvksxSRnQx76gclCIb7kdAd1Pw=
|
github.com/yuin/gopher-lua v1.1.1/go.mod h1:GBR0iDaNXjAgGg9zfCvksxSRnQx76gclCIb7kdAd1Pw=
|
||||||
golang.org/x/mod v0.21.0 h1:vvrHzRwRfVKSiLrG+d4FMl/Qi4ukBCE6kZlTUkDYRT0=
|
golang.org/x/mod v0.21.0 h1:vvrHzRwRfVKSiLrG+d4FMl/Qi4ukBCE6kZlTUkDYRT0=
|
||||||
|
2
main.go
2
main.go
@@ -58,6 +58,8 @@ func main() {
|
|||||||
fmt.Fprintf(os.Stderr, " If expression starts with an operator like *, /, +, -, =, etc., v1 is automatically prepended\n")
|
fmt.Fprintf(os.Stderr, " If expression starts with an operator like *, /, +, -, =, etc., v1 is automatically prepended\n")
|
||||||
fmt.Fprintf(os.Stderr, " You can use any valid Lua code, including if statements, loops, etc.\n")
|
fmt.Fprintf(os.Stderr, " You can use any valid Lua code, including if statements, loops, etc.\n")
|
||||||
fmt.Fprintf(os.Stderr, " Glob patterns are supported for file selection (*.xml, data/**.xml, etc.)\n")
|
fmt.Fprintf(os.Stderr, " Glob patterns are supported for file selection (*.xml, data/**.xml, etc.)\n")
|
||||||
|
fmt.Fprintf(os.Stderr, "\nLua Functions Available:\n")
|
||||||
|
fmt.Fprintf(os.Stderr, "%s\n", processor.GetLuaFunctionsHelp())
|
||||||
}
|
}
|
||||||
// TODO: Fix bed shitting when doing *.yml in barotrauma directory
|
// TODO: Fix bed shitting when doing *.yml in barotrauma directory
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
@@ -4,9 +4,13 @@ import (
|
|||||||
"cook/utils"
|
"cook/utils"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"sort"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
logger "git.site.quack-lab.dev/dave/cylogger"
|
logger "git.site.quack-lab.dev/dave/cylogger"
|
||||||
|
"github.com/tidwall/gjson"
|
||||||
lua "github.com/yuin/gopher-lua"
|
lua "github.com/yuin/gopher-lua"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -85,26 +89,462 @@ func ProcessJSON(content string, command utils.ModifyCommand, filename string) (
|
|||||||
return commands, fmt.Errorf("failed to convert Lua table back to Go: %v", err)
|
return commands, fmt.Errorf("failed to convert Lua table back to Go: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Marshal back to JSON
|
processJsonLogger.Debug("About to call applyChanges with original data and modified data")
|
||||||
modifiedJSON, err := json.MarshalIndent(goData, "", " ")
|
commands, err = applyChanges(content, jsonData, goData)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
processJsonLogger.Error("Failed to marshal modified data to JSON: %v", err)
|
processJsonLogger.Error("Failed to apply surgical JSON changes: %v", err)
|
||||||
return commands, fmt.Errorf("failed to marshal modified data to JSON: %v", err)
|
return commands, fmt.Errorf("failed to apply surgical JSON changes: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create replacement command for the entire file
|
|
||||||
// For JSON mode, we always replace the entire content
|
|
||||||
commands = append(commands, utils.ReplaceCommand{
|
|
||||||
From: 0,
|
|
||||||
To: len(content),
|
|
||||||
With: string(modifiedJSON),
|
|
||||||
})
|
|
||||||
|
|
||||||
processJsonLogger.Debug("Total JSON processing time: %v", time.Since(startTime))
|
processJsonLogger.Debug("Total JSON processing time: %v", time.Since(startTime))
|
||||||
processJsonLogger.Debug("Generated %d total modifications", len(commands))
|
processJsonLogger.Debug("Generated %d total modifications", len(commands))
|
||||||
return commands, nil
|
return commands, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// applyJSONChanges compares original and modified data and applies changes surgically
|
||||||
|
func applyJSONChanges(content string, originalData, modifiedData interface{}) ([]utils.ReplaceCommand, error) {
|
||||||
|
var commands []utils.ReplaceCommand
|
||||||
|
|
||||||
|
appliedCommands, err := applyChanges(content, originalData, modifiedData)
|
||||||
|
if err == nil && len(appliedCommands) > 0 {
|
||||||
|
return appliedCommands, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return commands, fmt.Errorf("failed to make any changes to the json")
|
||||||
|
}
|
||||||
|
|
||||||
|
// applyChanges attempts to make surgical changes while preserving exact formatting
|
||||||
|
func applyChanges(content string, originalData, modifiedData interface{}) ([]utils.ReplaceCommand, error) {
|
||||||
|
var commands []utils.ReplaceCommand
|
||||||
|
|
||||||
|
// Find all changes between original and modified data
|
||||||
|
changes := findDeepChanges("", originalData, modifiedData)
|
||||||
|
|
||||||
|
jsonLogger.Debug("applyChanges: Found %d changes: %v", len(changes), changes)
|
||||||
|
|
||||||
|
if len(changes) == 0 {
|
||||||
|
return commands, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sort removal operations by index in descending order to avoid index shifting
|
||||||
|
var removals []string
|
||||||
|
var additions []string
|
||||||
|
var valueChanges []string
|
||||||
|
|
||||||
|
for path := range changes {
|
||||||
|
if strings.HasSuffix(path, "@remove") {
|
||||||
|
removals = append(removals, path)
|
||||||
|
} else if strings.HasSuffix(path, "@add") {
|
||||||
|
additions = append(additions, path)
|
||||||
|
} else {
|
||||||
|
valueChanges = append(valueChanges, path)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
jsonLogger.Debug("applyChanges: %d removals, %d additions, %d value changes", len(removals), len(additions), len(valueChanges))
|
||||||
|
|
||||||
|
// Apply removals first (from end to beginning to avoid index shifting)
|
||||||
|
for _, removalPath := range removals {
|
||||||
|
actualPath := strings.TrimSuffix(removalPath, "@remove")
|
||||||
|
elementIndex := extractIndexFromRemovalPath(actualPath)
|
||||||
|
arrayPath := getArrayPathFromElementPath(actualPath)
|
||||||
|
|
||||||
|
jsonLogger.Debug("Processing removal: path=%s, index=%d, arrayPath=%s", actualPath, elementIndex, arrayPath)
|
||||||
|
|
||||||
|
// Find the exact byte range to remove
|
||||||
|
from, to := findArrayElementRemovalRange(content, arrayPath, elementIndex)
|
||||||
|
|
||||||
|
jsonLogger.Debug("Removing bytes %d-%d", from, to)
|
||||||
|
|
||||||
|
commands = append(commands, utils.ReplaceCommand{
|
||||||
|
From: from,
|
||||||
|
To: to,
|
||||||
|
With: "",
|
||||||
|
})
|
||||||
|
|
||||||
|
jsonLogger.Debug("Added removal command: From=%d, To=%d, With=\"\"", from, to)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply additions (new fields)
|
||||||
|
for _, additionPath := range additions {
|
||||||
|
actualPath := strings.TrimSuffix(additionPath, "@add")
|
||||||
|
newValue := changes[additionPath]
|
||||||
|
|
||||||
|
jsonLogger.Debug("Processing addition: path=%s, value=%v", actualPath, newValue)
|
||||||
|
|
||||||
|
// Find the parent object to add the field to
|
||||||
|
parentPath := getParentPath(actualPath)
|
||||||
|
fieldName := getFieldName(actualPath)
|
||||||
|
|
||||||
|
jsonLogger.Debug("Parent path: %s, field name: %s", parentPath, fieldName)
|
||||||
|
|
||||||
|
// Get the parent object
|
||||||
|
var parentResult gjson.Result
|
||||||
|
if parentPath == "" {
|
||||||
|
// Adding to root object - get the entire JSON
|
||||||
|
parentResult = gjson.Parse(content)
|
||||||
|
} else {
|
||||||
|
parentResult = gjson.Get(content, parentPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !parentResult.Exists() {
|
||||||
|
jsonLogger.Debug("Parent path %s does not exist, skipping", parentPath)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Find where to insert the new field (at the end of the object)
|
||||||
|
startPos := int(parentResult.Index + len(parentResult.Raw) - 1) // Before closing brace
|
||||||
|
|
||||||
|
jsonLogger.Debug("Inserting at pos %d", startPos)
|
||||||
|
|
||||||
|
// Convert the new value to JSON string
|
||||||
|
newValueStr := convertValueToJSONString(newValue)
|
||||||
|
|
||||||
|
|
||||||
|
// Insert the new field with pretty-printed formatting
|
||||||
|
// Format: ,"fieldName": { ... }
|
||||||
|
insertText := fmt.Sprintf(`,"%s": %s`, fieldName, newValueStr)
|
||||||
|
|
||||||
|
|
||||||
|
commands = append(commands, utils.ReplaceCommand{
|
||||||
|
From: startPos,
|
||||||
|
To: startPos,
|
||||||
|
With: insertText,
|
||||||
|
})
|
||||||
|
|
||||||
|
jsonLogger.Debug("Added addition command: From=%d, To=%d, With=%q", startPos, startPos, insertText)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply value changes (in reverse order to avoid position shifting)
|
||||||
|
sort.Slice(valueChanges, func(i, j int) bool {
|
||||||
|
// Get positions for comparison
|
||||||
|
resultI := gjson.Get(content, valueChanges[i])
|
||||||
|
resultJ := gjson.Get(content, valueChanges[j])
|
||||||
|
return resultI.Index > resultJ.Index // Descending order
|
||||||
|
})
|
||||||
|
|
||||||
|
for _, path := range valueChanges {
|
||||||
|
newValue := changes[path]
|
||||||
|
|
||||||
|
jsonLogger.Debug("Processing value change: path=%s, value=%v", path, newValue)
|
||||||
|
|
||||||
|
// Get the current value and its position in the original JSON
|
||||||
|
result := gjson.Get(content, path)
|
||||||
|
if !result.Exists() {
|
||||||
|
jsonLogger.Debug("Path %s does not exist, skipping", path)
|
||||||
|
continue // Skip if path doesn't exist
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get the exact byte positions of this value
|
||||||
|
startPos := result.Index
|
||||||
|
endPos := startPos + len(result.Raw)
|
||||||
|
|
||||||
|
jsonLogger.Debug("Found value at pos %d-%d: %q", startPos, endPos, result.Raw)
|
||||||
|
|
||||||
|
// Convert the new value to JSON string
|
||||||
|
newValueStr := convertValueToJSONString(newValue)
|
||||||
|
|
||||||
|
jsonLogger.Debug("Converting to: %q", newValueStr)
|
||||||
|
|
||||||
|
// Create a replacement command for this specific value
|
||||||
|
commands = append(commands, utils.ReplaceCommand{
|
||||||
|
From: int(startPos),
|
||||||
|
To: int(endPos),
|
||||||
|
With: newValueStr,
|
||||||
|
})
|
||||||
|
|
||||||
|
jsonLogger.Debug("Added command: From=%d, To=%d, With=%q", int(startPos), int(endPos), newValueStr)
|
||||||
|
}
|
||||||
|
|
||||||
|
return commands, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// extractIndexFromRemovalPath extracts the array index from a removal path like "Rows.0.Inputs.1@remove"
|
||||||
|
func extractIndexFromRemovalPath(path string) int {
|
||||||
|
parts := strings.Split(strings.TrimSuffix(path, "@remove"), ".")
|
||||||
|
if len(parts) > 0 {
|
||||||
|
lastPart := parts[len(parts)-1]
|
||||||
|
if index, err := strconv.Atoi(lastPart); err == nil {
|
||||||
|
return index
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return -1
|
||||||
|
}
|
||||||
|
|
||||||
|
// getArrayPathFromElementPath converts "Rows.0.Inputs.1" to "Rows.0.Inputs"
|
||||||
|
func getArrayPathFromElementPath(elementPath string) string {
|
||||||
|
parts := strings.Split(elementPath, ".")
|
||||||
|
if len(parts) > 0 {
|
||||||
|
return strings.Join(parts[:len(parts)-1], ".")
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// getParentPath extracts the parent path from a full path like "Rows.0.Inputs.1"
|
||||||
|
func getParentPath(fullPath string) string {
|
||||||
|
parts := strings.Split(fullPath, ".")
|
||||||
|
if len(parts) > 0 {
|
||||||
|
return strings.Join(parts[:len(parts)-1], ".")
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// getFieldName extracts the field name from a full path like "Rows.0.Inputs.1"
|
||||||
|
func getFieldName(fullPath string) string {
|
||||||
|
parts := strings.Split(fullPath, ".")
|
||||||
|
if len(parts) > 0 {
|
||||||
|
return parts[len(parts)-1]
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// convertValueToJSONString converts a Go interface{} to a JSON string representation
|
||||||
|
func convertValueToJSONString(value interface{}) string {
|
||||||
|
switch v := value.(type) {
|
||||||
|
case string:
|
||||||
|
return `"` + strings.ReplaceAll(v, `"`, `\"`) + `"`
|
||||||
|
case float64:
|
||||||
|
if v == float64(int64(v)) {
|
||||||
|
return strconv.FormatInt(int64(v), 10)
|
||||||
|
}
|
||||||
|
return strconv.FormatFloat(v, 'f', -1, 64)
|
||||||
|
case bool:
|
||||||
|
return strconv.FormatBool(v)
|
||||||
|
case nil:
|
||||||
|
return "null"
|
||||||
|
case map[string]interface{}:
|
||||||
|
// Handle maps specially to avoid double-escaping of keys
|
||||||
|
var pairs []string
|
||||||
|
for key, val := range v {
|
||||||
|
// The key might already have escaped quotes from Lua, so we need to be careful
|
||||||
|
// If the key already contains escaped quotes, we need to unescape them first
|
||||||
|
keyStr := key
|
||||||
|
if strings.Contains(key, `\"`) {
|
||||||
|
// Key already has escaped quotes, use it as-is
|
||||||
|
keyStr = `"` + key + `"`
|
||||||
|
} else {
|
||||||
|
// Normal key, escape quotes
|
||||||
|
keyStr = `"` + strings.ReplaceAll(key, `"`, `\"`) + `"`
|
||||||
|
}
|
||||||
|
valStr := convertValueToJSONString(val)
|
||||||
|
pairs = append(pairs, keyStr+":"+valStr)
|
||||||
|
}
|
||||||
|
return "{" + strings.Join(pairs, ",") + "}"
|
||||||
|
default:
|
||||||
|
// For other complex types (arrays), we need to use json.Marshal
|
||||||
|
jsonBytes, err := json.Marshal(v)
|
||||||
|
if err != nil {
|
||||||
|
return "null" // Fallback to null if marshaling fails
|
||||||
|
}
|
||||||
|
return string(jsonBytes)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// findArrayElementRemovalRange finds the exact byte range to remove for an array element
|
||||||
|
func findArrayElementRemovalRange(content, arrayPath string, elementIndex int) (int, int) {
|
||||||
|
// Get the array using gjson
|
||||||
|
arrayResult := gjson.Get(content, arrayPath)
|
||||||
|
if !arrayResult.Exists() || !arrayResult.IsArray() {
|
||||||
|
return -1, -1
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get all array elements
|
||||||
|
elements := arrayResult.Array()
|
||||||
|
if elementIndex >= len(elements) {
|
||||||
|
return -1, -1
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get the target element
|
||||||
|
elementResult := elements[elementIndex]
|
||||||
|
startPos := int(elementResult.Index)
|
||||||
|
endPos := int(elementResult.Index + len(elementResult.Raw))
|
||||||
|
|
||||||
|
// Handle comma removal properly
|
||||||
|
if elementIndex == 0 && len(elements) > 1 {
|
||||||
|
// First element but not the only one - remove comma after
|
||||||
|
for i := endPos; i < len(content) && i < endPos+50; i++ {
|
||||||
|
if content[i] == ',' {
|
||||||
|
endPos = i + 1
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if elementIndex == len(elements)-1 && len(elements) > 1 {
|
||||||
|
// Last element and not the only one - remove comma before
|
||||||
|
prevElementEnd := int(elements[elementIndex-1].Index + len(elements[elementIndex-1].Raw))
|
||||||
|
for i := prevElementEnd; i < startPos && i < len(content); i++ {
|
||||||
|
if content[i] == ',' {
|
||||||
|
startPos = i
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// If it's the only element, don't remove any commas
|
||||||
|
|
||||||
|
return startPos, endPos
|
||||||
|
}
|
||||||
|
|
||||||
|
// findDeepChanges recursively finds all paths that need to be changed
|
||||||
|
func findDeepChanges(basePath string, original, modified interface{}) map[string]interface{} {
|
||||||
|
changes := make(map[string]interface{})
|
||||||
|
|
||||||
|
switch orig := original.(type) {
|
||||||
|
case map[string]interface{}:
|
||||||
|
if mod, ok := modified.(map[string]interface{}); ok {
|
||||||
|
// Check for new keys added in modified data
|
||||||
|
for key, modValue := range mod {
|
||||||
|
var currentPath string
|
||||||
|
if basePath == "" {
|
||||||
|
currentPath = key
|
||||||
|
} else {
|
||||||
|
currentPath = basePath + "." + key
|
||||||
|
}
|
||||||
|
|
||||||
|
if origValue, exists := orig[key]; exists {
|
||||||
|
// Key exists in both, check if value changed
|
||||||
|
switch modValue.(type) {
|
||||||
|
case map[string]interface{}, []interface{}:
|
||||||
|
// Recursively check nested structures
|
||||||
|
nestedChanges := findDeepChanges(currentPath, origValue, modValue)
|
||||||
|
for nestedPath, nestedValue := range nestedChanges {
|
||||||
|
changes[nestedPath] = nestedValue
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
// Primitive value - check if changed
|
||||||
|
if !deepEqual(origValue, modValue) {
|
||||||
|
changes[currentPath] = modValue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// New key added - mark for addition
|
||||||
|
changes[currentPath+"@add"] = modValue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
case []interface{}:
|
||||||
|
if mod, ok := modified.([]interface{}); ok {
|
||||||
|
// Handle array changes by detecting specific element operations
|
||||||
|
if len(orig) != len(mod) {
|
||||||
|
// Array length changed - detect if it's element removal
|
||||||
|
if len(orig) > len(mod) {
|
||||||
|
// Element(s) removed - find which ones by comparing content
|
||||||
|
removedIndices := findRemovedArrayElements(orig, mod)
|
||||||
|
for _, removedIndex := range removedIndices {
|
||||||
|
var currentPath string
|
||||||
|
if basePath == "" {
|
||||||
|
currentPath = fmt.Sprintf("%d@remove", removedIndex)
|
||||||
|
} else {
|
||||||
|
currentPath = fmt.Sprintf("%s.%d@remove", basePath, removedIndex)
|
||||||
|
}
|
||||||
|
changes[currentPath] = nil // Mark for removal
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Elements added - more complex, skip for now
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// Same length - check individual elements for value changes
|
||||||
|
for i, modValue := range mod {
|
||||||
|
var currentPath string
|
||||||
|
if basePath == "" {
|
||||||
|
currentPath = strconv.Itoa(i)
|
||||||
|
} else {
|
||||||
|
currentPath = basePath + "." + strconv.Itoa(i)
|
||||||
|
}
|
||||||
|
|
||||||
|
if i < len(orig) {
|
||||||
|
// Index exists in both, check if value changed
|
||||||
|
switch modValue.(type) {
|
||||||
|
case map[string]interface{}, []interface{}:
|
||||||
|
// Recursively check nested structures
|
||||||
|
nestedChanges := findDeepChanges(currentPath, orig[i], modValue)
|
||||||
|
for nestedPath, nestedValue := range nestedChanges {
|
||||||
|
changes[nestedPath] = nestedValue
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
// Primitive value - check if changed
|
||||||
|
if !deepEqual(orig[i], modValue) {
|
||||||
|
changes[currentPath] = modValue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
// For primitive types, compare directly
|
||||||
|
if !deepEqual(original, modified) {
|
||||||
|
if basePath == "" {
|
||||||
|
changes[""] = modified
|
||||||
|
} else {
|
||||||
|
changes[basePath] = modified
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return changes
|
||||||
|
}
|
||||||
|
|
||||||
|
// findRemovedArrayElements compares two arrays and returns indices of removed elements
|
||||||
|
func findRemovedArrayElements(original, modified []interface{}) []int {
|
||||||
|
var removedIndices []int
|
||||||
|
|
||||||
|
// Simple approach: find elements in original that don't exist in modified
|
||||||
|
for i, origElement := range original {
|
||||||
|
found := false
|
||||||
|
for _, modElement := range modified {
|
||||||
|
if deepEqual(origElement, modElement) {
|
||||||
|
found = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
removedIndices = append(removedIndices, i)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return removedIndices
|
||||||
|
}
|
||||||
|
|
||||||
|
// deepEqual performs deep comparison of two values
|
||||||
|
func deepEqual(a, b interface{}) bool {
|
||||||
|
if a == nil && b == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if a == nil || b == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
switch av := a.(type) {
|
||||||
|
case map[string]interface{}:
|
||||||
|
if bv, ok := b.(map[string]interface{}); ok {
|
||||||
|
if len(av) != len(bv) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
for k, v := range av {
|
||||||
|
if !deepEqual(v, bv[k]) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
case []interface{}:
|
||||||
|
if bv, ok := b.([]interface{}); ok {
|
||||||
|
if len(av) != len(bv) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
for i, v := range av {
|
||||||
|
if !deepEqual(v, bv[i]) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
default:
|
||||||
|
return a == b
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// ToLuaTable converts a Go interface{} to a Lua table recursively
|
// ToLuaTable converts a Go interface{} to a Lua table recursively
|
||||||
func ToLuaTable(L *lua.LState, data interface{}) (*lua.LTable, error) {
|
func ToLuaTable(L *lua.LState, data interface{}) (*lua.LTable, error) {
|
||||||
toLuaTableLogger := jsonLogger.WithPrefix("ToLuaTable")
|
toLuaTableLogger := jsonLogger.WithPrefix("ToLuaTable")
|
||||||
|
@@ -16,46 +16,25 @@ func TestProcessJSON(t *testing.T) {
|
|||||||
expectedMods int
|
expectedMods int
|
||||||
}{
|
}{
|
||||||
{
|
{
|
||||||
name: "Basic JSON object modification",
|
name: "Basic JSON object modification",
|
||||||
input: `{"name": "test", "value": 42}`,
|
input: `{"name": "test", "value": 42}`,
|
||||||
luaExpression: `data.value = data.value * 2; return true`,
|
luaExpression: `data.value = data.value * 2; return true`,
|
||||||
expectedOutput: `{
|
expectedOutput: `{"name": "test", "value": 84}`,
|
||||||
"name": "test",
|
expectedMods: 1,
|
||||||
"value": 84
|
|
||||||
}`,
|
|
||||||
expectedMods: 1,
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "JSON array modification",
|
name: "JSON array modification",
|
||||||
input: `{"items": [{"id": 1, "value": 10}, {"id": 2, "value": 20}]}`,
|
input: `{"items": [{"name": "item1", "value": 10}, {"name": "item2", "value": 20}]}`,
|
||||||
luaExpression: `for i, item in ipairs(data.items) do data.items[i].value = item.value * 1.5 end; return true`,
|
luaExpression: `for i, item in ipairs(data.items) do item.value = item.value * 2 end modified = true`,
|
||||||
expectedOutput: `{
|
expectedOutput: `{"items": [{"name": "item1", "value": 20}, {"name": "item2", "value": 40}]}`,
|
||||||
"items": [
|
expectedMods: 2,
|
||||||
{
|
|
||||||
"id": 1,
|
|
||||||
"value": 15
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"id": 2,
|
|
||||||
"value": 30
|
|
||||||
}
|
|
||||||
]
|
|
||||||
}`,
|
|
||||||
expectedMods: 1,
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "JSON nested object modification",
|
name: "JSON nested object modification",
|
||||||
input: `{"config": {"settings": {"enabled": false, "timeout": 30}}}`,
|
input: `{"config": {"setting1": {"enabled": true, "value": 5}, "setting2": {"enabled": false, "value": 10}}}`,
|
||||||
luaExpression: `data.config.settings.enabled = true; data.config.settings.timeout = 60; return true`,
|
luaExpression: `data.config.setting1.enabled = false data.config.setting2.value = 15 modified = true`,
|
||||||
expectedOutput: `{
|
expectedOutput: `{"config": {"setting1": {"enabled": false, "value": 5}, "setting2": {"enabled": false, "value": 15}}}`,
|
||||||
"config": {
|
expectedMods: 2,
|
||||||
"settings": {
|
|
||||||
"enabled": true,
|
|
||||||
"timeout": 60
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}`,
|
|
||||||
expectedMods: 1,
|
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "JSON no modification",
|
name: "JSON no modification",
|
||||||
|
@@ -247,6 +247,7 @@ modified = false
|
|||||||
initLuaHelpersLogger.Debug("Setting up Lua print function to Go")
|
initLuaHelpersLogger.Debug("Setting up Lua print function to Go")
|
||||||
L.SetGlobal("print", L.NewFunction(printToGo))
|
L.SetGlobal("print", L.NewFunction(printToGo))
|
||||||
L.SetGlobal("fetch", L.NewFunction(fetch))
|
L.SetGlobal("fetch", L.NewFunction(fetch))
|
||||||
|
L.SetGlobal("re", L.NewFunction(EvalRegex))
|
||||||
initLuaHelpersLogger.Debug("Lua print and fetch functions bound to Go")
|
initLuaHelpersLogger.Debug("Lua print and fetch functions bound to Go")
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -481,3 +482,86 @@ func fetch(L *lua.LState) int {
|
|||||||
fetchLogger.Debug("Pushed response table to Lua stack")
|
fetchLogger.Debug("Pushed response table to Lua stack")
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func EvalRegex(L *lua.LState) int {
|
||||||
|
evalRegexLogger := processorLogger.WithPrefix("evalRegex")
|
||||||
|
evalRegexLogger.Debug("Lua evalRegex function called")
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
evalRegexLogger.Error("Panic in EvalRegex: %v", r)
|
||||||
|
// Push empty table on panic
|
||||||
|
emptyTable := L.NewTable()
|
||||||
|
L.Push(emptyTable)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
pattern := L.ToString(1)
|
||||||
|
input := L.ToString(2)
|
||||||
|
|
||||||
|
evalRegexLogger.Debug("Pattern: %q, Input: %q", pattern, input)
|
||||||
|
|
||||||
|
re := regexp.MustCompile(pattern)
|
||||||
|
matches := re.FindStringSubmatch(input)
|
||||||
|
|
||||||
|
evalRegexLogger.Debug("Go regex matches: %v (count: %d)", matches, len(matches))
|
||||||
|
|
||||||
|
matchesTable := L.NewTable()
|
||||||
|
for i, match := range matches {
|
||||||
|
matchesTable.RawSetInt(i, lua.LString(match))
|
||||||
|
evalRegexLogger.Debug("Set table[%d] = %q", i, match)
|
||||||
|
}
|
||||||
|
|
||||||
|
L.Push(matchesTable)
|
||||||
|
evalRegexLogger.Debug("Pushed matches table to Lua stack")
|
||||||
|
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetLuaFunctionsHelp returns a comprehensive help string for all available Lua functions
|
||||||
|
func GetLuaFunctionsHelp() string {
|
||||||
|
return `Lua Functions Available in Global Environment:
|
||||||
|
|
||||||
|
MATH FUNCTIONS:
|
||||||
|
min(a, b) - Returns the minimum of two numbers
|
||||||
|
max(a, b) - Returns the maximum of two numbers
|
||||||
|
round(x, n) - Rounds x to n decimal places (default 0)
|
||||||
|
floor(x) - Returns the floor of x
|
||||||
|
ceil(x) - Returns the ceiling of x
|
||||||
|
|
||||||
|
STRING FUNCTIONS:
|
||||||
|
upper(s) - Converts string to uppercase
|
||||||
|
lower(s) - Converts string to lowercase
|
||||||
|
format(s, ...) - Formats string using Lua string.format
|
||||||
|
trim(s) - Removes leading/trailing whitespace
|
||||||
|
strsplit(inputstr, sep) - Splits string by separator (default: whitespace)
|
||||||
|
num(str) - Converts string to number (returns 0 if invalid)
|
||||||
|
str(num) - Converts number to string
|
||||||
|
is_number(str) - Returns true if string is numeric
|
||||||
|
|
||||||
|
TABLE FUNCTIONS:
|
||||||
|
DumpTable(table, depth) - Prints table structure recursively
|
||||||
|
isArray(t) - Returns true if table is a sequential array
|
||||||
|
|
||||||
|
HTTP FUNCTIONS:
|
||||||
|
fetch(url, options) - Makes HTTP request, returns response table
|
||||||
|
options: {method="GET", headers={}, body=""}
|
||||||
|
returns: {status, statusText, ok, body, headers}
|
||||||
|
|
||||||
|
REGEX FUNCTIONS:
|
||||||
|
re(pattern, input) - Applies regex pattern to input string
|
||||||
|
returns: table with matches (index 0 = full match, 1+ = groups)
|
||||||
|
|
||||||
|
UTILITY FUNCTIONS:
|
||||||
|
print(...) - Prints arguments to Go logger
|
||||||
|
|
||||||
|
EXAMPLES:
|
||||||
|
round(3.14159, 2) -> 3.14
|
||||||
|
strsplit("a,b,c", ",") -> {"a", "b", "c"}
|
||||||
|
upper("hello") -> "HELLO"
|
||||||
|
min(5, 3) -> 3
|
||||||
|
num("123") -> 123
|
||||||
|
is_number("abc") -> false
|
||||||
|
fetch("https://api.example.com/data")
|
||||||
|
re("(\\w+)@(\\w+)", "user@domain.com") -> {"user@domain.com", "user", "domain.com"}`
|
||||||
|
}
|
||||||
|
162
processor/processor_test.go
Normal file
162
processor/processor_test.go
Normal file
@@ -0,0 +1,162 @@
|
|||||||
|
package processor_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
lua "github.com/yuin/gopher-lua"
|
||||||
|
|
||||||
|
"cook/processor"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Happy Path: Function correctly returns all regex capture groups as Lua table when given valid pattern and input.
|
||||||
|
func TestEvalRegex_CaptureGroupsReturned(t *testing.T) {
|
||||||
|
L := lua.NewState()
|
||||||
|
defer L.Close()
|
||||||
|
pattern := `(\w+)-(\d+)`
|
||||||
|
input := "test-42"
|
||||||
|
L.Push(lua.LString(pattern))
|
||||||
|
L.Push(lua.LString(input))
|
||||||
|
|
||||||
|
result := processor.EvalRegex(L)
|
||||||
|
|
||||||
|
assert.Equal(t, 0, result, "Expected return value to be 0")
|
||||||
|
|
||||||
|
out := L.Get(-1)
|
||||||
|
tbl, ok := out.(*lua.LTable)
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("Expected Lua table, got %T", out)
|
||||||
|
}
|
||||||
|
expected := []string{"test-42", "test", "42"}
|
||||||
|
for i, v := range expected {
|
||||||
|
val := tbl.RawGetString(fmt.Sprintf("%d", i))
|
||||||
|
assert.Equal(t, lua.LString(v), val, "Expected index %d to be %q", i, v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Happy Path: Function returns an empty Lua table when regex pattern does not match input string.
|
||||||
|
func TestEvalRegex_NoMatchReturnsEmptyTable(t *testing.T) {
|
||||||
|
L := lua.NewState()
|
||||||
|
defer L.Close()
|
||||||
|
L.Push(lua.LString(`(foo)(bar)`))
|
||||||
|
L.Push(lua.LString("no-match-here"))
|
||||||
|
|
||||||
|
result := processor.EvalRegex(L)
|
||||||
|
assert.Equal(t, 0, result)
|
||||||
|
|
||||||
|
out := L.Get(-1)
|
||||||
|
tbl, ok := out.(*lua.LTable)
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("Expected Lua table, got %T", out)
|
||||||
|
}
|
||||||
|
count := 0
|
||||||
|
tbl.ForEach(func(k, v lua.LValue) {
|
||||||
|
count++
|
||||||
|
})
|
||||||
|
assert.Zero(t, count, "Expected no items in the table for non-matching input")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Happy Path: Function handles patterns with no capture groups by returning the full match in the Lua table.
|
||||||
|
func TestEvalRegex_NoCaptureGroups(t *testing.T) {
|
||||||
|
L := lua.NewState()
|
||||||
|
defer L.Close()
|
||||||
|
pattern := `foo\d+`
|
||||||
|
input := "foo123"
|
||||||
|
L.Push(lua.LString(pattern))
|
||||||
|
L.Push(lua.LString(input))
|
||||||
|
|
||||||
|
result := processor.EvalRegex(L)
|
||||||
|
assert.Equal(t, 0, result)
|
||||||
|
|
||||||
|
out := L.Get(-1)
|
||||||
|
tbl, ok := out.(*lua.LTable)
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("Expected Lua table, got %T", out)
|
||||||
|
}
|
||||||
|
fullMatch := tbl.RawGetString("0")
|
||||||
|
assert.Equal(t, lua.LString("foo123"), fullMatch)
|
||||||
|
// There should be only the full match (index 0)
|
||||||
|
count := 0
|
||||||
|
tbl.ForEach(func(k, v lua.LValue) {
|
||||||
|
count++
|
||||||
|
})
|
||||||
|
assert.Equal(t, 1, count)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Edge Case: Function panics or errors when given an invalid regex pattern.
|
||||||
|
func TestEvalRegex_InvalidPattern(t *testing.T) {
|
||||||
|
L := lua.NewState()
|
||||||
|
defer L.Close()
|
||||||
|
pattern := `([a-z` // invalid regex
|
||||||
|
L.Push(lua.LString(pattern))
|
||||||
|
L.Push(lua.LString("someinput"))
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r == nil {
|
||||||
|
t.Error("Expected panic for invalid regex pattern, but did not panic")
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
processor.EvalRegex(L)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Edge Case: Function returns an empty Lua table when input string is empty.
|
||||||
|
func TestEvalRegex_EmptyInputString(t *testing.T) {
|
||||||
|
L := lua.NewState()
|
||||||
|
defer L.Close()
|
||||||
|
L.Push(lua.LString(`(foo)`))
|
||||||
|
L.Push(lua.LString(""))
|
||||||
|
|
||||||
|
result := processor.EvalRegex(L)
|
||||||
|
assert.Equal(t, 0, result)
|
||||||
|
|
||||||
|
out := L.Get(-1)
|
||||||
|
tbl, ok := out.(*lua.LTable)
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("Expected Lua table, got %T", out)
|
||||||
|
}
|
||||||
|
// Should be empty
|
||||||
|
count := 0
|
||||||
|
tbl.ForEach(func(k, v lua.LValue) {
|
||||||
|
count++
|
||||||
|
})
|
||||||
|
assert.Zero(t, count, "Expected empty table when input is empty")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Edge Case: Function handles nil or missing arguments gracefully without causing a runtime panic.
|
||||||
|
func TestEvalRegex_MissingArguments(t *testing.T) {
|
||||||
|
L := lua.NewState()
|
||||||
|
defer L.Close()
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
t.Errorf("Did not expect panic when arguments are missing, got: %v", r)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
// No arguments pushed at all
|
||||||
|
processor.EvalRegex(L)
|
||||||
|
// Should just not match anything or produce empty table, but must not panic
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEvalComplexRegex(t *testing.T) {
|
||||||
|
// 23:47:35.567068 processor.go:369 [g:22 ] [LUA] Pistol_Round ^((Bulk_)?(Pistol|Rifle).*?Round.*?)$
|
||||||
|
L := lua.NewState()
|
||||||
|
defer L.Close()
|
||||||
|
pattern := `^((Bulk_)?(Pistol|Rifle).*?Round.*?)$`
|
||||||
|
input := "Pistol_Round"
|
||||||
|
L.Push(lua.LString(pattern))
|
||||||
|
L.Push(lua.LString(input))
|
||||||
|
|
||||||
|
processor.EvalRegex(L)
|
||||||
|
|
||||||
|
out := L.Get(-1)
|
||||||
|
tbl, ok := out.(*lua.LTable)
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("Expected Lua table, got %T", out)
|
||||||
|
}
|
||||||
|
count := 0
|
||||||
|
tbl.ForEach(func(k, v lua.LValue) {
|
||||||
|
fmt.Println(k, v)
|
||||||
|
count++
|
||||||
|
})
|
||||||
|
assert.Equal(t, 1, count)
|
||||||
|
}
|
847
processor/surgical_json_test.go
Normal file
847
processor/surgical_json_test.go
Normal file
@@ -0,0 +1,847 @@
|
|||||||
|
package processor
|
||||||
|
|
||||||
|
import (
|
||||||
|
"cook/utils"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/google/go-cmp/cmp"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestSurgicalJSONEditing(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
content string
|
||||||
|
luaCode string
|
||||||
|
expected string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Modify single field",
|
||||||
|
content: `{
|
||||||
|
"name": "test",
|
||||||
|
"value": 42,
|
||||||
|
"description": "original"
|
||||||
|
}`,
|
||||||
|
luaCode: `
|
||||||
|
data.value = 84
|
||||||
|
modified = true
|
||||||
|
`,
|
||||||
|
expected: `{
|
||||||
|
"name": "test",
|
||||||
|
"value": 84,
|
||||||
|
"description": "original"
|
||||||
|
}`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Add new field",
|
||||||
|
content: `{
|
||||||
|
"name": "test",
|
||||||
|
"value": 42
|
||||||
|
}`,
|
||||||
|
luaCode: `
|
||||||
|
data.newField = "added"
|
||||||
|
modified = true
|
||||||
|
`,
|
||||||
|
expected: `{
|
||||||
|
"name": "test",
|
||||||
|
"value": 42
|
||||||
|
,"newField": "added"}`, // sjson.Set() adds new fields in compact format
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Modify nested field",
|
||||||
|
content: `{
|
||||||
|
"config": {
|
||||||
|
"settings": {
|
||||||
|
"enabled": false,
|
||||||
|
"timeout": 30
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}`,
|
||||||
|
luaCode: `
|
||||||
|
data.config.settings.enabled = true
|
||||||
|
data.config.settings.timeout = 60
|
||||||
|
modified = true
|
||||||
|
`,
|
||||||
|
expected: `{
|
||||||
|
"config": {
|
||||||
|
"settings": {
|
||||||
|
"enabled": true,
|
||||||
|
"timeout": 60
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
command := utils.ModifyCommand{
|
||||||
|
Name: "test",
|
||||||
|
Lua: tt.luaCode,
|
||||||
|
}
|
||||||
|
|
||||||
|
commands, err := ProcessJSON(tt.content, command, "test.json")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ProcessJSON failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(commands) == 0 {
|
||||||
|
t.Fatal("Expected at least one command")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply the commands
|
||||||
|
result := tt.content
|
||||||
|
for _, cmd := range commands {
|
||||||
|
result = result[:cmd.From] + cmd.With + result[cmd.To:]
|
||||||
|
}
|
||||||
|
|
||||||
|
diff := cmp.Diff(result, tt.expected)
|
||||||
|
if diff != "" {
|
||||||
|
t.Errorf("Differences:\n%s", diff)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check the actual result matches expected
|
||||||
|
if result != tt.expected {
|
||||||
|
t.Errorf("Expected:\n%s\n\nGot:\n%s", tt.expected, result)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSurgicalJSONPreservesFormatting(t *testing.T) {
|
||||||
|
// Test that surgical editing preserves the original formatting structure
|
||||||
|
content := `{
|
||||||
|
"Defaults": {
|
||||||
|
"Behaviour": "None",
|
||||||
|
"Description": "",
|
||||||
|
"DisplayName": "",
|
||||||
|
"FlavorText": "",
|
||||||
|
"Icon": "None",
|
||||||
|
"MaxStack": 1,
|
||||||
|
"Override_Glow_Icon": "None",
|
||||||
|
"Weight": 0,
|
||||||
|
"bAllowZeroWeight": false
|
||||||
|
},
|
||||||
|
"RowStruct": "/Script/Icarus.ItemableData",
|
||||||
|
"Rows": [
|
||||||
|
{
|
||||||
|
"Description": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-Description\", \"A bundle of soft fiber, highly useful.\")",
|
||||||
|
"DisplayName": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-DisplayName\", \"Fiber\")",
|
||||||
|
"FlavorText": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-FlavorText\", \"Fiber is collected from bast, the strong inner bark of certain flowering plants.\")",
|
||||||
|
"Icon": "/Game/Assets/2DArt/UI/Items/Item_Icons/Resources/ITEM_Fibre.ITEM_Fibre",
|
||||||
|
"MaxStack": 1000000,
|
||||||
|
"Name": "Item_Fiber",
|
||||||
|
"Weight": 10
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}`
|
||||||
|
|
||||||
|
expected := `{
|
||||||
|
"Defaults": {
|
||||||
|
"Behaviour": "None",
|
||||||
|
"Description": "",
|
||||||
|
"DisplayName": "",
|
||||||
|
"FlavorText": "",
|
||||||
|
"Icon": "None",
|
||||||
|
"MaxStack": 1,
|
||||||
|
"Override_Glow_Icon": "None",
|
||||||
|
"Weight": 0,
|
||||||
|
"bAllowZeroWeight": false
|
||||||
|
},
|
||||||
|
"RowStruct": "/Script/Icarus.ItemableData",
|
||||||
|
"Rows": [
|
||||||
|
{
|
||||||
|
"Description": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-Description\", \"A bundle of soft fiber, highly useful.\")",
|
||||||
|
"DisplayName": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-DisplayName\", \"Fiber\")",
|
||||||
|
"FlavorText": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-FlavorText\", \"Fiber is collected from bast, the strong inner bark of certain flowering plants.\")",
|
||||||
|
"Icon": "/Game/Assets/2DArt/UI/Items/Item_Icons/Resources/ITEM_Fibre.ITEM_Fibre",
|
||||||
|
"MaxStack": 1000000,
|
||||||
|
"Name": "Item_Fiber",
|
||||||
|
"Weight": 500
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}`
|
||||||
|
|
||||||
|
command := utils.ModifyCommand{
|
||||||
|
Name: "test",
|
||||||
|
Lua: `
|
||||||
|
-- Modify the weight of the first item
|
||||||
|
data.Rows[1].Weight = 500
|
||||||
|
modified = true
|
||||||
|
`,
|
||||||
|
}
|
||||||
|
|
||||||
|
commands, err := ProcessJSON(content, command, "test.json")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ProcessJSON failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(commands) == 0 {
|
||||||
|
t.Fatal("Expected at least one command")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply the commands
|
||||||
|
result := content
|
||||||
|
for _, cmd := range commands {
|
||||||
|
result = result[:cmd.From] + cmd.With + result[cmd.To:]
|
||||||
|
}
|
||||||
|
|
||||||
|
diff := cmp.Diff(result, expected)
|
||||||
|
if diff != "" {
|
||||||
|
t.Errorf("Differences:\n%s", diff)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check that the result matches expected (preserves formatting and changes weight)
|
||||||
|
if result != expected {
|
||||||
|
t.Errorf("Expected:\n%s\n\nGot:\n%s", expected, result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSurgicalJSONPreservesFormatting2(t *testing.T) {
|
||||||
|
// Test that surgical editing preserves the original formatting structure
|
||||||
|
content := `
|
||||||
|
{
|
||||||
|
"RowStruct": "/Script/Icarus.ProcessorRecipe",
|
||||||
|
"Defaults": {
|
||||||
|
"bForceDisableRecipe": false,
|
||||||
|
"Requirement": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_Talents"
|
||||||
|
},
|
||||||
|
"SessionRequirement": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_CharacterFlags"
|
||||||
|
},
|
||||||
|
"CharacterRequirement": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_CharacterFlags"
|
||||||
|
},
|
||||||
|
"RequiredMillijoules": 2500,
|
||||||
|
"RecipeSets": [],
|
||||||
|
"ResourceCostMultipliers": [],
|
||||||
|
"Inputs": [
|
||||||
|
{
|
||||||
|
"Element": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_ItemsStatic"
|
||||||
|
},
|
||||||
|
"Count": 1,
|
||||||
|
"DynamicProperties": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"Container": {
|
||||||
|
"Value": "None"
|
||||||
|
},
|
||||||
|
"ResourceInputs": [],
|
||||||
|
"bSelectOutputItemRandomly": false,
|
||||||
|
"bContainsContainer": false,
|
||||||
|
"ItemIconOverride": {
|
||||||
|
"ItemStaticData": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_ItemsStatic"
|
||||||
|
},
|
||||||
|
"ItemDynamicData": [],
|
||||||
|
"ItemCustomStats": [],
|
||||||
|
"CustomProperties": {
|
||||||
|
"StaticWorldStats": [],
|
||||||
|
"StaticWorldHeldStats": [],
|
||||||
|
"Stats": [],
|
||||||
|
"Alterations": [],
|
||||||
|
"LivingItemSlots": []
|
||||||
|
},
|
||||||
|
"DatabaseGUID": "",
|
||||||
|
"ItemOwnerLookupId": -1,
|
||||||
|
"RuntimeTags": {
|
||||||
|
"GameplayTags": []
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"Outputs": [
|
||||||
|
{
|
||||||
|
"Element": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_ItemTemplate"
|
||||||
|
},
|
||||||
|
"Count": 1,
|
||||||
|
"DynamicProperties": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"ResourceOutputs": [],
|
||||||
|
"Refundable": "Inherit",
|
||||||
|
"ExperienceMultiplier": 1,
|
||||||
|
"Audio": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_CraftingAudioData"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"Rows": [
|
||||||
|
{
|
||||||
|
"Name": "Biofuel1",
|
||||||
|
"RecipeSets": [
|
||||||
|
{
|
||||||
|
"RowName": "Composter",
|
||||||
|
"DataTableName": "D_RecipeSets"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"Inputs": [
|
||||||
|
{
|
||||||
|
"Element": {
|
||||||
|
"RowName": "Raw_Meat",
|
||||||
|
"DataTableName": "D_ItemsStatic"
|
||||||
|
},
|
||||||
|
"Count": 2,
|
||||||
|
"DynamicProperties": []
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"Element": {
|
||||||
|
"RowName": "Tree_Sap",
|
||||||
|
"DataTableName": "D_ItemsStatic"
|
||||||
|
},
|
||||||
|
"Count": 1,
|
||||||
|
"DynamicProperties": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"Outputs": [],
|
||||||
|
"Audio": {
|
||||||
|
"RowName": "Composter"
|
||||||
|
},
|
||||||
|
"ResourceOutputs": [
|
||||||
|
{
|
||||||
|
"Type": {
|
||||||
|
"Value": "Biofuel"
|
||||||
|
},
|
||||||
|
"RequiredUnits": 100
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
expected := `
|
||||||
|
{
|
||||||
|
"RowStruct": "/Script/Icarus.ProcessorRecipe",
|
||||||
|
"Defaults": {
|
||||||
|
"bForceDisableRecipe": false,
|
||||||
|
"Requirement": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_Talents"
|
||||||
|
},
|
||||||
|
"SessionRequirement": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_CharacterFlags"
|
||||||
|
},
|
||||||
|
"CharacterRequirement": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_CharacterFlags"
|
||||||
|
},
|
||||||
|
"RequiredMillijoules": 2500,
|
||||||
|
"RecipeSets": [],
|
||||||
|
"ResourceCostMultipliers": [],
|
||||||
|
"Inputs": [
|
||||||
|
{
|
||||||
|
"Element": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_ItemsStatic"
|
||||||
|
},
|
||||||
|
"Count": 1,
|
||||||
|
"DynamicProperties": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"Container": {
|
||||||
|
"Value": "None"
|
||||||
|
},
|
||||||
|
"ResourceInputs": [],
|
||||||
|
"bSelectOutputItemRandomly": false,
|
||||||
|
"bContainsContainer": false,
|
||||||
|
"ItemIconOverride": {
|
||||||
|
"ItemStaticData": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_ItemsStatic"
|
||||||
|
},
|
||||||
|
"ItemDynamicData": [],
|
||||||
|
"ItemCustomStats": [],
|
||||||
|
"CustomProperties": {
|
||||||
|
"StaticWorldStats": [],
|
||||||
|
"StaticWorldHeldStats": [],
|
||||||
|
"Stats": [],
|
||||||
|
"Alterations": [],
|
||||||
|
"LivingItemSlots": []
|
||||||
|
},
|
||||||
|
"DatabaseGUID": "",
|
||||||
|
"ItemOwnerLookupId": -1,
|
||||||
|
"RuntimeTags": {
|
||||||
|
"GameplayTags": []
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"Outputs": [
|
||||||
|
{
|
||||||
|
"Element": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_ItemTemplate"
|
||||||
|
},
|
||||||
|
"Count": 1,
|
||||||
|
"DynamicProperties": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"ResourceOutputs": [],
|
||||||
|
"Refundable": "Inherit",
|
||||||
|
"ExperienceMultiplier": 1,
|
||||||
|
"Audio": {
|
||||||
|
"RowName": "None",
|
||||||
|
"DataTableName": "D_CraftingAudioData"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"Rows": [
|
||||||
|
{
|
||||||
|
"Name": "Biofuel1",
|
||||||
|
"RecipeSets": [
|
||||||
|
{
|
||||||
|
"RowName": "Composter",
|
||||||
|
"DataTableName": "D_RecipeSets"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"Inputs": [
|
||||||
|
{
|
||||||
|
"Element": {
|
||||||
|
"RowName": "Raw_Meat",
|
||||||
|
"DataTableName": "D_ItemsStatic"
|
||||||
|
},
|
||||||
|
"Count": 2,
|
||||||
|
"DynamicProperties": []
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"Outputs": [],
|
||||||
|
"Audio": {
|
||||||
|
"RowName": "Composter"
|
||||||
|
},
|
||||||
|
"ResourceOutputs": [
|
||||||
|
{
|
||||||
|
"Type": {
|
||||||
|
"Value": "Biofuel"
|
||||||
|
},
|
||||||
|
"RequiredUnits": 100
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
command := utils.ModifyCommand{
|
||||||
|
Name: "test",
|
||||||
|
Lua: `
|
||||||
|
-- Define regex patterns for matching recipe names
|
||||||
|
local function matchesPattern(name, pattern)
|
||||||
|
local matches = re(pattern, name)
|
||||||
|
-- Check if matches table has any content (index 0 or 1 should exist if there's a match)
|
||||||
|
return matches and (matches[0] or matches[1])
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Selection pattern for recipes that get multiplied
|
||||||
|
local selectionPattern = "(?-s)(Bulk_)?(Pistol|Rifle).*?Round.*?|(Carbon|Composite)_Paste.*|(Gold|Copper)_Wire|(Ironw|Copper)_Nail|(Platinum|Steel|Cold_Steel|Titanium)_Ingot|.*?Shotgun_Shell.*?|.*_Arrow|.*_Bolt|.*_Fertilizer_?\\d*|.*_Grenade|.*_Pill|.*_Tonic|Aluminum|Ammo_Casing|Animal_Fat|Carbon_Fiber|Composites|Concrete_Mix|Cured_Leather_?\\d?|Electronics|Epoxy_?\\d?|Glass\\d?|Gunpowder\\w*|Health_.*|Titanium_Plate|Organic_Resin|Platinum_Sheath|Refined_[a-zA-Z]+|Rope|Shotgun_Casing|Steel_Bloom\\d?|Tree_Sap\\w*"
|
||||||
|
|
||||||
|
-- Ingot pattern for recipes that get count set to 1
|
||||||
|
local ingotPattern = "(?-s)(Platinum|Steel|Cold_Steel|Titanium)_Ingot|Aluminum|Refined_[a-zA-Z]+|Glass\\d?"
|
||||||
|
|
||||||
|
local factor = 16
|
||||||
|
local bonus = 0.5
|
||||||
|
|
||||||
|
for _, row in ipairs(data.Rows) do
|
||||||
|
local recipeName = row.Name
|
||||||
|
|
||||||
|
-- Special case: Biofuel recipes - remove Tree_Sap input
|
||||||
|
if string.find(recipeName, "Biofuel") then
|
||||||
|
if row.Inputs then
|
||||||
|
for i = #row.Inputs, 1, -1 do
|
||||||
|
local input = row.Inputs[i]
|
||||||
|
if input.Element and input.Element.RowName and string.find(input.Element.RowName, "Tree_Sap") then
|
||||||
|
table.remove(row.Inputs, i)
|
||||||
|
print("Removing input 'Tree_Sap' from processor recipe '" .. recipeName .. "'")
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Ingot recipes: set input and output counts to 1
|
||||||
|
if matchesPattern(recipeName, ingotPattern) then
|
||||||
|
if row.Inputs then
|
||||||
|
for _, input in ipairs(row.Inputs) do
|
||||||
|
input.Count = 1
|
||||||
|
end
|
||||||
|
end
|
||||||
|
if row.Outputs then
|
||||||
|
for _, output in ipairs(row.Outputs) do
|
||||||
|
output.Count = 1
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
-- Selected recipes: multiply inputs by factor, outputs by factor * (1 + bonus)
|
||||||
|
if matchesPattern(recipeName, selectionPattern) then
|
||||||
|
if row.Inputs then
|
||||||
|
for _, input in ipairs(row.Inputs) do
|
||||||
|
local oldCount = input.Count
|
||||||
|
input.Count = input.Count * factor
|
||||||
|
print("Recipe " .. recipeName .. " Input.Count: " .. oldCount .. " -> " .. input.Count)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
if row.Outputs then
|
||||||
|
for _, output in ipairs(row.Outputs) do
|
||||||
|
local oldCount = output.Count
|
||||||
|
output.Count = math.floor(output.Count * factor * (1 + bonus))
|
||||||
|
print("Recipe " .. recipeName .. " Output.Count: " .. oldCount .. " -> " .. output.Count)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
`,
|
||||||
|
}
|
||||||
|
|
||||||
|
commands, err := ProcessJSON(content, command, "test.json")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ProcessJSON failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(commands) == 0 {
|
||||||
|
t.Fatal("Expected at least one command")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply the commands
|
||||||
|
result := content
|
||||||
|
for _, cmd := range commands {
|
||||||
|
result = result[:cmd.From] + cmd.With + result[cmd.To:]
|
||||||
|
}
|
||||||
|
|
||||||
|
diff := cmp.Diff(result, expected)
|
||||||
|
if diff != "" {
|
||||||
|
t.Errorf("Differences:\n%s", diff)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check that the result matches expected (preserves formatting and changes weight)
|
||||||
|
if result != expected {
|
||||||
|
t.Errorf("Expected:\n%s\n\nGot:\n%s", expected, result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRetardedJSONEditing(t *testing.T) {
|
||||||
|
original := `{
|
||||||
|
"RowStruct": "/Script/Icarus.ItemableData",
|
||||||
|
"Defaults": {
|
||||||
|
"Behaviour": "None",
|
||||||
|
"DisplayName": "",
|
||||||
|
"Icon": "None",
|
||||||
|
"Override_Glow_Icon": "None",
|
||||||
|
"Description": "",
|
||||||
|
"FlavorText": "",
|
||||||
|
"Weight": 0,
|
||||||
|
"bAllowZeroWeight": false,
|
||||||
|
"MaxStack": 1
|
||||||
|
},
|
||||||
|
"Rows": [
|
||||||
|
{
|
||||||
|
"DisplayName": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-DisplayName\", \"Fiber\")",
|
||||||
|
"Icon": "/Game/Assets/2DArt/UI/Items/Item_Icons/Resources/ITEM_Fibre.ITEM_Fibre",
|
||||||
|
"Description": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-Description\", \"A bundle of soft fiber, highly useful.\")",
|
||||||
|
"FlavorText": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-FlavorText\", \"Fiber is collected from bast, the strong inner bark of certain flowering plants.\")",
|
||||||
|
"Weight": 10,
|
||||||
|
"MaxStack": 200,
|
||||||
|
"Name": "Item_Fiber"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}`
|
||||||
|
|
||||||
|
expected := `{
|
||||||
|
"RowStruct": "/Script/Icarus.ItemableData",
|
||||||
|
"Defaults": {
|
||||||
|
"Behaviour": "None",
|
||||||
|
"DisplayName": "",
|
||||||
|
"Icon": "None",
|
||||||
|
"Override_Glow_Icon": "None",
|
||||||
|
"Description": "",
|
||||||
|
"FlavorText": "",
|
||||||
|
"Weight": 0,
|
||||||
|
"bAllowZeroWeight": false,
|
||||||
|
"MaxStack": 1
|
||||||
|
},
|
||||||
|
"Rows": [
|
||||||
|
{
|
||||||
|
"DisplayName": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-DisplayName\", \"Fiber\")",
|
||||||
|
"Icon": "/Game/Assets/2DArt/UI/Items/Item_Icons/Resources/ITEM_Fibre.ITEM_Fibre",
|
||||||
|
"Description": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-Description\", \"A bundle of soft fiber, highly useful.\")",
|
||||||
|
"FlavorText": "NSLOCTEXT(\"D_Itemable\", \"Item_Fiber-FlavorText\", \"Fiber is collected from bast, the strong inner bark of certain flowering plants.\")",
|
||||||
|
"Weight": 10,
|
||||||
|
"MaxStack": 1000000,
|
||||||
|
"Name": "Item_Fiber"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}`
|
||||||
|
|
||||||
|
command := utils.ModifyCommand{
|
||||||
|
Name: "test",
|
||||||
|
Lua: `
|
||||||
|
for _, row in ipairs(data.Rows) do
|
||||||
|
if row.MaxStack then
|
||||||
|
if string.find(row.Name, "Carrot") or string.find(row.Name, "Potato") then
|
||||||
|
row.MaxStack = 25
|
||||||
|
else
|
||||||
|
row.MaxStack = row.MaxStack * 10000
|
||||||
|
if row.MaxStack > 1000000 then
|
||||||
|
row.MaxStack = 1000000
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
`,
|
||||||
|
}
|
||||||
|
|
||||||
|
commands, err := ProcessJSON(original, command, "test.json")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ProcessJSON failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(commands) == 0 {
|
||||||
|
t.Fatal("Expected at least one command")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply the commands
|
||||||
|
result := original
|
||||||
|
for _, cmd := range commands {
|
||||||
|
result = result[:cmd.From] + cmd.With + result[cmd.To:]
|
||||||
|
}
|
||||||
|
|
||||||
|
diff := cmp.Diff(result, expected)
|
||||||
|
if diff != "" {
|
||||||
|
t.Errorf("Differences:\n%s", diff)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check that the weight was changed
|
||||||
|
if result != expected {
|
||||||
|
t.Errorf("Expected:\n%s\nGot:\n%s", expected, result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRetardedJSONEditing2(t *testing.T) {
|
||||||
|
original := `
|
||||||
|
{
|
||||||
|
"Rows": [
|
||||||
|
{
|
||||||
|
"Name": "Deep_Mining_Drill_Biofuel",
|
||||||
|
"Meshable": {
|
||||||
|
"RowName": "Mesh_Deep_Mining_Drill_Biofuel"
|
||||||
|
},
|
||||||
|
"Itemable": {
|
||||||
|
"RowName": "Item_Deep_Mining_Drill_Biofuel"
|
||||||
|
},
|
||||||
|
"Interactable": {
|
||||||
|
"RowName": "Deployable"
|
||||||
|
},
|
||||||
|
"Focusable": {
|
||||||
|
"RowName": "Focusable_1H"
|
||||||
|
},
|
||||||
|
"Highlightable": {
|
||||||
|
"RowName": "Generic"
|
||||||
|
},
|
||||||
|
"Actionable": {
|
||||||
|
"RowName": "Deployable"
|
||||||
|
},
|
||||||
|
"Usable": {
|
||||||
|
"RowName": "Place"
|
||||||
|
},
|
||||||
|
"Deployable": {
|
||||||
|
"RowName": "Deep_Mining_Drill_Biofuel"
|
||||||
|
},
|
||||||
|
"Durable": {
|
||||||
|
"RowName": "Deployable_750"
|
||||||
|
},
|
||||||
|
"Inventory": {
|
||||||
|
"RowName": "Deep_Mining_Drill_Biofuel"
|
||||||
|
},
|
||||||
|
"Decayable": {
|
||||||
|
"RowName": "Decay_MetaItem"
|
||||||
|
},
|
||||||
|
"Generator": {
|
||||||
|
"RowName": "Deep_Mining_Biofuel_Drill"
|
||||||
|
},
|
||||||
|
"Resource": {
|
||||||
|
"RowName": "Simple_Internal_Flow_Only"
|
||||||
|
},
|
||||||
|
"Manual_Tags": {
|
||||||
|
"GameplayTags": [
|
||||||
|
{
|
||||||
|
"TagName": "Item.Machine"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"Generated_Tags": {
|
||||||
|
"GameplayTags": [
|
||||||
|
{
|
||||||
|
"TagName": "Item.Machine"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Meshable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Itemable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Interactable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Highlightable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Actionable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Usable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Deployable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Durable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Inventory"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"ParentTags": []
|
||||||
|
}
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
expected := `
|
||||||
|
{
|
||||||
|
"Rows": [
|
||||||
|
{
|
||||||
|
"Name": "Deep_Mining_Drill_Biofuel",
|
||||||
|
"Meshable": {
|
||||||
|
"RowName": "Mesh_Deep_Mining_Drill_Biofuel"
|
||||||
|
},
|
||||||
|
"Itemable": {
|
||||||
|
"RowName": "Item_Deep_Mining_Drill_Biofuel"
|
||||||
|
},
|
||||||
|
"Interactable": {
|
||||||
|
"RowName": "Deployable"
|
||||||
|
},
|
||||||
|
"Focusable": {
|
||||||
|
"RowName": "Focusable_1H"
|
||||||
|
},
|
||||||
|
"Highlightable": {
|
||||||
|
"RowName": "Generic"
|
||||||
|
},
|
||||||
|
"Actionable": {
|
||||||
|
"RowName": "Deployable"
|
||||||
|
},
|
||||||
|
"Usable": {
|
||||||
|
"RowName": "Place"
|
||||||
|
},
|
||||||
|
"Deployable": {
|
||||||
|
"RowName": "Deep_Mining_Drill_Biofuel"
|
||||||
|
},
|
||||||
|
"Durable": {
|
||||||
|
"RowName": "Deployable_750"
|
||||||
|
},
|
||||||
|
"Inventory": {
|
||||||
|
"RowName": "Deep_Mining_Drill_Biofuel"
|
||||||
|
},
|
||||||
|
"Decayable": {
|
||||||
|
"RowName": "Decay_MetaItem"
|
||||||
|
},
|
||||||
|
"Generator": {
|
||||||
|
"RowName": "Deep_Mining_Biofuel_Drill"
|
||||||
|
},
|
||||||
|
"Resource": {
|
||||||
|
"RowName": "Simple_Internal_Flow_Only"
|
||||||
|
},
|
||||||
|
"Manual_Tags": {
|
||||||
|
"GameplayTags": [
|
||||||
|
{
|
||||||
|
"TagName": "Item.Machine"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"Generated_Tags": {
|
||||||
|
"GameplayTags": [
|
||||||
|
{
|
||||||
|
"TagName": "Item.Machine"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Meshable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Itemable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Interactable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Highlightable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Actionable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Usable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Deployable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Durable"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"TagName": "Traits.Inventory"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"ParentTags": []
|
||||||
|
}
|
||||||
|
,"AdditionalStats": {"(Value=\"BaseDeepMiningDrillSpeed_+%\")":4000}}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
command := utils.ModifyCommand{
|
||||||
|
Name: "test",
|
||||||
|
Lua: `
|
||||||
|
for i, row in ipairs(data.Rows) do
|
||||||
|
-- Special case: Deep_Mining_Drill_Biofuel
|
||||||
|
if string.find(row.Name, "Deep_Mining_Drill_Biofuel") then
|
||||||
|
print("[DEBUG] Special case: Deep_Mining_Drill_Biofuel")
|
||||||
|
if not row.AdditionalStats then
|
||||||
|
print("[DEBUG] Creating AdditionalStats table for Deep_Mining_Drill_Biofuel")
|
||||||
|
row.AdditionalStats = {}
|
||||||
|
end
|
||||||
|
print("[DEBUG] Setting BaseDeepMiningDrillSpeed_+% to 4000")
|
||||||
|
row.AdditionalStats["(Value=\\\"BaseDeepMiningDrillSpeed_+%\\\")"] = 4000
|
||||||
|
end
|
||||||
|
end
|
||||||
|
`,
|
||||||
|
}
|
||||||
|
|
||||||
|
commands, err := ProcessJSON(original, command, "test.json")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ProcessJSON failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(commands) == 0 {
|
||||||
|
t.Fatal("Expected at least one command")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply the commands
|
||||||
|
result := original
|
||||||
|
for _, cmd := range commands {
|
||||||
|
result = result[:cmd.From] + cmd.With + result[cmd.To:]
|
||||||
|
}
|
||||||
|
|
||||||
|
diff := cmp.Diff(result, expected)
|
||||||
|
if diff != "" {
|
||||||
|
t.Errorf("Differences:\n%s", diff)
|
||||||
|
}
|
||||||
|
|
||||||
|
if result != expected {
|
||||||
|
t.Errorf("Expected:\n%s\nGot:\n%s", expected, result)
|
||||||
|
}
|
||||||
|
}
|
11
test_surgical.yml
Normal file
11
test_surgical.yml
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
- name: SurgicalWeightTest
|
||||||
|
json: true
|
||||||
|
lua: |
|
||||||
|
-- This demonstrates surgical JSON editing
|
||||||
|
-- Only the Weight field of Item_Fiber will be modified
|
||||||
|
data.Rows[1].Weight = 999
|
||||||
|
modified = true
|
||||||
|
files:
|
||||||
|
- 'D_Itemable.json'
|
||||||
|
reset: false
|
||||||
|
loglevel: INFO
|
Reference in New Issue
Block a user