Deploy guest upgrade flow

Dockerfile

@@ -10,6 +10,9 @@ ENV ENABLE_FOLLOW_UP_QUESTIONS=false \
     ENABLE_EVALUATION_ARENA_MODELS=false \
     ENABLE_USER_STATUS=false \
     ENABLE_PUBLIC_ACTIVE_USERS_COUNT=false \
+    ENABLE_FORWARD_USER_INFO_HEADERS=true \
+    ENABLE_LOGIN_FORM=true \
+    ENABLE_SIGNUP=true \
     ENABLE_EASTER_EGGS=false
 
 # Montaigne config and entrypoint

config/minimal_ui_patch.js

@@ -1,4 +1,5 @@
 (() => {
+  // ── Prompt suggestions ──────────────────────────────────────────
   const MONTAIGNE_SUGGESTIONS = [
     {
       title: ['What occupies your thoughts', 'most lately?'],
@@ -26,6 +27,7 @@
     }
   ];
 
+  // ── Elements to hide ────────────────────────────────────────────
   const HIDE_TEXT = new Set([
     'Settings',
     'Archived Chats',
@@ -41,53 +43,188 @@
   const HIDE_HREF = new Set(['/playground', '/admin']);
   const HIDE_ARIA = new Set(['Controls', 'Add Model', 'Remove Model', 'Voice Input', 'Voice mode']);
 
+  // ── Guest upgrade constants ─────────────────────────────────────
+  const GUEST_TOKEN_KEY = 'montaigne_guest_upgrade_token';
+  const CHAT_COMPLETIONS_PATH = '/api/chat/completions';
+  const GUEST_SIGNUP_NOTICE_ID = 'montaigne-guest-signup-notice';
+
   function normalize(text) {
     return (text || '').replace(/\s+/g, ' ').trim();
   }
 
-  function installConfigFetchPatch() {
-    if (window.__montaigneConfigFetchPatched) {
+  // ── Guest upgrade: CTA banner ───────────────────────────────────
+  function showGuestBanner() {
+    if (document.getElementById('montaigne-guest-banner')) return;
+
+    const banner = document.createElement('div');
+    banner.id = 'montaigne-guest-banner';
+    banner.style.cssText =
+      'position:fixed;bottom:0;left:0;right:0;z-index:99999;' +
+      'background:linear-gradient(135deg,#3d2b1f,#5c3d2e);color:#f5e6d3;' +
+      'padding:14px 20px;display:flex;align-items:center;justify-content:center;' +
+      'gap:16px;font-family:Inter,system-ui,sans-serif;font-size:14px;' +
+      'box-shadow:0 -2px 12px rgba(0,0,0,.3);';
+
+    const msg = document.createElement('span');
+    msg.textContent =
+      'You\u2019ve used your 5 free messages. Create a free account to continue chatting with Montaigne.';
+
+    const btn = document.createElement('a');
+    btn.textContent = 'Create free account';
+    btn.style.cssText =
+      'background:#c9a96e;color:#2b1d0e;padding:8px 18px;border-radius:6px;' +
+      'text-decoration:none;font-weight:600;white-space:nowrap;cursor:pointer;' +
+      'transition:background .15s;';
+    btn.onmouseenter = () => (btn.style.background = '#d4b87a');
+    btn.onmouseleave = () => (btn.style.background = '#c9a96e');
+
+    const currentPath = window.location.pathname + window.location.search;
+    const authUrl = '/auth?form=true&redirect=' + encodeURIComponent(currentPath) + '&guestUpgrade=1';
+    btn.href = authUrl;
+
+    const close = document.createElement('button');
+    close.textContent = '\u00d7';
+    close.style.cssText =
+      'background:none;border:none;color:#f5e6d3;font-size:20px;cursor:pointer;' +
+      'padding:0 4px;line-height:1;';
+    close.onclick = () => banner.remove();
+
+    banner.appendChild(msg);
+    banner.appendChild(btn);
+    banner.appendChild(close);
+    document.body.appendChild(banner);
+  }
+
+  function applyGuestUpgradeAuthPrompt() {
+    const url = new URL(window.location.href);
+    if (url.pathname !== '/auth' || url.searchParams.get('guestUpgrade') !== '1') {
       return;
     }
 
-    window.__montaigneConfigFetchPatched = true;
+    const authContainer = document.getElementById('auth-container');
+    if (!authContainer) {
+      return;
+    }
+
+    if (!document.getElementById(GUEST_SIGNUP_NOTICE_ID)) {
+      const notice = document.createElement('div');
+      notice.id = GUEST_SIGNUP_NOTICE_ID;
+      notice.textContent = 'Create an account to keep this conversation and continue where you left off.';
+      notice.style.cssText =
+        'margin:18px auto 14px;max-width:28rem;padding:12px 14px;border-radius:10px;' +
+        'background:rgba(201,169,110,.12);border:1px solid rgba(201,169,110,.35);' +
+        'color:inherit;font-size:14px;line-height:1.45;';
+      const target = authContainer.querySelector('.w-full.px-10') || authContainer.firstElementChild;
+      if (target) {
+        target.insertBefore(notice, target.firstChild || null);
+      }
+    }
+
+    if (document.documentElement.dataset.montaigneGuestSignupActivated === '1') {
+      return;
+    }
+
+    const toggleButtons = Array.from(authContainer.querySelectorAll('button[type="button"]'));
+    const signupToggle = toggleButtons.find((button) => normalize(button.textContent) === 'Sign up');
+    if (signupToggle) {
+      document.documentElement.dataset.montaigneGuestSignupActivated = '1';
+      signupToggle.click();
+    }
+  }
+
+  // ── Unified fetch patch (config + guest upgrade) ───────────────
+  function installFetchPatch() {
+    if (window.__montaigneFetchPatched) return;
+    window.__montaigneFetchPatched = true;
 
     const originalFetch = window.fetch.bind(window);
+
     window.fetch = async (...args) => {
+      // ── Outgoing request: inject guest upgrade token ──────────
+      try {
+        const input = args[0];
+        const rawUrl = input instanceof Request ? input.url : String(input);
+        const url = new URL(rawUrl, window.location.origin);
+
+        if (url.pathname === CHAT_COMPLETIONS_PATH) {
+          const storedToken = localStorage.getItem(GUEST_TOKEN_KEY);
+          if (storedToken) {
+            let body = args[1] && args[1].body;
+            if (typeof body === 'string') {
+              try {
+                const parsed = JSON.parse(body);
+                if (!parsed.variables) parsed.variables = {};
+                parsed.variables.guest_upgrade_token = storedToken;
+                args[1] = { ...args[1], body: JSON.stringify(parsed) };
+              } catch (_) {
+                // Body not JSON — skip injection
+              }
+            }
+          }
+        }
+      } catch (_) {
+        // URL parsing failed — proceed with original args
+      }
+
+      // ── Perform request ───────────────────────────────────────
       const response = await originalFetch(...args);
 
+      // ── Incoming response handling ────────────────────────────
       try {
         const input = args[0];
         const rawUrl = input instanceof Request ? input.url : String(input);
         const url = new URL(rawUrl, window.location.origin);
 
-        if (url.pathname !== '/api/config') {
-          return response;
+        // Patch 1: inject prompt suggestions into /api/config
+        if (url.pathname === '/api/config') {
+          const payload = await response.clone().json();
+          payload.default_prompt_suggestions = MONTAIGNE_SUGGESTIONS;
+          const headers = new Headers(response.headers);
+          headers.set('content-type', 'application/json');
+          headers.delete('content-length');
+          return new Response(JSON.stringify(payload), {
+            status: response.status,
+            statusText: response.statusText,
+            headers
+          });
         }
 
-        const payload = await response.clone().json();
-        payload.default_prompt_suggestions = MONTAIGNE_SUGGESTIONS;
+        // Patch 2: intercept 403 guest_limit_reached on chat completions
+        if (url.pathname === CHAT_COMPLETIONS_PATH && response.status === 403) {
+          const clone = response.clone();
+          try {
+            const body = await clone.json();
+            const code = typeof body === 'object' ? body.code : null;
+            const token = typeof body === 'object' ? body.guest_upgrade_token : null;
 
-        const headers = new Headers(response.headers);
-        headers.set('content-type', 'application/json');
-        headers.delete('content-length');
+            if (code === 'guest_limit_reached' && token) {
+              localStorage.setItem(GUEST_TOKEN_KEY, token);
+              showGuestBanner();
+            }
+          } catch (_) {
+            // Not JSON — fall through
+          }
 
-        return new Response(JSON.stringify(payload), {
-          status: response.status,
-          statusText: response.statusText,
-          headers
-        });
-      } catch (_error) {
-        return response;
+          // Patch 3: on successful chat response, clear consumed token
+        } else if (
+          url.pathname === CHAT_COMPLETIONS_PATH &&
+          response.ok
+        ) {
+          localStorage.removeItem(GUEST_TOKEN_KEY);
+          const banner = document.getElementById('montaigne-guest-banner');
+          if (banner) banner.remove();
+        }
+      } catch (_) {
+        // Swallow — never break the page
       }
+
+      return response;
     };
   }
 
+  // ── Minimal UI: hide power-user elements ───────────────────────
   function hide(el) {
-    if (!el || el.dataset.montaigneHidden === '1') {
-      return;
-    }
-
+    if (!el || el.dataset.montaigneHidden === '1') return;
     el.dataset.montaigneHidden = '1';
     el.style.display = 'none';
   }
@@ -98,9 +235,7 @@
 
   function hideModelSelector(trigger) {
     hide(trigger);
-
     let node = trigger;
-
     for (let i = 0; i < 6 && node; i += 1) {
       const className = typeof node.className === 'string' ? node.className : '';
       if (className.includes('overflow-hidden')) {
@@ -109,11 +244,12 @@
       }
       node = node.parentElement;
     }
-
     hide(trigger);
   }
 
   function applyMinimalUi() {
+    applyGuestUpgradeAuthPrompt();
+
     document
       .querySelectorAll('[id^="model-selector-"][id$="-button"]')
       .forEach(hideModelSelector);
@@ -125,9 +261,7 @@
       const text = normalize(el.textContent);
       if (text.includes("What's New") || text.includes('Release Notes')) {
         const closeButton = el.querySelector('button[aria-label="Close"]');
-        if (closeButton) {
-          closeButton.click();
-        }
+        if (closeButton) closeButton.click();
         hide(el);
       }
     });
@@ -136,7 +270,6 @@
       const text = normalize(el.textContent);
       const aria = normalize(el.getAttribute('aria-label'));
       const href = el.getAttribute('href');
-
       if (HIDE_TEXT.has(text) || HIDE_ARIA.has(aria) || HIDE_HREF.has(href)) {
         hideMenuItem(el);
       }
@@ -145,7 +278,8 @@
     document.querySelectorAll('#temporary-chat-button').forEach(hide);
   }
 
-  installConfigFetchPatch();
+  // ── Boot ───────────────────────────────────────────────────────
+  installFetchPatch();
 
   const run = () => window.requestAnimationFrame(applyMinimalUi);