🐛 Site > Viewer > Coloration > Fix format

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

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>