FrontmatterHolder Class — astro Architecture

Source Code

packages/language-tools/language-server/src/core/frontmatterHolders.ts lines 93–166

export class FrontmatterHolder implements VirtualCode {
	id = 'frontmatter-holder';
	mappings: CodeMapping[];
	embeddedCodes: VirtualCode[];
	public hasFrontmatter = false;

	constructor(
		public fileName: string,
		public languageId: string,
		public snapshot: ts.IScriptSnapshot,
		public collection: string | undefined,
	) {
		this.mappings = [
			{
				sourceOffsets: [0],
				generatedOffsets: [0],
				lengths: [this.snapshot.getLength()],
				data: {
					verification: true,
					completion: true,
					semantic: true,
					navigation: true,
					structure: true,
					format: true,
				},
			},
		];

		this.embeddedCodes = [];
		this.snapshot = snapshot;

		// If the file is not part of a collection, we don't need to do anything
		if (!this.collection) {
			return;
		}

		const frontmatterContent =
			frontmatterRE
				.exec(this.snapshot.getText(0, this.snapshot.getLength()))?.[0]
				.replaceAll('---', '   ') ?? '';

		this.hasFrontmatter = frontmatterContent.length > 0;

		this.embeddedCodes.push({
			id: `yaml_frontmatter_${this.collection}`,
			languageId: 'yaml',
			snapshot: {
				getText: (start, end) => frontmatterContent.substring(start, end),
				getLength: () => frontmatterContent.length,
				getChangeRange: () => undefined,
			},
			mappings: [
				{
					sourceOffsets: [0],
					generatedOffsets: [0],
					lengths: [frontmatterContent.length],
					data: {
						verification: true,
						completion: true,
						semantic: true,
						navigation: true,
						structure: true,
						format: false,
					},
				},
			],
		});

		if (this.hasFrontmatter) {
			const yaml2tsResult = yaml2ts(frontmatterContent, this.collection);
			this.embeddedCodes.push(yaml2tsResult.virtualCode);
		}
	}
}