105 lines
3.8 KiB
Python
105 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
|
import re
|
|
|
|
def fix_tab_file(file_path, component_name):
|
|
"""Add DebugName import and wrap the main return with DebugName"""
|
|
print(f"Processing {file_path} for {component_name}...")
|
|
|
|
with open(file_path, 'r') as f:
|
|
content = f.read()
|
|
|
|
# Check if DebugName is already imported
|
|
if "from '@/lib/game/debug-context'" in content:
|
|
print(f" - DebugName already imported")
|
|
else:
|
|
# Find the last import line and add the DebugName import after it
|
|
lines = content.split('\n')
|
|
last_import_idx = -1
|
|
for i, line in enumerate(lines):
|
|
if line.startswith('import ') or line.strip().startswith('import {'):
|
|
last_import_idx = i
|
|
|
|
if last_import_idx >= 0:
|
|
# Check if next line is also part of import
|
|
if i + 1 < len(lines) and (lines[i+1].strip().startswith('} from') or lines[i+1].strip() == '}'):
|
|
# Multi-line import, find the closing
|
|
for j in range(i+1, len(lines)):
|
|
if '} from' in lines[j]:
|
|
last_import_idx = j
|
|
break
|
|
|
|
if last_import_idx >= 0:
|
|
lines.insert(last_import_idx + 1, "import { DebugName } from '@/lib/game/debug-context';")
|
|
content = '\n'.join(lines)
|
|
print(f" - Added DebugName import")
|
|
else:
|
|
print(f" - WARNING: No import found")
|
|
return False
|
|
|
|
# Now find the main return statement (not early returns)
|
|
# Look for "return (" followed by newline and then some JSX
|
|
pattern = r'(export function \w+\(\)\s*\{.*?return\s*\(\s*\n)(\s*)(<)'
|
|
|
|
match = re.search(pattern, content, re.DOTALL)
|
|
if not match:
|
|
print(f" - WARNING: Could not find main return pattern")
|
|
return False
|
|
|
|
# Get the indentation
|
|
indent = match.group(2)
|
|
|
|
# Add opening DebugName tag after the return (
|
|
before_return = content[:match.end(1)]
|
|
after_return = content[match.end(1):]
|
|
|
|
# Add <DebugName name="..."> after the return (
|
|
modified = before_return + f'{indent}<DebugName name="{component_name}">\n{indent}{after_return[0]}'
|
|
|
|
# Now find the closing ); for the main return
|
|
# Find "displayName" to locate the end of the component
|
|
display_pattern = re.escape(component_name) + r'\.displayName\s*=\s*[\'"].*?[\'"];\s*\n'
|
|
display_match = re.search(display_pattern, modified)
|
|
|
|
if not display_match:
|
|
print(f" - WARNING: Could not find displayName")
|
|
return False
|
|
|
|
# Find the ); before displayName
|
|
before_display = modified[:display_match.start()]
|
|
after_display = modified[display_match.start():]
|
|
|
|
# Find the last ); in before_display that's at the start of a line
|
|
lines_before = before_display.split('\n')
|
|
close_paren_line = -1
|
|
close_paren_indent = ''
|
|
|
|
for i in range(len(lines_before) - 1, -1, -1):
|
|
line = lines_before[i]
|
|
if line.strip() == ');':
|
|
close_paren_line = i
|
|
close_paren_indent = line[:len(line) - len(line.lstrip())]
|
|
break
|
|
|
|
if close_paren_line == -1:
|
|
print(f" - WARNING: Could not find closing );\")
|
|
return False
|
|
|
|
# Insert </DebugName> before );
|
|
lines_before.insert(close_paren_line, f"{close_paren_indent}</DebugName>")
|
|
before_display_fixed = '\n'.join(lines_before)
|
|
|
|
modified = before_display_fixed + after_display
|
|
|
|
# Write back
|
|
with open(file_path, 'w') as f:
|
|
f.write(modified)
|
|
|
|
print(f" - Successfully wrapped {component_name} with DebugName")
|
|
return True
|
|
|
|
# Fix the remaining files
|
|
fix_tab_file('src/components/game/tabs/GolemancyTab.tsx', 'GolemancyTab')
|
|
fix_tab_file('src/components/game/tabs/SpellsTab.tsx', 'SpellsTab')
|
|
|
|
print("\nDone!")
|