yaml.ts — astro Source File

Source Code

import { Diagnostic, DiagnosticSeverity, MarkupContent, Range } from '@volar/language-server';
import type { LanguageServicePlugin } from '@volar/language-service';
import type { Provide } from 'volar-service-yaml';
import { create as createYAMLService } from 'volar-service-yaml';
import { URI, Utils } from 'vscode-uri';
import type { CollectionConfig } from '../core/frontmatterHolders.js';
import {
	FrontmatterHolder,
	SUPPORTED_FRONTMATTER_EXTENSIONS_KEYS,
} from '../core/frontmatterHolders.js';

type LanguageSettings = Parameters<ReturnType<Provide['yaml/languageService']>['configure']>['0'];

export function getSettings(collectionConfig: CollectionConfig): LanguageSettings {
	const schemas = collectionConfig.configs.flatMap((workspaceCollectionConfig) => {
		return workspaceCollectionConfig.config.collections.flatMap((collection) => {
			return {
				fileMatch: SUPPORTED_FRONTMATTER_EXTENSIONS_KEYS.map(
					(ext) => `volar-embedded-content://yaml_frontmatter_${collection.name}/**/*${ext}`,
				),
				uri: Utils.joinPath(
					workspaceCollectionConfig.folder,
					'.astro/collections',
					`${collection.name}.schema.json`,
				).toString(),
			};
		});
	});

	return {
		completion: true,
		format: false,
		hover: true,
		validate: true,
		customTags: [],
		yamlVersion: '1.2',
		isKubernetes: false,
		parentSkeletonSelectedFirst: false,
		disableDefaultProperties: false,
		schemas: schemas,
	};
}

export const create = (collectionConfig: CollectionConfig): LanguageServicePlugin => {
	const yamlPlugin = createYAMLService({
		getLanguageSettings: () => getSettings(collectionConfig),
	}) as LanguageServicePlugin<Provide>;

	return {
		...yamlPlugin,
		capabilities: {
			...yamlPlugin.capabilities,
			codeLensProvider: undefined,
		},
		create(context) {
			const yamlPluginInstance = yamlPlugin.create(context);

			const languageService = yamlPluginInstance.provide?.['yaml/languageService']();
			if (languageService && context.env.onDidChangeWatchedFiles) {
				context.env.onDidChangeWatchedFiles(async (events) => {
// ... (89 more lines)