@ifc-lite/viewer 1.17.3 → 1.17.4

package/src/components/viewer/CesiumSettingsDialog.tsx

@@ -0,0 +1,100 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+/**
+ * Settings dialog for configuring Cesium 3D overlay options:
+ * - Data source selection (OSM Buildings, Bing Aerial, Google Photorealistic)
+ * - Terrain toggle
+ * - Advanced: custom Cesium ion token override (optional — default is bundled)
+ */
+
+import { useState } from 'react';
+import {
+  Dialog,
+  DialogContent,
+  DialogHeader,
+  DialogTitle,
+  DialogTrigger,
+} from '@/components/ui/dialog';
+import { Button } from '@/components/ui/button';
+import { Label } from '@/components/ui/label';
+import { useViewerStore } from '@/store';
+import type { CesiumDataSource } from '@/store/slices/cesiumSlice';
+import { Settings2 } from 'lucide-react';
+
+const DATA_SOURCE_OPTIONS: { value: CesiumDataSource; label: string; description: string }[] = [
+  { value: 'osm-buildings', label: 'OSM Buildings', description: 'OpenStreetMap 3D buildings worldwide' },
+  { value: 'bing-aerial', label: 'Bing Aerial', description: 'Satellite imagery draped on terrain' },
+  { value: 'google-photorealistic', label: 'Google Photorealistic', description: 'High-fidelity 3D photogrammetry (where available)' },
+];
+
+export function CesiumSettingsDialog() {
+  const dataSource = useViewerStore((s) => s.cesiumDataSource);
+  const setCesiumDataSource = useViewerStore((s) => s.setCesiumDataSource);
+  const terrainEnabled = useViewerStore((s) => s.cesiumTerrainEnabled);
+  const setCesiumTerrainEnabled = useViewerStore((s) => s.setCesiumTerrainEnabled);
+  const [open, setOpen] = useState(false);
+
+  return (
+    <Dialog open={open} onOpenChange={setOpen}>
+      <DialogTrigger asChild>
+        <Button variant="ghost" size="icon-sm" className="h-6 w-6" aria-label="Cesium settings">
+          <Settings2 className="h-3 w-3" />
+        </Button>
+      </DialogTrigger>
+      <DialogContent className="sm:max-w-md">
+        <DialogHeader>
+          <DialogTitle className="text-sm font-bold">3D World Context</DialogTitle>
+        </DialogHeader>
+
+        <div className="space-y-4 mt-2">
+          {/* Data Source */}
+          <div className="space-y-1.5">
+            <Label className="text-xs font-medium">3D Data Source</Label>
+            <div className="space-y-1">
+              {DATA_SOURCE_OPTIONS.map((opt) => (
+                <label
+                  key={opt.value}
+                  className={`flex items-start gap-2 p-2 rounded border cursor-pointer transition-colors ${
+                    dataSource === opt.value
+                      ? 'border-teal-500 bg-teal-50 dark:bg-teal-950/30'
+                      : 'border-zinc-200 dark:border-zinc-700 hover:border-zinc-300 dark:hover:border-zinc-600'
+                  }`}
+                >
+                  <input
+                    type="radio"
+                    name="cesium-data-source"
+                    checked={dataSource === opt.value}
+                    onChange={() => setCesiumDataSource(opt.value)}
+                    className="mt-0.5 accent-teal-500"
+                  />
+                  <div>
+                    <div className="text-xs font-medium">{opt.label}</div>
+                    <div className="text-[10px] text-zinc-400">{opt.description}</div>
+                  </div>
+                </label>
+              ))}
+            </div>
+          </div>
+
+          {/* Terrain Toggle */}
+          <label className="flex items-center gap-2 cursor-pointer">
+            <input
+              type="checkbox"
+              checked={terrainEnabled}
+              onChange={(e) => setCesiumTerrainEnabled(e.target.checked)}
+              className="accent-teal-500"
+            />
+            <span className="text-xs">Enable 3D terrain (Cesium World Terrain)</span>
+          </label>
+
+          <p className="text-[10px] text-zinc-400 border-t border-zinc-100 dark:border-zinc-800 pt-3">
+            Powered by Cesium ion. Real-world 3D tiles are overlaid on your georeferenced
+            IFC model. Only available when the model contains IfcMapConversion data.
+          </p>
+        </div>
+      </DialogContent>
+    </Dialog>
+  );
+}

package/src/components/viewer/ChatPanel.tsx

@@ -31,6 +31,7 @@ import {
 } from 'lucide-react';
 import { SignInButton, SignedIn, SignedOut, UserButton } from '@clerk/clerk-react';
 import { Button } from '@/components/ui/button';
+import { toast } from '@/components/ui/toast';
 import { Tooltip, TooltipContent, TooltipTrigger } from '@/components/ui/tooltip';
 import { useViewerStore } from '@/store';
 import { buildErrorFeedbackContent } from '@/store/slices/chatSlice';
@@ -50,6 +51,8 @@ import type { ChatMessage, ChatRepairRequest, FileAttachment } from '@/lib/llm/t
 import { canUsePlainCodeBlockFallback, type ScriptMutationIntent } from '@/lib/llm/script-preservation';
 import { Image as ImageIcon } from 'lucide-react';
 import { isClerkConfigured } from '@/lib/llm/clerk-auth';
+import { buildDesktopUpgradeUrl, hasDesktopFeatureAccess } from '@/lib/desktop-product';
+import { navigateToPath } from '@/services/app-navigation';
 import { getModelById } from '@/lib/llm/models';
 import { useSandbox } from '@/hooks/useSandbox';
 
@@ -212,7 +215,9 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
   const hasPro = useViewerStore((s) => s.chatHasPro);
   const usage = useViewerStore((s) => s.chatUsage);
   const setChatUsage = useViewerStore((s) => s.setChatUsage);
+  const desktopEntitlement = useViewerStore((s) => s.desktopEntitlement);
   const { execute } = useSandbox();
+  const canUseAiAssistant = hasDesktopFeatureAccess(desktopEntitlement, 'ai_assistant');
   const displayUsage: UsageInfo | null = usage ?? (hasPro
     ? {
       type: 'credits',
@@ -233,6 +238,14 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
   const [showScrollBtn, setShowScrollBtn] = useState(false);
   const [userScrolledUp, setUserScrolledUp] = useState(false);
   const [lastFinishReason, setLastFinishReason] = useState<string | null>(null);
+  const openUpgradePage = useCallback(() => {
+    navigateToPath(buildDesktopUpgradeUrl());
+  }, []);
+  const promptAiUpgrade = useCallback(() => {
+    setChatError('AI assistant is available with Desktop Pro.');
+    toast.info('AI assistant is available with Desktop Pro');
+    openUpgradePage();
+  }, [openUpgradePage, setChatError]);
 
   const inputRef = useRef<HTMLTextAreaElement>(null);
   const scrollRef = useRef<HTMLDivElement>(null);
@@ -385,6 +398,10 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
   // ── Core send logic ──
   const doSend = useCallback(async (text: string, options?: ChatSendOptions) => {
     if (!text.trim() || status === 'streaming' || status === 'sending') return;
+    if (!canUseAiAssistant) {
+      setChatError('AI assistant is available with Desktop Pro.');
+      return;
+    }
 
     const continuationBase = options?.continuationBase;
     const responseIntent = options?.intent ?? 'create';
@@ -780,15 +797,19 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
       }
     }
   }, [
-    status, activeModel, attachments, authToken,
+    canUseAiAssistant, status, activeModel, attachments, authToken,
     addMessage, setChatStatus, updateStreaming, finalizeAssistant,
     setChatError, setChatAbortController, clearAttachments, setChatUsage, resizeInput,
     buildRepairPromptFromLiveState, triggerAutoRepair, execute,
   ]);
 
   const handleSend = useCallback(() => {
+    if (!canUseAiAssistant) {
+      promptAiUpgrade();
+      return;
+    }
     doSend(inputText);
-  }, [inputText, doSend]);
+  }, [canUseAiAssistant, doSend, inputText, promptAiUpgrade]);
 
   // Allow other panels (e.g. ScriptPanel errors) to trigger a chat repair turn.
   useEffect(() => {
@@ -817,6 +838,10 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
   }, [pendingRepairRequest, status, consumePendingRepairRequest, buildRepairPromptFromLiveState, doSend]);
 
   const handleContinue = useCallback(() => {
+    if (!canUseAiAssistant) {
+      promptAiUpgrade();
+      return;
+    }
     const state = useViewerStore.getState();
     const partial = state.chatStreamingContent.trim();
     const lastAssistant = [...state.chatMessages].reverse().find((m) => m.role === 'assistant');
@@ -829,7 +854,7 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
     }
     setChatError(null);
     doSend(CONTINUE_PROMPT, { continuationBase });
-  }, [doSend, finalizeAssistant, setChatError]);
+  }, [canUseAiAssistant, doSend, finalizeAssistant, promptAiUpgrade, setChatError]);
 
   const handleStop = useCallback(() => {
     const controller = useViewerStore.getState().chatAbortController;
@@ -854,6 +879,10 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
 
   // ── Error feedback (Fix this) ──
   const handleFixError = useCallback((code: string, errorMsg: string) => {
+    if (!canUseAiAssistant) {
+      promptAiUpgrade();
+      return;
+    }
     const diagnostics = useViewerStore.getState().scriptLastDiagnostics;
     const liveCode = useViewerStore.getState().scriptEditorContent;
     const staleCode = code.trim() !== liveCode.trim() ? code : undefined;
@@ -868,7 +897,7 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
       requestedRepairScope: getPrimaryRootCause(diagnostics)?.repairScope,
       rootCauseKey: getPrimaryRootCause(diagnostics)?.rootCauseKey,
     });
-  }, [buildRepairPromptFromLiveState, doSend]);
+  }, [buildRepairPromptFromLiveState, canUseAiAssistant, doSend, promptAiUpgrade]);
 
   // ── Clickable example prompts ──
   const handleExampleClick = useCallback((prompt: string) => {
@@ -898,6 +927,10 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
 
   // ── File upload (button + drag-drop + paste) ──
   const processFiles = useCallback(async (files: FileList | File[]) => {
+    if (!canUseAiAssistant) {
+      promptAiUpgrade();
+      return;
+    }
     const model = getModelById(activeModel);
     const supportsImages = model?.supportsImages ?? false;
     const supportsFileAttachments = model?.supportsFileAttachments ?? true;
@@ -965,7 +998,7 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
         setChatError(`Could not read ${file.name}: ${error instanceof Error ? error.message : String(error)}`);
       }
     }
-  }, [activeModel, addAttachment, attachments.length, setChatError]);
+  }, [activeModel, addAttachment, attachments.length, canUseAiAssistant, promptAiUpgrade, setChatError]);
 
   const handleFileUpload = useCallback(async (e: React.ChangeEvent<HTMLInputElement>) => {
     const files = e.target.files;
@@ -1044,13 +1077,6 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
   const canContinue = Boolean(
     !isActive && (streamingContent.trim().length > 0 || lastFinishReason === 'length'),
   );
-  const openUpgradePage = useCallback(() => {
-    const currentPath = `${window.location.pathname}${window.location.search}`;
-    const target = `/upgrade?returnTo=${encodeURIComponent(currentPath)}`;
-    window.history.pushState({}, '', target);
-    window.dispatchEvent(new PopStateEvent('popstate'));
-  }, []);
-
   return (
     <div
       className="h-full flex flex-col bg-background relative"
@@ -1134,6 +1160,15 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
         )}
       </div>
 
+      {!canUseAiAssistant && (
+        <div className="border-b bg-muted/40 px-3 py-2 text-xs text-muted-foreground flex items-center justify-between gap-3">
+          <span>AI assistant is available with Desktop Pro. Core viewing and scripting stay available without it.</span>
+          <Button variant="outline" size="sm" className="h-6 px-2 text-xs" onClick={openUpgradePage}>
+            Upgrade
+          </Button>
+        </div>
+      )}
+
       {/* Clear confirmation */}
       {showClearConfirm && (
         <div className="px-3 py-2 bg-destructive/5 border-b flex items-center gap-2 text-xs">
@@ -1308,14 +1343,16 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
                 variant="ghost"
                 size="icon-xs"
                 onClick={() => fileInputRef.current?.click()}
-                disabled={!canAttachInput}
+                disabled={!canAttachInput || !canUseAiAssistant}
                 className="shrink-0 mb-0.5"
               >
                 <Paperclip className="h-3.5 w-3.5" />
               </Button>
             </TooltipTrigger>
             <TooltipContent>
-              {canAttachInput
+              {!canUseAiAssistant
+                ? 'Desktop Pro required for AI assistant'
+                : canAttachInput
                 ? 'Attach file or image (paste, drag & drop)'
                 : 'Selected model does not support attachments'}
             </TooltipContent>
@@ -1330,10 +1367,11 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
             }}
             onKeyDown={handleKeyDown}
             onPaste={handlePaste}
-            placeholder="Ask anything..."
+            placeholder={canUseAiAssistant ? 'Ask anything...' : 'Desktop Pro required for AI assistant'}
             rows={1}
             className="flex-1 resize-none rounded-md border border-input bg-background text-foreground placeholder:text-muted-foreground px-3 py-1.5 text-sm min-h-[32px] max-h-[120px] focus:outline-none focus:ring-1 focus:ring-ring"
             style={{ height: 'auto', overflow: 'hidden' }}
+            disabled={!canUseAiAssistant}
           />
 
           {isActive ? (
@@ -1357,7 +1395,7 @@ export function ChatPanel({ onClose }: ChatPanelProps) {
                   variant="default"
                   size="icon-xs"
                   onClick={handleSend}
-                  disabled={!inputText.trim()}
+                  disabled={!inputText.trim() || !canUseAiAssistant}
                   className="shrink-0 mb-0.5"
                 >
                   <Send className="h-3.5 w-3.5" />

package/src/components/viewer/CommandPalette.tsx

@@ -70,6 +70,7 @@ import { SCRIPT_TEMPLATES } from '@/lib/scripts/templates';
 import { GLTFExporter, CSVExporter } from '@ifc-lite/export';
 import { getRecentFiles, formatFileSize, getCachedFile } from '@/lib/recent-files';
 import type { RecentFileEntry } from '@/lib/recent-files';
+import { closeActiveAnalysisExtension } from '@/services/analysis-extensions';
 
 // ── Types ──────────────────────────────────────────────────────────────
 
@@ -189,6 +190,8 @@ function activateRightPanel(panel: 'bcf' | 'ids' | 'lens') {
     panel === 'ids' ? s.idsPanelVisible :
     s.lensPanelVisible;
 
+  closeActiveAnalysisExtension();
+
   // Close all content panels
   s.setBcfPanelVisible(false);
   s.setIdsPanelVisible(false);
@@ -211,6 +214,8 @@ function activateBottomPanel(panel: 'script' | 'list') {
   const s = useViewerStore.getState();
   const isActive = panel === 'script' ? s.scriptPanelVisible : s.listPanelVisible;
 
+  closeActiveAnalysisExtension();
+
   // Close all bottom panels
   s.setScriptPanelVisible(false);
   s.setListPanelVisible(false);
@@ -270,7 +275,7 @@ export function CommandPalette({ open, onOpenChange }: CommandPaletteProps) {
         detail: formatFileSize(rf.size),
         action: () => {
           // Try loading from IndexedDB blob cache → dispatches to MainToolbar's loadFile
-          getCachedFile(fileName).then(file => {
+          getCachedFile(rf).then(file => {
             if (file) {
               window.dispatchEvent(new CustomEvent('ifc-lite:load-file', { detail: file }));
             } else {

package/src/components/viewer/DesktopEntitlementBanner.tsx

@@ -0,0 +1,74 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+import { AlertTriangle, Clock3, WifiOff } from 'lucide-react';
+import { Button } from '@/components/ui/button';
+import { useViewerStore } from '@/store';
+import { buildDesktopUpgradeUrl, hasDesktopPro } from '@/lib/desktop-product';
+import { navigateToPath } from '@/services/app-navigation';
+
+function formatDate(timestamp: number | null): string | null {
+  if (!timestamp || !Number.isFinite(timestamp)) {
+    return null;
+  }
+  return new Date(timestamp).toLocaleDateString();
+}
+
+export function DesktopEntitlementBanner() {
+  const entitlement = useViewerStore((s) => s.desktopEntitlement);
+
+  if (entitlement.status === 'active' || entitlement.status === 'anonymous') {
+    return null;
+  }
+
+  if (entitlement.status === 'trial') {
+    return (
+      <div className="flex items-center justify-between gap-3 border-b bg-amber-50 px-4 py-2 text-sm text-amber-900 dark:bg-amber-950/60 dark:text-amber-200">
+        <div className="flex items-center gap-2 min-w-0">
+          <Clock3 className="h-4 w-4 shrink-0" />
+          <span className="truncate">
+            Desktop Pro trial active{formatDate(entitlement.trialEndsAt) ? ` until ${formatDate(entitlement.trialEndsAt)}` : ''}.
+          </span>
+        </div>
+        <Button variant="outline" size="sm" onClick={() => navigateToPath('/settings')}>
+          View Plan
+        </Button>
+      </div>
+    );
+  }
+
+  if (entitlement.status === 'grace_offline') {
+    return (
+      <div className="flex items-center justify-between gap-3 border-b bg-blue-50 px-4 py-2 text-sm text-blue-900 dark:bg-blue-950/60 dark:text-blue-200">
+        <div className="flex items-center gap-2 min-w-0">
+          <WifiOff className="h-4 w-4 shrink-0" />
+          <span className="truncate">
+            Offline grace active{formatDate(entitlement.graceUntil) ? ` until ${formatDate(entitlement.graceUntil)}` : ''}. Pro features remain available from the last validated plan.
+          </span>
+        </div>
+        <Button variant="outline" size="sm" onClick={() => navigateToPath('/settings')}>
+          View Plan
+        </Button>
+      </div>
+    );
+  }
+
+  if (!hasDesktopPro(entitlement)) {
+    return (
+      <div className="flex items-center justify-between gap-3 border-b bg-muted/60 px-4 py-2 text-sm text-foreground">
+        <div className="flex items-center gap-2 min-w-0">
+          <AlertTriangle className="h-4 w-4 shrink-0 text-amber-600 dark:text-amber-400" />
+          <span className="truncate">
+            Desktop Pro is not active. Core viewing stays available; advanced features are locked.
+          </span>
+        </div>
+        <Button variant="default" size="sm" onClick={() => navigateToPath(buildDesktopUpgradeUrl())}>
+          Upgrade
+        </Button>
+      </div>
+    );
+  }
+
+  return null;
+}

package/src/components/viewer/ExportChangesButton.tsx

@@ -18,6 +18,7 @@ import { StepExporter } from '@ifc-lite/export';
 import { MutablePropertyView } from '@ifc-lite/mutations';
 import type { IfcDataStore } from '@ifc-lite/parser';
 import { toast } from '@/components/ui/toast';
+import { ensureModelExportReady } from '@/services/desktop-export';
 
 interface ExportChangesButtonProps {
   /** Optional custom class name */
@@ -124,6 +125,10 @@ export function ExportChangesButton({ className }: ExportChangesButtonProps) {
 
     try {
       const mutationView = getMutationView(modelInfo.id);
+      const exportDataStore = await ensureModelExportReady(modelInfo.id);
+      if (!exportDataStore) {
+        throw new Error('Model data is unavailable for export');
+      }
 
       // Determine schema version
       const schemaVersion = modelInfo.schemaVersion || 'IFC4';
@@ -131,7 +136,7 @@ export function ExportChangesButton({ className }: ExportChangesButtonProps) {
                    : schemaVersion.includes('4X3') ? 'IFC4X3'
                    : 'IFC4';
 
-      const exporter = new StepExporter(modelInfo.ifcDataStore, mutationView || undefined);
+      const exporter = new StepExporter(exportDataStore, mutationView || undefined);
       const georefMutations = useViewerStore.getState().georefMutations?.get(modelInfo.id) ?? undefined;
       const result = exporter.export({
         schema: schema as 'IFC2X3' | 'IFC4' | 'IFC4X3',

package/src/components/viewer/ExportDialog.tsx

@@ -52,6 +52,7 @@ import { Progress } from '@/components/ui/progress';
 import { useViewerStore } from '@/store';
 import { configureMutationView } from '@/utils/configureMutationView';
 import { toast } from '@/components/ui/toast';
+import { ensureModelExportReady } from '@/services/desktop-export';
 import { StepExporter, MergedExporter, Ifc5Exporter, IFC5_KNOWN_PROP_NAMES, type MergeModelInput, type ExportProgress, type StepExportProgress } from '@ifc-lite/export';
 import { MutablePropertyView } from '@ifc-lite/mutations';
 import type { IfcDataStore } from '@ifc-lite/parser';
@@ -320,11 +321,21 @@ export function ExportDialog({ trigger }: ExportDialogProps) {
     try {
       // Handle merged export of all models (STEP only, not IFC5)
       if (!isIfc5 && exportScope === 'merged' && !changesOnly) {
-        const mergeInputs: MergeModelInput[] = Array.from(models.values()).map((m) => ({
-          id: m.id,
-          name: m.name,
-          dataStore: m.ifcDataStore,
-        }));
+        const hydratedModels = await Promise.all(Array.from(models.values()).map(async (model) => ({
+          model,
+          dataStore: await ensureModelExportReady(model.id),
+        })));
+        const mergeInputs: MergeModelInput[] = [];
+        for (const entry of hydratedModels) {
+          if (!entry.dataStore) {
+            continue;
+          }
+          mergeInputs.push({
+            id: entry.model.id,
+            name: entry.model.name,
+            dataStore: entry.dataStore,
+          });
+        }
 
         const mergedExporter = new MergedExporter(mergeInputs);
 
@@ -467,7 +478,12 @@ export function ExportDialog({ trigger }: ExportDialogProps) {
 
       // ── Pre-IFC5 full export → STEP ──────────────────────────────────
       } else {
-        const exporter = new StepExporter(selectedModel.ifcDataStore, mutationView || undefined);
+        const exportDataStore = await ensureModelExportReady(selectedModelId);
+        if (!exportDataStore) {
+          throw new Error('Model data is unavailable for export');
+        }
+
+        const exporter = new StepExporter(exportDataStore, mutationView || undefined);
 
         const localHidden = visibleOnly ? getLocalHiddenIds(selectedModelId) : undefined;
         const localIsolated = visibleOnly ? getLocalIsolatedIds(selectedModelId) : undefined;