327 lines
10 KiB
JavaScript
327 lines
10 KiB
JavaScript
|
const fs = require('fs')
|
||
|
|
const path = require('path')
|
||
|
|
const prettier = require('prettier')
|
||
|
|
const {
|
||
|
|
snakeToCamel,
|
||
|
|
camelToPascal,
|
||
|
|
camelToSnake,
|
||
|
|
} = require('../../tl/scripts/common')
|
||
|
|
|
||
|
|
function parseUpdateTypes() {
|
||
|
|
const lines = fs
|
||
|
|
.readFileSync(path.join(__dirname, 'update-types.txt'), 'utf-8')
|
||
|
|
.split('\n')
|
||
|
|
.map((it) => it.trim())
|
||
|
|
.filter((it) => it && it[0] !== '#')
|
||
|
|
|
||
|
|
const ret = []
|
||
|
|
|
||
|
|
for (const line of lines) {
|
||
|
|
const m = line.match(/^([a-z_]+)(?:: ([a-zA-Z]+))? = ([a-zA-Z]+)( \+ State)?$/)
|
||
|
|
if (!m) throw new Error(`invalid syntax: ${line}`)
|
||
|
|
ret.push({
|
||
|
|
typeName: m[1],
|
||
|
|
handlerTypeName: m[2] || camelToPascal(snakeToCamel(m[1])),
|
||
|
|
updateType: m[3],
|
||
|
|
funcName: m[2]
|
||
|
|
? m[2][0].toLowerCase() + m[2].substr(1)
|
||
|
|
: snakeToCamel(m[1]),
|
||
|
|
state: !!m[4]
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
function replaceSections(filename, sections, dir = __dirname) {
|
||
|
|
let lines = fs
|
||
|
|
.readFileSync(path.join(dir, '../src', filename), 'utf-8')
|
||
|
|
.split('\n')
|
||
|
|
|
||
|
|
const findMarker = (marker) => {
|
||
|
|
const idx = lines.findIndex((line) => line.trim() === `// ${marker}`)
|
||
|
|
if (idx === -1) throw new Error(marker + ' not found')
|
||
|
|
return idx
|
||
|
|
}
|
||
|
|
|
||
|
|
for (const [name, content] of Object.entries(sections)) {
|
||
|
|
const start = findMarker(`begin-${name}`)
|
||
|
|
const end = findMarker(`end-${name}`)
|
||
|
|
|
||
|
|
if (start > end) throw new Error('begin is after end')
|
||
|
|
|
||
|
|
lines.splice(start + 1, end - start - 1, content)
|
||
|
|
}
|
||
|
|
|
||
|
|
fs.writeFileSync(path.join(dir, '../src', filename), lines.join('\n'))
|
||
|
|
}
|
||
|
|
|
||
|
|
const types = parseUpdateTypes()
|
||
|
|
|
||
|
|
async function formatFile(filename, dir = __dirname) {
|
||
|
|
const targetFile = path.join(dir, '../src/', filename)
|
||
|
|
const prettierConfig = await prettier.resolveConfig(targetFile)
|
||
|
|
let fullSource = await fs.promises.readFile(targetFile, 'utf-8')
|
||
|
|
fullSource = await prettier.format(fullSource, {
|
||
|
|
...(prettierConfig || {}),
|
||
|
|
filepath: targetFile,
|
||
|
|
})
|
||
|
|
await fs.promises.writeFile(targetFile, fullSource)
|
||
|
|
}
|
||
|
|
|
||
|
|
function toSentence(type, stype = 'inline') {
|
||
|
|
const name = camelToSnake(type.handlerTypeName)
|
||
|
|
.toLowerCase()
|
||
|
|
.replace(/_/g, ' ')
|
||
|
|
|
||
|
|
if (stype === 'inline') {
|
||
|
|
return `${name[0].match(/[aeiouy]/i) ? 'an' : 'a'} ${name} handler`
|
||
|
|
} else if (stype === 'plain') {
|
||
|
|
return `${name} handler`
|
||
|
|
} else {
|
||
|
|
return `${name[0].toUpperCase()}${name.substr(1)} handler`
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// function generateHandler() {
|
||
|
|
// const lines = []
|
||
|
|
// const names = ['RawUpdateHandler']
|
||
|
|
//
|
||
|
|
// // imports must be added manually because yeah
|
||
|
|
//
|
||
|
|
// types.forEach((type) => {
|
||
|
|
// if (type.updateType === 'IGNORE') return
|
||
|
|
//
|
||
|
|
// lines.push(
|
||
|
|
// `export type ${type.handlerTypeName}Handler<T = ${type.updateType}` +
|
||
|
|
// `${type.state ? ', S = never' : ''}> = ParsedUpdateHandler<` +
|
||
|
|
// `'${type.typeName}', T${type.state ? ', S' : ''}>`
|
||
|
|
// )
|
||
|
|
// names.push(`${type.handlerTypeName}Handler`)
|
||
|
|
// })
|
||
|
|
//
|
||
|
|
// replaceSections('handler.ts', {
|
||
|
|
// codegen:
|
||
|
|
// lines.join('\n') +
|
||
|
|
// '\n\nexport type UpdateHandler = \n' +
|
||
|
|
// names.map((i) => ` | ${i}\n`).join(''),
|
||
|
|
// })
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// function generateBuilders() {
|
||
|
|
// const lines = []
|
||
|
|
// const imports = ['UpdateHandler']
|
||
|
|
//
|
||
|
|
// types.forEach((type) => {
|
||
|
|
// imports.push(`${type.handlerTypeName}Handler`)
|
||
|
|
//
|
||
|
|
// if (type.updateType === 'IGNORE') {
|
||
|
|
// lines.push(`
|
||
|
|
// /**
|
||
|
|
// * Create ${toSentence(type)}
|
||
|
|
// *
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// */
|
||
|
|
// export function ${type.funcName}(
|
||
|
|
// handler: ${type.handlerTypeName}Handler['callback']
|
||
|
|
// ): ${type.handlerTypeName}Handler
|
||
|
|
//
|
||
|
|
// /**
|
||
|
|
// * Create ${toSentence(type)} with a filter
|
||
|
|
// *
|
||
|
|
// * @param filter Predicate to check the update against
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// */
|
||
|
|
// export function ${type.funcName}(
|
||
|
|
// filter: ${type.handlerTypeName}Handler['check'],
|
||
|
|
// handler: ${type.handlerTypeName}Handler['callback']
|
||
|
|
// ): ${type.handlerTypeName}Handler
|
||
|
|
//
|
||
|
|
// /** @internal */
|
||
|
|
// export function ${type.funcName}(filter: any, handler?: any): ${
|
||
|
|
// type.handlerTypeName
|
||
|
|
// }Handler {
|
||
|
|
// return _create('${type.typeName}', filter, handler)
|
||
|
|
// }
|
||
|
|
// `)
|
||
|
|
// } else {
|
||
|
|
// lines.push(`
|
||
|
|
// /**
|
||
|
|
// * Create ${toSentence(type)}
|
||
|
|
// *
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// */
|
||
|
|
// export function ${type.funcName}(
|
||
|
|
// handler: ${type.handlerTypeName}Handler['callback']
|
||
|
|
// ): ${type.handlerTypeName}Handler
|
||
|
|
//
|
||
|
|
// /**
|
||
|
|
// * Create ${toSentence(type)} with a filter
|
||
|
|
// *
|
||
|
|
// * @param filter Update filter
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// */
|
||
|
|
// export function ${type.funcName}<Mod>(
|
||
|
|
// filter: UpdateFilter<${type.updateType}, Mod>,
|
||
|
|
// handler: ${type.handlerTypeName}Handler<
|
||
|
|
// filters.Modify<${type.updateType}, Mod>
|
||
|
|
// >['callback']
|
||
|
|
// ): ${type.handlerTypeName}Handler
|
||
|
|
//
|
||
|
|
// /** @internal */
|
||
|
|
// export function ${type.funcName}(
|
||
|
|
// filter: any,
|
||
|
|
// handler?: any
|
||
|
|
// ): ${type.handlerTypeName}Handler {
|
||
|
|
// return _create('${type.typeName}', filter, handler)
|
||
|
|
// }
|
||
|
|
// `)
|
||
|
|
// }
|
||
|
|
// })
|
||
|
|
//
|
||
|
|
// replaceSections('builders.ts', {
|
||
|
|
// codegen: lines.join('\n'),
|
||
|
|
// 'codegen-imports':
|
||
|
|
// 'import {\n' +
|
||
|
|
// imports.map((i) => ` ${i},\n`).join('') +
|
||
|
|
// "} from './handler'",
|
||
|
|
// })
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// function generateDispatcher() {
|
||
|
|
// const lines = []
|
||
|
|
// const declareLines = []
|
||
|
|
// const imports = ['UpdateHandler']
|
||
|
|
//
|
||
|
|
// types.forEach((type) => {
|
||
|
|
// imports.push(`${type.handlerTypeName}Handler`)
|
||
|
|
//
|
||
|
|
// if (type.updateType === 'IGNORE') {
|
||
|
|
// declareLines.push(`
|
||
|
|
// /**
|
||
|
|
// * Register a plain old ${toSentence(type, 'plain')}
|
||
|
|
// *
|
||
|
|
// * @param name Event name
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// */
|
||
|
|
// on(name: '${type.typeName}', handler: ${type.handlerTypeName}Handler['callback']): this
|
||
|
|
// `)
|
||
|
|
//
|
||
|
|
// lines.push(`
|
||
|
|
// /**
|
||
|
|
// * Register ${toSentence(type)} without any filters
|
||
|
|
// *
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// * @param group Handler group index
|
||
|
|
// */
|
||
|
|
// on${type.handlerTypeName}(handler: ${type.handlerTypeName}Handler['callback'], group?: number): void
|
||
|
|
//
|
||
|
|
// /**
|
||
|
|
// * Register ${toSentence(type)} with a filter
|
||
|
|
// *
|
||
|
|
// * @param filter Update filter function
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// * @param group Handler group index
|
||
|
|
// */
|
||
|
|
// on${type.handlerTypeName}(
|
||
|
|
// filter: ${type.handlerTypeName}Handler['check'],
|
||
|
|
// handler: ${type.handlerTypeName}Handler['callback'],
|
||
|
|
// group?: number
|
||
|
|
// ): void
|
||
|
|
//
|
||
|
|
// /** @internal */
|
||
|
|
// on${type.handlerTypeName}(filter: any, handler?: any, group?: number): void {
|
||
|
|
// this._addKnownHandler('${type.funcName}', filter, handler, group)
|
||
|
|
// }
|
||
|
|
// `)
|
||
|
|
// } else {
|
||
|
|
// declareLines.push(`
|
||
|
|
// /**
|
||
|
|
// * Register a plain old ${toSentence(type, 'plain')}
|
||
|
|
// *
|
||
|
|
// * @param name Event name
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// */
|
||
|
|
// on(name: '${type.typeName}', handler: ${type.handlerTypeName}Handler['callback']): this
|
||
|
|
//
|
||
|
|
// `)
|
||
|
|
// lines.push(`
|
||
|
|
// /**
|
||
|
|
// * Register ${toSentence(type)} without any filters
|
||
|
|
// *
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// * @param group Handler group index
|
||
|
|
// */
|
||
|
|
// on${type.handlerTypeName}(handler: ${type.handlerTypeName}Handler${type.state ? `<${type.updateType}, State extends never ? never : UpdateState<State, SceneName>>` : ''}['callback'], group?: number): void
|
||
|
|
//
|
||
|
|
// ${type.state ? `
|
||
|
|
// /**
|
||
|
|
// * Register ${toSentence(type)} with a filter
|
||
|
|
// *
|
||
|
|
// * @param filter Update filter
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// * @param group Handler group index
|
||
|
|
// */
|
||
|
|
// on${type.handlerTypeName}<Mod>(
|
||
|
|
// filter: UpdateFilter<${type.updateType}, Mod, State>,
|
||
|
|
// handler: ${type.handlerTypeName}Handler<filters.Modify<${type.updateType}, Mod>, State extends never ? never : UpdateState<State, SceneName>>['callback'],
|
||
|
|
// group?: number
|
||
|
|
// ): void
|
||
|
|
// ` : ''}
|
||
|
|
//
|
||
|
|
// /**
|
||
|
|
// * Register ${toSentence(type)} with a filter
|
||
|
|
// *
|
||
|
|
// * @param filter Update filter
|
||
|
|
// * @param handler ${toSentence(type, 'full')}
|
||
|
|
// * @param group Handler group index
|
||
|
|
// */
|
||
|
|
// on${type.handlerTypeName}<Mod>(
|
||
|
|
// filter: UpdateFilter<${type.updateType}, Mod>,
|
||
|
|
// handler: ${type.handlerTypeName}Handler<filters.Modify<${type.updateType}, Mod>${type.state ? ', State extends never ? never : UpdateState<State, SceneName>' : ''}>['callback'],
|
||
|
|
// group?: number
|
||
|
|
// ): void
|
||
|
|
//
|
||
|
|
// /** @internal */
|
||
|
|
// on${type.handlerTypeName}(filter: any, handler?: any, group?: number): void {
|
||
|
|
// this._addKnownHandler('${type.funcName}', filter, handler, group)
|
||
|
|
// }
|
||
|
|
// `)
|
||
|
|
// }
|
||
|
|
// })
|
||
|
|
//
|
||
|
|
// replaceSections('dispatcher.ts', {
|
||
|
|
// codegen: lines.join('\n'),
|
||
|
|
// 'codegen-declare': declareLines.join('\n'),
|
||
|
|
// 'codegen-imports':
|
||
|
|
// 'import {\n' +
|
||
|
|
// imports.map((i) => ` ${i},\n`).join('') +
|
||
|
|
// "} from './handler'",
|
||
|
|
// })
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
|
||
|
|
function generateParsedUpdate() {
|
||
|
|
replaceSections('types/updates/index.ts', {
|
||
|
|
codegen: 'export type ParsedUpdate =\n'
|
||
|
|
+ types.map((typ) => ` | { name: '${typ.typeName}', data: ${typ.updateType} }\n`).join(''),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
async function main() {
|
||
|
|
generateParsedUpdate()
|
||
|
|
// generateBuilders()
|
||
|
|
// generateHandler()
|
||
|
|
// generateDispatcher()
|
||
|
|
|
||
|
|
// await formatFile('builders.ts')
|
||
|
|
// await formatFile('handler.ts')
|
||
|
|
// await formatFile('dispatcher.ts')
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = { types, toSentence, replaceSections, formatFile }
|
||
|
|
|
||
|
|
if (require.main === module) {
|
||
|
|
main().catch(console.error)
|
||
|
|
}
|