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

fix(client/codegen): fix rare crash when parsing default arguments

Browse source 
i have no idea why having .parent makes it crash but whatever lol.
This commit is contained in:
teidesu 2021-04-10 23:12:07 +03:00
parent 9edaa438fd
commit eb9b004b44
1 changed files with 74 additions and 41 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -38,6 +38,8 @@ async function addSingleMethod(state, fileName) {
)
const relPath = path.relative(targetDir, fileName).replace(/\\/g, '/') // replace path delim to unix
state.files[relPath] = fileFullText
function getLeadingComments(ast) {
return (ts.getLeadingCommentRanges(fileFullText, ast.pos) || [])
.map((range) => fileFullText.substring(range.pos, range.end))
@ -195,7 +197,7 @@ async function addSingleMethod(state, fileName) {
isPrivate,
func: stmt,
comment: getLeadingComments(stmt),
aliases
aliases,
})
const module = `./${relPath.replace(/\.ts$/, '')}`
@ -255,6 +257,7 @@ async function main() {
list: [],
},
copy: [],
files: {},
}
for await (const file of getFiles(path.join(__dirname, '../src/methods'))) {
@ -296,7 +299,8 @@ async function main() {
const printer = ts.createPrinter()
state.methods.list.forEach(({ name: origName, isPrivate, func, comment, aliases }) => {
state.methods.list.forEach(
({ name: origName, isPrivate, func, comment, aliases }) => {
// create method that calls that function and passes `this`
// first let's determine the signature
const returnType = func.type ? ': ' + func.type.getText() : ''
@ -317,43 +321,70 @@ async function main() {
// no explicit type.
// infer from initializer
if (
it.initializer.kind === ts.SyntaxKind.TrueKeyword ||
it.initializer.kind === ts.SyntaxKind.FalseKeyword
it.initializer.kind ===
ts.SyntaxKind.TrueKeyword ||
it.initializer.kind ===
ts.SyntaxKind.FalseKeyword
) {
it.type = { kind: ts.SyntaxKind.BooleanKeyword }
} else if (
it.initializer.kind === ts.SyntaxKind.StringLiteral
it.initializer.kind ===
ts.SyntaxKind.StringLiteral
) {
it.type = { kind: ts.SyntaxKind.StringKeyword }
} else if (
it.initializer.kind ===
ts.SyntaxKind.NumericLiteral ||
(it.initializer.kind === ts.SyntaxKind.Identifier &&
(it.initializer.kind ===
ts.SyntaxKind.Identifier &&
it.initializer.escapedText === 'NaN')
) {
it.type = { kind: ts.SyntaxKind.NumberKeyword }
} else {
throwError(
it,
state.methods.used[name],
state.methods.used[origName],
'Cannot infer parameter type'
)
}
}
it.initializer = undefined
const deleteParents = (obj) => {
if (Array.isArray(obj)) return obj.forEach((it) => deleteParents(it))
if (obj.parent) delete obj.parent
for (const k of Object.keys(obj)) {
if (obj[k] && typeof obj[k] === 'object') {
deleteParents(obj[k])
}
}
}
deleteParents(it)
it.questionToken = { kind: ts.SyntaxKind.QuestionToken }
return printer.printNode(ts.EmitHint.Unspecified, it)
return printer.printNode(
ts.EmitHint.Unspecified,
it,
// state.files[state.methods.used[origName]]
)
}
return it.getFullText()
}).join(', ')
})
.join(', ')
// remove @internal mark and set default values for parameters
comment = comment
.replace(/^\s*\/\/+\s*@alias.*$/m, '')
.replace(/(\n^|\/\*)\s*\*\s*@internal.*/m, '')
.replace(/((?:\n^|\/\*)\s*\*\s*@param )([^\s]+?)($|\s+)/gm, (_, pref, arg, post) => {
const param = rawParams.find(it => it.name.escapedText === arg)
.replace(
/((?:\n^|\/\*)\s*\*\s*@param )([^\s]+?)($|\s+)/gm,
(_, pref, arg, post) => {
const param = rawParams.find(
(it) => it.name.escapedText === arg
)
if (!param) return _
if (!param._savedDefault) return _
if (post) {
@ -361,7 +392,8 @@ async function main() {
} else {
return `${pref}${arg}\n* (default: \`${param._savedDefault.trim()}\`)`
}
})
}
)
for (const name of [origName, ...aliases]) {
if (!comment.match(/\/\*\*?\s*\*\//))
@ -380,7 +412,8 @@ return ${origName}.apply(this, arguments)
}`
)
}
})
}
)
output.untab()
output.write('}')