Skip to content

Commit 7e306c9

Browse files
committed
Add support for manual functions
1 parent 5001a01 commit 7e306c9

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

packages/core/babel/core/memo-variable.ts

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,11 @@ export function memoVariable(
1111
stateIdentifier: t.Expression,
1212
optionsIdentifier?: t.Expression,
1313
): t.VariableDeclarator {
14-
const normalIdentifier = t.arrowFunctionExpression([], stateIdentifier);
14+
const normalIdentifier =
15+
t.isArrowFunctionExpression(stateIdentifier) ||
16+
t.isFunctionExpression(stateIdentifier)
17+
? stateIdentifier
18+
: t.arrowFunctionExpression([], stateIdentifier);
1519

1620
const exprs: t.Expression[] = [normalIdentifier];
1721

packages/core/babel/transform-ctf.ts

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -378,7 +378,12 @@ const CTF_TRAVERSE: babel.Visitor<State> = {
378378
t.expressionStatement(
379379
t.callExpression(
380380
getImportIdentifier(state, path, 'createEffect', 'solid-js'),
381-
[t.arrowFunctionExpression([], argument)],
381+
[
382+
t.isArrowFunctionExpression(argument) ||
383+
t.isFunctionExpression(argument)
384+
? argument
385+
: t.arrowFunctionExpression([], argument),
386+
],
382387
),
383388
),
384389
);

0 commit comments

Comments
 (0)