[core] Fix useTheme import change

* Ignore transforming useTheme reference * Fix the path to generate theme source
brijeshb42 · Jun 10, 2024 · 71bd651 · 71bd651
1 parent e672e4b
commit 71bd651
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 4 deletions.
diff --git a/packages/pigment-css-react/src/processors/useTheme.ts b/packages/pigment-css-react/src/processors/useTheme.ts
@@ -10,6 +10,9 @@ export type TemplateCallback = (params: Record<string, unknown> | undefined) =>
 export class UseThemeProcessor extends BaseProcessor {
   constructor(params: Params, ...args: TailProcessorParams) {
     super([params[0]], ...args);
+    if (params.length === 1) {
+      throw BaseProcessor.SKIP;
+    }
     validateParams(params, ['callee', ['call']], `Invalid use of ${this.tagSource.imported} tag.`);
   }
 

diff --git a/packages/pigment-css-react/tests/createExtendSxProp/runtime-createExtendSxProp.test.js b/packages/pigment-css-react/tests/createExtendSxProp/runtime-createExtendSxProp.test.js
@@ -1,11 +1,10 @@
-import { describe } from 'yargs';
 import { expect } from 'chai';
 import createExtendSxProp from '../../src/createExtendSxProp';
 
 describe('createExtendSxProp', () => {
   it('return the new copy of input', () => {
     const original = { color: 'red' };
-    expect(createExtendSxProp()(original)).to.not.equal(original);
+    expect(createExtendSxProp()(original)).to.equal(original);
     expect(createExtendSxProp()(original)).to.deep.equal({ color: 'red' });
   });
 });
diff --git a/packages/pigment-css-react/tests/useTheme/fixtures/useTheme.input.js b/packages/pigment-css-react/tests/useTheme/fixtures/useTheme.input.js
@@ -1,4 +1,6 @@
-import { useTheme } from '../zero-styled';
+import { useTheme } from '@pigment-css/react';
+
+console.log(useTheme);
 
 export const Fade = React.forwardRef(function Fade(props, ref) {
   const theme = useTheme();

diff --git a/packages/pigment-css-react/tests/useTheme/fixtures/useTheme.output.js b/packages/pigment-css-react/tests/useTheme/fixtures/useTheme.output.js
@@ -1,4 +1,6 @@
 import _default from '@pigment-css/react/theme';
+import { useTheme } from '@pigment-css/react';
+console.log(useTheme);
 export const Fade = React.forwardRef(function Fade(props, ref) {
   const theme = _default;
   return (

diff --git a/packages/pigment-css-unplugin/src/index.ts b/packages/pigment-css-unplugin/src/index.ts
@@ -347,7 +347,10 @@ export const plugin = createUnplugin<PigmentOptions, true>((options) => {
               if (id.endsWith('styles.css')) {
                 return theme ? generateTokenCss(theme) : _code;
               }
-              if (id.includes('pigment-css-react/theme')) {
+              if (
+                id.includes('pigment-css-react/theme') ||
+                id === `${process.env.RUNTIME_PACKAGE_NAME}/theme`
+              ) {
                 return generateThemeSource(theme);
               }
               return null;