removing hx-lsp attributes

examples/web/templates/example.j2

@@ -1,2 +1,2 @@
-<p class="text-2xl bg-red-400 rounded">hello world</p>
+<p class="text-2xl bg-red-400 rounded" hx-lsp="abc def">hello world</p>
 <p class="rounded-xl text-3xl">hello world 2</p>

src/short_classes.rs

@@ -31,30 +31,26 @@ impl Default for ClassIter {
 impl ClassContainer {
     pub fn add(&mut self, key: String, token: CSSToken) -> Option<String> {
         let index = token as usize;
-        if let Some(map) = self.container.get_mut(index) {
-            let oldclass = css_to_html(&key);
-            match map.entry(oldclass) {
-                Entry::Vacant(entry) => {
-                    if let Some(new) = self.class_name[index].next() {
-                        return Some(entry.insert(new_class(&new)).to_string());
-                    }
-                }
-                Entry::Occupied(entry) => {
-                    return Some(entry.get().clone());
+        let map = self.container.get_mut(index)?;
+        let oldclass = css_to_html(&key);
+        match map.entry(oldclass) {
+            Entry::Vacant(entry) => {
+                if let Some(new) = self.class_name[index].next() {
+                    return Some(entry.insert(new_class(&new)).to_string());
                 }
             }
+            Entry::Occupied(entry) => {
+                return Some(entry.get().clone());
+            }
         }
         None
     }
 
     pub fn get(&self, key: String, container: CSSToken) -> Option<String> {
         let index = container as usize;
-        if let Some(map) = self.container.get(index) {
-            if let Some(v) = map.get(&key) {
-                return Some(v.to_string());
-            }
-        }
-        None
+        let map = self.container.get(index)?;
+        let v = map.get(&key)?;
+        Some(v.to_string())
     }
 
     pub fn into_file(&self, stylesheet: StyleSheet) {

src/visit_class.rs

@@ -25,6 +25,7 @@ pub fn check_html(
                     .collect::<Vec<String>>()
                     .join(" ");
                 let _r = el.set_attribute("class", &classes);
+                el.remove_attribute("hx-lsp");
 
                 Ok(())
             })],

src/visit_map.rs

@@ -46,10 +46,9 @@ pub fn check_js(
                         Some(&mut srcmap),
                     ),
                 };
-                if let Some(module) = program.as_script() {
-                    if emitter.emit_script(module).is_ok() {
-                        return Some(code);
-                    }
+                let module = program.as_script()?;
+                if emitter.emit_script(module).is_ok() {
+                    return Some(code);
                 }
                 None
             }