Refactor part length logic into its own function

Author: runem

src/parsing/virtual-document/virtual-ast-document.ts

@@ -4,6 +4,11 @@ import { Range } from "../../types/range";
 import { intersects } from "../../util/util";
 import { VirtualDocument } from "./virtual-document";
 
+function getPartLength(part: Node): number {
+	const end = part.parent && tsModule.ts.isTemplateSpan(part.parent) ? part.parent.literal.getStart() : part.getEnd();
+	return end - part.getFullStart();
+}
+
 export class VirtualAstDocument implements VirtualDocument {
 	readonly fileName: string;
 	readonly location: Range;
@@ -23,8 +28,7 @@ export class VirtualAstDocument implements VirtualDocument {
 					str += part.substring(i === 0 ? 0 : 1, part.length - (isLastPart ? 0 : 2));
 					prevPart = part;
 				} else {
-					const end = part.parent && tsModule.ts.isTemplateSpan(part.parent) ? part.parent.literal.getStart() : part.getEnd();
-					const length = end - part.getFullStart() + 3;
+					const length = getPartLength(part) + 3;
 					const substitution = this.substituteExpression(length, part, prevPart, this.parts[i + 1] as string);
 					str += substitution;
 				}
@@ -67,7 +71,7 @@ export class VirtualAstDocument implements VirtualDocument {
 					resultParts.push(substr);
 				}
 			} else {
-				offset += part.getEnd() - part.getStart();
+				offset += getPartLength(part);
 
 				const expressionPartRange: Range = {
 					start: startOffset,