integrate clustertool sourcecode

clustertool/fix.py

@@ -0,0 +1,63 @@
+import re
+import os
+
+# Path to the directory with Go files
+directory = "."
+
+# Updated regex pattern to find incorrect Msgf statements
+pattern = re.compile(r'log\.Info\(\)\.Msgf\((\"[^\"]*\")((?:,\s*(?:\"[^\"]*\"|\w+))*)\)')
+
+# Function to embed string literals and handle existing %s or %v placeholders
+def replacer(match):
+    message = match.group(1)
+    args = match.group(2).strip().split(',')[1:]  # Extract arguments
+
+    formatted_message = message[:-1]  # Remove the closing quote for now
+    new_args = []
+    placeholders = []
+
+    # Count the number of existing %s and %v in the message
+    existing_placeholders = re.findall(r'%[sv]', formatted_message)
+
+    for arg in args:
+        arg = arg.strip()
+        if re.match(r'\"[^\"]*\"', arg):  # If the argument is a string literal
+            formatted_message += " " + arg[1:-1]  # Embed the string literal without quotes
+        else:  # If the argument is a variable
+            # Determine which placeholder to use
+            if "str" in arg.lower():  # Customize this check as needed
+                placeholders.append('%s')
+            else:
+                placeholders.append('%v')
+
+            new_args.append(arg)
+
+    # Insert additional placeholders, only if we haven't exceeded existing placeholders
+    for _ in range(len(placeholders)):
+        if len(existing_placeholders) > 0:
+            existing_placeholders.pop(0)  # Consume an existing placeholder
+        else:
+            formatted_message += " " + placeholders.pop(0)
+
+    # Close the format string and reassemble the log statement
+    formatted_message += '",'
+
+    # Add any remaining arguments
+    return f'log.Info().Msgf({formatted_message} {", ".join(new_args)})'
+
+# Iterate over Go files and apply the regex replacement
+for subdir, _, files in os.walk(directory):
+    for file in files:
+        if file.endswith(".go"):
+            filepath = os.path.join(subdir, file)
+            with open(filepath, 'r') as f:
+                content = f.read()
+
+            # Apply the replacement
+            new_content = pattern.sub(replacer, content)
+
+            # Write the modified content back to the file
+            with open(filepath, 'w') as f:
+                f.write(new_content)
+
+print("Replacement completed!")