1 files changed, 30 insertions, 2 deletions

diff --git a/code/native/src/lib.rs b/code/native/src/lib.rs
index aae7ad2f3..cb304a141 100644
--- a/code/native/src/lib.rs
+++ b/code/native/src/lib.rs
@@ -3,6 +3,7 @@ extern crate neon;
 extern crate libeditor;
 
 use neon::prelude::*;
+use libeditor::TextRange;
 
 pub struct Wrapper {
     inner: libeditor::File,
@@ -19,8 +20,8 @@ declare_types! {
         }
 
         method syntaxTree(mut cx) {
-            let this = cx.this();
             let tree = {
+            let this = cx.this();
                 let guard = cx.lock();
                 let wrapper = this.borrow(&guard);
                 wrapper.inner.syntax_tree()
@@ -29,8 +30,8 @@ declare_types! {
         }
 
         method highlight(mut cx) {
-            let this = cx.this();
             let highlights = {
+                let this = cx.this();
                 let guard = cx.lock();
                 let wrapper = this.borrow(&guard);
                 wrapper.inner.highlight()
@@ -51,6 +52,33 @@ declare_types! {
 
             Ok(res.upcast())
         }
+
+        method extendSelection(mut cx) {
+            let from_offset = cx.argument::<JsNumber>(0)?.value() as u32;
+            let to_offset = cx.argument::<JsNumber>(1)?.value() as u32;
+            let text_range = TextRange::from_to(from_offset.into(), to_offset.into());
+            let extended_range = {
+                let this = cx.this();
+                let guard = cx.lock();
+                let wrapper = this.borrow(&guard);
+                wrapper.inner.extend_selection(text_range)
+            };
+
+            match extended_range {
+                None => Ok(cx.null().upcast()),
+                Some(range) => {
+                    let start: u32 = range.start().into();
+                    let end: u32 = range.end().into();
+                    let start = cx.number(start);
+                    let end = cx.number(end);
+                    let arr = cx.empty_array();
+                    arr.set(&mut cx, 0, start)?;
+                    arr.set(&mut cx, 1, end)?;
+                    Ok(arr.upcast())
+                }
+            }
+
+        }
     }
 
 }