"use client";
import { useRef, useState, useEffect, useCallback, useMemo } from "react";
import { useUpdateEffect } from "react-use";
import classNames from "classnames";
import { cn } from "@/lib/utils";
import { GridPattern } from "@/components/magic-ui/grid-pattern";
import { useEditor } from "@/hooks/useEditor";
import { useAi } from "@/hooks/useAi";
import { htmlTagToText } from "@/lib/html-tag-to-text";
import { AnimatedBlobs } from "@/components/animated-blobs";
import { AiLoading } from "../ask-ai/loading";
import { defaultHTML } from "@/lib/consts";
import { HistoryNotification } from "../history-notification";
import { api } from "@/lib/api";
import { toast } from "sonner";
import { RefreshCcw, TriangleAlert } from "lucide-react";
import { Page } from "@/types";
export const Preview = ({
isNew,
namespace,
repoId,
}: {
isNew: boolean;
namespace?: string;
repoId?: string;
}) => {
const {
project,
device,
isLoadingProject,
currentTab,
currentCommit,
setCurrentCommit,
currentPageData,
pages,
setPages,
setCurrentPage,
previewPage,
setPreviewPage,
setLastSavedPages,
hasUnsavedChanges,
} = useEditor();
const {
isEditableModeEnabled,
setSelectedElement,
isAiWorking,
globalAiLoading,
} = useAi();
const iframeRef = useRef(null);
const [hoveredElement, setHoveredElement] = useState<{
tagName: string;
rect: { top: number; left: number; width: number; height: number };
} | null>(null);
const [isPromotingVersion, setIsPromotingVersion] = useState(false);
const [stableHtml, setStableHtml] = useState("");
const [throttledHtml, setThrottledHtml] = useState("");
const lastUpdateTimeRef = useRef(0);
const [iframeKey, setIframeKey] = useState(0);
const [commitPages, setCommitPages] = useState([]);
const [isLoadingCommitPages, setIsLoadingCommitPages] = useState(false);
const prevCommitRef = useRef(null);
useEffect(() => {
if (!previewPage && pages.length > 0) {
const indexPage = pages.find(
(p) => p.path === "index.html" || p.path === "index" || p.path === "/"
);
const firstHtmlPage = pages.find((p) => p.path.endsWith(".html"));
setPreviewPage(indexPage?.path || firstHtmlPage?.path || "index.html");
}
}, [pages, previewPage]);
const pagesToUse = currentCommit ? commitPages : pages;
const previewPageData = useMemo(() => {
const found = pagesToUse.find((p) => {
const normalizedPagePath = p.path.replace(/^\.?\//, "");
const normalizedPreviewPage = previewPage.replace(/^\.?\//, "");
return normalizedPagePath === normalizedPreviewPage;
});
return found || (pagesToUse.length > 0 ? pagesToUse[0] : currentPageData);
}, [pagesToUse, previewPage, currentPageData]);
// Fetch commit pages when currentCommit changes
useEffect(() => {
if (currentCommit && namespace && repoId) {
setIsLoadingCommitPages(true);
api
.get(`/me/projects/${namespace}/${repoId}/commits/${currentCommit}`)
.then((res) => {
if (res.data.ok) {
setCommitPages(res.data.pages);
// Set preview page to index.html if available
const indexPage = res.data.pages.find(
(p: Page) =>
p.path === "index.html" || p.path === "index" || p.path === "/"
);
if (indexPage) {
setPreviewPage(indexPage.path);
}
// Refresh iframe to show commit version
setIframeKey((prev) => prev + 1);
}
})
.catch((err) => {
toast.error(
err.response?.data?.error || "Failed to fetch commit pages"
);
})
.finally(() => {
setIsLoadingCommitPages(false);
});
} else if (!currentCommit && prevCommitRef.current !== null) {
// Only clear commitPages when transitioning from a commit to no commit
setCommitPages([]);
}
prevCommitRef.current = currentCommit;
}, [currentCommit, namespace, repoId]);
// Create navigation interception script
const createNavigationScript = useCallback((availablePages: Page[]) => {
const pagePaths = availablePages.map((p) => p.path.replace(/^\.?\//, ""));
return `
(function() {
const availablePages = ${JSON.stringify(pagePaths)};
function normalizePath(path) {
let normalized = path.replace(/^\.?\//, "");
if (normalized === "" || normalized === "/") {
normalized = "index.html";
}
const hashIndex = normalized.indexOf("#");
if (hashIndex !== -1) {
normalized = normalized.substring(0, hashIndex);
}
if (!normalized.includes(".")) {
normalized = normalized + ".html";
}
return normalized;
}
function handleNavigation(url) {
if (!url) return;
// Handle hash-only navigation
if (url.startsWith("#")) {
const targetElement = document.querySelector(url);
if (targetElement) {
targetElement.scrollIntoView({ behavior: "smooth" });
}
// Search in shadow DOM
const searchInShadows = (root) => {
const elements = root.querySelectorAll("*");
for (const el of elements) {
if (el.shadowRoot) {
const found = el.shadowRoot.querySelector(url);
if (found) {
found.scrollIntoView({ behavior: "smooth" });
return;
}
searchInShadows(el.shadowRoot);
}
}
};
searchInShadows(document);
return;
}
// Handle external URLs
if (url.startsWith("http://") || url.startsWith("https://") || url.startsWith("//")) {
window.open(url, "_blank");
return;
}
const normalizedPath = normalizePath(url);
if (availablePages.includes(normalizedPath)) {
// Dispatch custom event to notify parent
window.parent.postMessage({ type: 'navigate', path: normalizedPath }, '*');
} else {
console.warn('Page not found:', normalizedPath);
}
}
// Intercept window.location methods
const originalAssign = window.location.assign;
const originalReplace = window.location.replace;
window.location.assign = function(url) {
handleNavigation(url);
};
window.location.replace = function(url) {
handleNavigation(url);
};
// Intercept window.location.href setter
try {
let currentHref = window.location.href;
Object.defineProperty(window.location, 'href', {
get: function() {
return currentHref;
},
set: function(url) {
handleNavigation(url);
},
configurable: true
});
} catch (e) {
// Fallback: use proxy if defineProperty fails
console.warn('Could not intercept location.href:', e);
}
// Intercept link clicks
document.addEventListener('click', function(e) {
const anchor = e.target.closest('a');
if (anchor && anchor.href) {
const href = anchor.getAttribute('href');
if (href && !href.startsWith('http://') && !href.startsWith('https://') && !href.startsWith('//') && !href.startsWith('mailto:') && !href.startsWith('tel:')) {
e.preventDefault();
handleNavigation(href);
}
}
}, true);
// Intercept form submissions
document.addEventListener('submit', function(e) {
const form = e.target;
if (form.action && !form.action.startsWith('http://') && !form.action.startsWith('https://') && !form.action.startsWith('//')) {
e.preventDefault();
handleNavigation(form.action);
}
}, true);
})();
`;
}, []);
const injectAssetsIntoHtml = useCallback(
(html: string, pagesToUse: Page[] = pages): string => {
if (!html) return html;
const cssFiles = pagesToUse.filter(
(p) => p.path.endsWith(".css") && p.path !== previewPageData?.path
);
const jsFiles = pagesToUse.filter(
(p) => p.path.endsWith(".js") && p.path !== previewPageData?.path
);
const jsonFiles = pagesToUse.filter(
(p) => p.path.endsWith(".json") && p.path !== previewPageData?.path
);
let modifiedHtml = html;
// Inject navigation script for srcDoc
const navigationScript = createNavigationScript(pagesToUse);
// Inject all CSS files
if (cssFiles.length > 0) {
const allCssContent = cssFiles
.map(
(file) =>
``
)
.join("\n");
if (modifiedHtml.includes("")) {
modifiedHtml = modifiedHtml.replace(
"",
`${allCssContent}\n`
);
} else if (modifiedHtml.includes("")) {
modifiedHtml = modifiedHtml.replace(
"",
`\n${allCssContent}`
);
} else {
modifiedHtml = allCssContent + "\n" + modifiedHtml;
}
cssFiles.forEach((file) => {
const escapedPath = file.path.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
modifiedHtml = modifiedHtml.replace(
new RegExp(
`]*href=["'][\\.\/]*${escapedPath}["'][^>]*>`,
"gi"
),
""
);
});
}
if (jsFiles.length > 0) {
const allJsContent = jsFiles
.map(
(file) =>
``
)
.join("\n");
if (modifiedHtml.includes("")) {
modifiedHtml = modifiedHtml.replace(
"",
`${allJsContent}\n`
);
} else if (modifiedHtml.includes("")) {
modifiedHtml = modifiedHtml + allJsContent;
} else {
modifiedHtml = modifiedHtml + "\n" + allJsContent;
}
jsFiles.forEach((file) => {
const escapedPath = file.path.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
modifiedHtml = modifiedHtml.replace(
new RegExp(
``
)
.join("\n");
if (modifiedHtml.includes("")) {
modifiedHtml = modifiedHtml.replace(
"
")) {
modifiedHtml = modifiedHtml + allJsonContent;
} else {
modifiedHtml = modifiedHtml + "\n" + allJsonContent;
}
jsonFiles.forEach((file) => {
const escapedPath = file.path.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
modifiedHtml = modifiedHtml.replace(
new RegExp(
``
);
} else if (modifiedHtml.includes("")) {
modifiedHtml = modifiedHtml.replace(
"",
`\n`
);
} else if (modifiedHtml.includes("")) {
modifiedHtml = modifiedHtml.replace(
"",
`\n`
);
} else {
modifiedHtml =
`\n` + modifiedHtml;
}
}
return modifiedHtml;
},
[pages, previewPageData?.path, createNavigationScript]
);
useEffect(() => {
if (isNew && previewPageData?.html) {
const now = Date.now();
const timeSinceLastUpdate = now - lastUpdateTimeRef.current;
if (lastUpdateTimeRef.current === 0 || timeSinceLastUpdate >= 3000) {
const processedHtml = injectAssetsIntoHtml(
previewPageData.html,
pagesToUse
);
setThrottledHtml(processedHtml);
lastUpdateTimeRef.current = now;
} else {
const timeUntilNextUpdate = 3000 - timeSinceLastUpdate;
const timer = setTimeout(() => {
const processedHtml = injectAssetsIntoHtml(
previewPageData.html,
pagesToUse
);
setThrottledHtml(processedHtml);
lastUpdateTimeRef.current = Date.now();
}, timeUntilNextUpdate);
return () => clearTimeout(timer);
}
}
}, [isNew, previewPageData?.html, injectAssetsIntoHtml, pagesToUse]);
useEffect(() => {
if (!isAiWorking && !globalAiLoading && previewPageData?.html) {
const processedHtml = injectAssetsIntoHtml(
previewPageData.html,
pagesToUse
);
setStableHtml(processedHtml);
}
}, [
isAiWorking,
globalAiLoading,
previewPageData?.html,
injectAssetsIntoHtml,
previewPage,
pagesToUse,
]);
useEffect(() => {
if (
previewPageData?.html &&
!stableHtml &&
!isAiWorking &&
!globalAiLoading
) {
const processedHtml = injectAssetsIntoHtml(
previewPageData.html,
pagesToUse
);
setStableHtml(processedHtml);
}
}, [
previewPageData?.html,
stableHtml,
isAiWorking,
globalAiLoading,
injectAssetsIntoHtml,
pagesToUse,
]);
const setupIframeListeners = () => {
if (iframeRef?.current?.contentDocument) {
const iframeDocument = iframeRef.current.contentDocument;
iframeDocument.addEventListener(
"click",
handleCustomNavigation as any,
true
);
if (isEditableModeEnabled) {
iframeDocument.addEventListener("mouseover", handleMouseOver);
iframeDocument.addEventListener("mouseout", handleMouseOut);
iframeDocument.addEventListener("click", handleClick);
}
}
};
// Listen for navigation messages from iframe
useEffect(() => {
const handleMessage = (event: MessageEvent) => {
if (event.data?.type === "navigate" && event.data?.path) {
setPreviewPage(event.data.path);
}
};
window.addEventListener("message", handleMessage);
return () => window.removeEventListener("message", handleMessage);
}, [setPreviewPage]);
useEffect(() => {
const cleanupListeners = () => {
if (iframeRef?.current?.contentDocument) {
const iframeDocument = iframeRef.current.contentDocument;
iframeDocument.removeEventListener(
"click",
handleCustomNavigation as any,
true
);
iframeDocument.removeEventListener("mouseover", handleMouseOver);
iframeDocument.removeEventListener("mouseout", handleMouseOut);
iframeDocument.removeEventListener("click", handleClick);
}
};
const timer = setTimeout(() => {
if (iframeRef?.current?.contentDocument) {
cleanupListeners();
setupIframeListeners();
}
}, 50);
return () => {
clearTimeout(timer);
cleanupListeners();
};
}, [isEditableModeEnabled, stableHtml, throttledHtml, previewPage]);
const refreshIframe = () => {
setIframeKey((prev) => prev + 1);
};
const promoteVersion = async () => {
setIsPromotingVersion(true);
await api
.post(
`/me/projects/${project?.space_id}/commits/${currentCommit}/promote`
)
.then((res) => {
if (res.data.ok) {
setCurrentCommit(null);
setPages(res.data.pages);
setCurrentPage(res.data.pages[0].path);
setLastSavedPages(res.data.pages);
setPreviewPage(res.data.pages[0].path);
toast.success("Version promoted successfully");
}
})
.catch((err) => {
toast.error(err.response.data.error);
});
setIsPromotingVersion(false);
};
const handleMouseOver = (event: MouseEvent) => {
if (iframeRef?.current) {
const iframeDocument = iframeRef.current.contentDocument;
if (iframeDocument) {
const targetElement = event.target as HTMLElement;
if (
hoveredElement?.tagName !== targetElement.tagName ||
hoveredElement?.rect.top !==
targetElement.getBoundingClientRect().top ||
hoveredElement?.rect.left !==
targetElement.getBoundingClientRect().left ||
hoveredElement?.rect.width !==
targetElement.getBoundingClientRect().width ||
hoveredElement?.rect.height !==
targetElement.getBoundingClientRect().height
) {
if (targetElement !== iframeDocument.body) {
const rect = targetElement.getBoundingClientRect();
setHoveredElement({
tagName: targetElement.tagName,
rect: {
top: rect.top,
left: rect.left,
width: rect.width,
height: rect.height,
},
});
targetElement.classList.add("hovered-element");
} else {
return setHoveredElement(null);
}
}
}
}
};
const handleMouseOut = () => {
setHoveredElement(null);
};
const handleClick = (event: MouseEvent) => {
if (iframeRef?.current) {
const iframeDocument = iframeRef.current.contentDocument;
if (iframeDocument) {
const path = event.composedPath();
const targetElement = path[0] as HTMLElement;
const findClosestAnchor = (
element: HTMLElement
): HTMLAnchorElement | null => {
let current: HTMLElement | null = element;
while (current) {
if (current.tagName?.toUpperCase() === "A") {
return current as HTMLAnchorElement;
}
if (current === iframeDocument.body) {
break;
}
const parent: Node | null = current.parentNode;
if (parent && parent.nodeType === 11) {
current = (parent as ShadowRoot).host as HTMLElement;
} else if (parent && parent.nodeType === 1) {
current = parent as HTMLElement;
} else {
break;
}
}
return null;
};
const anchorElement = findClosestAnchor(targetElement);
if (anchorElement) {
return;
}
if (targetElement !== iframeDocument.body) {
setSelectedElement(targetElement);
}
}
}
};
const handleCustomNavigation = (event: MouseEvent) => {
if (iframeRef?.current) {
const iframeDocument = iframeRef.current.contentDocument;
if (iframeDocument) {
const path = event.composedPath();
const actualTarget = path[0] as HTMLElement;
const findClosestAnchor = (
element: HTMLElement
): HTMLAnchorElement | null => {
let current: HTMLElement | null = element;
while (current) {
if (current.tagName?.toUpperCase() === "A") {
return current as HTMLAnchorElement;
}
if (current === iframeDocument.body) {
break;
}
const parent: Node | null = current.parentNode;
if (parent && parent.nodeType === 11) {
current = (parent as ShadowRoot).host as HTMLElement;
} else if (parent && parent.nodeType === 1) {
current = parent as HTMLElement;
} else {
break;
}
}
return null;
};
const anchorElement = findClosestAnchor(actualTarget);
if (anchorElement) {
let href = anchorElement.getAttribute("href");
if (href) {
event.stopPropagation();
event.preventDefault();
if (href.startsWith("#")) {
let targetElement = iframeDocument.querySelector(href);
if (!targetElement) {
const searchInShadows = (
root: Document | ShadowRoot
): Element | null => {
const elements = root.querySelectorAll("*");
for (const el of elements) {
if (el.shadowRoot) {
const found = el.shadowRoot.querySelector(href);
if (found) return found;
const nested = searchInShadows(el.shadowRoot);
if (nested) return nested;
}
}
return null;
};
targetElement = searchInShadows(iframeDocument);
}
if (targetElement) {
targetElement.scrollIntoView({ behavior: "smooth" });
}
return;
}
let normalizedHref = href.replace(/^\.?\//, "");
if (normalizedHref === "" || normalizedHref === "/") {
normalizedHref = "index.html";
}
const hashIndex = normalizedHref.indexOf("#");
if (hashIndex !== -1) {
normalizedHref = normalizedHref.substring(0, hashIndex);
}
if (!normalizedHref.includes(".")) {
normalizedHref = normalizedHref + ".html";
}
const isPageExist = pagesToUse.some((page) => {
const pagePath = page.path.replace(/^\.?\//, "");
return pagePath === normalizedHref;
});
if (isPageExist) {
setPreviewPage(normalizedHref);
}
}
}
}
}
};
return (
{/* Preview page indicator */}
{!isAiWorking && hoveredElement && isEditableModeEnabled && (
{htmlTagToText(hoveredElement.tagName.toLowerCase())}
)}
{isLoadingProject ? (
) : (
<>
{isLoadingCommitPages && (
)}
{!isNew && !currentCommit && (
Preview version of the project. Try refreshing the preview if
you experience any issues.
)}
);
};