fix: Detect leetcode file by the comments (LeetCode-OpenSource#417)

jdneo · web-flow · commit a6d34bc98994 · 2019-09-14T15:16:11.000+08:00
diff --git a/src/codelens/CustomCodeLensProvider.ts b/src/codelens/CustomCodeLensProvider.ts
@@ -22,8 +22,8 @@ export class CustomCodeLensProvider implements vscode.CodeLensProvider {
             return;
         }
 
-        const fileName: string = document.fileName.trim();
-        const matchResult: RegExpMatchArray | null = fileName.match(/\d+\..*\.(.+)/);
+        const content: string = document.getText();
+        const matchResult: RegExpMatchArray | null = content.match(/@lc app=.* id=.* lang=.*/);
         if (!matchResult) {
             return undefined;
         }

Original file line number	Diff line number	Diff line change
`@@ -22,8 +22,8 @@ export class CustomCodeLensProvider implements vscode.CodeLensProvider {`
`22`	`22`	`return;`
`23`	`23`	`}`
`24`	`24`
`25`		`- const fileName: string = document.fileName.trim();`
`26`		`- const matchResult: RegExpMatchArray \| null = fileName.match(/\d+\..*\.(.+)/);`
	`25`	`+ const content: string = document.getText();`
	`26`	`+ const matchResult: RegExpMatchArray \| null = content.match(/@lc app=.* id=.* lang=.*/);`
`27`	`27`	`if (!matchResult) {`
`28`	`28`	`return undefined;`
`29`	`29`	`}`