From 41247fcfa0733204fdfea68eb2721cf85510ffca Mon Sep 17 00:00:00 2001
From: Anthony Fu <anthonyfu117@hotmail.com>
Date: Mon, 30 May 2022 15:09:54 +0800
Subject: [PATCH] feat: support dark/light color themes (#682)

Co-authored-by: Kia Ishii <kia.king.08@gmail.com>
---
 docs/config/app-configs.md                    |  2 +-
 docs/guide/using-vue.md                       |  6 ++---
 src/client/theme-default/index.ts             |  1 +
 .../styles/components/vp-code.css             |  7 +++++
 .../styles/components/vp-doc.css              |  6 ++---
 src/client/theme-default/styles/vars.css      | 13 +++++++---
 src/node/markdown/markdown.ts                 |  4 ++-
 src/node/markdown/plugins/highlight.ts        | 26 ++++++++++++++-----
 8 files changed, 48 insertions(+), 17 deletions(-)
 create mode 100644 src/client/theme-default/styles/components/vp-code.css

diff --git a/docs/config/app-configs.md b/docs/config/app-configs.md
index 47adbec8..ee355c88 100644
--- a/docs/config/app-configs.md
+++ b/docs/config/app-configs.md
@@ -102,7 +102,7 @@ Below shows the the full option you may define within this object.
 interface MarkdownOptions extends MarkdownIt.Options {
   // Syntax highlight theme for Shiki.
   // See: https://github.com/shikijs/shiki/blob/main/docs/themes.md#all-themes
-  theme?: Shiki.Theme
+  theme?: Shiki.Theme | { light: Shiki.Theme, dark: Shiki.Theme }
 
   // Enable line numbers in code block.
   lineNumbers?: boolean
diff --git a/docs/guide/using-vue.md b/docs/guide/using-vue.md
index 8ed6b0aa..998d3aa8 100644
--- a/docs/guide/using-vue.md
+++ b/docs/guide/using-vue.md
@@ -26,8 +26,8 @@ Directives also work:
 
 **Input**
 
-```md
-<span v-for="i in 3">{{ i }} </span>
+```html
+<span v-for="i in 3">{{ i }}</span>
 ```
 
 **Output**
@@ -40,7 +40,7 @@ You can use the [`useData` helper](./api#usedata) in a `<script>` block and expo
 
 **Input**
 
-```md
+```html
 <script setup>
 import { useData } from 'vitepress'
 
diff --git a/src/client/theme-default/index.ts b/src/client/theme-default/index.ts
index 66f2708c..ba3dc291 100644
--- a/src/client/theme-default/index.ts
+++ b/src/client/theme-default/index.ts
@@ -3,6 +3,7 @@ import './styles/vars.css'
 import './styles/base.css'
 import './styles/utils.css'
 import './styles/components/custom-block.css'
+import './styles/components/vp-code.css'
 import './styles/components/vp-doc.css'
 import './styles/components/vp-sponsor.css'
 
diff --git a/src/client/theme-default/styles/components/vp-code.css b/src/client/theme-default/styles/components/vp-code.css
new file mode 100644
index 00000000..ee74e5b2
--- /dev/null
+++ b/src/client/theme-default/styles/components/vp-code.css
@@ -0,0 +1,7 @@
+.dark .vp-code-light {
+  display: none;
+}
+
+html:not(.dark) .vp-code-dark {
+  display: none;
+}
diff --git a/src/client/theme-default/styles/components/vp-doc.css b/src/client/theme-default/styles/components/vp-doc.css
index 99ce571d..212141aa 100644
--- a/src/client/theme-default/styles/components/vp-doc.css
+++ b/src/client/theme-default/styles/components/vp-doc.css
@@ -376,7 +376,7 @@
   background-color: var(--vp-code-block-bg);
   opacity: 0;
   cursor: pointer;
-  background-image: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' fill='none' height='20' width='20' stroke='rgba(235,235,235,0.38)' stroke-width='2' class='h-6 w-6' viewBox='0 0 24 24'%3E%3Cpath stroke-linecap='round' stroke-linejoin='round' d='M9 5H7a2 2 0 0 0-2 2v12a2 2 0 0 0 2 2h10a2 2 0 0 0 2-2V7a2 2 0 0 0-2-2h-2M9 5a2 2 0 0 0 2 2h2a2 2 0 0 0 2-2M9 5a2 2 0 0 1 2-2h2a2 2 0 0 1 2 2'/%3E%3C/svg%3E");
+  background-image: var(--vp-icon-copy);
   background-position: 50%;
   background-size: 20px;
   background-repeat: no-repeat;
@@ -395,7 +395,7 @@
 .vp-doc [class*='language-'] > span.copy:hover.copied {
   border-radius: 0 4px 4px 0;
   background-color: var(--vp-code-copy-code-hover-bg);
-  background-image: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' fill='none' height='20' width='20' stroke='rgba(235,235,235,0.5)' stroke-width='2' class='h-6 w-6' viewBox='0 0 24 24'%3E%3Cpath stroke-linecap='round' stroke-linejoin='round' d='M9 5H7a2 2 0 0 0-2 2v12a2 2 0 0 0 2 2h10a2 2 0 0 0 2-2V7a2 2 0 0 0-2-2h-2M9 5a2 2 0 0 0 2 2h2a2 2 0 0 0 2-2M9 5a2 2 0 0 1 2-2h2a2 2 0 0 1 2 2m-6 9 2 2 4-4'/%3E%3C/svg%3E");
+  background-image: var(--vp-icon-copied);
 }
 
 .vp-doc [class*='language-'] > span.copy.copied::before,
@@ -410,7 +410,7 @@
   text-align: center;
   font-size: 12px;
   font-weight: 500;
-  color: var(--vp-c-text-dark-2);
+  color: var(--vp-code-copy-code-active-text);
   background-color: var(--vp-code-copy-code-hover-bg);
   white-space: nowrap;
   content: "Copied";
diff --git a/src/client/theme-default/styles/vars.css b/src/client/theme-default/styles/vars.css
index 00939405..b85d13b4 100644
--- a/src/client/theme-default/styles/vars.css
+++ b/src/client/theme-default/styles/vars.css
@@ -175,6 +175,15 @@
   --vp-z-index-footer: 50;
 }
 
+/**
+ * Icons
+ * -------------------------------------------------------------------------- */
+
+ :root {
+  --vp-icon-copy: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' fill='none' height='20' width='20' stroke='rgba(128,128,128,1)' stroke-width='2' class='h-6 w-6' viewBox='0 0 24 24'%3E%3Cpath stroke-linecap='round' stroke-linejoin='round' d='M9 5H7a2 2 0 0 0-2 2v12a2 2 0 0 0 2 2h10a2 2 0 0 0 2-2V7a2 2 0 0 0-2-2h-2M9 5a2 2 0 0 0 2 2h2a2 2 0 0 0 2-2M9 5a2 2 0 0 1 2-2h2a2 2 0 0 1 2 2'/%3E%3C/svg%3E");
+  --vp-icon-copied: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' fill='none' height='20' width='20' stroke='rgba(128,128,128,1)' stroke-width='2' class='h-6 w-6' viewBox='0 0 24 24'%3E%3Cpath stroke-linecap='round' stroke-linejoin='round' d='M9 5H7a2 2 0 0 0-2 2v12a2 2 0 0 0 2 2h10a2 2 0 0 0 2-2V7a2 2 0 0 0-2-2h-2M9 5a2 2 0 0 0 2 2h2a2 2 0 0 0 2-2M9 5a2 2 0 0 1 2-2h2a2 2 0 0 1 2 2m-6 9 2 2 4-4'/%3E%3C/svg%3E");
+}
+
 /**
  * Layouts
  * -------------------------------------------------------------------------- */
@@ -195,16 +204,14 @@
   --vp-code-block-bg: #292d3e;
 
   --vp-code-line-highlight-color: rgba(0, 0, 0, 0.5);
-
   --vp-code-line-number-color: var(--vp-c-text-dark-3);
 
   --vp-code-copy-code-hover-bg: rgba(255, 255, 255, 0.05);
+  --vp-code-copy-code-active-text: var(--vp-c-text-dark-2);
 }
 
 .dark {
   --vp-code-block-bg: var(--vp-c-bg-alt);
-
-  --vp-code-line-number-color: var(--vp-c-text-dark-3);
 }
 
 /**
diff --git a/src/node/markdown/markdown.ts b/src/node/markdown/markdown.ts
index 2a2d3287..1ed38025 100644
--- a/src/node/markdown/markdown.ts
+++ b/src/node/markdown/markdown.ts
@@ -19,6 +19,8 @@ import attrs from 'markdown-it-attrs'
 import emoji from 'markdown-it-emoji'
 import toc from 'markdown-it-toc-done-right'
 
+export type ThemeOptions = Theme | { light: Theme; dark: Theme }
+
 export interface MarkdownOptions extends MarkdownIt.Options {
   lineNumbers?: boolean
   config?: (md: MarkdownIt) => void
@@ -31,7 +33,7 @@ export interface MarkdownOptions extends MarkdownIt.Options {
     allowedAttributes?: string[]
     disable?: boolean
   }
-  theme?: Theme
+  theme?: ThemeOptions
   // https://github.com/nagaozen/markdown-it-toc-done-right
   toc?: any
   externalLinks?: Record<string, string>
diff --git a/src/node/markdown/plugins/highlight.ts b/src/node/markdown/plugins/highlight.ts
index 291267c3..eb27b559 100644
--- a/src/node/markdown/plugins/highlight.ts
+++ b/src/node/markdown/plugins/highlight.ts
@@ -1,14 +1,28 @@
-import escapeHtml from 'escape-html'
 import { getHighlighter } from 'shiki'
+import type { ThemeOptions } from '../markdown'
 
-export const highlight = async (theme = 'material-palenight') => {
-  const highlighter = await getHighlighter({ theme })
+export async function highlight(theme: ThemeOptions = 'material-palenight') {
+  const themes = typeof theme === 'string' ? [theme] : [theme.dark, theme.light]
+  const highlighter = await getHighlighter({ themes })
+  const preRE = /^<pre.*?>/
 
   return (str: string, lang: string) => {
-    if (!lang || lang === 'text') {
-      return `<pre v-pre><code>${escapeHtml(str)}</code></pre>`
+    lang = lang || 'text'
+
+    if (typeof theme === 'string') {
+      return highlighter
+        .codeToHtml(str, { lang, theme })
+        .replace(preRE, '<pre v-pre>')
     }
 
-    return highlighter.codeToHtml(str, lang).replace(/^<pre.*?>/, '<pre v-pre>')
+    const dark = highlighter
+      .codeToHtml(str, { lang, theme: theme.dark })
+      .replace(preRE, '<pre v-pre class="vp-code-dark">')
+
+    const light = highlighter
+      .codeToHtml(str, { lang, theme: theme.light })
+      .replace(preRE, '<pre v-pre class="vp-code-light">')
+
+    return dark + light
   }
 }