import fs from 'node:fs';

interface IResolveArgs {
    importer: string;
    path: string;
}

interface ILoadArgs {
    path: string;
}

interface IPluginBuild {
    onResolve(
        options: { filter: RegExp; namespace?: string },
        callback: (args: IResolveArgs) => unknown
    ): void;
    onLoad(
        options: { filter: RegExp; namespace?: string },
        callback: (args: ILoadArgs) => unknown
    ): void;
}

interface IEsbuildPlugin {
    name: string;
    setup: (build: IPluginBuild) => void;
}

export function ignoreGlobalCssPlugin(): IEsbuildPlugin {
    return {
        name: 'ignore-global-css',
        setup(build: IPluginBuild) {
            build.onResolve({ filter: /\/global\.css$/ }, (args: IResolveArgs) => {
                if (args.importer.includes('packages')) {
                    return {
                        path: args.path,
                        namespace: 'ignore-global-css',
                        pluginData: {
                            importer: args.importer,
                        },
                    };
                }
            });

            build.onLoad({ filter: /\/global\.css$/, namespace: 'ignore-global-css' }, () => {
                return { contents: '' };
            });
        },
    };
}

export function removeClassnameNewlinesPlugin(): IEsbuildPlugin {
    return {
        name: 'remove-classname-newlines',
        setup(build: IPluginBuild) {
            build.onLoad({ filter: /\.(tsx)$/ }, (args: ILoadArgs) => {
                const source = fs.readFileSync(args.path, 'utf8');

                const transformedSource = source.replace(
                    /className\s*=\s*{([^}]*?)}/gs,
                    (_match: string, classNameValue: string) => {
                        const cleanedValue = classNameValue.replace(/`([^`]*?)`/gs, (_templateMatch: string, templateContent: string) => {
                            return `\`${templateContent.replace(/\s*\n\s*/g, ' ').trim()}\``;
                        });

                        return `className={${cleanedValue.trim()}}`;
                    }
                );

                return {
                    contents: transformedSource,
                    loader: 'tsx',
                };
            });
        },
    };
}