Babel plugin that injects twin.macro

plugin.ts

import template from '@babel/template';

// template
const buildImport = template(`
  import 'twin.macro';
`);

// const debug = console.log;
const debug = (args: any) => {};

// const styledComponentsMacroImport = 'styled-components/macro';

// plugin
export default function importTwinMarcoPlugin(babel) {
    const importDeclaration = buildImport();
    return {
        visitor: {
            Program: (path, state, fileName) => {
                let shouldAddImport = true;
                state.file.path.traverse({
                    ImportDeclaration(path) {
                        // Find the twin import path
                        if (path.node.source.value === 'twin.macro') {
                            shouldAddImport = false;
                        }
                    },
                });

                if (shouldAddImport) {
                    debug(`Injected \'import "twin.macro"\' [${state.file.opts.sourceFileName}]`);
                    path.unshiftContainer('body', importDeclaration);
                } else {
                    debug(`Skipped twin.macro injection [${state.file.opts.sourceFileName}]`);
                }
            },
        },
    };
}