Pvb/discover code actions

Jakefile.js

@@ -96,7 +96,13 @@ var servicesSources = [
     "formatting/rulesMap.ts",
     "formatting/rulesProvider.ts",
     "formatting/smartIndenter.ts",
-    "formatting/tokenRange.ts"
+    "formatting/tokenRange.ts",
+    "codeFixes/changeExtendsToImplementsFix.ts",
+    "codeFixes/codeFixProvider.ts",
+    "codeFixes/interfaceFixes.ts",
+    "codeFixes/references.ts",
+    "codeFixes/superFixes.ts",
+    "codeFixes/unusedIdentifierFixes.ts"
 ].map(function (f) {
     return path.join(servicesDirectory, f);
 }));

src/compiler/core.ts

@@ -115,6 +115,15 @@ namespace ts {
         return -1;
     }
 
+    export function firstOrUndefined<T>(array: T[], predicate: (x: T) => boolean): T {
+        for (let i = 0, len = array.length; i < len; i++) {
+            if (predicate(array[i])) {
+                return array[i];
+            }
+        }
+        return undefined;
+    }
+
     export function indexOfAnyCharCode(text: string, charCodes: number[], start?: number): number {
         for (let i = start || 0, len = text.length; i < len; i++) {
             if (contains(charCodes, text.charCodeAt(i))) {

src/compiler/diagnosticMessages.json

@@ -3031,5 +3031,33 @@
     "Unknown typing option '{0}'.": {
         "category": "Error",
         "code": 17010
+    },
+    "Add missing 'super()' call.": {
+        "category": "CodeFix",
+        "code": 90001
+    },
+    "Make 'super()' call the first statement in the constructor.": {
+        "category": "CodeFix",
+        "code": 90002
+    },
+    "Change 'extends' to 'implements'": {
+        "category": "CodeFix",
+        "code": 90003
+    },
+    "Remove unused identifiers": {
+        "category": "CodeFix",
+        "code": 90004
+    },
+    "Implement interface on reference": {
+        "category": "CodeFix",
+        "code": 90005
+    },
+    "Implement interface on class": {
+        "category": "CodeFix",
+        "code": 90006
+    },
+    "Implement inherited abstract class": {
+        "category": "CodeFix",
+        "code": 90007
     }
 }

src/compiler/types.ts

@@ -2547,6 +2547,7 @@ namespace ts {
         Warning,
         Error,
         Message,
+        CodeFix,
     }
 
     export enum ModuleResolutionKind {

src/harness/fourslash.ts

@@ -384,7 +384,7 @@ namespace FourSlash {
 
             if (exists !== negative) {
                 this.printErrorLog(negative, this.getAllDiagnostics());
-                throw new Error("Failure between markers: " + startMarkerName + ", " + endMarkerName);
+                throw new Error(`Failure between markers: '${startMarkerName}', '${endMarkerName}'`);
             }
         }
 
@@ -638,7 +638,6 @@ namespace FourSlash {
             }
         }
 
-
         public verifyCompletionListAllowsNewIdentifier(negative: boolean) {
             const completions = this.getCompletionListAtCaret();
 
@@ -1479,7 +1478,7 @@ namespace FourSlash {
             if (isFormattingEdit) {
                 const newContent = this.getFileContent(fileName);
 
-                if (newContent.replace(/\s/g, "") !== oldContent.replace(/\s/g, "")) {
+                if (this.removeWhitespace(newContent) !== this.removeWhitespace(oldContent)) {
                     this.raiseError("Formatting operation destroyed non-whitespace content");
                 }
             }
@@ -1545,6 +1544,10 @@ namespace FourSlash {
             }
         }
 
+        private removeWhitespace(text: string): string {
+            return text.replace(/\s/g, "");
+        }
+
         public goToBOF() {
             this.goToPosition(0);
         }
@@ -1862,6 +1865,44 @@ namespace FourSlash {
             }
         }
 
+        public verifyCodeFixAtPosition(expectedText: string, errorCode?: number) {
+
+            const ranges = this.getRanges();
+            if (ranges.length == 0) {
+                this.raiseError("At least one range should be specified in the testfile.");
+            }
+
+            const fileName = this.activeFile.fileName;
+            const diagnostics = this.getDiagnostics(fileName);
+
+            if (diagnostics.length === 0) {
+                this.raiseError("Errors expected.");
+            }
+
+            if (diagnostics.length > 1 && !errorCode) {
+                this.raiseError("When there's more than one error, you must specify the errror to fix.");
+            }
+
+            const diagnostic = !errorCode ? diagnostics[0] : ts.firstOrUndefined(diagnostics, d => d.code == errorCode);
+
+            const actual = this.languageService.getCodeFixesAtPosition(fileName, diagnostic.start, diagnostic.length, [`TS${diagnostic.code}`]);
+
+            if (!actual || actual.length == 0) {
+                this.raiseError("No codefixes returned.");
+            }
+
+            if (actual.length > 1) {
+                this.raiseError("More than 1 codefix returned.");
+            }
+
+            this.applyEdits(actual[0].changes[0].fileName, actual[0].changes[0].textChanges, /*isFormattingEdit*/ false);
+            const actualText = this.rangeText(ranges[0]);
+
+            if (this.removeWhitespace(actualText) !== this.removeWhitespace(expectedText)) {
+                this.raiseError(`Actual text doesn't match expected text. Actual: '${actualText}' Expected: '${expectedText}'`);
+            }
+        }
+
         public verifyDocCommentTemplate(expected?: ts.TextInsertion) {
             const name = "verifyDocCommentTemplate";
             const actual = this.languageService.getDocCommentTemplateAtPosition(this.activeFile.fileName, this.currentCaretPosition);
@@ -3066,6 +3107,10 @@ namespace FourSlashInterface {
             this.DocCommentTemplate(/*expectedText*/ undefined, /*expectedOffset*/ undefined, /*empty*/ true);
         }
 
+        public codeFixAtPosition(expectedText: string, errorCode?: number): void {
+            this.state.verifyCodeFixAtPosition(expectedText, errorCode);
+        }
+
         public navigationBar(json: any) {
             this.state.verifyNavigationBar(json);
         }

src/harness/harnessLanguageService.ts

@@ -453,6 +453,9 @@ namespace Harness.LanguageService {
         isValidBraceCompletionAtPosition(fileName: string, position: number, openingBrace: number): boolean {
             return unwrapJSONCallResult(this.shim.isValidBraceCompletionAtPosition(fileName, position, openingBrace));
         }
+        getCodeFixesAtPosition(fileName: string, start: number, end: number, errorCodes: string[]): ts.CodeAction[] {
+            return unwrapJSONCallResult(this.shim.getCodeFixesAtPosition(fileName, start, end, JSON.stringify(errorCodes)));
+        }
         getEmitOutput(fileName: string): ts.EmitOutput {
             return unwrapJSONCallResult(this.shim.getEmitOutput(fileName));
         }

src/server/client.ts

@@ -592,6 +592,10 @@ namespace ts.server {
             throw new Error("Not Implemented Yet.");
         }
 
+        getCodeFixesAtPosition(fileName: string, start: number, end: number, errorCodes: string[]): ts.CodeAction[] {
+            throw new Error("Not Implemented Yet.");
+        }
+
         getBraceMatchingAtPosition(fileName: string, position: number): TextSpan[] {
             const lineOffset = this.positionToOneBasedLineOffset(fileName, position);
             const args: protocol.FileLocationRequestArgs = {

src/services/codefixes/changeExtendsToImplementsFix.ts

@@ -0,0 +1,35 @@
+/* @internal */
+namespace ts.codeFix {
+    registerCodeFix({
+        name: "changeExtendsToImplementsFix",
+        errorCodes: ["TS2689"],
+        getCodeActions: (context: CodeFixContext) => {
+            const sourceFile = context.sourceFile;
+            const start = context.span.start;
+            const token = getTokenAtPosition(sourceFile, start);
+
+            const textChanges: TextChange[] = [];
+
+            if (token.kind === SyntaxKind.Identifier && token.parent.parent.kind === SyntaxKind.HeritageClause) {
+                const children = (<HeritageClause>token.parent.parent).getChildren();
+                for (const child of children) {
+                    if (child.kind === SyntaxKind.ExtendsKeyword) {
+                        textChanges.push({ newText: " implements", span: { start: child.pos, length: child.end - child.pos } });
+                    }
+                }
+            }
+
+            if (textChanges.length > 0) {
+                return [{
+                    description: getLocaleSpecificMessage(Diagnostics.Change_extends_to_implements),
+                    changes: [{
+                        fileName: sourceFile.fileName,
+                        textChanges: textChanges
+                    }]
+                }];
+            }
+
+            Debug.fail("Failed to construct a fix.");
+        }
+    });
+}

src/services/codefixes/codeFixProvider.ts

@@ -0,0 +1,53 @@
+/* @internal */
+namespace ts {
+    export interface CodeFix {
+        name: string;
+        errorCodes: string[];
+        getCodeActions(context: CodeFixContext): CodeAction[];
+    }
+
+    export interface CodeFixContext {
+        errorCode: string;
+        sourceFile: SourceFile;
+        span: TextSpan;
+        checker: TypeChecker;
+        newLineCharacter: string;
+    }
+
+    export namespace codeFix {
+        const codeFixes: Map<CodeFix[]> = {};
+
+        export function registerCodeFix(action: CodeFix) {
+            forEach(action.errorCodes, error => {
+                let fixes = codeFixes[error];
+                if (!fixes) {
+                    fixes = [];
+                    codeFixes[error] = fixes;
+                }
+                fixes.push(action);
+            });
+        }
+
+        export class CodeFixProvider {
+            public static getSupportedErrorCodes() {
+                return getKeys(codeFixes);
+            }
+
+            public getFixes(context: CodeFixContext): CodeAction[] {
+                const fixes = codeFixes[context.errorCode];
+                let allActions: CodeAction[] = [];
+
+                Debug.assert(fixes && fixes.length > 0, "No fixes found for error: '${errorCode}'.");
+
+                forEach(fixes, f => {
+                    const actions = f.getCodeActions(context);
+                    if (actions && actions.length > 0) {
+                        allActions = allActions.concat(actions);
+                    }
+                });
+
+                return allActions;
+            }
+        }
+    }
+}