feat(napi/parer): add get_line_column_number API

oxc-project · Dec 10, 2024 · bd36325 · bd36325
1 parent 52a4180
commit bd36325
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/napi/parser/Cargo.toml b/napi/parser/Cargo.toml
@@ -23,6 +23,7 @@ doctest = false
 [dependencies]
 oxc = { workspace = true }
 oxc_ast = { workspace = true, features = ["serialize"] } # enable feature only
+oxc_data_structures = { workspace = true }
 oxc_napi = { workspace = true }
 
 rustc-hash = { workspace = true }

diff --git a/napi/parser/src/magic_string.rs b/napi/parser/src/magic_string.rs
@@ -2,6 +2,7 @@
 // use std::sync::Arc;
 
 use napi_derive::napi;
+use oxc_data_structures::rope::{get_line_column, Rope};
 
 // use oxc_sourcemap::napi::SourceMap;
 use self_cell::self_cell;
@@ -10,6 +11,7 @@ use string_wizard::MagicString as MS;
 #[napi]
 pub struct MagicString {
     cell: MagicStringImpl,
+    rope: Option<Rope>,
 }
 
 self_cell!(
@@ -26,6 +28,12 @@ impl MagicString {
     }
 }
 
+#[napi(object)]
+pub struct LineColumn {
+    pub line: u32,
+    pub column: u32,
+}
+
 #[napi(object)]
 pub struct OverwriteOptions {
     pub content_only: bool,
@@ -45,6 +53,14 @@ impl MagicString {
         &self.cell.borrow_owner()[start as usize..end as usize]
     }
 
+    #[napi]
+    pub fn get_line_column(&self, offest: u32) -> LineColumn {
+        let source_text = self.cell.borrow_owner();
+        let rope = self.rope.get_or_insert_with(|| Rope::from_str(source_text));
+        let (line, column) = get_line_column(rope, offset, source_text);
+        LineColumn { line, column }
+    }
+
     #[napi]
     pub fn length(&self) -> u32 {
         self.cell.borrow_dependent().len() as u32