Implement a more better logging solution

processor/json.go

@@ -3,7 +3,7 @@ package processor
 import (
 	"encoding/json"
 	"fmt"
-	"log"
+	"modify/logger"
 	"modify/processor/jsonpath"
 
 	lua "github.com/yuin/gopher-lua"
@@ -14,80 +14,98 @@ type JSONProcessor struct{}
 
 // ProcessContent implements the Processor interface for JSONProcessor
 func (p *JSONProcessor) ProcessContent(content string, pattern string, luaExpr string) (string, int, int, error) {
+	logger.Debug("Processing JSON content with JSONPath: %s", pattern)
+
 	// Parse JSON document
+	logger.Trace("Parsing JSON document")
 	var jsonData interface{}
 	err := json.Unmarshal([]byte(content), &jsonData)
 	if err != nil {
+		logger.Error("Failed to parse JSON: %v", err)
 		return content, 0, 0, fmt.Errorf("error parsing JSON: %v", err)
 	}
 
 	// Find nodes matching the JSONPath pattern
+	logger.Debug("Executing JSONPath query: %s", pattern)
 	nodes, err := jsonpath.Get(jsonData, pattern)
 	if err != nil {
+		logger.Error("Failed to execute JSONPath: %v", err)
 		return content, 0, 0, fmt.Errorf("error getting nodes: %v", err)
 	}
 
 	matchCount := len(nodes)
+	logger.Debug("Found %d nodes matching JSONPath", matchCount)
 	if matchCount == 0 {
+		logger.Warning("No nodes matched the JSONPath pattern: %s", pattern)
 		return content, 0, 0, nil
 	}
 
 	modCount := 0
-	for _, node := range nodes {
-		log.Printf("Processing node at path: %s with value: %v", node.Path, node.Value)
+	for i, node := range nodes {
+		logger.Trace("Processing node #%d at path: %s with value: %v", i+1, node.Path, node.Value)
 
 		// Initialize Lua
 		L, err := NewLuaState()
 		if err != nil {
+			logger.Error("Failed to create Lua state: %v", err)
 			return content, len(nodes), 0, fmt.Errorf("error creating Lua state: %v", err)
 		}
 		defer L.Close()
-		log.Println("Lua state initialized successfully.")
+		logger.Trace("Lua state initialized successfully")
 
 		err = p.ToLua(L, node.Value)
 		if err != nil {
+			logger.Error("Failed to convert value to Lua: %v", err)
 			return content, len(nodes), 0, fmt.Errorf("error converting to Lua: %v", err)
 		}
-		log.Printf("Converted node value to Lua: %v", node.Value)
+		logger.Trace("Converted node value to Lua: %v", node.Value)
 
 		originalScript := luaExpr
 		fullScript := BuildLuaScript(luaExpr)
-		log.Printf("Original script: %q, Full script: %q", originalScript, fullScript)
+		logger.Debug("Original script: %q, Full script: %q", originalScript, fullScript)
 
 		// Execute Lua script
-		log.Printf("Executing Lua script: %q", fullScript)
+		logger.Trace("Executing Lua script: %q", fullScript)
 		if err := L.DoString(fullScript); err != nil {
+			logger.Error("Failed to execute Lua script: %v", err)
 			return content, len(nodes), 0, fmt.Errorf("error executing Lua %q: %v", fullScript, err)
 		}
-		log.Println("Lua script executed successfully.")
+		logger.Trace("Lua script executed successfully")
 
 		// Get modified value
 		result, err := p.FromLua(L)
 		if err != nil {
+			logger.Error("Failed to get result from Lua: %v", err)
 			return content, len(nodes), 0, fmt.Errorf("error getting result from Lua: %v", err)
 		}
-		log.Printf("Retrieved modified value from Lua: %v", result)
+		logger.Trace("Retrieved modified value from Lua: %v", result)
 
 		modified := false
 		modified = L.GetGlobal("modified").String() == "true"
 		if !modified {
-			log.Printf("No changes made to node at path: %s", node.Path)
+			logger.Debug("No changes made to node at path: %s", node.Path)
 			continue
 		}
 
 		// Apply the modification to the JSON data
+		logger.Debug("Updating JSON at path: %s with new value: %v", node.Path, result)
 		err = p.updateJSONValue(jsonData, node.Path, result)
 		if err != nil {
+			logger.Error("Failed to update JSON at path %s: %v", node.Path, err)
 			return content, len(nodes), 0, fmt.Errorf("error updating JSON: %v", err)
 		}
-		log.Printf("Updated JSON at path: %s with new value: %v", node.Path, result)
+		logger.Debug("Updated JSON at path: %s successfully", node.Path)
 		modCount++
 	}
 
+	logger.Info("JSON processing complete: %d modifications from %d matches", modCount, matchCount)
+
 	// Convert the modified JSON back to a string with same formatting
+	logger.Trace("Marshalling JSON data back to string")
 	var jsonBytes []byte
 	jsonBytes, err = json.MarshalIndent(jsonData, "", "  ")
 	if err != nil {
+		logger.Error("Failed to marshal JSON: %v", err)
 		return content, modCount, matchCount, fmt.Errorf("error marshalling JSON: %v", err)
 	}
 	return string(jsonBytes), modCount, matchCount, nil
@@ -95,8 +113,11 @@ func (p *JSONProcessor) ProcessContent(content string, pattern string, luaExpr s
 
 // updateJSONValue updates a value in the JSON structure based on its JSONPath
 func (p *JSONProcessor) updateJSONValue(jsonData interface{}, path string, newValue interface{}) error {
+	logger.Trace("Updating JSON value at path: %s", path)
+
 	// Special handling for root node
 	if path == "$" {
+		logger.Debug("Handling special case for root node update")
 		// For the root node, we'll copy the value to the jsonData reference
 		// This is a special case since we can't directly replace the interface{} variable
 
@@ -108,15 +129,18 @@ func (p *JSONProcessor) updateJSONValue(jsonData interface{}, path string, newVa
 			if !ok {
 				// If the original wasn't a map, completely replace it with the new map
 				// This is handled by the jsonpath.Set function
+				logger.Debug("Root was not a map, replacing entire root")
 				return jsonpath.Set(jsonData, path, newValue)
 			}
 
 			// Clear the original map
+			logger.Trace("Clearing original root map")
 			for k := range rootMap {
 				delete(rootMap, k)
 			}
 
 			// Copy all keys from the new map
+			logger.Trace("Copying keys to root map")
 			for k, v := range rootValue {
 				rootMap[k] = v
 			}
@@ -127,22 +151,27 @@ func (p *JSONProcessor) updateJSONValue(jsonData interface{}, path string, newVa
 			rootArray, ok := jsonData.([]interface{})
 			if !ok {
 				// If the original wasn't an array, use jsonpath.Set
+				logger.Debug("Root was not an array, replacing entire root")
 				return jsonpath.Set(jsonData, path, newValue)
 			}
 
 			// Clear and recreate the array
+			logger.Trace("Replacing root array")
 			*&rootArray = rootValue
 			return nil
 
 		default:
 			// For other types, use jsonpath.Set
+			logger.Debug("Replacing root with primitive value")
 			return jsonpath.Set(jsonData, path, newValue)
 		}
 	}
 
 	// For non-root paths, use the regular Set method
+	logger.Trace("Using regular Set method for non-root path")
 	err := jsonpath.Set(jsonData, path, newValue)
 	if err != nil {
+		logger.Error("Failed to set JSON value at path %s: %v", path, err)
 		return fmt.Errorf("failed to update JSON value at path '%s': %w", path, err)
 	}
 	return nil