teidesu/mtcute
1
0
Fork 0
Code Issues 15 Pull requests 1 Projects Releases Packages Wiki Activity Actions

feat(client/codegen): aliases for methods

Browse source
This commit is contained in:
teidesu 2021-04-10 20:51:45 +03:00
parent 8acad15d7c
commit c0103441d3
1 changed files with 81 additions and 63 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

144
packages/client/scripts/generate-client.js
View file

@ -48,7 +48,7 @@ async function addSingleMethod(state, fileName) {
return getLeadingComments(ast) return getLeadingComments(ast)
.split('\n') .split('\n')
.map((i) => i.replace(/^(\/\/|\s*\*+|\/\*\*+\s*)/g, '').trim()) .map((i) => i.replace(/^(\/\/|\s*\*+|\/\*\*+\s*)/g, '').trim())
.some((i) => i.startsWith(flag)) .find((i) => i.startsWith(flag))
} }
for (const stmt of program.statements) { for (const stmt of program.statements) {
@ -108,6 +108,15 @@ async function addSingleMethod(state, fileName) {
(mod) => mod.kind === 92 /* ExportKeyword */ (mod) => mod.kind === 92 /* ExportKeyword */
) )
const isInitialize = checkForFlag(stmt, '@initialize') const isInitialize = checkForFlag(stmt, '@initialize')
const aliases = (function() {
const flag = checkForFlag(stmt, '@alias')
if (!flag) return []
const [, aliases] = flag.split('=')
if (!aliases || !aliases.length) return []
return aliases.split(',')
})()
if (!isExported && !isPrivate) { if (!isExported && !isPrivate) {
throwError( throwError(
@ -186,6 +195,7 @@ async function addSingleMethod(state, fileName) {
isPrivate, isPrivate,
func: stmt, func: stmt,
comment: getLeadingComments(stmt), comment: getLeadingComments(stmt),
aliases
}) })
const module = `./${relPath.replace(/\.ts$/, '')}` const module = `./${relPath.replace(/\.ts$/, '')}`
@ -286,82 +296,90 @@ async function main() {
const printer = ts.createPrinter() const printer = ts.createPrinter()
state.methods.list.forEach(({ name, isPrivate, func, comment }) => { state.methods.list.forEach(({ name: origName, isPrivate, func, comment, aliases }) => {
// create method that calls that function and passes `this` // create method that calls that function and passes `this`
// first let's determine the signature // first let's determine the signature
const returnType = func.type ? ': ' + func.type.getText() : '' const returnType = func.type ? ': ' + func.type.getText() : ''
const generics = func.typeParameters const generics = func.typeParameters
? `<${func.typeParameters ? `<${func.typeParameters
.map((it) => it.getFullText()) .map((it) => it.getFullText())
.join(', ')}>` .join(', ')}>`
: '' : ''
const rawParams = (func.parameters || []).filter( const rawParams = (func.parameters || []).filter(
(it) => !it.type || it.type.getText() !== 'TelegramClient' (it) => !it.type || it.type.getText() !== 'TelegramClient'
) )
const parameters = rawParams const parameters = rawParams
.map((it) => { .map((it) => {
if (it.initializer) { if (it.initializer) {
// has default value // has default value
it._savedDefault = it.initializer.getFullText() it._savedDefault = it.initializer.getFullText()
if (!it.type) { if (!it.type) {
// no explicit type. // no explicit type.
// infer from initializer // infer from initializer
if ( if (
it.initializer.kind === ts.SyntaxKind.TrueKeyword || it.initializer.kind === ts.SyntaxKind.TrueKeyword ||
it.initializer.kind === ts.SyntaxKind.FalseKeyword it.initializer.kind === ts.SyntaxKind.FalseKeyword
) { ) {
it.type = { kind: ts.SyntaxKind.BooleanKeyword } it.type = { kind: ts.SyntaxKind.BooleanKeyword }
} else if ( } else if (
it.initializer.kind === ts.SyntaxKind.StringLiteral it.initializer.kind === ts.SyntaxKind.StringLiteral
) { ) {
it.type = { kind: ts.SyntaxKind.StringKeyword } it.type = { kind: ts.SyntaxKind.StringKeyword }
} else if ( } else if (
it.initializer.kind === it.initializer.kind ===
ts.SyntaxKind.NumericLiteral || ts.SyntaxKind.NumericLiteral ||
(it.initializer.kind === ts.SyntaxKind.Identifier && (it.initializer.kind === ts.SyntaxKind.Identifier &&
it.initializer.escapedText === 'NaN') it.initializer.escapedText === 'NaN')
) { ) {
it.type = { kind: ts.SyntaxKind.NumberKeyword } it.type = { kind: ts.SyntaxKind.NumberKeyword }
} else { } else {
throwError( throwError(
it, it,
state.methods.used[name], state.methods.used[name],
'Cannot infer parameter type' 'Cannot infer parameter type'
) )
}
} }
it.initializer = undefined
it.questionToken = { kind: ts.SyntaxKind.QuestionToken }
return printer.printNode(ts.EmitHint.Unspecified, it)
} }
it.initializer = undefined
it.questionToken = { kind: ts.SyntaxKind.QuestionToken }
return printer.printNode(ts.EmitHint.Unspecified, it)
}
return it.getFullText() return it.getFullText()
}).join(', ') }).join(', ')
// write comment, but remove @internal mark and set default values for parameters // remove @internal mark and set default values for parameters
comment = comment comment = comment
.replace(/^\s*\/\/+\s*@alias.*$/m, '')
.replace(/(\n^|\/\*)\s*\*\s*@internal.*/m, '') .replace(/(\n^|\/\*)\s*\*\s*@internal.*/m, '')
.replace(/((?:\n^|\/\*)\s*\*\s*@param )([^\s]+?)($|\s+)/gm, (_, pref, arg, post) => { .replace(/((?:\n^|\/\*)\s*\*\s*@param )([^\s]+?)($|\s+)/gm, (_, pref, arg, post) => {
const param = rawParams.find(it => it.name.escapedText === arg) const param = rawParams.find(it => it.name.escapedText === arg)
if (!param) return _ if (!param) return _
if (!param._savedDefault) return _ if (!param._savedDefault) return _
return `${pref}${arg}${post}(default: \`${param._savedDefault.trim()}\`) ` if (post) {
return `${pref}${arg}${post}(default: \`${param._savedDefault.trim()}\`) `
} else {
return `${pref}${arg}\n* (default: \`${param._savedDefault.trim()}\`)`
}
}) })
if (!comment.match(/\/\*\*?\s*\*\//))
// empty comment, no need to write it
output.write(comment)
output.write( for (const name of [origName, ...aliases]) {
`${ if (!comment.match(/\/\*\*?\s*\*\//))
isPrivate ? 'protected ' : '' // empty comment, no need to write it
}${name}${generics}(${parameters})${returnType}${ output.write(comment)
func.body
? `{ output.write(
return ${name}.apply(this, arguments) `${
isPrivate ? 'protected ' : ''
}${name}${generics}(${parameters})${returnType}${
func.body
? `{
return ${origName}.apply(this, arguments)
}` }`
: '' : ''
}` }`
) )
}
}) })
output.untab() output.untab()
output.write('}') output.write('}')