---
import Prism from 'prismjs';
import { addAstro } from '../astro-prism/index.mjs';
import * as loadLanguages from 'prismjs/components/';

export let lang;
export let code;

const languageMap = new Map([
  ['ts', 'typescript']
]);

if(lang == null) {
  console.warn('Prism.astro: No language provided.');
}

const ensureLoaded = lang => {
  if(!Prism.languages[lang]) {
    loadLanguages([lang]);
  }
};

if(languageMap.has(lang)) {
  ensureLoaded(languageMap.get(lang));
} else if(lang === 'astro') {
  ensureLoaded('typescript');
  addAstro(Prism);
} else {
  ensureLoaded(lang);
}

if(!Prism.languages[lang]) {
  console.warn(`Unable to load the language: ${lang}`);
}

const grammar = Prism.languages[lang];
let html = Prism.highlight(code, grammar, lang);

let className = `language-${lang}`;
---

<pre class={className}><code class={className}>{html}</code></pre>