[rjolaverria]

Fixes #10.

What changed

useMemo is replaced by the factory's result, not the factory. useMemo(factory, deps) evaluates to what the factory returns, but the codemod substituted the factory function itself, so every memoized value became a function. Now:

• an expression-bodied arrow becomes its body: useMemo(() => ({ a }), [a]) → { a }. Outside declarators, arguments, returns, pairs, arrays, parentheses and JSX expressions the body is parenthesised so precedence survives: 1 + useMemo(() => a * 2) → 1 + (a * 2)
• a block-bodied or async factory is invoked in place: useMemo(() => { ... }, deps) → (() => { ... })()
• a bare identifier is called: useMemo(compute, [compute]) → compute()

Import rewriting is narrowed.

• an import that loses no specifier keeps its original text, which stops import type { X } from 'react' becoming a value import and stops ; being appended in semicolon-free files
• only the memoization specifiers are candidates for removal; an unrelated unused React import is left alone
• a rewritten import keeps the file's semicolon style

Fixtures

Six existing expected.tsx files encoded the useMemo bug (a memoized boolean became () => boolean) and are corrected. Three new fixtures cover the expression-body forms, precedence wrapping, async and identifier factories, a type-only import beside a rewritten value import in a semicolon-free file, and an inline type specifier surviving a rewrite.

Verification

• pnpm test in codemods/remove-memoization: 16 passed
• pnpm check-types: clean
• Ran the patched codemod over 13 files of a React 19 + React Compiler app that the published version had broken (18 type errors, 169 failing tests): zero type errors, 687/687 tests passing

🤖 Generated with Claude Code

[alexbit-codemod]

thank you for your contributions @rjolaverria. will check this out soon

[alexbit-codemod]

Thanks for this fix — the core useMemo unwrap looks good.

I pushed three additional fixtures that currently fail (cd7b212). Could you update the transform so pnpm test in codemods/remove-memoization is green again?

1. tests/use-memo-parenthesized-factory

Parenthesized expression-bodied arrows should unwrap like bare ones:

useMemo((() => computeValue()), [computeValue])
// expected: computeValue()
// actual today: ((() => computeValue()))()

2. tests/use-memo-block-return

A block body that only returns an expression should become that expression (not an IIFE):

useMemo(() => { return computeValue(); }, [computeValue])
// expected: computeValue()
// actual today: (() => { return computeValue(); })()

3. tests/unused-default-import-preserved

Per the PR description (“an unrelated unused React import is left alone”), an unused default import in a separate statement should stay when only useMemo is removed from another import:

import React from 'react';
import { useMemo, useState } from 'react';
// expected: keep `import React from 'react';`, drop only `useMemo`
// actual today: deletes the unused default import

No rush on polish beyond making these three pass — happy to re-review after.

[rjolaverria]

Thanks for the fixtures! All three pass now (f22bb65), plus a few more edge-case tests. Ready for another look whenever you have time.

[alexbit-codemod]

can you also resolve the conflicts? i recent made some improvements to the codemods by utilizing jssg utils... @rjolaverria

[alexbit-codemod]

/codex-review

[github-actions]

Codex Review

Reviewed all 46 changed files statically. Five correctness issues remain. No PR code or tests were executed.

Findings

• [MEDIUM] codemods/remove-memoization/scripts/codemod.ts:195 Keep parameterized arrow factories as invocations
• [MEDIUM] codemods/remove-memoization/scripts/codemod.ts:198 Preserve TypeScript typing when extracting factory results
• [MEDIUM] codemods/remove-memoization/scripts/codemod.ts:211 Preserve unbound invocation of member references
• [MEDIUM] codemods/remove-memoization/scripts/codemod.ts:224 Apply result conversion at every nested memoization layer
• [MEDIUM] codemods/remove-memoization/scripts/codemod.ts:356 Remove individual import bindings instead of all matching names

[alexbit-codemod]

i'll take care of the comments above, push the fix, and merge the PR