Grab previous className from the node.sel if available

sliptype · Jun 11, 2018 · 7b06852 · 7b06852
1 parent f83f3d9
commit 7b06852
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/src/utils.ts b/src/utils.ts
@@ -3,16 +3,26 @@ import { style } from 'typestyle';
 
 import { StyledVNodeData } from './types';
 
-export const makeClassName = (oldClassName: string, newClassName: string): string => {
+const makeClassName = (oldClassName: string, newClassName: string): string => {
   return `${oldClassName} ${newClassName}`.trim();
 };
 
 export const updateVNode = (node: VNode, attributeAccessor: (attribute: string, className: string) => void): void => {
   const data: StyledVNodeData = node.data as StyledVNodeData;
 
   if (data.css) {
-    data.props = data.props || {};
-    attributeAccessor('class', makeClassName(data.props.className || '', style(data.css)));
+    let previousClassName;
+
+    if (data.props && data.props.className) {
+      previousClassName = data.props.className;
+    } else if (node.sel) {
+      const dotIdx = node.sel.indexOf('.');
+      if (dotIdx >= 0) {
+        previousClassName = node.sel.slice(dotIdx + 1);
+      }
+    }
+
+    attributeAccessor('class', makeClassName(previousClassName || '', style(data.css)));
   }
 };