add support for prop value in mapToTheme

scf4 · May 9, 2019 · 5b2cca7 · 5b2cca7
1 parent 6274cbe
commit 5b2cca7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -68,7 +68,7 @@ export const _dotProp = (string, object) => string
   .split('.')
   .reduce((acc, key) => acc[key], object);
 
-export const mapToTheme = (key) => 
-  (props) => styledMap(_dotProp(key, props.theme));
+export const mapToTheme = (key, prop) =>
+  (props) => prop ? styledMap(prop, _dotProp(key, props.theme)) : styledMap(_dotProp(key, props.theme));
 
 export default styledMap;
diff --git a/lib/index.test.js b/lib/index.test.js
@@ -144,6 +144,12 @@ describe('mapToTheme', () => {
     });
     expect(result).toEqual(_dotProp('button.foreground.primary', nestedTheme));
   });
+  it('uses the prop value when passed a second argument', () => {
+    const result = mapToTheme('buttonColor', 'kind')({theme})({
+      kind: 'other'
+    });
+    expect(result).toEqual('#aab');
+  })
 });
 
 describe('_convertToObject', () => {