fix sx types

mui · Aug 8, 2024 · a5a1d0e · a5a1d0e
1 parent 5b3f475
commit a5a1d0e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/packages/pigment-css-react/src/sx.d.ts b/packages/pigment-css-react/src/sx.d.ts
@@ -1,9 +1,11 @@
 import type { CSSObjectNoCallback } from './base';
 import type { ThemeArgs } from './theme';
 
+type GetTheme<Argument> = Argument extends { theme: infer Theme } ? Theme : never;
+
 export type SxProp =
   | CSSObjectNoCallback
-  | ((themeArgs: ThemeArgs['theme']) => CSSObjectNoCallback)
-  | ReadonlyArray<CSSObjectNoCallback | ((themeArgs: ThemeArgs['theme']) => CSSObjectNoCallback)>;
+  | ((theme: GetTheme<ThemeArgs>) => CSSObjectNoCallback)
+  | ReadonlyArray<CSSObjectNoCallback | ((theme: GetTheme<ThemeArgs>) => CSSObjectNoCallback)>;
 
-export default function sx(arg: SxProp | Array<SxProp>, componentClass?: string): string;
+export default function sx(arg: SxProp, componentClass?: string): string;