I wrote about extending gatsby-remark-prismjs with your own extension.

https://marcelhauri.ch/blog/how-to-colorize-a-tree-output-in-gatsbys-prism-plugin/