Implement loading lua from separate files via @

processor/lua_external_integration_test.go

@@ -0,0 +1,191 @@
+package processor
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"cook/utils"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestProcessRegexWithExternalLuaFile(t *testing.T) {
+	tmpDir, err := os.MkdirTemp("", "lua-external-integration-test-*")
+	assert.NoError(t, err)
+	defer os.RemoveAll(tmpDir)
+
+	// Create a test Lua file with replacement variable
+	luaFile := filepath.Join(tmpDir, "multiply.lua")
+	luaContent := `v1 = v1 * 2
+replacement = format("<value>%s</value>", v1)
+return true`
+	err = os.WriteFile(luaFile, []byte(luaContent), 0644)
+	assert.NoError(t, err)
+
+	// Create test content
+	content := `<value>10</value>`
+
+	// Create command with external Lua reference
+	command := utils.ModifyCommand{
+		Name:      "test",
+		Regex:     `<value>(\d+)</value>`,
+		Lua:       "@" + filepath.Base(luaFile),
+		SourceDir: tmpDir,
+	}
+
+	// Process
+	modifications, err := ProcessRegex(content, command, "test.xml")
+	assert.NoError(t, err)
+	assert.Greater(t, len(modifications), 0)
+	
+	// Apply modifications
+	result := content
+	for _, mod := range modifications {
+		result = result[:mod.From] + mod.With + result[mod.To:]
+	}
+	assert.Contains(t, result, "<value>20</value>")
+}
+
+func TestProcessJSONWithExternalLuaFile(t *testing.T) {
+	tmpDir, err := os.MkdirTemp("", "lua-external-json-integration-test-*")
+	assert.NoError(t, err)
+	defer os.RemoveAll(tmpDir)
+
+	// Create a test Lua file
+	luaFile := filepath.Join(tmpDir, "json_modify.lua")
+	luaContent := `data.value = 84
+modified = true`
+	err = os.WriteFile(luaFile, []byte(luaContent), 0644)
+	assert.NoError(t, err)
+
+	// Create test JSON content
+	content := `{"value": 42}`
+
+	// Create command with external Lua reference
+	command := utils.ModifyCommand{
+		Name:      "test",
+		JSON:      true,
+		Lua:       "@" + filepath.Base(luaFile),
+		SourceDir: tmpDir,
+	}
+
+	// Process
+	modifications, err := ProcessJSON(content, command, "test.json")
+	assert.NoError(t, err)
+	assert.Greater(t, len(modifications), 0)
+
+	// Apply modifications to verify
+	result := content
+	for _, mod := range modifications {
+		result = result[:mod.From] + mod.With + result[mod.To:]
+	}
+	// Check that value was changed to 84 (formatting may vary)
+	assert.Contains(t, result, `"value"`)
+	assert.Contains(t, result, `84`)
+	assert.NotContains(t, result, `"value": 42`)
+	assert.NotContains(t, result, `"value":42`)
+}
+
+func TestProcessXMLWithExternalLuaFile(t *testing.T) {
+	tmpDir, err := os.MkdirTemp("", "lua-external-xml-integration-test-*")
+	assert.NoError(t, err)
+	defer os.RemoveAll(tmpDir)
+
+	// Create a test Lua file (XML uses 'root' not 'data')
+	luaFile := filepath.Join(tmpDir, "xml_modify.lua")
+	luaContent := `visitElements(root, function(elem)
+    if elem._tag == "Item" then
+        modifyNumAttr(elem, "Weight", function(val) return val * 2 end)
+    end
+end)
+modified = true`
+	err = os.WriteFile(luaFile, []byte(luaContent), 0644)
+	assert.NoError(t, err)
+
+	// Create test XML content
+	content := `<Items><Item Weight="10" /></Items>`
+
+	// Create command with external Lua reference
+	command := utils.ModifyCommand{
+		Name:      "test",
+		Lua:       "@" + filepath.Base(luaFile),
+		SourceDir: tmpDir,
+	}
+
+	// Process
+	modifications, err := ProcessXML(content, command, "test.xml")
+	assert.NoError(t, err)
+	assert.Greater(t, len(modifications), 0)
+
+	// Apply modifications to verify
+	result := content
+	for _, mod := range modifications {
+		result = result[:mod.From] + mod.With + result[mod.To:]
+	}
+	assert.Contains(t, result, `Weight="20"`)
+}
+
+func TestExternalLuaFileWithVariables(t *testing.T) {
+	tmpDir, err := os.MkdirTemp("", "lua-external-vars-integration-test-*")
+	assert.NoError(t, err)
+	defer os.RemoveAll(tmpDir)
+
+	// Create a test Lua file with variable reference
+	luaFile := filepath.Join(tmpDir, "with_vars.lua")
+	luaContent := `v1 = v1 * $multiply
+replacement = format("<value>%s</value>", v1)
+return true`
+	err = os.WriteFile(luaFile, []byte(luaContent), 0644)
+	assert.NoError(t, err)
+
+	// Set global variable
+	SetVariables(map[string]interface{}{"multiply": 1.5})
+	defer SetVariables(map[string]interface{}{})
+
+	// Create test content
+	content := `<value>10</value>`
+
+	// Create command with external Lua reference
+	command := utils.ModifyCommand{
+		Name:      "test",
+		Regex:     `<value>(\d+)</value>`,
+		Lua:       "@" + filepath.Base(luaFile),
+		SourceDir: tmpDir,
+	}
+
+	// Process
+	modifications, err := ProcessRegex(content, command, "test.xml")
+	assert.NoError(t, err)
+	assert.Greater(t, len(modifications), 0)
+	
+	// Apply modifications
+	result := content
+	for _, mod := range modifications {
+		result = result[:mod.From] + mod.With + result[mod.To:]
+	}
+	assert.Contains(t, result, "<value>15</value>")
+}
+
+func TestExternalLuaFileErrorHandling(t *testing.T) {
+	tmpDir, err := os.MkdirTemp("", "lua-external-error-test-*")
+	assert.NoError(t, err)
+	defer os.RemoveAll(tmpDir)
+
+	// Create command with non-existent external Lua file
+	command := utils.ModifyCommand{
+		Name:      "test",
+		Regex:     `<value>(\d+)</value>`,
+		Lua:       "@nonexistent.lua",
+		SourceDir: tmpDir,
+	}
+
+	// Process - the error script will be generated but execution will fail
+	// ProcessRegex continues on Lua errors, so no modifications will be made
+	content := `<value>10</value>`
+	modifications, err := ProcessRegex(content, command, "test.xml")
+	
+	// No error returned (ProcessRegex continues on Lua errors), but no modifications made
+	assert.NoError(t, err)
+	assert.Empty(t, modifications)
+}