refactor: Use glob pattern to find litegraph files with declare keywords

Instead of hardcoding the list of files, dynamically find all TypeScript files in litegraph that contain 'declare' keywords using glob pattern matching.
This commit is contained in:
snomiao
2025-09-02 14:24:31 +00:00
parent 4e56bae7f1
commit 2403b47b5c

View File

@@ -5,52 +5,72 @@
import * as fs from 'fs' import * as fs from 'fs'
import * as path from 'path' import * as path from 'path'
import { fileURLToPath } from 'url' import { fileURLToPath } from 'url'
import { globSync } from 'glob'
const __dirname = path.dirname(fileURLToPath(import.meta.url)) const __dirname = path.dirname(fileURLToPath(import.meta.url))
const rootDir = path.resolve(__dirname, '..') const rootDir = path.resolve(__dirname, '..')
const litegraphSrcDir = path.join(rootDir, 'src/lib/litegraph/src') const litegraphSrcDir = path.join(rootDir, 'src/lib/litegraph/src')
const filesToProcess = [
'LGraphNode.ts',
'widgets/BaseWidget.ts',
'subgraph/SubgraphInput.ts',
'subgraph/SubgraphNode.ts',
'subgraph/SubgraphOutput.ts',
'subgraph/EmptySubgraphInput.ts',
'subgraph/EmptySubgraphOutput.ts'
]
const backupMap = new Map<string, string>() const backupMap = new Map<string, string>()
/**
* Find all TypeScript files in litegraph that contain 'declare' keywords
*/
function findFilesWithDeclare(): string[] {
// Search for all .ts files in litegraph src directory
const pattern = path.join(litegraphSrcDir, '**/*.ts')
const files = globSync(pattern, {
ignore: ['**/*.test.ts', '**/*.spec.ts', '**/node_modules/**']
})
// Filter to only files that actually contain 'declare' keyword
return files.filter(filePath => {
try {
const content = fs.readFileSync(filePath, 'utf-8')
// Check for class property declarations with 'declare' keyword
return /^\s*declare\s+/m.test(content)
} catch (error) {
console.warn(` ⚠ Could not read ${filePath}: ${error.message}`)
return false
}
})
}
export async function preprocessLitegraph() { export async function preprocessLitegraph() {
console.log('Preprocessing litegraph files for i18n collection...') console.log('Preprocessing litegraph files for i18n collection...')
for (const relativePath of filesToProcess) { const filesToProcess = findFilesWithDeclare()
const filePath = path.join(litegraphSrcDir, relativePath)
if (filesToProcess.length === 0) {
console.log(' No files with declare keywords found')
return
}
console.log(` Found ${filesToProcess.length} files with declare keywords`)
for (const filePath of filesToProcess) {
const originalContent = fs.readFileSync(filePath, 'utf-8')
if (fs.existsSync(filePath)) { // Store original content in memory
const originalContent = fs.readFileSync(filePath, 'utf-8') backupMap.set(filePath, originalContent)
// Only process if file contains 'declare' keywords // Remove 'declare' keyword from class properties
if (originalContent.includes('declare ')) { const modifiedContent = originalContent.replace(
// Store original content in memory /^(\s*)declare\s+/gm,
backupMap.set(filePath, originalContent) '$1// @ts-ignore - removed declare for Playwright\n$1'
)
// Remove 'declare' keyword from class properties
const modifiedContent = originalContent.replace( // Write modified content
/^(\s*)declare\s+/gm, fs.writeFileSync(filePath, modifiedContent)
'$1// @ts-ignore - removed declare for Playwright\n$1' console.log(` ✓ Processed ${path.relative(litegraphSrcDir, filePath)}`)
)
// Write modified content
fs.writeFileSync(filePath, modifiedContent)
console.log(` ✓ Processed ${relativePath}`)
}
}
} }
} }
export async function restoreLitegraph() { export async function restoreLitegraph() {
if (backupMap.size === 0) {
return
}
console.log('Restoring original litegraph files...') console.log('Restoring original litegraph files...')
for (const [filePath, originalContent] of backupMap.entries()) { for (const [filePath, originalContent] of backupMap.entries()) {