fix(styleMap): Support numbers as keys on 'styleMap' objects (#88)

seek-oss · Jan 31, 2020 · 1fd5f1f · 1fd5f1f
1 parent defbf43
commit 1fd5f1f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 8 deletions.
diff --git a/packages/treat/src/builder.ts b/packages/treat/src/builder.ts
@@ -159,14 +159,14 @@ export function style(
   }
 }
 
-type StyleMapParam<ClassName extends string> = ThemedStyle<
-  StyleMap<ClassName, Style>,
+type StyleMapParam<StyleName extends string | number> = ThemedStyle<
+  StyleMap<StyleName, Style>,
   ThemeOrAny
 >;
-export function styleMap<ClassName extends string>(
-  stylesheet: StyleMapParam<ClassName>,
+export function styleMap<StyleName extends string | number>(
+  stylesheet: StyleMapParam<StyleName>,
   localDebugName?: string,
-): StylesMap<ClassName> {
+): StylesMap<StyleName> {
   const classRefs: { [className: string]: ClassRef } = {};
   const createLocalName = (classIdentifier: string) => {
     if (localDebugName) {

diff --git a/packages/treat/src/types.ts b/packages/treat/src/types.ts
@@ -38,8 +38,8 @@ export type Style = StyleWithSelectors & MediaQueries<StyleWithSelectors>;
 
 export type GlobalStyle = CSSProperties & MediaQueries<CSSProperties>;
 
-export type StyleMap<ClassName extends string, StyleType> = Record<
-  ClassName,
+export type StyleMap<StyleName extends string | number, StyleType> = Record<
+  StyleName,
   StyleType
 >;
 
@@ -54,7 +54,10 @@ export interface TreatTheme<Tokens> {
 
 export type ClassRef = string;
 
-export type StylesMap<ClassName extends string> = Record<ClassName, ClassRef>;
+export type StylesMap<StyleName extends string | number> = Record<
+  StyleName,
+  ClassRef
+>;
 
 type TreatModuleValue =
   | string