Function statementsToFunction

src/common/sourceCodeManipulations.ts:61–99

(
  parameterNames: ReadonlyArray<string>,
  statements: ReadonlyArray<Statement>,
  catchAndReturnErrors: boolean,
)

 * statement; statement => function () { catchAndReturnErrors?(statement; return statement;) }
 */
export function statementsToFunction(
  parameterNames: ReadonlyArray<string>,
  statements: ReadonlyArray<Statement>,
  catchAndReturnErrors: boolean,
): AnyFunctionExpression {
  if (statements.length > 1 || statements[0].type !== 'FunctionDeclaration') {
    return statementToFunction(parameterNames, statements, true, catchAndReturnErrors);
  }

  return codeToFunctionExecutingCode(
    parameterNames,
    [
      {
        type: 'ReturnStatement',
        argument: {
          type: 'CallExpression',
          optional: false,
          arguments: [
            { type: 'ThisExpression' },
            ...parameterNames.map(name => ({ type: 'Identifier' as const, name })),
          ],
          callee: {
            type: 'MemberExpression',
            property: { type: 'Identifier', name: 'call' },
            computed: false,
            optional: false,
            object: {
              type: 'FunctionExpression',
              params: statements[0].params,
              body: statements[0].body,
            },
          },
        },
      },
    ],
    true,
    catchAndReturnErrors,
  );
}

/**
 * code => (parameterNames) => return catchAndReturnErrors?(code)