From 257b5a422078d9828de2d5e87b498388938bac26 Mon Sep 17 00:00:00 2001
From: Brijesh Bittu <brijeshb42@gmail.com>
Date: Fri, 15 Mar 2024 15:26:08 +0530
Subject: [PATCH] [pigment-css][nextjs-plugin] Follow-up to #41494 (#41502)

Signed-off-by: Brijesh Bittu <brijeshb42@gmail.com>
Co-authored-by: Siriwat K <siriwatkunaporn@gmail.com>
---
 .../pigment-css-nextjs-plugin/src/index.ts    | 23 +++++++++++--------
 packages/pigment-css-unplugin/src/index.ts    | 11 ++++-----
 2 files changed, 17 insertions(+), 17 deletions(-)

diff --git a/packages/pigment-css-nextjs-plugin/src/index.ts b/packages/pigment-css-nextjs-plugin/src/index.ts
index 92b0ddee1dcf21..ecf7b1598687b2 100644
--- a/packages/pigment-css-nextjs-plugin/src/index.ts
+++ b/packages/pigment-css-nextjs-plugin/src/index.ts
@@ -1,15 +1,9 @@
 import * as path from 'node:path';
 import type { NextConfig } from 'next';
 import { findPagesDir } from 'next/dist/lib/find-pages-dir';
-import {
-  webpack as webpackPlugin,
-  extendTheme,
-  type PigmentOptions as BasePigmentOptions,
-} from '@pigment-css/unplugin';
+import { webpack as webpackPlugin, extendTheme, type PigmentOptions } from '@pigment-css/unplugin';
 
-export type PigmentOptions = BasePigmentOptions & {
-  asyncResolve?: (what: string) => string | null;
-};
+export { type PigmentOptions };
 
 const extractionFile = path.join(
   path.dirname(require.resolve('../package.json')),
@@ -60,14 +54,23 @@ export function withPigment(nextConfig: NextConfig, pigmentConfig?: PigmentOptio
           outputCss: dev || hasAppDir || !isServer,
           placeholderCssFile: extractionFile,
         },
-        asyncResolve(what) {
+        async asyncResolve(what: string, importer: string, stack: string[]) {
+          // Need to point to the react from node_modules during eval time.
+          // Otherwise, next makes it point to its own version of react that
+          // has a lot of RSC specific logic which is not actually needed.
+          if (what.startsWith('react') || what.startsWith('next')) {
+            return require.resolve(what);
+          }
           if (what === 'next/image') {
             return require.resolve('../next-image');
           }
           if (what.startsWith('next/font')) {
             return require.resolve('../next-font');
           }
-          return asyncResolve?.(what) ?? null;
+          if (asyncResolve) {
+            return asyncResolve(what, importer, stack);
+          }
+          return null;
         },
         babelOptions: {
           ...babelOptions,
diff --git a/packages/pigment-css-unplugin/src/index.ts b/packages/pigment-css-unplugin/src/index.ts
index 38ef63479089ee..262577e8056808 100644
--- a/packages/pigment-css-unplugin/src/index.ts
+++ b/packages/pigment-css-unplugin/src/index.ts
@@ -41,6 +41,7 @@ type WebpackMeta = {
 };
 
 type Meta = NextMeta | ViteMeta | WebpackMeta;
+export type AsyncResolver = (what: string, importer: string, stack: string[]) => Promise<string>;
 
 export type PigmentOptions<Theme extends BaseTheme = BaseTheme> = {
   theme?: Theme;
@@ -49,7 +50,7 @@ export type PigmentOptions<Theme extends BaseTheme = BaseTheme> = {
   debug?: IFileReporterOptions | false;
   sourceMap?: boolean;
   meta?: Meta;
-  asyncResolve?: (what: string) => string | null;
+  asyncResolve?: (...args: Parameters<AsyncResolver>) => Promise<string | null>;
   transformSx?: boolean;
 } & Partial<WywInJsPluginOptions>;
 
@@ -62,8 +63,6 @@ function hasCorectExtension(fileName: string) {
 const VIRTUAL_CSS_FILE = `\0zero-runtime-styles.css`;
 const VIRTUAL_THEME_FILE = `\0zero-runtime-theme.js`;
 
-type AsyncResolver = (what: string, importer: string, stack: string[]) => Promise<string>;
-
 function isZeroRuntimeThemeFile(fileName: string) {
   return fileName === VIRTUAL_CSS_FILE || fileName === VIRTUAL_THEME_FILE;
 }
@@ -142,13 +141,11 @@ export const plugin = createUnplugin<PigmentOptions, true>((options) => {
   let webpackResolver: AsyncResolver;
 
   const asyncResolve: AsyncResolver = async (what, importer, stack) => {
-    const result = asyncResolveOpt?.(what);
+    const result = await asyncResolveOpt?.(what, importer, stack);
     if (typeof result === 'string') {
       return result;
     }
-    // Use Webpack's resolver to resolve actual path but
-    // ignore next.js files during evaluation phase of WyW
-    if (webpackResolver && !what.startsWith('next')) {
+    if (webpackResolver) {
       return webpackResolver(what, importer, stack);
     }
     return asyncResolveFallback(what, importer, stack);