From efe082937b64ec048bb496030ed9af0cae24a13c Mon Sep 17 00:00:00 2001
From: Ulysse ARNAUD <mexanga@gmail.com>
Date: Wed, 11 Jan 2023 01:24:02 +0100
Subject: [PATCH] Better class removing

---
 package.json |  2 +-
 src/index.js | 18 +++++++++++++++---
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/package.json b/package.json
index 9c8812c..a330502 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@ulyssear/css-in-js-in-html",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "",
   "main": "index.js",
   "scripts": {
diff --git a/src/index.js b/src/index.js
index dcde67a..d4543b1 100644
--- a/src/index.js
+++ b/src/index.js
@@ -258,16 +258,17 @@ function apply_custom_class(element, className) {
 				value: class_value.substring(1, class_value.length - 1),
 			});
 		}
-		element.className = element.className.replace(className, '').trim();
+		const original_class = className; 
+		// element.className = element.className.replace(className, '').trim();
 		// console.log({classes_to_apply})
 
 		// console.log({ selectors, classes_to_apply, events, media_query });
 
-		do_apply(element, selectors, classes_to_apply, events, media_query);
+		do_apply(element, selectors, classes_to_apply, events, media_query, original_class);
 	}
 }
 
-function do_apply(element, selectors, classes, events, media_query) {
+function do_apply(element, selectors, classes, events, media_query, original_class) {
 	if (media_query) {
 		if ('(' !== media_query[0]) {
 			media_query = `(${media_query})`;
@@ -353,14 +354,25 @@ function do_apply(element, selectors, classes, events, media_query) {
 		return;
 	}
 
+	let to_remove_class = false;
+
 	for (let j = 0; j < elements_to_apply.length; j++) {
 		const element_to_apply = elements_to_apply[j];
 		if (!element_to_apply) continue;
 		for (let k = 0; k < classes.length; k++) {
 			const { name, value } = classes[k];
+			if (!name) continue;
 			element_to_apply.style.setProperty(name, value);
+			if (to_remove_class) continue;
+			if (element.className.includes(original_class)) {
+				to_remove_class = true;
+			}
 		}
 	}
+
+	if (to_remove_class) {
+		element.className = element.className.replace(original_class, '');
+	}
 }
 
 function init(document, event = undefined) {