Merge pull request #2 from hannut91/fix-letter-cut-bug

Fix letter cut bug

package.json

@@ -2,7 +2,7 @@
   "name": "korean-spell-checker-vs-code",
   "displayName": "Korean Spell Checker VS Code",
   "description": "",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "engines": {
     "vscode": "^1.38.0"
   },
@@ -36,13 +36,15 @@
     "lint": "./node_modules/.bin/tslint -p ."
   },
   "devDependencies": {
+    "@types/lodash": "^4.14.144",
     "@types/node": "^12.11.1",
     "@types/vscode": "^1.38.0",
     "tslint": "^5.20.0",
     "typescript": "^3.6.4"
   },
   "dependencies": {
-    "axios": "^0.19.0"
+    "axios": "^0.19.0",
+    "lodash": "^4.17.15"
   },
   "publisher": "Yunseok"
 }

src/commands/spell-fix.ts

@@ -1,27 +1,74 @@
-import { window, TextEditorEdit } from 'vscode';
+import {
+  window, TextEditorEdit, Position, Range, Selection, TextLine
+} from 'vscode';
+import { range, curry } from 'lodash';
+import * as _ from 'lodash';
 
 import { SpellCheck } from '../services/spell-checker';
 
+const MAX_TEXT_COUNT = 500;
+
+const getText = curry((textline: TextLine, selection: Selection) => {
+  const { lineNumber } = textline;
+  if (lineNumber === selection.start.line) {
+    return new Range(
+      selection.start, textline.rangeIncludingLineBreak.end)
+  }
+
+  if (lineNumber === selection.end.line) {
+    return new Range(textline.range.start, selection.end);
+  }
+
+  return textline.rangeIncludingLineBreak;
+})
+
 export const spellFix = async () => {
   const editor = window.activeTextEditor;
   if (!editor) {
     window.showInformationMessage('선택된 에디터가 없습니다.');
     return;
   }
 
-  const text = editor.document.getText(editor.selection);
-  if (!text) {
+  if (!editor.document.getText(editor.selection)) {
     window.showInformationMessage('선택된 텍스트가 없습니다.');
     return;
   }
 
+  let { selection, document } = editor;
+
+  let originText = '';
+
+  if (selection.isSingleLine) {
+    originText = document.getText(selection);
+  } else {
+    range(selection.start.line, selection.end.line + 1)
+      .map(document.lineAt)
+      .map(getText(_, selection))
+      .map(document.getText)
+      .every((text: string, index: number) => {
+        let result = true;
+        if (originText.length + text.length >= MAX_TEXT_COUNT) {
+          text = text.slice(0, MAX_TEXT_COUNT - originText.length);
+          selection = new Selection(
+            selection.start,
+            new Position(selection.start.line + index, text.length),
+          );
+          result = false;
+        }
+
+        originText += text;
+        return result;
+      });
+  }
+
   try {
-    const { notag_html } = await SpellCheck(text);
+    const { notag_html } = await SpellCheck(originText);
 
     const fixedText = notag_html.replace(/<br>/g, '\n');
 
     editor.edit((editBuilder: TextEditorEdit) => {
-      editBuilder.replace(editor.selection, fixedText);
+      editBuilder.replace(selection, fixedText);
+      editor.selection = selection;
     });
   } catch (err) {
     window.showInformationMessage(