Fixes msiniscalchi#176 (#1)

asinghvi17 · Nov 16, 2019 · e4455c7 · e4455c7
1 parent f4d6c61
commit e4455c7
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/lib/snippet-manager.coffee b/lib/snippet-manager.coffee
@@ -151,10 +151,10 @@ class SnippetManager extends LTool
     te = atom.workspace.getActiveTextEditor()
 
     # First, check if there is a selection, and if so, add $..$ around it
-    if text =  te.getSelectedText()
-      text = text.replace(/\\/g, "\\\\")
-      range = te.getSelectedBufferRange()
-      te.setSelectedBufferRange(range, '')
+    range = te.getSelectedBufferRange()
+    text = te.getTextInBufferRange(range)
+    text = text.replace(/\\/g, "\\\\")
+    if text
       @snippetService.insertSnippet("\$#{text}\$")
       return
 
@@ -186,11 +186,10 @@ class SnippetManager extends LTool
     te = atom.workspace.getActiveTextEditor()
 
     # First, check if there is a selection, and if so, add quotes around it
-    if text =  te.getSelectedText()
-      text = text.replace(/\\/g, "\\\\")
-      range = te.getSelectedBufferRange()
-      te.setSelectedBufferRange(range, '')
-      # Use snippet to leave selection on (same as ST)
+    range = te.getSelectedBufferRange()
+    text = te.getTextInBufferRange(range)
+    text = text.replace(/\\/g, "\\\\")
+    if text
       @snippetService.insertSnippet("#{left}${1:#{text}}#{right}")
       return