162 lines
6.8 KiB
Python
162 lines
6.8 KiB
Python
|
"""
|
|||
|
|
Fix script for i18n batch transformation issues:
|
|||
|
|
1. Remove misplaced `const { t } = useTranslation();` lines (inside object literals, etc.)
|
|||
|
|
2. Fix JSX attributes: `=t('...')` -> `={t('...')}`
|
|||
|
|
3. Re-add the hook correctly in the main function component
|
|||
|
|
"""
|
|||
|
|
import re
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
target_dirs = ['src/pages', 'src/components']
|
|||
|
|
skip_files = [
|
|||
|
|
'src/components/LanguageSwitcher.tsx',
|
|||
|
|
'src/components/Layout.tsx',
|
|||
|
|
'src/i18n/index.ts',
|
|||
|
|
'src/locales/zh/index.ts',
|
|||
|
|
'src/locales/en/index.ts',
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
def get_target_files():
|
|||
|
|
files = []
|
|||
|
|
for target_dir in target_dirs:
|
|||
|
|
for root, dirs, filenames in os.walk(target_dir):
|
|||
|
|
for fname in filenames:
|
|||
|
|
if fname.endswith(('.tsx', '.ts', '.jsx', '.js')):
|
|||
|
|
filepath = os.path.join(root, fname).replace('\\', '/')
|
|||
|
|
if filepath not in skip_files:
|
|||
|
|
files.append(filepath)
|
|||
|
|
return files
|
|||
|
|
|
|||
|
|
target_files = get_target_files()
|
|||
|
|
total_fixed = 0
|
|||
|
|
|
|||
|
|
for filepath in target_files:
|
|||
|
|
try:
|
|||
|
|
with open(filepath, 'r', encoding='utf-8') as f:
|
|||
|
|
content = f.read()
|
|||
|
|
except:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
original = content
|
|||
|
|
fixes = 0
|
|||
|
|
|
|||
|
|
# Step 1: Remove ALL `const { t } = useTranslation();` lines
|
|||
|
|
# These might be misplaced (inside object literals, etc.)
|
|||
|
|
old_hook_line = '\tconst { t } = useTranslation();\n'
|
|||
|
|
old_hook_line2 = '\n\tconst { t } = useTranslation();\n'
|
|||
|
|
|
|||
|
|
hook_count = content.count('const { t } = useTranslation();')
|
|||
|
|
if hook_count > 0:
|
|||
|
|
# Remove all instances
|
|||
|
|
content = content.replace(old_hook_line, '')
|
|||
|
|
content = content.replace(old_hook_line2, '\n')
|
|||
|
|
# Also handle variations
|
|||
|
|
content = re.sub(r'\n\s*const \{ t \} = useTranslation\(\);\n', '\n', content)
|
|||
|
|
fixes += hook_count
|
|||
|
|
|
|||
|
|
# Step 2: Fix JSX attributes: `=t('...')` -> `={t('...')}`
|
|||
|
|
# Pattern: word=t('...') where word is a JSX attribute name
|
|||
|
|
# But NOT: key: t('...') (JS object property) or return t('...') or = t('...')
|
|||
|
|
# JSX attributes look like: placeholder=t('key') or title=t('key')
|
|||
|
|
# We need to match: identifier immediately followed by =t('
|
|||
|
|
|
|||
|
|
# Match patterns like: placeholder=t('some.key') or title=t('some.key')
|
|||
|
|
# But not patterns like: :t(' or {t(' or (t(' or = t(' (with space)
|
|||
|
|
def fix_jsx_attr(match):
|
|||
|
|
prefix = match.group(1)
|
|||
|
|
key = match.group(2)
|
|||
|
|
return f'{prefix}={{t(\'{key}\')}}'
|
|||
|
|
|
|||
|
|
# This regex matches: word=t('key') where word is preceded by space or <
|
|||
|
|
# and NOT preceded by : or { or (
|
|||
|
|
content = re.sub(
|
|||
|
|
r'(?<![{:（(])(\b\w+)=t\(\'([^\']+)\'\)',
|
|||
|
|
fix_jsx_attr,
|
|||
|
|
content
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Also fix: word=t("key") with double quotes
|
|||
|
|
content = re.sub(
|
|||
|
|
r'(?<![{:（(])(\b\w+)=t\("([^"]+)"\)',
|
|||
|
|
lambda m: f'{m.group(1)}={{t(\'{m.group(2)}\')}}',
|
|||
|
|
content
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Step 3: Re-add useTranslation import if file uses t() but doesn't have the import
|
|||
|
|
has_t_calls = bool(re.search(r"\bt\(['\"]", content))
|
|||
|
|
has_import = 'useTranslation' in content and 'import' in content
|
|||
|
|
|
|||
|
|
if has_t_calls and not has_import:
|
|||
|
|
# Add import after last import line
|
|||
|
|
import_pattern = r"((?:import .+?;\n)+)"
|
|||
|
|
matches = list(re.finditer(import_pattern, content))
|
|||
|
|
if matches:
|
|||
|
|
last_import = matches[-1]
|
|||
|
|
insert_pos = last_import.end()
|
|||
|
|
content = content[:insert_pos] + "import { useTranslation } from 'react-i18next';\n" + content[insert_pos:]
|
|||
|
|
fixes += 1
|
|||
|
|
|
|||
|
|
# Step 4: Re-add the hook in the correct place (main function component)
|
|||
|
|
if has_t_calls and has_import or (has_t_calls and 'useTranslation' in content):
|
|||
|
|
# Check if hook is already present
|
|||
|
|
if 'const { t } = useTranslation();' not in content:
|
|||
|
|
# Find the main function component and add the hook after its opening brace
|
|||
|
|
# Pattern 1: export default function ComponentName() {
|
|||
|
|
match = re.search(r'(export default function \w+\([^)]*\)\s*\{)', content)
|
|||
|
|
if match:
|
|||
|
|
insert_pos = match.end()
|
|||
|
|
content = content[:insert_pos] + '\n\tconst { t } = useTranslation();' + content[insert_pos:]
|
|||
|
|
fixes += 1
|
|||
|
|
else:
|
|||
|
|
# Pattern 2: function ComponentName() {
|
|||
|
|
match = re.search(r'(function \w+\([^)]*\)\s*\{)', content)
|
|||
|
|
if match:
|
|||
|
|
insert_pos = match.end()
|
|||
|
|
content = content[:insert_pos] + '\n\tconst { t } = useTranslation();' + content[insert_pos:]
|
|||
|
|
fixes += 1
|
|||
|
|
else:
|
|||
|
|
# Pattern 3: const ComponentName = () => {
|
|||
|
|
# or: const ComponentName = (props) => {
|
|||
|
|
match = re.search(r'(const \w+\s*=\s*(?:\([^)]*\))?\s*=>\s*\{)', content)
|
|||
|
|
if match:
|
|||
|
|
insert_pos = match.end()
|
|||
|
|
content = content[:insert_pos] + '\n\tconst { t } = useTranslation();' + content[insert_pos:]
|
|||
|
|
fixes += 1
|
|||
|
|
else:
|
|||
|
|
# Pattern 4: export default () => {
|
|||
|
|
match = re.search(r'(export default\s*(?:\([^)]*\))?\s*=>\s*\{)', content)
|
|||
|
|
if match:
|
|||
|
|
insert_pos = match.end()
|
|||
|
|
content = content[:insert_pos] + '\n\tconst { t } = useTranslation();' + content[insert_pos:]
|
|||
|
|
fixes += 1
|
|||
|
|
|
|||
|
|
# Step 5: Also check for files that have the import but the hook was removed
|
|||
|
|
# Re-add if needed
|
|||
|
|
if 'useTranslation' in content and 'import' in content and 'const { t } = useTranslation();' not in content:
|
|||
|
|
if has_t_calls or re.search(r"\bt\(['\"]", content):
|
|||
|
|
match = re.search(r'(export default function \w+\([^)]*\)\s*\{)', content)
|
|||
|
|
if match:
|
|||
|
|
insert_pos = match.end()
|
|||
|
|
content = content[:insert_pos] + '\n\tconst { t } = useTranslation();' + content[insert_pos:]
|
|||
|
|
fixes += 1
|
|||
|
|
else:
|
|||
|
|
match = re.search(r'(function \w+\([^)]*\)\s*\{)', content)
|
|||
|
|
if match:
|
|||
|
|
insert_pos = match.end()
|
|||
|
|
content = content[:insert_pos] + '\n\tconst { t } = useTranslation();' + content[insert_pos:]
|
|||
|
|
fixes += 1
|
|||
|
|
else:
|
|||
|
|
match = re.search(r'(const \w+\s*=\s*(?:\([^)]*\))?\s*=>\s*\{)', content)
|
|||
|
|
if match:
|
|||
|
|
insert_pos = match.end()
|
|||
|
|
content = content[:insert_pos] + '\n\tconst { t } = useTranslation();' + content[insert_pos:]
|
|||
|
|
fixes += 1
|
|||
|
|
|
|||
|
|
if content != original:
|
|||
|
|
with open(filepath, 'w', encoding='utf-8') as f:
|
|||
|
|
f.write(content)
|
|||
|
|
total_fixed += 1
|
|||
|
|
print(f" Fixed: {filepath} ({fixes} issues)")
|
|||
|
|
|
|||
|
|
print(f"\nDone! Fixed {total_fixed} files")
|