2 files changed, 21 insertions, 2 deletions
diff --git a/pages/login/page.js b/pages/login/page.js
index dcc6f20..573c382 100644
--- a/pages/login/page.js
+++ b/pages/login/page.js
@@ -4,14 +4,23 @@ async function fetchJSON(url, opts) {
 }
 
 export function onload() {
+    if (localStorage.getItem('token')) {
+        window.stores.sdlib.ws.addEventListener('open', () => {
+            window.stores.sdlib.loginToken(localStorage.getItem('token'), localStorage.getItem('username'));
+            pages.goToPage('main')
+        })
+    }
     document.getElementById("loginForm").addEventListener("submit", async function (ev) {
         ev.preventDefault();
         const username = document.getElementById('username').value;
         const password = document.getElementById('password').value;
         document.getElementById('topbar').classList.remove('hidden')
         try {
-            await stores.sdlib.login(username, password)
+            const token = await stores.sdlib.login(username, password)
+            localStorage.setItem('token', token)
+            localStorage.setItem('username', username)
         } catch (error) {
+            console.error(error)
             return document.getElementById('error').innerText = 'An error occured\n' + error
         }
         pages.goToPage("main")
diff --git a/pages/main/page.js b/pages/main/page.js
index f1d40f2..ebdaab2 100644
--- a/pages/main/page.js
+++ b/pages/main/page.js
@@ -5,7 +5,17 @@ import { openPopup } from "../../lib/popups.js";
 
 window.html = html // debug
 
-const md = markdwonits()
+const md = markdwonits({
+    highlight: function (str, lang) {
+        if (lang && hljs.getLanguage(lang)) {
+            try {
+                return hljs.highlight(str, { language: lang }).value;
+            } catch (__) {}
+        }
+    
+        return ''; // use external default escaping
+    }
+})
 
 async function fetchJSON(url, opts) {
     let resp = await fetch(url, opts);