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 = 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}( // 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>` : ''}['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}( // filter: UpdateFilter<${type.updateType}, Mod, State>, // handler: ${type.handlerTypeName}Handler, State extends never ? never : UpdateState>['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}( // filter: UpdateFilter<${type.updateType}, Mod>, // handler: ${type.handlerTypeName}Handler${type.state ? ', State extends never ? never : UpdateState' : ''}>['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) }