From 2530f6ba5a875ec0a88e42126ab84f1be21a0289 Mon Sep 17 00:00:00 2001
From: brunosimon <simon.bruno.77@gmail.com>
Date: Tue, 29 Aug 2017 19:11:08 +0100
Subject: [PATCH] :bug: Site > Viewer > Coloration > Fix format

---
 site/src/components/viewer/script.js     | 5 +++++
 site/src/components/viewer/template.html | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/site/src/components/viewer/script.js b/site/src/components/viewer/script.js
index ad9e6f9..948adc1 100644
--- a/site/src/components/viewer/script.js
+++ b/site/src/components/viewer/script.js
@@ -13,6 +13,11 @@ export default
 
     computed:
     {
+        extension()
+        {
+            return this.$store.state.files.current.data.extension
+        },
+
         version()
         {
             return this.$store.state.files.currentVersion
diff --git a/site/src/components/viewer/template.html b/site/src/components/viewer/template.html
index 4863a04..e2c5ce9 100644
--- a/site/src/components/viewer/template.html
+++ b/site/src/components/viewer/template.html
@@ -1,5 +1,5 @@
 <div class="viewer" v-if="version">
-    <pre v-highlightjs="version.content" contenteditable @keydown="onCodeKeyDown($event)"><code></code></pre>
+    <pre v-highlightjs="version.content" contenteditable @keydown="onCodeKeyDown($event)"><code :class="[ extension ]"></code></pre>
     <div class="lines">
         <div class="line" v-for="i in linesCount">{{ i }}</div>
     </div>