Reemus Icon

Code highlighting in React using Prism.js

👍🔥❤️😂😢😕
views
comments

Prism.js is a fantastic code highlighting library. Probably the best I’ve seen and used. However, it has one serious problem. The documentation while simple feels very confusing at the same time. I can’t help but feel a bit lost every time I try to use it with React and enable a plugin, add a theme or support more languages.

Since I just went through the process of implementing Prism.js in this website, I decided to write a simple tutorial on how to use it with React. All the code examples you are about to see use Prism.js for code highlighting.

This tutorial assumes you are using standard React bundlers such as webpack or parcel and use Babel to transform your code.

Install and configure Prism.js

First thing we need to do is install the library with NPM or yarn. You can also load the library from a CDN like jsDelivr.

npm install --save prismjs
# OR
yarn add prismjs
shell-icon

The next thing we need to do is add a babel plugin responsible for loading the CSS and language support for Prism.js. You can load them manually from the Prism.js package but the issue is that package bundlers will load support for all languages. This can be detrimental to your final bundle size if you are only using a small subset of the languages provided.

To install this plugin, we simply run the following in our shell

npm install --save-dev babel-plugin-prismjs
# OR
yarn add --dev babel-plugin-prismjs
shell-icon

Next we need to add the plugin configuration to our .babelrc file. If you do not already have a .babelrc file, create one in the root folder of your source code. Then add the following JSON to the .babelrc file.

{
  "plugins": [
    [
      "prismjs",
      {
        "languages": ["javascript", "css", "html"],
        "plugins": ["line-numbers", "show-language"],
        "theme": "okaidia",
        "css": true
      }
    ]
  ]
}
json-icon

As you may have noticed, you can specify which theme and plugins to use along with what languages to support. The loader will automatically add any CSS required by the plugins as well. If setting CSS to true, make sure your bundler is able to handle CSS imports.

The simplest way to use Prism.js in React

This way is great if your code snippets to be highlighted don't change. However, if they do change, read the component method below instead.

By default, Prism.js is said to automatically highlight any code blocks on your page. However, I’ve found with React, code blocks are not automatically highlighted. You will need to use the Prism.js API to highlight all the code on your page.

Doing this is fairly straightforward, just add your code embed and call the Prism.js highlight all API.

import React from "react";
import ReactDOM from "react-dom";
import Prism from "prismjs";
 
const code = `
const foo = 'foo';
const bar = 'bar';
console.log(foo + bar);
`.trim();
 
class Page extends React.Component {
  componentDidMount() {
    // You can call the Prism.js API here
    // Use setTimeout to push onto callback queue so it runs after the DOM is updated
    setTimeout(() => Prism.highlightAll(), 0);
  }
 
  render() {
    return (
      <pre className="line-numbers">
        <code className="language-js">{code}</code>
      </pre>
    );
  }
}
 
ReactDOM.render(<Page />, document.getElementById("root"));
 
// Or call the Prism.js API here
setTimeout(() => Prism.highlightAll(), 0);
jsx-icon

Notice how I wrote the code snippet to be embedded. The string is called with the function trim to remove the whitespace around the code. This is important to be able to write our code snippets neatly.

Creating a Prism.js component

This is a more robust and contained way to highlight code using Prism.js. It will also update if the code snippet or settings have changed. The component itself is relatively simple and looks like this.

import React from "react";
import Prism from "prismjs";
 
export class PrismCode extends React.Component {
  constructor(props) {
    super(props);
    this.ref = React.createRef();
  }
 
  componentDidMount() {
    this.highlight();
  }
 
  componentDidUpdate() {
    this.highlight();
  }
 
  highlight = () => {
    if (this.ref && this.ref.current) {
      Prism.highlightElement(this.ref.current);
    }
  };
 
  render() {
    const {code, plugins, language} = this.props;
    return (
      <pre className={!plugins ? "" : plugins.join(" ")}>
        <code ref={this.ref} className={`language-${language}`}>
          {code.trim()}
        </code>
      </pre>
    );
  }
}
jsx-icon

To use the component, we can simply do

import React from "react";
import {PrismCode} from "./component";
 
const code = `
const foo = 'foo';
const bar = 'bar';
console.log(foo + bar);
`;
 
const Example = () => (
  <PrismCode code={code} language="js" plugins={["line-numbers"]} />
);
jsx-icon

Advanced usage of Prism.js in React

There have been instances where I have had issues using Prism.js inside tabs or components that destroy and re-create parts of the DOM when their state changes.

The most common issue I've noticed is React complaining that it was not able to remove certain DOM nodes as the Prism.js library modified the DOM causing React to lose track.

If you are running into these issues, you will need to make use of the Prism.tokenise API. This function is what Prism.js uses under the hood to construct the HTML for the highlighted code section. You can find more info on this function at the bottom of this page.

Using this functionality, you can construct the highlighted code markup within a React component. This way, React can keep track of the DOM correctly and won't hit any errors.

Thankfully, Formidable Labs created prism-react-renderer which is a React component to help you do exactly this. Save yourself the trouble of re-implementing the Prism.tokenise API and use this component.

There are a few potential issues when using this plugin that should be noted:

  • Cannot easily use existing Prism.js plugins
  • Cannot easily use existing Prism.js theme
  • Doesn't support all Prism.js languages, see supported languages

If these issues are a problem for you, then it's best to implement your own component.


That about sums it up for this tutorial. If you think there is anything important that can be added to this, let me know in the comments.

👍🔥❤️😂😢😕

Comments

...

Your name will be displayed publicly

Loading comments...