Stylesheet partial parse

extensions/emmet/src/defaultCompletionProvider.ts

@@ -6,7 +6,7 @@
 import * as vscode from 'vscode';
 import { Node } from 'EmmetNode';
 import { isValidLocationForEmmetAbbreviation } from './abbreviationActions';
-import { getEmmetHelper, getNode, getMappingForIncludedLanguages, parseDocument, getEmmetConfiguration, getEmmetMode, isStyleSheet } from './util';
+import { getEmmetHelper, getNode, getMappingForIncludedLanguages, parsePartialStylesheet, getEmmetConfiguration, getEmmetMode, isStyleSheet, parseDocument } from './util';
 
 export class DefaultCompletionItemProvider implements vscode.CompletionItemProvider {
 
@@ -38,7 +38,7 @@ export class DefaultCompletionItemProvider implements vscode.CompletionItemProvi
 
 		// If document can be css parsed, get currentNode
 		if (isStyleSheet(document.languageId)) {
-			const rootNode = parseDocument(document, false);
+			const rootNode = document.lineCount > 1000 ? parsePartialStylesheet(document, position) : parseDocument(document, false);
 			if (!rootNode) {
 				return;
 			}

extensions/emmet/src/util.ts

@@ -128,6 +128,52 @@ export function parseDocument(document: vscode.TextDocument, showError: boolean
 	return undefined;
 }
 
+export function parsePartialStylesheet(document: vscode.TextDocument, position: vscode.Position): Node | undefined {
+
+	let startPosition = new vscode.Position(0, 0);
+	let endPosition = new vscode.Position(document.lineCount - 1, document.lineAt(document.lineCount - 1).text.length);
+	const closeBrace = 125;
+	const openBrace = 123;
+
+	// Go forward until we found a closing or opening brace
+	let stream = new DocumentStreamReader(document, position);
+	stream.eatWhile(char => { return char !== closeBrace && char !== openBrace; });
+	stream.eat(closeBrace);
+	endPosition = stream.pos;
+
+	// Go back until we found an opening brace. If we find a closing one, we first find its opening brace and then we continue.
+	stream.pos = position;
+	let openBracesRemaining = 1;
+	while (openBracesRemaining > 0 && !stream.sof()) {
+		if (position.line - stream.pos.line > 1000) {
+			return parseStylesheet(new DocumentStreamReader(document, startPosition, new vscode.Range(startPosition, endPosition)));
+		}
+		let ch = stream.backUp(1);
+		if (ch === openBrace) {
+			openBracesRemaining--;
+		} else if (ch === closeBrace) {
+			if (document.languageId === 'css') {
+				stream.next();
+				return parseStylesheet(new DocumentStreamReader(document, stream.pos, new vscode.Range(stream.pos, endPosition)));
+			}
+			openBracesRemaining++;
+		}
+	}
+	// We are at an opening brace. We need to include its selector
+	while (stream.pos.character > 0) {
+		let ch = stream.backUp(1);
+		if (ch === closeBrace || ch === openBrace) {
+			stream.next();
+			break;
+		}
+	}
+	startPosition = stream.pos;
+	try {
+		return parseStylesheet(new DocumentStreamReader(document, startPosition, new vscode.Range(startPosition, endPosition)));
+	} catch (e) {
+	}
+}
+
 /**
  * Returns node corresponding to given position in the given root node
  */