pref(transformer/styled-components): remove quasis and expressions in batch

Author: Dunqing

crates/oxc_transformer/src/plugins/styled_components.rs

@@ -868,6 +868,7 @@ fn is_valid_styled_component_source(source: &str) -> bool {
 /// ```
 fn minify_template_literal<'a>(lit: &mut TemplateLiteral<'a>, ast: AstBuilder<'a>) {
     const NOT_IN_STRING: u8 = 0;
+    const QUASI_DELETION_MARKER: Atom = Atom::new_const("__QUASI_DELETION_MARKER__");
 
     let TemplateLiteral { quasis, expressions, .. } = lit;
 
@@ -883,6 +884,8 @@ fn minify_template_literal<'a>(lit: &mut TemplateLiteral<'a>, ast: AstBuilder<'a
     let mut string_quote = NOT_IN_STRING;
     // Parentheses depth. `((x)` -> 1, `(x))` -> -1.
     let mut paren_depth: isize = 0;
+    // A state that indicates whether we should delete quasi and expression.
+    let mut has_deletion_marker = false;
 
     // Current minified quasi being built
     let mut output = Vec::new();
@@ -911,11 +914,10 @@ fn minify_template_literal<'a>(lit: &mut TemplateLiteral<'a>, ast: AstBuilder<'a
                 // //         ^^^^^^^^^^^^^^^^^^^^^^
                 // ```
                 quasis[quasi_index - 1].span.end = quasi.span.end;
-
-                // Remove this quasi and the preceding expression.
-                // TODO: Remove scopes, symbols, and references for removed `Expression`.
-                quasis.remove(quasi_index);
-                expressions.remove(quasi_index - 1);
+                // Mark previous quasi for deletion, it will be deleted all of quasis are processed.
+                quasis[quasi_index - 1].value.raw = QUASI_DELETION_MARKER;
+                has_deletion_marker = true;
+                quasi_index += 1;
 
                 // Find end of comment
                 let start_index = if is_block_comment {
@@ -1087,6 +1089,14 @@ fn minify_template_literal<'a>(lit: &mut TemplateLiteral<'a>, ast: AstBuilder<'a
     // SAFETY: Output is all picked from the original `raw` values and is guaranteed to be valid UTF-8.
     let output_str = unsafe { std::str::from_utf8_unchecked(&output) };
     quasis.last_mut().unwrap().value.raw = ast.atom(output_str);
+
+    // Remove quasis that were marked for deletion as well as their preceding expressions.
+    if has_deletion_marker {
+        let mut quasis_iter = quasis.iter();
+        // TODO: Remove scopes, symbols, and references for removed `Expression`.
+        expressions.retain(|_| quasis_iter.next().unwrap().value.raw != QUASI_DELETION_MARKER);
+        quasis.retain(|quasi| quasi.value.raw != QUASI_DELETION_MARKER);
+    }
 }
 
 #[cfg(test)]