Add support for custom outlining regions

src/compiler/types.ts

@@ -2037,6 +2037,10 @@ namespace ts {
         end: -1;
     }
 
+    export interface RegionRange extends TextRange {
+        name?: string;
+    }
+
     // represents a top level: { type } expression in a JSDoc comment.
     export interface JSDocTypeExpression extends TypeNode {
         kind: SyntaxKind.JSDocTypeExpression;

src/services/outliningElementsCollector.ts

@@ -6,9 +6,14 @@ namespace ts.OutliningElementsCollector {
     export function collectElements(sourceFile: SourceFile, cancellationToken: CancellationToken): OutliningSpan[] {
         const elements: OutliningSpan[] = [];
         let depth = 0;
+        const regions: RegionRange[] = [];
+        const regionText = "#region";
+        const regionStart = new RegExp("^\\s*//\\s*#region(\\s+.*)?$", "gm");
+        const regionEnd = new RegExp("^\\s*//\\s*#endregion(\\s|$)", "gm");
 
         walk(sourceFile);
-        return elements;
+        gatherRegions();
+        return elements.sort((span1, span2) => span1.textSpan.start - span2.textSpan.start);
 
         /** If useFullStart is true, then the collapsing span includes leading whitespace, including linebreaks. */
         function addOutliningSpan(hintSpanNode: Node, startElement: Node, endElement: Node, autoCollapse: boolean, useFullStart: boolean) {
@@ -35,6 +40,18 @@ namespace ts.OutliningElementsCollector {
             }
         }
 
+        function addOutliningSpanRegions(regionSpan: RegionRange) {
+            if (regionSpan) {
+                const span: OutliningSpan = {
+                    textSpan: createTextSpanFromBounds(regionSpan.pos, regionSpan.end),
+                    hintSpan: createTextSpanFromBounds(regionSpan.pos, regionSpan.end),
+                    bannerText: regionSpan.name,
+                    autoCollapse: false,
+                };
+                elements.push(span);
+            }
+        }
+
         function addOutliningForLeadingCommentsForNode(n: Node) {
             const comments = ts.getLeadingCommentRangesOfNode(n, sourceFile);
 
@@ -89,6 +106,69 @@ namespace ts.OutliningElementsCollector {
             return isFunctionBlock(node) && node.parent.kind !== SyntaxKind.ArrowFunction;
         }
 
+        function isRegionStart(start: number, end: number) {
+            if (!ts.formatting.getRangeOfEnclosingComment(sourceFile, start, /*onlyMultiLine*/ true)) {
+                const comment = sourceFile.text.substring(start, end);
+                const result = comment.match(regionStart);
+
+                if (result && result.length > 0) {
+                    const sections = result[0].split(" ").filter(function (s) { return s !== ""; });
+
+                    if (sections[0] === "//") {
+                        if (sections.length > 2) {
+                            return result[0].substring(result[0].indexOf(sections[2]));
+                        }
+                        else {
+                            return regionText;
+                        }
+                    }
+                    else {
+                        if (sections.length > 1) {
+                            return result[0].substring(result[0].indexOf(sections[1]));
+                        }
+                        else {
+                            return regionText;
+                        }
+                    }
+                }
+            }
+            return "";
+        }
+
+        function isRegionEnd(start: number, end: number) {
+            if (!ts.formatting.getRangeOfEnclosingComment(sourceFile, start, /*onlyMultiLine*/ true)) {
+                const comment = sourceFile.text.substring(start, end);
+                return comment.match(regionEnd);
+            }
+            return undefined;
+        }
+
+        function gatherRegions(): void {
+            const lineStarts = sourceFile.getLineStarts();
+
+            for (const currentLineStart of lineStarts) {
+                const lineEnd = sourceFile.getLineEndOfPosition(currentLineStart);
+
+                const name = isRegionStart(currentLineStart, lineEnd);
+                if (name) {
+                    const region: RegionRange = {
+                        pos: currentLineStart,
+                        end: lineEnd,
+                        name,
+                    };
+                    regions.push(region);
+                }
+                else if (isRegionEnd(currentLineStart, lineEnd)) {
+                    const region = regions.pop();
+
+                    if (region) {
+                        region.end = lineEnd;
+                        addOutliningSpanRegions(region);
+                    }
+                }
+            }
+        }
+
         function walk(n: Node): void {
             cancellationToken.throwIfCancellationRequested();
             if (depth > maxDepth) {

tests/cases/fourslash/getOutliningSpansForRegions.ts

@@ -0,0 +1,46 @@
+/// <reference path="fourslash.ts"/>
+
+////// basic region
+////[|// #region
+////
+////// #endregion|]
+////
+////// region with label
+////[|// #region label1
+////
+////// #endregion|]
+////
+////// region with extra whitespace in all valid locations
+////[|             //              #region          label2    label3
+////
+////        //        #endregion|]
+////
+////// No space before directive
+////[|//#region label4
+////
+//////#endregion|]
+////
+////// Nested regions
+////[|// #region outer
+////
+////[|// #region inner
+////
+////// #endregion inner|]
+////
+////// #endregion outer|]
+////
+////// region delimiters not valid when preceding text on line
+//// test // #region invalid1
+////
+////test // #endregion
+////
+////// region delimiters not valid when in multiline comment
+/////*
+////// #region invalid2
+////*/
+////
+/////*
+////// #endregion
+////*/
+
+verify.outliningSpansInCurrentFile(test.ranges());