diff options
Diffstat (limited to 'packages/astro-prism/index.mjs')
| -rw-r--r-- | packages/astro-prism/index.mjs | 111 |
1 files changed, 58 insertions, 53 deletions
diff --git a/packages/astro-prism/index.mjs b/packages/astro-prism/index.mjs index bb4fc53e4..05f1ef2db 100644 --- a/packages/astro-prism/index.mjs +++ b/packages/astro-prism/index.mjs @@ -1,72 +1,82 @@ - export function addAstro(Prism) { - if(Prism.languages.astro) { + if (Prism.languages.astro) { return; } let scriptLang; - if(Prism.languages.typescript) { + if (Prism.languages.typescript) { scriptLang = 'typescript'; } else { scriptLang = 'javascript'; console.warn('Prism TypeScript language not loaded, Astro scripts will be treated as JavaScript.'); } - let script = Prism.util.clone(Prism.languages[scriptLang]); - + let space = /(?:\s|\/\/.*(?!.)|\/\*(?:[^*]|\*(?!\/))\*\/)/.source; let braces = /(?:\{(?:\{(?:\{[^{}]*\}|[^{}])*\}|[^{}])*\})/.source; let spread = /(?:\{<S>*\.{3}(?:[^{}]|<BRACES>)*\})/.source; - + /** * @param {string} source * @param {string} [flags] */ function re(source, flags) { source = source - .replace(/<S>/g, function () { return space; }) - .replace(/<BRACES>/g, function () { return braces; }) - .replace(/<SPREAD>/g, function () { return spread; }); + .replace(/<S>/g, function () { + return space; + }) + .replace(/<BRACES>/g, function () { + return braces; + }) + .replace(/<SPREAD>/g, function () { + return spread; + }); return RegExp(source, flags); } - + spread = re(spread).source; - - + Prism.languages.astro = Prism.languages.extend('markup', script); - Prism.languages.astro.tag.pattern = re( - /<\/?(?:[\w.:-]+(?:<S>+(?:[\w.:$-]+(?:=(?:"(?:\\[^]|[^\\"])*"|'(?:\\[^]|[^\\'])*'|[^\s{'"/>=]+|<BRACES>))?|<SPREAD>))*<S>*\/?)?>/.source - ); - + Prism.languages.astro.tag.pattern = re(/<\/?(?:[\w.:-]+(?:<S>+(?:[\w.:$-]+(?:=(?:"(?:\\[^]|[^\\"])*"|'(?:\\[^]|[^\\'])*'|[^\s{'"/>=]+|<BRACES>))?|<SPREAD>))*<S>*\/?)?>/.source); + Prism.languages.astro.tag.inside['tag'].pattern = /^<\/?[^\s>\/]*/i; Prism.languages.astro.tag.inside['attr-value'].pattern = /=(?!\{)(?:"(?:\\[^]|[^\\"])*"|'(?:\\[^]|[^\\'])*'|[^\s'">]+)/i; Prism.languages.astro.tag.inside['tag'].inside['class-name'] = /^[A-Z]\w*(?:\.[A-Z]\w*)*$/; Prism.languages.astro.tag.inside['comment'] = script['comment']; - - Prism.languages.insertBefore('inside', 'attr-name', { - 'spread': { - pattern: re(/<SPREAD>/.source), - inside: Prism.languages.astro - } - }, Prism.languages.astro.tag); - - Prism.languages.insertBefore('inside', 'special-attr', { - 'script': { - // Allow for two levels of nesting - pattern: re(/=<BRACES>/.source), - inside: { - 'script-punctuation': { - pattern: /^=(?={)/, - alias: 'punctuation' + + Prism.languages.insertBefore( + 'inside', + 'attr-name', + { + spread: { + pattern: re(/<SPREAD>/.source), + inside: Prism.languages.astro, + }, + }, + Prism.languages.astro.tag + ); + + Prism.languages.insertBefore( + 'inside', + 'special-attr', + { + script: { + // Allow for two levels of nesting + pattern: re(/=<BRACES>/.source), + inside: { + 'script-punctuation': { + pattern: /^=(?={)/, + alias: 'punctuation', + }, + rest: Prism.languages.astro, }, - rest: Prism.languages.astro + alias: `language-${scriptLang}`, }, - 'alias': `language-${scriptLang}` - } - }, Prism.languages.astro.tag); + }, + Prism.languages.astro.tag + ); - // The following will handle plain text inside tags let stringifyToken = function (token) { if (!token) { @@ -80,23 +90,23 @@ export function addAstro(Prism) { } return token.content.map(stringifyToken).join(''); }; - + let walkTokens = function (tokens) { let openedTags = []; for (let i = 0; i < tokens.length; i++) { let token = tokens[i]; // This breaks styles, not sure why - if(token.type === 'style') { + if (token.type === 'style') { return; } let notTagNorBrace = false; - + if (typeof token !== 'string') { if (token.type === 'tag' && token.content[0] && token.content[0].type === 'tag') { // We found a tag, now find its kind - + if (token.content[0].content[0].content === '</') { // Closing tag if (openedTags.length > 0 && openedTags[openedTags.length - 1].tagName === stringifyToken(token.content[0].content[1])) { @@ -110,22 +120,18 @@ export function addAstro(Prism) { // Opening tag openedTags.push({ tagName: stringifyToken(token.content[0].content[1]), - openedBraces: 0 + openedBraces: 0, }); } } } else if (openedTags.length > 0 && token.type === 'punctuation' && token.content === '{') { - // Here we might have entered a Astro context inside a tag openedTags[openedTags.length - 1].openedBraces++; - } else if (openedTags.length > 0 && openedTags[openedTags.length - 1].openedBraces > 0 && token.type === 'punctuation' && token.content === '}') { - // Here we might have left a Astro context inside a tag openedTags[openedTags.length - 1].openedBraces--; - } else { - notTagNorBrace = true + notTagNorBrace = true; } } if (notTagNorBrace || typeof token === 'string') { @@ -133,7 +139,7 @@ export function addAstro(Prism) { // Here we are inside a tag, and not inside a Astro context. // That's plain text: drop any tokens matched. let plainText = stringifyToken(token); - + // And merge text with adjacent text if (i < tokens.length - 1 && (typeof tokens[i + 1] === 'string' || tokens[i + 1].type === 'plain-text')) { plainText += stringifyToken(tokens[i + 1]); @@ -144,22 +150,21 @@ export function addAstro(Prism) { tokens.splice(i - 1, 1); i--; } - + tokens[i] = new Prism.Token('plain-text', plainText, null, plainText); } } - + if (token.content && typeof token.content !== 'string') { walkTokens(token.content); } } }; - + Prism.hooks.add('after-tokenize', function (env) { if (env.language !== 'astro') { return; } walkTokens(env.tokens); }); - -}
\ No newline at end of file +} |