add supports of completion label list

Author: Kingwl

src/services/completions.ts

@@ -39,6 +39,10 @@ namespace ts.Completions {
             return getStringLiteralCompletionEntries(sourceFile, position, typeChecker, compilerOptions, host, log);
         }
 
+        if (isInBreakOrContinue(sourceFile, position)) {
+            return getLabelCompletionAtPosition(sourceFile, position);
+        }
+
         const completionData = getCompletionData(typeChecker, log, sourceFile, position, allSourceFiles, options, compilerOptions.target);
         if (!completionData) {
             return undefined;
@@ -211,6 +215,19 @@ namespace ts.Completions {
         return uniques;
     }
 
+    function getLabelCompletionAtPosition(sourceFile: SourceFile, position: number): CompletionInfo | undefined {
+        const contextToken: Node = findPrecedingToken(position, sourceFile);
+        if (!contextToken || !contextToken.parent || !isBreakOrContinueStatement(contextToken.parent)) {
+            return undefined;
+        }
+        const entries: CompletionEntry[] = [];
+        addLabelStatementCompletions(contextToken.parent, entries);
+
+        if (entries.length) {
+            return { isGlobalCompletion: false, isMemberCompletion: false, isNewIdentifierLocation: false, entries };
+        }
+    }
+
     function getStringLiteralCompletionEntries(sourceFile: SourceFile, position: number, typeChecker: TypeChecker, compilerOptions: CompilerOptions, host: LanguageServiceHost, log: Log): CompletionInfo | undefined {
         const node = findPrecedingToken(position, sourceFile);
         if (!node || node.kind !== SyntaxKind.StringLiteral) {
@@ -346,6 +363,28 @@ namespace ts.Completions {
         return undefined;
     }
 
+    function addLabelStatementCompletions(node: Node, entries: CompletionEntry[], uniques = createMap<true>()): void {
+        let current: Node = node;
+        while (current) {
+            if (isFunctionLike(current)) {
+                break;
+            }
+            if (isLabeledStatement(current)) {
+                const name = current.label.escapedText as string;
+                if (!uniques.has(name)) {
+                    uniques.set(name, true);
+                    entries.push({
+                        name,
+                        kindModifiers: ScriptElementKindModifier.none,
+                        kind: ScriptElementKind.label,
+                        sortText: "0"
+                    });
+                }
+            }
+            current = current.parent;
+        }
+    }
+
     function addStringLiteralCompletionsFromType(type: Type, result: Push<CompletionEntry>, typeChecker: TypeChecker, uniques = createMap<true>()): void {
         if (type && type.flags & TypeFlags.TypeParameter) {
             type = typeChecker.getBaseConstraintOfType(type);

src/services/utilities.ts

@@ -857,6 +857,11 @@ namespace ts {
         return false;
     }
 
+    export function isInBreakOrContinue(sourceFile: SourceFile, position: number): boolean {
+        const contextToken = findPrecedingToken(position, sourceFile);
+        return contextToken && contextToken.parent && isBreakOrContinueStatement(contextToken.parent);
+    }
+
     /**
      * returns true if the position is in between the open and close elements of an JSX expression.
      */

tests/cases/fourslash/completionListWithLabel.ts

@@ -0,0 +1,44 @@
+/// <reference path="fourslash.ts"/>
+
+//// label: while (true) {
+////    break /*1*/
+////    continue /*2*/
+////    testlabel: while (true) {
+////        break /*3*/
+////        continue /*4*/
+////        break tes/*5*/
+////        continue tes/*6*/
+////    }
+////    break /*7*/
+////    break; /*8*/
+////}
+
+goTo.marker("1");
+verify.completionListContains("label");
+
+goTo.marker("2");
+verify.completionListContains("label");
+verify.not.completionListContains("testlabel");
+
+goTo.marker("3");
+verify.completionListContains("label");
+verify.completionListContains("testlabel");
+
+goTo.marker("4");
+verify.completionListContains("label");
+verify.completionListContains("testlabel");
+
+goTo.marker("5");
+verify.completionListContains("testlabel");
+verify.completionListContains("label");
+
+goTo.marker("6");
+verify.completionListContains("testlabel");
+verify.completionListContains("label");
+
+goTo.marker("7");
+verify.completionListContains("label");
+verify.not.completionListContains("testlabel");
+
+goTo.marker("8");
+verify.not.completionListContains("label");