From 3a8811e115c17d170b709dca829d8991dfdd4f15 Mon Sep 17 00:00:00 2001
From: Stefan van der Walt <stefan@sun.ac.za>
Date: Sun, 21 Aug 2011 14:14:16 -0700
Subject: [PATCH] Add code highlighting to markdown cells.

---
 IPython/frontend/html/notebook/static/js/textcell.js | 12 ++++++++++++
 .../frontend/html/notebook/templates/notebook.html   |  4 ++++
 2 files changed, 16 insertions(+)

diff --git a/IPython/frontend/html/notebook/static/js/textcell.js b/IPython/frontend/html/notebook/static/js/textcell.js
index 535c226bb..7732c3ea9 100644
--- a/IPython/frontend/html/notebook/static/js/textcell.js
+++ b/IPython/frontend/html/notebook/static/js/textcell.js
@@ -202,6 +202,18 @@ var IPython = (function (IPython) {
             MathJax.Hub.Queue(["Typeset",MathJax.Hub]);
             this.element.find('div.text_cell_input').hide();
             this.element.find("div.text_cell_render").show();
+            var code_snippets = this.element.find("pre > code");
+            code_snippets.replaceWith(function () {
+                var code = $(this).html();
+                /* Substitute br for newlines and &nbsp; for spaces
+                   before highlighting, since prettify doesn't
+                   preserve those on all browsers */
+                code = code.replace(/(\r\n|\n|\r)/gm, "<br/>");
+                code = code.replace(/ /gm, '&nbsp;');
+                code = prettyPrintOne(code);
+
+                return '<code class="prettyprint">' + code + '</code>';
+            });
             this.rendered = true;
         };
     };
diff --git a/IPython/frontend/html/notebook/templates/notebook.html b/IPython/frontend/html/notebook/templates/notebook.html
index a114023a6..97bfe603e 100644
--- a/IPython/frontend/html/notebook/templates/notebook.html
+++ b/IPython/frontend/html/notebook/templates/notebook.html
@@ -24,6 +24,8 @@
     <link rel="stylesheet" href="static/codemirror-2.12/theme/ipython.css">
     <link rel="stylesheet" href="static/codemirror-2.12/theme/default.css">
 
+    <link rel="stylesheet" href="static/prettify/prettify.css"/>
+
     <link rel="stylesheet" href="static/css/boilerplate.css" type="text/css" />
     <link rel="stylesheet" href="static/css/layout.css" type="text/css" />
     <link rel="stylesheet" href="static/css/base.css" type="text/css" />
@@ -214,6 +216,8 @@
 
 <script src="static/pagedown/Markdown.Converter.js" charset="utf-8"></script>
 
+<script src="static/prettify/prettify.js" charset="utf-8"></script>
+
 <script src="static/js/namespace.js" type="text/javascript" charset="utf-8"></script>
 <script src="static/js/utils.js" type="text/javascript" charset="utf-8"></script>
 <script src="static/js/cell.js" type="text/javascript" charset="utf-8"></script>