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

LeetCode-OpenSource · Sep 14, 2019 · a6d34bc · a6d34bc
1 parent ff9e364
commit a6d34bc
Showing 1 changed file with 2 additions and 2 deletions.
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;
         }