parseJS.ts — astro Source File

Source Code

import type { TSXExtractedScript } from '@astrojs/compiler/types';
import type { CodeInformation, VirtualCode } from '@volar/language-core';
import type { Segment } from 'muggle-string';
import { toString } from 'muggle-string';
import { buildMappings } from '../buildMappings';

export function extractScriptTags(scripts: TSXExtractedScript[]): VirtualCode[] {
	const embeddedJSCodes: VirtualCode[] = [];

	const moduleScripts = scripts
		.filter((script) => script.type === 'module' || script.type === 'processed-module')
		.map(moduleScriptToVirtualCode) satisfies VirtualCode[];

	const inlineScripts = scripts
		.filter(
			(script) =>
				// TODO: Change this at some point so that unknown scripts are not included
				// We can't guarantee that they are JavaScript, so we shouldn't treat them as such, even if it might work in some cases
				// Perhaps we should make it so that the user has to specify the language of the script if it's not a known type (ex: lang="js"), not sure.
				script.type === 'event-attribute' || script.type === 'inline' || script.type === 'unknown',
		)
		.sort((a, b) => a.position.start - b.position.start);

	embeddedJSCodes.push(...moduleScripts);
	const mergedJSContext = mergeJSContexts(inlineScripts);
	if (mergedJSContext) {
		embeddedJSCodes.push(mergedJSContext);
	}

	const JSONScripts = scripts
		.filter((script) => script.type === 'json')
		.map(jsonScriptToVirtualCode) satisfies VirtualCode[];
	embeddedJSCodes.push(...JSONScripts);

	return embeddedJSCodes;
}

function moduleScriptToVirtualCode(script: TSXExtractedScript, index: number): VirtualCode {
	let extension = 'mts';
	let languageId = 'typescript';
	if (script.type === 'module') {
		extension = 'mjs';
		languageId = 'javascript';
	}

	return {
		id: `${index}.${extension}`,
		languageId,
		snapshot: {
			getText: (start, end) => script.content.substring(start, end),
			getLength: () => script.content.length,
			getChangeRange: () => undefined,
		},
		mappings: [
			{
				sourceOffsets: [script.position.start],
				generatedOffsets: [0],
				lengths: [script.content.length],
				data: {
					verification: true,
// ... (86 more lines)