summaryrefslogtreecommitdiff
path: root/packages/astro-prism/index.mjs
diff options
context:
space:
mode:
authorGravatar Nate Moore <natemoo-re@users.noreply.github.com> 2021-04-30 16:33:35 -0500
committerGravatar GitHub <noreply@github.com> 2021-04-30 16:33:35 -0500
commit4df1347156cf2632ea2f3475d3a5f8f08d197cc3 (patch)
tree9d50de89dfe62827c32a8a4046120af4ab61dc0c /packages/astro-prism/index.mjs
parent1d498facc8f78a3ffbfecd05cc6ecd45e8a4a1ae (diff)
downloadastro-4df1347156cf2632ea2f3475d3a5f8f08d197cc3.tar.gz
astro-4df1347156cf2632ea2f3475d3a5f8f08d197cc3.tar.zst
astro-4df1347156cf2632ea2f3475d3a5f8f08d197cc3.zip
Migrate to `yarn` monorepo (#157)
* chore: use monorepo * chore: scaffold astro-scripts * chore: move tests inside packages/astro * chore: refactor tests, add scripts * chore: move parser to own module * chore: move runtime to packages/astro * fix: move parser to own package * test: fix prettier-plugin-astro tests * fix: tests * chore: update package-lock * chore: add changesets * fix: cleanup examples * fix: starter example * chore: update changeset config * chore: update changeset config * chore: setup changeset release workflow * chore: bump lockfiles * chore: prism => astro-prism * fix: tsc --emitDeclarationOnly * chore: final cleanup, switch to yarn * chore: add lerna * chore: update workflows to yarn * chore: update workflows * chore: remove lint workflow * chore: add astro-dev script * chore: add symlinked README
Diffstat (limited to 'packages/astro-prism/index.mjs')
-rw-r--r--packages/astro-prism/index.mjs165
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