Fix for files directly inside the lib folder

bluefireteam · Jul 12, 2018 · a0d6348 · a0d6348
1 parent 110dab1
commit a0d6348
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/extension.ts b/src/extension.ts
@@ -58,12 +58,12 @@ export async function activate(context: vscode.ExtensionContext) {
         }
 
     const currentPath = editor.document.fileName.replace(/\/[^\/]*.dart$/, '');
-        const libFolder = packageInfo.projectRoot + '/lib/';
+        const libFolder = packageInfo.projectRoot + '/lib';
         if (!currentPath.startsWith(libFolder)) {
             vscode.window.showErrorMessage('Current file is not on project root or not on lib folder? File must be on $root/lib.');
             return;
         }
-        const relativePath = currentPath.substring(libFolder.length);
+        const relativePath = currentPath.substring(libFolder.length + 1);
         let count = 0;
         for (let currentLine = 0;; currentLine++) {
             const line : vscode.TextLine = editor.document.lineAt(currentLine);