Fix Add Import location discovery

Fixes #160.
microsoft · Dec 16, 2015 · c14dfb2 · c14dfb2
1 parent a7ca66a
commit c14dfb2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/util.ts b/src/util.ts
@@ -32,9 +32,11 @@ export function parseFilePrelude(text: string): Prelude {
 			ret.imports.push({kind: "single", start: i, end: i});
 		}
 		if (line.match(/^(\s)*\)/)) {
-			ret.imports[ret.imports.length - 1].end = i;
+			if(ret.imports[ret.imports.length - 1].end == -1) {
+				ret.imports[ret.imports.length - 1].end = i;
+			}
 		}
-		if (line.match(/^(\s)(func|const|type|var)/)) {
+		if (line.match(/^(\s)*(func|const|type|var)/)) {
 			break;
 		}
 	}