cha0s/avocado
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: types

Browse Source
This commit is contained in:
cha0s 2022-04-02 06:37:02 -05:00
parent 852c9782d5
commit 7416cec896
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
packages/code/src/function.js
View File

@ -3,8 +3,7 @@ import {
isExpression,
isExpressionStatement,
isFile,
isFunctionDeclaration,
isFunctionExpression,
isFunction,
isMemberExpression,
isProgram,
isReturnStatement,
@ -21,12 +20,12 @@ export function extractFunction(ast) {
) {
return undefined;
}
if (isFunctionDeclaration(ast.program.body[0])) {
if (isFunction(ast.program.body[0])) {
return ast.program.body[0];
}
if (
isExpressionStatement(ast.program.body[0])
&& isFunctionExpression(ast.program.body[0].expression)
&& isFunction(ast.program.body[0].expression)
) {
return ast.program.body[0].expression;
}
@ -35,7 +34,7 @@ export function extractFunction(ast) {
export function extractExpressionFromFunction(ast) {
if (
isFunctionDeclaration(ast)
isFunction(ast)
&& isBlockStatement(ast.body)
&& isReturnStatement(ast.body.body[0])
&& isExpression(ast.body.body[0].argument)
@ -43,13 +42,13 @@ export function extractExpressionFromFunction(ast) {
return ast.body.body[0].argument;
}
if (
isFunctionExpression(ast)
isFunction(ast)
&& isExpression(ast.body)
) {
return ast.body;
}
if (
isFunctionExpression(ast)
isFunction(ast)
&& isBlockStatement(ast.body)
&& isReturnStatement(ast.body.body[0])
) {