Plugin does not inject the __collectModule hook into ArrowFunction declarations.

[DanDroryAu]

If a component is defined like so:

// MyComponent.tsx
const MyComponent = () => {
  return (<div>hello, world!</div>);
}

export default MyComponent;

The plugin doesn't inject the __collectModule hook.

This is because the ExportDefaultDeclaration ast traverse has a small missing step for arrowFunctionExpressions.

I've provided some code that I think can resolve the issue. Let me know if you want me to make a forkPR.

ExportDefaultDeclaration(path) {
    const declaration = path.node.declaration;

    // Insert hook in `export default function() { ... }`
    if (isReactFunctionComponent(declaration)) {
        injectImport(
            ast,
            __internal_importHelperModuleName
        );
        injectHook(path.get('declaration'), relative);
        injected = true;
    } else if (t.isIdentifier(declaration)) {
        // Insert hook in `function Component() { ... }; export default Component;`
        const binding = path.scope.getBinding(
            declaration.name
        );
        if (binding) {
            // Right here we need to check if the binding is a declarator for the ArrowFunctionExpression. 
            // This code creates the correct NodePath for the if statement and the injectHook function.
            const expressionPath = t.isVariableDeclarator(binding.path.node) ? binding.path.get('init') : binding.path;
            if (isReactFunctionComponent(expressionPath.node)) {
                injectImport(
                    ast,
                    __internal_importHelperModuleName
                );
                injectHook(expressionPath, relative);
                injected = true;
            }
        }
    }
},

Let me know if I can help resolve this or if more information is required.

[wille]

Awesome, please make a PR!