Function rewriteCtxMethods

packages/update/src/update.ts:632–690

(
  importState: ImportState,
  nodes: (tsmorph.Node<tsmorph.ts.Node>)[],
)

}

function rewriteCtxMethods(
  importState: ImportState,
  nodes: (tsmorph.Node<tsmorph.ts.Node>)[],
) {
  for (let i = 0; i < nodes.length; i++) {
    const node = nodes[i];

    if (node.wasForgotten()) continue;

    if (
      node.isKind(SyntaxKind.ExpressionStatement) &&
      node.getText() === "ctx.renderNotFound();"
    ) {
      importState.httpError++;
      node.replaceWithText("throw new HttpError(404);");
      continue;
    }

    if (node.isKind(SyntaxKind.PropertyAccessExpression)) {
      rewriteCtxMemberName(node);
    } else if (node.isKind(SyntaxKind.ReturnStatement)) {
      if (node.getText() === "return ctx.renderNotFound();") {
        importState.httpError++;
        node.replaceWithText(`throw new HttpError(404)`);
        continue;
      } else {
        const expr = node.getExpression();
        if (expr !== undefined) {
          rewriteCtxMethods(importState, [expr]);
        }
      }
    } else if (node.isKind(SyntaxKind.VariableStatement)) {
      const decls = node.getDeclarations();
      for (let i = 0; i < decls.length; i++) {
        const decl = decls[i];
        const init = decl.getInitializer();
        if (init !== undefined) {
          rewriteCtxMethods(importState, [init]);
        }
      }
    } else if (
      node.isKind(SyntaxKind.ExpressionStatement) ||
      node.isKind(SyntaxKind.AwaitExpression) ||
      node.isKind(SyntaxKind.CallExpression)
    ) {
      const expr = node.getExpression();
      rewriteCtxMethods(importState, [expr]);
    } else if (node.isKind(SyntaxKind.BinaryExpression)) {
      rewriteCtxMethods(importState, [node.getLeft()]);
      rewriteCtxMethods(importState, [node.getRight()]);
    } else if (
      !node.isKind(SyntaxKind.ExpressionStatement) &&
      node.getKindName().endsWith("Statement")
    ) {
      const inner = node.getDescendantStatements();
      rewriteCtxMethods(importState, inner);
    }
  }