✨ Add "toggle block quote"

package.json

@@ -65,6 +65,11 @@
                 "title": "%command.editing.toggleCodeSpan.title%",
                 "category": "Markdown All in One"
             },
+            {
+                "command": "markdown.extension.editing.toggleQuote",
+                "title": "%command.editing.toggleQuote.title%",
+                "category": "Markdown All in One"
+            },
             {
                 "command": "markdown.extension.editing.toggleMath",
                 "title": "%command.editing.toggleMath.title%",

package.nls.json

@@ -12,6 +12,7 @@
     "command.editing.toggleMathReverse.title": "Toggle math environment (in reverse order)",
     "command.editing.toggleList.title": "Toggle list",
     "command.editing.toggleCodeBlock.title": "Toggle code block",
+    "command.editing.toggleQuote.title": "Toggle quote",
     "config.title": "Markdown All in One",
     "config.toc.levels.description": "Range of levels for table of contents. Use `x..y` for level x to y",
     "config.toc.orderedList.description": "Use ordered list (1. ..., 2. ...)",

package.nls.zh-cn.json

@@ -10,6 +10,7 @@
     "command.editing.toggleMath.title": "触发数学环境",
     "command.editing.toggleMathReverse.title": "触发数学环境（反向）",
     "command.editing.toggleList.title": "触发列表",
+    "command.editing.toggleQuote.title": "触发引用",
     "config.title": "Markdown All in One",
     "config.toc.levels.description": "目录级别的范围. 例如 `2..5` 表示在目录中只包含 2 到 5 级标题",
     "config.toc.orderedList.description": "使用有序列表 (1. ..., 2. ...)",

src/formatting.ts

@@ -1,6 +1,6 @@
 'use strict';
 
-import { commands, env, ExtensionContext, Position, Range, Selection, SnippetString, TextDocument, TextEditor, window, workspace, WorkspaceEdit } from 'vscode';
+import { commands, env, EndOfLine, ExtensionContext, Position, Range, Selection, SnippetString, TextDocument, TextEditor, window, workspace, WorkspaceEdit } from 'vscode';
 import { fixMarker } from './listEditing';
 
 export function activate(context: ExtensionContext) {
@@ -15,6 +15,7 @@ export function activate(context: ExtensionContext) {
         commands.registerCommand('markdown.extension.editing.toggleHeadingDown', toggleHeadingDown),
         commands.registerCommand('markdown.extension.editing.toggleList', toggleList),
         commands.registerCommand('markdown.extension.editing.toggleCodeBlock', toggleCodeBlock),
+        commands.registerCommand('markdown.extension.editing.toggleQuote', toggleQuote),
         commands.registerCommand('markdown.extension.editing.paste', paste),
         commands.registerCommand('markdown.extension.editing._wrapBy', args => styleByWrapping(args['before'], args['after']))
     );
@@ -45,6 +46,70 @@ function toggleCodeBlock() {
     return editor.insertSnippet(new SnippetString('```$0\n$TM_SELECTED_TEXT\n```'));
 }
 
+enum QuoteState {
+    // State 1: part of lines has been quoted, need to be quoted
+    PARTIALQUOTED,
+    // State 2: all of lines has been quoted, need to be unquoted
+    FULLQUOTED,
+}
+
+function getQuoteState(lineArray: Array<string>): QuoteState {
+    return lineArray.every(line => line.startsWith('>')) ? QuoteState.FULLQUOTED : QuoteState.PARTIALQUOTED;
+}
+
+function quoteLine(line: string): string {
+    if(!line.startsWith(">")){
+        return "> " + line;
+    } else {
+        return line;
+    }
+}
+
+function unquoteLine(line: string): string {
+    if (!line.startsWith("> ") && line.startsWith(">")) {
+        return line.slice(2);
+    }
+    else if (line.startsWith(">")) {
+        return line.slice(1);
+    } else {
+        return line;
+    }
+}
+
+function setQuote(lineArray: Array<string>, state: QuoteState, eol: string): string {
+    let resultArray;
+    switch(state) {
+        case QuoteState.PARTIALQUOTED:
+            resultArray = lineArray.map(quoteLine);
+            break;
+        case QuoteState.FULLQUOTED:
+            resultArray = lineArray.map(unquoteLine); 
+            break;
+        default:
+            resultArray = lineArray;
+    }
+    return resultArray.join(eol);
+}
+
+function toggleQuote() {
+    let editor = window.activeTextEditor;
+    let start = editor.selection.start;
+    let end = editor.selection.end;
+    start = editor.document.lineAt(start.line).range.start;
+    end = editor.document.lineAt(end.line).range.end;
+    let linesText = editor.document.getText(new Range(start, end));;
+    let eol = editor.document.eol === EndOfLine.CRLF ? '\r\n' : '\n';
+
+    let line_array = linesText.split(eol);
+    let state = getQuoteState(line_array);
+    console.log("state", state);
+    let resultText = setQuote(line_array, state, eol);
+
+    return editor.edit(editBuilder => {
+        editBuilder.replace(new Range(start, end), resultText);
+    });
+}
+
 function toggleStrikethrough() {
     return styleByWrapping('~~');
 }