refactor: try unsafe cell

Author: SyMind

src/replace_source.rs

@@ -1,6 +1,6 @@
 use std::{
-  borrow::Cow,
-  cell::RefCell,
+  borrow::{Borrow, Cow},
+  cell::{RefCell, UnsafeCell},
   hash::{Hash, Hasher},
   sync::{
     atomic::{AtomicBool, Ordering},
@@ -265,7 +265,7 @@ impl<'a, T: Source> StreamChunks<'a> for ReplaceSource<T> {
     on_name: crate::helpers::OnName<'_, 'a>,
   ) -> crate::helpers::GeneratedInfo {
     self.sort_replacement();
-    let on_name = RefCell::new(on_name);
+    let on_name = UnsafeCell::new(on_name);
     let repls = self.replacements();
     let mut pos: u32 = 0;
     let mut i: usize = 0;
@@ -274,12 +274,12 @@ impl<'a, T: Source> StreamChunks<'a> for ReplaceSource<T> {
     let mut generated_line_offset: i64 = 0;
     let mut generated_column_offset: i64 = 0;
     let mut generated_column_offset_line = 0;
-    let source_content_lines: RefCell<Vec<Option<Vec<&str>>>> =
-      RefCell::new(Vec::new());
-    let name_mapping: RefCell<HashMap<Cow<str>, u32>> =
-      RefCell::new(HashMap::default());
-    let name_index_mapping: RefCell<HashMap<u32, u32>> =
-      RefCell::new(HashMap::default());
+    let source_content_lines: UnsafeCell<Vec<Option<Vec<&str>>>> =
+    UnsafeCell::new(Vec::new());
+    let name_mapping: UnsafeCell<HashMap<Cow<str>, u32>> =
+    UnsafeCell::new(HashMap::default());
+    let name_index_mapping: UnsafeCell<HashMap<u32, u32>> =
+    UnsafeCell::new(HashMap::default());
 
     // check if source_content[line][col] is equal to expect
     // Why this is needed?
@@ -309,8 +309,9 @@ impl<'a, T: Source> StreamChunks<'a> for ReplaceSource<T> {
     // webpack-sources also have this function, refer https://github.com/webpack/webpack-sources/blob/main/lib/ReplaceSource.js#L158
     let check_original_content =
       |source_index: u32, line: u32, column: u32, expected_chunk: &str| {
+        let source_content_lines = unsafe { &mut *source_content_lines.get() }; 
         if let Some(Some(content_lines)) =
-          source_content_lines.borrow().get(source_index as usize)
+        source_content_lines.get(source_index as usize)
         {
           if let Some(content_line) = content_lines.get(line as usize - 1) {
             WithIndices::new(content_line).substring(
@@ -405,7 +406,8 @@ impl<'a, T: Source> StreamChunks<'a> for ReplaceSource<T> {
                     original_line: original.original_line,
                     original_column: original.original_column,
                     name_index: original.name_index.and_then(|name_index| {
-                      name_index_mapping.borrow().get(&name_index).copied()
+                      let name_index_mapping = unsafe { &mut *name_index_mapping.get() }; 
+                      name_index_mapping.get(&name_index).copied()
                     }),
                   }
                 }),
@@ -437,12 +439,13 @@ impl<'a, T: Source> StreamChunks<'a> for ReplaceSource<T> {
           if let Some(name) =
             repl.name.as_ref().filter(|_| mapping.original.is_some())
           {
-            let mut name_mapping = name_mapping.borrow_mut();
+            let name_mapping = unsafe { &mut *name_mapping.get() };
             let mut global_index = name_mapping.get(name.as_str()).copied();
             if global_index.is_none() {
               let len = name_mapping.len() as u32;
               name_mapping.insert(Cow::Borrowed(name), len);
-              on_name.borrow_mut()(len, Cow::Owned(name.to_string()));
+              let on_name = unsafe { &mut *on_name.get() }; 
+              on_name(len, Cow::Owned(name.to_string()));
               global_index = Some(len);
             }
             replacement_name_index = global_index;
@@ -582,7 +585,8 @@ impl<'a, T: Source> StreamChunks<'a> for ReplaceSource<T> {
                   original_line: original.original_line,
                   original_column: original.original_column,
                   name_index: original.name_index.and_then(|name_index| {
-                    name_index_mapping.borrow().get(&name_index).copied()
+                    let name_index_mapping = unsafe { &mut *name_index_mapping.get() };
+                    name_index_mapping.get(&name_index).copied()
                   }),
                 }
               }),
@@ -592,7 +596,7 @@ impl<'a, T: Source> StreamChunks<'a> for ReplaceSource<T> {
         pos = end_pos;
       },
       &mut |source_index, source, source_content| {
-        let mut source_content_lines = source_content_lines.borrow_mut();
+        let source_content_lines = unsafe { &mut *source_content_lines.get() };
         while source_content_lines.len() <= source_index as usize {
           source_content_lines.push(None);
         }
@@ -601,16 +605,17 @@ impl<'a, T: Source> StreamChunks<'a> for ReplaceSource<T> {
         on_source(source_index, source, source_content);
       },
       &mut |name_index, name| {
-        let mut name_mapping = name_mapping.borrow_mut();
+        let name_mapping = unsafe { &mut *name_mapping.get() };
         let mut global_index = name_mapping.get(&name).copied();
         if global_index.is_none() {
           let len = name_mapping.len() as u32;
           name_mapping.insert(name.clone(), len);
-          on_name.borrow_mut()(len, name);
+          let on_name = unsafe { &mut *on_name.get() }; 
+          on_name(len, name);
           global_index = Some(len);
         }
+        let name_index_mapping = unsafe { &mut *name_index_mapping.get() };
         name_index_mapping
-          .borrow_mut()
           .insert(name_index, global_index.unwrap());
       },
     );