Update index.html

index.html

@@ -316,34 +316,79 @@ fetch('/tools').then(r => r.json()).then(tools => {
 }).catch(() => {});
 
 // ── AUTO-INIT ──────────────────────────────────────────────────────
+const LS_TOKEN_KEY = 'supercity_session_token';
+const LS_RAW_KEY   = 'supercity_raw';
+
 (function initFromServer() {
   const wrap  = document.getElementById('progressWrap');
   const bar   = document.getElementById('progressBar');
   const label = document.getElementById('progressLabel');
-  wrap.style.display = 'block';
-  bar.style.width = '30%';
-  label.textContent = 'Loading APK…';
-
-  fetch('/init')
-    .then(r => { if (!r.ok) return r.json().then(e => { throw new Error(e.error); }); return r.json(); })
-    .then(data => {
-      bar.style.width = '100%';
-      sessionToken = data.token;
-      originalRaw  = data.characters;
-      parseCharacters(data.characters);
-      wrap.style.display = 'none';
-      document.getElementById('fileInfoBar').style.display = 'flex';
-      document.getElementById('fileInfoBar').innerHTML = `
-        <div class="file-info">📦 <span class="file-info-name">super-city-2.010.64-mod-t-5play.apk</span>
-        &nbsp;·&nbsp; ${characters.length} characters loaded</div>`;
-      document.getElementById('editorLayout').style.display = 'grid';
-      setStep(2);
-      showToast(`✔ Loaded ${characters.length} characters`);
-    })
-    .catch(err => {
-      wrap.style.display = 'none';
-      showToast('Failed to load APK: ' + err.message, true);
-    });
+
+  // Try to restore a previous session from localStorage
+  const savedToken = localStorage.getItem(LS_TOKEN_KEY);
+  const savedRaw   = localStorage.getItem(LS_RAW_KEY);
+
+  if (savedToken && savedRaw) {
+    // Verify the session is still alive on the server
+    label.textContent = 'Resuming session…';
+    wrap.style.display = 'block';
+    bar.style.width = '50%';
+
+    fetch('/session/' + savedToken)
+      .then(r => r.json())
+      .then(data => {
+        if (data.alive) {
+          bar.style.width = '100%';
+          sessionToken = savedToken;
+          originalRaw  = savedRaw;
+          parseCharacters(savedRaw);
+          wrap.style.display = 'none';
+          document.getElementById('fileInfoBar').style.display = 'flex';
+          document.getElementById('fileInfoBar').innerHTML = `
+            <div class="file-info">♻ <span class="file-info-name">Session restored</span>
+            &nbsp;·&nbsp; ${characters.length} characters loaded</div>`;
+          document.getElementById('editorLayout').style.display = 'grid';
+          setStep(2);
+          showToast('✔ Session restored — your edits are safe');
+        } else {
+          localStorage.removeItem(LS_TOKEN_KEY);
+          localStorage.removeItem(LS_RAW_KEY);
+          loadFresh(wrap, bar, label);
+        }
+      })
+      .catch(() => loadFresh(wrap, bar, label));
+  } else {
+    loadFresh(wrap, bar, label);
+  }
+
+  function loadFresh(wrap, bar, label) {
+    wrap.style.display = 'block';
+    bar.style.width = '30%';
+    label.textContent = 'Loading APK…';
+
+    fetch('/init')
+      .then(r => { if (!r.ok) return r.json().then(e => { throw new Error(e.error); }); return r.json(); })
+      .then(data => {
+        bar.style.width = '100%';
+        sessionToken = data.token;
+        originalRaw  = data.characters;
+        localStorage.setItem(LS_TOKEN_KEY, data.token);
+        localStorage.setItem(LS_RAW_KEY, data.characters);
+        parseCharacters(data.characters);
+        wrap.style.display = 'none';
+        document.getElementById('fileInfoBar').style.display = 'flex';
+        document.getElementById('fileInfoBar').innerHTML = `
+          <div class="file-info">📦 <span class="file-info-name">super-city-2.010.64-mod-t-5play.apk</span>
+          &nbsp;·&nbsp; ${characters.length} characters loaded</div>`;
+        document.getElementById('editorLayout').style.display = 'grid';
+        setStep(2);
+        showToast(`✔ Loaded ${characters.length} characters`);
+      })
+      .catch(err => {
+        wrap.style.display = 'none';
+        showToast('Failed to load APK: ' + err.message, true);
+      });
+  }
 })();
 
 // ── PARSE ──────────────────────────────────────────────────────────