refs #16 Extract texts from multi cursors

Author: ryu1kn

lib/editor-text-extractor.js

@@ -2,8 +2,23 @@
 class EditorTextExtractor {
 
     extract(editor) {
-        const selection = !editor.selection.isEmpty ? editor.selection : null;
-        return editor.document.getText(selection);
+        return this._isTextSelected(editor.selections) ?
+            this._extractTexts(editor) : this._getEntireText(editor);
+    }
+
+    _isTextSelected(selections) {
+        return selections.some(selection => !selection.isEmpty);
+    }
+
+    _extractTexts(editor) {
+        return editor.selections
+            .filter(selection => !selection.isEmpty)
+            .map(selection => editor.document.getText(selection))
+            .join('\n');
+    }
+
+    _getEntireText(editor) {
+        return editor.document.getText();
     }
 
 }

test/lib/editor-text-extractor.test.js

@@ -5,25 +5,42 @@ suite('EditorTextExtractor', () => {
 
     test('it extracts a text from editor for comparison', () => {
         const extractor = new EditorTextExtractor();
-        const editor = fakeEditor('SELECTED_TEXT');
+        const editor = fakeEditor(['SELECTED_TEXT']);
         expect(extractor.extract(editor)).to.eql('SELECTED_TEXT');
     });
 
     test('it extracts a whole text if no text is currently selected', () => {
         const extractor = new EditorTextExtractor();
-        const editor = fakeEditor('');
-        expect(extractor.extract(editor)).to.eql('ENTIRE TEXT ');
+        const editor = fakeEditor(['']);
+        expect(extractor.extract(editor)).to.eql('ENTIRE TEXT');
     });
 
-    function fakeEditor(selectedText) {
+    test('it extracts texts selected by all cursors and join them with newline character', () => {
+        const extractor = new EditorTextExtractor();
+        const editor = fakeEditor(['SELECTED_TEXT_1', 'SELECTED_TEXT_2']);
+        expect(extractor.extract(editor)).to.eql('SELECTED_TEXT_1\nSELECTED_TEXT_2');
+    });
+
+    test('it ignores cursor that is not selecting text if others are selecting one', () => {
+        const extractor = new EditorTextExtractor();
+        const editor = fakeEditor(['', 'SELECTED_TEXT_1', '', 'SELECTED_TEXT_2']);
+        expect(extractor.extract(editor)).to.eql('SELECTED_TEXT_1\nSELECTED_TEXT_2');
+    });
+
+    test('it extracts the whole text if no cursors are selecting text', () => {
+        const extractor = new EditorTextExtractor();
+        const editor = fakeEditor(['', '', '']);
+        expect(extractor.extract(editor)).to.eql('ENTIRE TEXT');
+    });
+
+    function fakeEditor(selectedTexts) {
+        const selections = selectedTexts.map(text => ({text, isEmpty: !text}));
         return {
-            selection: {
-                text: selectedText,
-                isEmpty: !selectedText
-            },
+            selections,
+            selection: selections[0],
             document: {
-                _entireText: `ENTIRE TEXT ${selectedText}`,
-                getText: function (selection) {
+                _entireText: 'ENTIRE TEXT',
+                getText(selection) {
                     return selection ? selection.text : this._entireText;
                 }
             }