@idealyst/theme 1.2.104 → 1.2.105

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@idealyst/theme",
-  "version": "1.2.104",
+  "version": "1.2.105",
   "description": "Theming system for Idealyst Framework",
   "readme": "README.md",
   "main": "src/index.ts",
@@ -63,7 +63,7 @@
     "publish:npm": "npm publish"
   },
   "dependencies": {
-    "@idealyst/tooling": "^1.2.104"
+    "@idealyst/tooling": "^1.2.105"
   },
   "peerDependencies": {
     "react-native-unistyles": ">=3.0.0"

package/src/babel/plugin.js

@@ -945,16 +945,24 @@ module.exports = function idealystStylesPlugin({ types: t }) {
                 if (t.isIdentifier(node.callee, { name: 'extendStyle' })) {
                     debug(`FOUND extendStyle in: ${filename}`);
 
-                    const [componentNameArg, stylesCallback] = node.arguments;
+                    const [componentNameArg, stylesArg] = node.arguments;
 
                     if (!t.isStringLiteral(componentNameArg)) {
                         debug(`  SKIP - componentName is not a string literal`);
                         return;
                     }
 
-                    if (!t.isArrowFunctionExpression(stylesCallback) &&
-                        !t.isFunctionExpression(stylesCallback)) {
-                        debug(`  SKIP - callback is not a function`);
+                    // Accept either a function callback or a plain object
+                    let stylesCallback = stylesArg;
+                    if (t.isObjectExpression(stylesArg)) {
+                        // Wrap plain object in an arrow function: (theme) => ({ ... })
+                        stylesCallback = t.arrowFunctionExpression(
+                            [t.identifier('theme')],
+                            t.parenthesizedExpression(stylesArg)
+                        );
+                    } else if (!t.isArrowFunctionExpression(stylesArg) &&
+                        !t.isFunctionExpression(stylesArg)) {
+                        debug(`  SKIP - second argument is not a function or object`);
                         return;
                     }
 
@@ -991,16 +999,24 @@ module.exports = function idealystStylesPlugin({ types: t }) {
                 if (t.isIdentifier(node.callee, { name: 'overrideStyle' })) {
                     debug(`FOUND overrideStyle in: ${filename}`);
 
-                    const [componentNameArg, stylesCallback] = node.arguments;
+                    const [componentNameArg, stylesArg] = node.arguments;
 
                     if (!t.isStringLiteral(componentNameArg)) {
                         debug(`  SKIP - componentName is not a string literal`);
                         return;
                     }
 
-                    if (!t.isArrowFunctionExpression(stylesCallback) &&
-                        !t.isFunctionExpression(stylesCallback)) {
-                        debug(`  SKIP - callback is not a function`);
+                    // Accept either a function callback or a plain object
+                    let stylesCallback = stylesArg;
+                    if (t.isObjectExpression(stylesArg)) {
+                        // Wrap plain object in an arrow function: (theme) => ({ ... })
+                        stylesCallback = t.arrowFunctionExpression(
+                            [t.identifier('theme')],
+                            t.parenthesizedExpression(stylesArg)
+                        );
+                    } else if (!t.isArrowFunctionExpression(stylesArg) &&
+                        !t.isFunctionExpression(stylesArg)) {
+                        debug(`  SKIP - second argument is not a function or object`);
                         return;
                     }
 

package/src/styleBuilder.ts

@@ -102,18 +102,27 @@ export function defineStyle<TTheme, TStyles extends Record<string, unknown>>(
 /**
  * Extend existing component styles (merged at build time).
  * Import BEFORE components for extensions to apply.
+ *
+ * Accepts either a plain style object or a theme callback:
+ * ```typescript
+ * // Plain object (no theme access needed)
+ * extendStyle('Text', { text: { fontFamily: 'MyFont' } });
+ *
+ * // Theme callback (when you need theme tokens)
+ * extendStyle('Text', (theme) => ({ text: { color: theme.colors.text.primary } }));
+ * ```
  */
 export function extendStyle<K extends keyof ComponentStyleRegistry>(
   componentName: K,
-  styles: (theme: any) => ExtendStyleDef<K>
+  styles: ExtendStyleDef<K> | ((theme: any) => ExtendStyleDef<K>)
 ): void;
 export function extendStyle<K extends string>(
   componentName: K,
-  styles: (theme: any) => Record<string, any>
+  styles: Record<string, any> | ((theme: any) => Record<string, any>)
 ): void;
 export function extendStyle(
   _componentName: string,
-  _styles: (theme: any) => any
+  _styles: any
 ): void {
   // Babel removes this call and merges into defineStyle
 }
@@ -121,18 +130,27 @@ export function extendStyle(
 /**
  * Override component styles completely (replaces base at build time).
  * Import BEFORE components for overrides to apply.
+ *
+ * Accepts either a plain style object or a theme callback:
+ * ```typescript
+ * // Plain object
+ * overrideStyle('Text', { text: { fontFamily: 'MyFont' } });
+ *
+ * // Theme callback
+ * overrideStyle('Text', (theme) => ({ text: { color: theme.colors.text.primary } }));
+ * ```
  */
 export function overrideStyle<K extends keyof ComponentStyleRegistry>(
   componentName: K,
-  styles: (theme: any) => OverrideStyleDef<K>
+  styles: OverrideStyleDef<K> | ((theme: any) => OverrideStyleDef<K>)
 ): void;
 export function overrideStyle<K extends string>(
   componentName: K,
-  styles: (theme: any) => Record<string, any>
+  styles: Record<string, any> | ((theme: any) => Record<string, any>)
 ): void;
 export function overrideStyle(
   _componentName: string,
-  _styles: (theme: any) => any
+  _styles: any
 ): void {
   // Babel removes this call and replaces defineStyle
 }