51 lines
1.6 KiB
TypeScript
51 lines
1.6 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
|
|
import { TextDocument } from 'vscode';
|
|
import { Node as FlatNode } from 'EmmetFlatNode';
|
|
import parse from '@emmetio/html-matcher';
|
|
import parseStylesheet from '@emmetio/css-parser';
|
|
import { isStyleSheet } from './util';
|
|
|
|
type Pair<K, V> = {
|
|
key: K;
|
|
value: V;
|
|
};
|
|
|
|
// Map(filename, Pair(fileVersion, rootNodeOfParsedContent))
|
|
const _parseCache = new Map<string, Pair<number, FlatNode> | undefined>();
|
|
|
|
export function getRootNode(document: TextDocument, useCache: boolean): FlatNode {
|
|
const key = document.uri.toString();
|
|
const result = _parseCache.get(key);
|
|
const documentVersion = document.version;
|
|
if (useCache && result) {
|
|
if (documentVersion === result.key) {
|
|
return result.value;
|
|
}
|
|
}
|
|
|
|
const parseContent = isStyleSheet(document.languageId) ? parseStylesheet : parse;
|
|
const rootNode = parseContent(document.getText());
|
|
if (useCache) {
|
|
_parseCache.set(key, { key: documentVersion, value: rootNode });
|
|
}
|
|
return rootNode;
|
|
}
|
|
|
|
export function addFileToParseCache(document: TextDocument) {
|
|
const filename = document.uri.toString();
|
|
_parseCache.set(filename, undefined);
|
|
}
|
|
|
|
export function removeFileFromParseCache(document: TextDocument) {
|
|
const filename = document.uri.toString();
|
|
_parseCache.delete(filename);
|
|
}
|
|
|
|
export function clearParseCache() {
|
|
_parseCache.clear();
|
|
}
|