74 lines
2.0 KiB
JavaScript
74 lines
2.0 KiB
JavaScript
#!/usr/bin/env node
|
|
|
|
import fs from 'fs'
|
|
import path from 'path'
|
|
import process from 'process'
|
|
|
|
const root = process.cwd()
|
|
const sourceExtensions = new Set(['.ts', '.tsx', '.js', '.jsx', '.mjs', '.cjs'])
|
|
const scanRoots = ['src']
|
|
const allowConstantDefinitionsIn = new Set([
|
|
'src/lib/constants.ts',
|
|
])
|
|
const forbiddenCapabilityConstants = [
|
|
'VIDEO_MODELS',
|
|
'FIRST_LAST_FRAME_MODELS',
|
|
'AUDIO_SUPPORTED_MODELS',
|
|
'BANANA_MODELS',
|
|
'BANANA_RESOLUTION_OPTIONS',
|
|
]
|
|
|
|
function fail(title, details = []) {
|
|
console.error(`\n[no-hardcoded-model-capabilities] ${title}`)
|
|
for (const line of details) {
|
|
console.error(` - ${line}`)
|
|
}
|
|
process.exit(1)
|
|
}
|
|
|
|
function toRel(fullPath) {
|
|
return path.relative(root, fullPath).split(path.sep).join('/')
|
|
}
|
|
|
|
function walk(dir, out = []) {
|
|
if (!fs.existsSync(dir)) return out
|
|
const entries = fs.readdirSync(dir, { withFileTypes: true })
|
|
for (const entry of entries) {
|
|
if (entry.name === '.git' || entry.name === '.next' || entry.name === 'node_modules') continue
|
|
const fullPath = path.join(dir, entry.name)
|
|
if (entry.isDirectory()) {
|
|
walk(fullPath, out)
|
|
continue
|
|
}
|
|
if (sourceExtensions.has(path.extname(entry.name))) {
|
|
out.push(fullPath)
|
|
}
|
|
}
|
|
return out
|
|
}
|
|
|
|
const files = scanRoots.flatMap((scanRoot) => walk(path.join(root, scanRoot)))
|
|
const violations = []
|
|
|
|
for (const fullPath of files) {
|
|
const relPath = toRel(fullPath)
|
|
if (allowConstantDefinitionsIn.has(relPath)) continue
|
|
|
|
const lines = fs.readFileSync(fullPath, 'utf8').split('\n')
|
|
for (let index = 0; index < lines.length; index += 1) {
|
|
const line = lines[index]
|
|
for (const token of forbiddenCapabilityConstants) {
|
|
const tokenPattern = new RegExp(`\\b${token}\\b`)
|
|
if (tokenPattern.test(line)) {
|
|
violations.push(`${relPath}:${index + 1} forbidden hardcoded model capability token ${token}`)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (violations.length > 0) {
|
|
fail('Found hardcoded model capability usage', violations)
|
|
}
|
|
|
|
console.log('[no-hardcoded-model-capabilities] OK')
|