yaml2ts.ts — astro Source File

Source Code

import type { CodeMapping, VirtualCode } from '@volar/language-core';
import type { YAMLError, YAMLMap, YAMLSeq } from 'yaml';
import YAML, { CST, isCollection, isMap, isScalar, isSeq, parseDocument } from 'yaml';

export const VIRTUAL_CODE_ID = 'frontmatter-ts';

export type YAML2TSResult = {
	errors: YAMLError[];
	virtualCode: VirtualCode;
};

const ONLY_NAV_CODE_INFO: CodeMapping['data'] = {
	verification: false,
	completion: false,
	semantic: false,
	navigation: true,
	structure: false,
	format: false,
};

export function yaml2ts(frontmatter: string, collection: string): YAML2TSResult {
	const frontmatterMappings: CodeMapping[] = [];
	const frontmatterContent = parseDocument(frontmatter, {
		keepSourceTokens: true,
		customTags: ['timestamp'], // Handle YAML timestamps
		// Those two options prevent `yaml` from throwing errors when it encounters parsing errors, which is useful for handling incomplete content
		strict: false,
		logLevel: 'silent',
	});

	let fullResult = 'import type { InferEntrySchema } from "astro:content";\n\n(\n';
	let objectContent = frontmatter.trim().length > 0 ? '' : '{}'; // If there's no content, provide an empty object so that there's no syntax error

	YAML.visit(frontmatterContent, {
		Value(key, value) {
			if (isCollection(value)) {
				if (isMap(value)) {
					mapMap(value, key);
				}

				if (isSeq(value)) {
					mapSeq(value);
				}
			}

			// If we didn't hit any of the above, we have a scalar value which in almost all cases is a Pair that's just not fully written yet
			if (isScalar(value)) {
				const itemKey = mapScalarKey(value);
				// We don't care about values, just keys, since we're only interested in the structure
				objectContent += `${itemKey}: null\n`;
			}

			return YAML.visit.REMOVE;
		},
	});

	function mapMap(map: YAMLMap, key?: string | number | null) {
		objectContent += '{\n';

		// Go through all the items in the map
// ... (95 more lines)