add support specify new languages for Prism

facebook/docusaurus#2250

README.md

@@ -2,6 +2,10 @@
 
 The classic theme for Docusaurus.
 
+## 语法高亮
+
+https://github.com/facebook/docusaurus/pull/2250
+
 ## Installation
 
 Add `docusaurus/theme-classic` to your package:

package.json

@@ -15,7 +15,8 @@
     "infima": "0.2.0-alpha.3",
     "parse-numeric-range": "^0.0.2",
     "prism-react-renderer": "^1.0.2",
-    "react-toggle": "^4.1.1"
+    "react-toggle": "^4.1.1",
+    "webpack": "^4.42.0"
   },
   "peerDependencies": {
     "@docusaurus/core": "^2.0.0",

src/index.js

@@ -6,6 +6,7 @@
  */
 
 const path = require('path');
+const ContextReplacementPlugin = require('webpack/lib/ContextReplacementPlugin');
 
 // Need to be inlined to prevent dark mode FOUC
 // Make sure that the 'storageKey' is the same as the one in `/theme/hooks/useTheme.js`
@@ -45,7 +46,10 @@ module.exports = function(context, options) {
   const {
     siteConfig: {themeConfig},
   } = context;
-  const {disableDarkMode = false} = themeConfig || {};
+
+  const {disableDarkMode = false, prism: {additionalLanguages = []} = {}} =
+  themeConfig || {};
+
   const {customCss} = options || {};
   return {
     name: 'docusaurus-theme-classic',
@@ -55,7 +59,25 @@ module.exports = function(context, options) {
     },
 
     getClientModules() {
-      return ['infima/dist/css/default/default.css', customCss];
+      return ['infima/dist/css/default/default.css',
+      customCss,
+      path.resolve(__dirname, './prism-include-languages'),
+      ];
+    },
+
+    configureWebpack() {
+      const prismLanguages = additionalLanguages
+        .map(lang => `prism-${lang}`)
+        .join('|');
+
+      return {
+        plugins: [
+          new ContextReplacementPlugin(
+            /prismjs[\\/]components$/,
+            new RegExp(`^./(${prismLanguages})$`),
+          ),
+        ],
+      };
     },
 
     injectHtmlTags() {

src/prism-include-languages.js

@@ -0,0 +1,26 @@
+/**
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+import Prism from 'prism-react-renderer/prism';
+import ExecutionEnvironment from '@docusaurus/ExecutionEnvironment';
+import siteConfig from '@generated/docusaurus.config';
+
+(() => {
+  if (ExecutionEnvironment.canUseDOM) {
+    const {
+      themeConfig: {prism: {additionalLanguages = []} = {}},
+    } = siteConfig;
+
+    window.Prism = Prism;
+
+    additionalLanguages.forEach(lang => {
+      require(`prismjs/components/prism-${lang}`); // eslint-disable-line
+    });
+
+    delete window.Prism;
+  }
+})();