feat: adapt to pre-processed styles (#243)

mui · Sep 19, 2024 · 3cfa69c · 3cfa69c
1 parent 40da08b
commit 3cfa69c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/packages/pigment-css-react/src/processors/styled.ts b/packages/pigment-css-react/src/processors/styled.ts
@@ -607,7 +607,7 @@ export class StyledProcessor extends BaseProcessor {
     themeImportIdentifier?: string,
   ) {
     const { themeArgs = {} } = this.options as IOptions;
-    const styleObj = typeof styleObjOrFn === 'function' ? styleObjOrFn(themeArgs) : styleObjOrFn;
+    let styleObj = typeof styleObjOrFn === 'function' ? styleObjOrFn(themeArgs) : styleObjOrFn;
     if (!styleObj) {
       return '';
     }
@@ -619,7 +619,11 @@ export class StyledProcessor extends BaseProcessor {
         })),
       );
     }
-    delete styleObj.variants;
+    if (styleObj.isProcessed) {
+      styleObj = styleObj.style;
+    } else {
+      delete styleObj.variants;
+    }
     const res = cssFnValueToVariable({
       styleObj,
       expressionValue: styleArg,