import DefaultTheme from 'vitepress/theme';
import type { Theme } from 'vitepress';

import MermaidDiagram from './components/MermaidDiagram.vue';
import NamedMermaidDiagram from './components/NamedMermaidDiagram.vue';

const theme: Theme = {
  ...DefaultTheme,
  enhanceApp({ app }) {
    DefaultTheme.enhanceApp?.({ app });
    app.component('MermaidDiagram', MermaidDiagram);
    app.component('NamedMermaidDiagram', NamedMermaidDiagram);
  },
};

export default theme;