fix: add result of property modification to returned classes with colors

muhammadsammy · Jul 3, 2020 · fe458c9 · fe458c9
1 parent d9663d6
commit fe458c9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/generation/ClassesGenerator.ts b/src/generation/ClassesGenerator.ts
@@ -300,14 +300,15 @@ export class ClassesGenerator implements IGenerator {
     return propertyNames.flatMap((propertyValue, i) => {
       const variant = propertyVariants[i];
 
-      property.replace('Color', '').replace('background', 'bg');
+      const propertyResult = property.replace('Color', '').replace('background', 'bg');
 
       if (typeof variant === 'object' && variant !== null) {
         return Object.keys(variant).map(
-          v => `${property}-${propertyValue}${v === 'default' ? '' : `-${v}`}`,
+          v => `${propertyResult}-${propertyValue}${v === 'default' ? '' : `-${v}`}`,
         );
+      } else {
+        return `${propertyResult}-${propertyValue}`;
       }
-      return `${property}-${propertyValue}`;
     });
   };