Update codeit.js

Author: benhatsor

lib/codeit.js

@@ -586,48 +586,63 @@ class CodeitElement extends HTMLElement {
           // get current selection
           const s = window.getSelection();
 
-          // if there's no selection
-          if (s.isCollapsed) {
+          let selContents = s.toString();
+          
+          // if selection exists
+          if (!s.isCollapsed) {
 
-            // delete a tab from selection
-  
+            let lines = selContents.split('\n');
+            
+            // run on all lines
+            lines.forEach((line, index) => {
+              
+              // if line contains a tab
+              if (line.startsWith(cd.options.tab)) {
+                
+                // remove tab from line
+                lines[index] = line.slice(cd.options.tab.length);
+                
+              }
+              
+            });
+            
+            // join lines
+            selContents = lines.join('\n');
+            
+            
+            // delete selection
+            cd.deleteCurrentSelection();
+              
+            // insert tabbed selection
+            cd.insert(selContents, { moveToEnd: false });
+              
             // get caret pos in text
             const pos = cd.getSelection();
-  
-            let startPos = Math.min(pos.start, pos.end);
-            let endPos = Math.max(pos.start, pos.end);
-            
-            // get text before cursor
+              
+            // restore pos in text
+            cd.setSelection(pos.start, (pos.start + selContents.length));
 
-            cd.setSelection(endPos);
+          } else {
 
-            const before = cd.beforeCursor();
-  
-            // get padding of line
-            let [linePadding, lineStartPos] = getPadding(before);
+            let lastLine = cd.beforeCursor().split('\n');
+            lastLine = lastLine[lastLine.length-1];
 
-            // if line has tabs
-            if (linePadding.length > 0) {
-  
-              // delete a tab
+            // if current line contains a tab
+            if (lastLine.startsWith(cd.options.tab)) {
 
-              const tabLength = cd.options.tab.length;
-  
-              cd.setSelection(lineStartPos, lineStartPos + tabLength);
-  
+              // remove tab from line
+              
+              // get caret pos in text
+              const pos = cd.getSelection();
+              
+              // select the tab
+              cd.setSelection((pos.start - lastLine.length), (pos.start - lastLine.length + cd.options.tab.length));
+              
+              // delete selection
               cd.deleteCurrentSelection();
 
               // restore pos in text
-              cd.setSelection(lineStartPos + linePadding.length - tabLength, endPos - tabLength);
-  
-              const selectedTab = window.getSelection().toString().startsWith(cd.options.tab);
-              
-              if (selectedTab) {
-                
-                // restore pos in text
-                cd.setSelection(lineStartPos + linePadding.length, endPos - tabLength);
-                
-              }
+              cd.setSelection(pos.start);
 
             }