diff options
Diffstat (limited to 'packages/astro-prism/index.mjs')
| -rw-r--r-- | packages/astro-prism/index.mjs | 165 |
1 files changed, 165 insertions, 0 deletions
diff --git a/packages/astro-prism/index.mjs b/packages/astro-prism/index.mjs new file mode 100644 index 000000000..bb4fc53e4 --- /dev/null +++ b/packages/astro-prism/index.mjs @@ -0,0 +1,165 @@ + +export function addAstro(Prism) { + if(Prism.languages.astro) { + return; + } + + let scriptLang; + 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; }); + 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.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' + }, + rest: Prism.languages.astro + }, + 'alias': `language-${scriptLang}` + } + }, Prism.languages.astro.tag); + + + // The following will handle plain text inside tags + let stringifyToken = function (token) { + if (!token) { + return ''; + } + if (typeof token === 'string') { + return token; + } + if (typeof token.content === 'string') { + return token.content; + } + 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') { + 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])) { + // Pop matching opening tag + openedTags.pop(); + } + } else { + if (token.content[token.content.length - 1].content === '/>') { + // Autoclosed tag, ignore + } else { + // Opening tag + openedTags.push({ + tagName: stringifyToken(token.content[0].content[1]), + 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 + } + } + if (notTagNorBrace || typeof token === 'string') { + if (openedTags.length > 0 && openedTags[openedTags.length - 1].openedBraces === 0) { + // 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]); + tokens.splice(i + 1, 1); + } + if (i > 0 && (typeof tokens[i - 1] === 'string' || tokens[i - 1].type === 'plain-text')) { + plainText = stringifyToken(tokens[i - 1]) + plainText; + 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 |