@adminide-stack/yantra-mobile 12.0.142 → 12.0.144

package/lib/screens/WebBuilderNative/useWebBuilderHost.js

@@ -1,4 +1,4 @@
-import {useRef,useState,useMemo,useEffect,useCallback,useSyncExternalStore}from'react';import {useAiCodeDeployToCloudflareMutation}from'common/graphql';import {v4}from'uuid';import AsyncStorage from'@react-native-async-storage/async-storage';import {useCdecliConnection}from'../../contexts/CdecliConnectionContext.js';import {useChatStream}from'../../hooks/useChatStream.js';import {useHostChatChannel}from'../../hooks/useHostChatChannel.js';import {config}from'../../config/env-config.js';import {useChatMessages,subscribeRememberedHistory,historyRunningChannelIds}from'../../hooks/useChatApi.js';import {usePrerequisiteIds}from'../../hooks/usePrerequisiteIds.js';import {useCanvasSession}from'../../features/canvas/useCanvasSession.js';import {tokenClaims}from'../../features/canvas/sessionIdentity.js';import {buildsFromPosts,hasWorkflowPosts,titleFromPrompt,stripAgentNote,projectFromPosts,turnState,pendingWork,transcriptFromPosts,displayAgentText}from'../../features/webbuilder/buildFragments.js';import {handEditsNote,savedCodeSource}from'../../features/webbuilder/codeFiles.js';import {createCodeWorkspace}from'./codeWorkspaceApi.js';import {WEBBUILDER_CHAT_SKILL,buildTriggerParams,templateRemixNote,newObjectId}from'../../features/webbuilder/workflow.js';import {useWebBuilderGallery}from'./useWebBuilderGallery.js';import {agentEditNote,agentPublishMessage}from'../../features/webbuilder/webbuilderPrompts.js';import {useCodeAssistantCatalog}from'./useCodeAssistantCatalog.js';import {uploadAttachment,triggerBuild,createSandboxRouter,directSandboxRun,restoreSandbox}from'./webbuilderTransport.js';import {useSandboxProxyApi}from'./sandboxProxyApi.js';import {useSandboxKeepAlive}from'./useSandboxKeepAlive.js';import {pendingAskUser}from'../../features/chat/askUser.js';import {runningBuild,useRunningBuilds,markBuildDone}from'./runningBuilds.js';import {isAgentSite}from'./siteLane.js';import {undoLastChange,redoChange}from'./versionUndo.js';import {WEBBUILDER_EXPORT_ZIP_ACTION,exportZipParams,exportZipUrl}from'../../features/webbuilder/exportCode.js';import {useRestoredSandboxes,applyRestored}from'./useRestoredSandboxes.js';import {useWebBuilderJobs}from'./useWebBuilderJobs.js';var __defProp = Object.defineProperty;
+import {useRef,useState,useCallback,useMemo,useEffect,useSyncExternalStore}from'react';import {useApolloClient,gql}from'@apollo/client/index.js';import {useAiCodeDeployToCloudflareMutation}from'common/graphql';import {v4}from'uuid';import AsyncStorage from'@react-native-async-storage/async-storage';import {projectInstructionsNote,readDeleteReply,readSettingsUpdateReply,parseProjectSettings}from'../../features/webbuilder/projectTools.js';import {useCdecliConnection}from'../../contexts/CdecliConnectionContext.js';import {useChatStream}from'../../hooks/useChatStream.js';import {useHostChatChannel}from'../../hooks/useHostChatChannel.js';import {config}from'../../config/env-config.js';import {useChatMessages,subscribeRememberedHistory,historyRunningChannelIds}from'../../hooks/useChatApi.js';import {usePrerequisiteIds}from'../../hooks/usePrerequisiteIds.js';import {useCanvasSession}from'../../features/canvas/useCanvasSession.js';import {tokenClaims}from'../../features/canvas/sessionIdentity.js';import {buildsFromPosts,hasWorkflowPosts,siteNameFromRequest,titleFromPrompt,stripAgentNote,projectFromPosts,turnState,pendingWork,transcriptFromPosts,userBubbleText,displayAgentText}from'../../features/webbuilder/buildFragments.js';import {handEditsNote,savedCodeSource}from'../../features/webbuilder/codeFiles.js';import {createCodeWorkspace}from'./codeWorkspaceApi.js';import {WEBBUILDER_CHAT_SKILL,buildTriggerParams,templateRemixNote,newObjectId}from'../../features/webbuilder/workflow.js';import {useWebBuilderGallery}from'./useWebBuilderGallery.js';import {agentEditNote,agentPublishMessage}from'../../features/webbuilder/webbuilderPrompts.js';import {useCodeAssistantCatalog}from'./useCodeAssistantCatalog.js';import {useConnectorTemplates}from'./useConnectorTemplates.js';import {directChannels,uploadAttachment,triggerBuild,createSandboxRouter,directSandboxRun,wakeSandbox,restoreFromSavedCopy,OUTDATED_SERVER_RE}from'./webbuilderTransport.js';import {useSandboxProxyApi}from'./sandboxProxyApi.js';import {useSandboxKeepAlive}from'./useSandboxKeepAlive.js';import {pendingAskUser}from'../../features/chat/askUser.js';import {runningBuild,useRunningBuilds,markBuildDone}from'./runningBuilds.js';import {isAgentSite}from'./siteLane.js';import {undoLastChange,redoChange}from'./versionUndo.js';import {WEBBUILDER_EXPORT_ZIP_ACTION,exportZipParams,exportZipUrl}from'../../features/webbuilder/exportCode.js';import {useRestoredSandboxes,applyRestored}from'./useRestoredSandboxes.js';import {useWebBuilderJobs}from'./useWebBuilderJobs.js';var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
 var __getOwnPropSymbols = Object.getOwnPropertySymbols;
@@ -21,6 +21,72 @@ const POLL_MS = 2500;
 const ADOPTED_KEY = "yantra.webbuilder.adopted.v1";
 const TURN_TIMEOUT_MS = 15 * 60 * 1e3;
 const NO_REPLY_TIMEOUT_MS = 3 * 60 * 1e3;
+const PROJECT_SETTINGS_QUERY = gql`
+    query aiCodeProjectSettings($channelId: String!) {
+        aiCodeProjectSettings(channelId: $channelId)
+    }
+`;
+const UPDATE_PROJECT_SETTINGS_MUTATION = gql`
+    mutation aiCodeUpdateProjectSettings($channelId: String!, $input: JSON!) {
+        aiCodeUpdateProjectSettings(channelId: $channelId, input: $input)
+    }
+`;
+const DELETE_PROJECT_MUTATION = gql`
+    mutation aiCodeDeleteProject($channelId: String!) {
+        aiCodeDeleteProject(channelId: $channelId)
+    }
+`;
+const DELETE_CHANNEL_MUTATION = gql`
+    mutation deleteChannel($id: ID!) {
+        deleteChannel(id: $id)
+    }
+`;
+const PUSH_TO_GITHUB_MUTATION = gql`
+    mutation aiCodePushToGithub($sandboxId: String!, $repoName: String!, $channelId: String) {
+        aiCodePushToGithub(sandboxId: $sandboxId, repoName: $repoName, channelId: $channelId) {
+            success
+            message
+        }
+    }
+`;
+const INTEGRATION_CATALOG_QUERY = gql`
+    query WebbuilderIntegrationConnectors($search: String, $category: String) {
+        integrationConnectors(search: $search, category: $category) {
+            id
+            title
+            description
+            logo
+            authMethods
+            meta {
+                categories
+            }
+        }
+    }
+`;
+const DEPLOY_WITH_DOMAIN_MUTATION = gql`
+    mutation aiCodeDeployToCloudflareWithDomain(
+        $sandboxId: String!
+        $projectName: String
+        $channelId: String
+        $customDomain: String
+    ) {
+        aiCodeDeployToCloudflare(
+            sandboxId: $sandboxId
+            projectName: $projectName
+            channelId: $channelId
+            customDomain: $customDomain
+        ) {
+            success
+            message
+            eventId
+        }
+    }
+`;
+const PRE_PUBLISH_REVIEW_QUERY = gql`
+    query aiCodePrePublishReview($channelId: String!, $sandboxId: String!) {
+        aiCodePrePublishReview(channelId: $channelId, sandboxId: $sandboxId)
+    }
+`;
 function notSetUp(org) {
   return `Builds are not set up for ${org || "this workspace"} yet. Ask an admin to configure the Code Assistant templates.`;
 }
@@ -46,7 +112,13 @@ function useWebBuilderHost(routeOrg) {
   const tokenRef = useRef(token);
   tokenRef.current = token;
   const catalog = useCodeAssistantCatalog(effectiveOrg);
-  const reloadTemplates = catalog.reload;
+  const [templatesReload, setTemplatesReload] = useState(0);
+  const galleryTemplates = useConnectorTemplates(token, indexUrl, catalog.templates, templatesReload);
+  const reloadCatalog = catalog.reload;
+  const reloadTemplates = useCallback(() => {
+    reloadCatalog();
+    setTemplatesReload((n) => n + 1);
+  }, [reloadCatalog]);
   const cdnOrigin = config.AWS_CDN_URL;
   const gallery = useWebBuilderGallery(effectiveOrg, cdnOrigin);
   const {
@@ -135,6 +207,7 @@ function useWebBuilderHost(routeOrg) {
     const run = adoptAgentProject(channelId, title).then((adopted) => {
       if (adopted) {
         adoptedRef.current.add(channelId);
+        directChannels.delete(channelId);
         AsyncStorage.setItem(ADOPTED_KEY, JSON.stringify([...adoptedRef.current].slice(-500))).catch(() => void 0);
       }
       return adopted;
@@ -142,6 +215,7 @@ function useWebBuilderHost(routeOrg) {
     adoptingRef.current.set(channelId, run);
     return run;
   }, [adoptAgentProject]);
+  const instructionsByChannel = useRef(/* @__PURE__ */ new Map());
   const jobs = useWebBuilderJobs(effectiveOrg);
   const agentJob = useRef(null);
   const sendAgentTurn = useCallback((channelId, text, files, task) => {
@@ -154,14 +228,18 @@ function useWebBuilderHost(routeOrg) {
       lane: "agent",
       startedAt: Date.now()
     }) : Promise.resolve(null);
-    const adopted = ready.then(() => ensureAdopted(channelId, (task == null ? void 0 : task.siteTitle) || titleFromPrompt(stripAgentNote(text)) || "New site").catch(() => false));
+    const adopted = ready.then(() => ensureAdopted(channelId, (task == null ? void 0 : task.siteTitle) || siteNameFromRequest(text) || titleFromPrompt(stripAgentNote(text)) || "New site").catch(() => false));
     Promise.all([adopted, recorded]).then(([, job]) => {
       unsavedAgentChannels.current.delete(channelId);
       if (job) agentJob.current = {
         channelId,
         job
       };
-      return chat.sendMessage(text, files, channelId, false, void 0, {
+      const knowledge = instructionsByChannel.current.get(channelId);
+      const turn2 = knowledge ? `${text}
+
+${projectInstructionsNote(knowledge)}` : text;
+      return chat.sendMessage(turn2, files, channelId, false, void 0, {
         job: job == null ? void 0 : job.ref
       });
     }).then((outcome) => {
@@ -496,9 +574,9 @@ ${templateRemixNote(request.template)}` : request.text,
     const rows = transcriptFromPosts(posts).map((m) => ({
       id: m.id,
       role: m.role,
-      content: m.role === "user" ? stripAgentNote(m.content) : displayAgentText(m.content)
+      content: m.role === "user" ? userBubbleText(m.content) : displayAgentText(m.content)
     })).filter((m) => m.role === "user" || m.content.trim());
-    const pending = running && (turn == null ? void 0 : turn.text) ? turn.text.replace(/^build\s*me\s*:\s*/i, "") : "";
+    const pending = running && (turn == null ? void 0 : turn.text) ? userBubbleText(turn.text) : "";
     if (pending && !rows.some((m) => m.role === "user" && pending.startsWith(m.content))) {
       rows.push({
         id: `pending-${turn == null ? void 0 : turn.startedAt}`,
@@ -564,7 +642,7 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
     id: `${m.role}-${index}`,
     role: m.role,
     // The template note is for the agent; the person sees what they typed.
-    content: m.role === "user" ? stripAgentNote(m.content).replace(/^build\s*me\s*:\s*/i, "") : displayAgentText(m.content),
+    content: m.role === "user" ? userBubbleText(m.content) : displayAgentText(m.content),
     attachments: m.attachments
   })).filter((m) => m.role === "user" || m.content.trim()), [chat.messages]);
   const withOpenRun = useCallback((rows) => {
@@ -636,10 +714,15 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
     }),
     proxyMissing: sandboxApi.proxyMissing
   }), [activeChannelId, token, indexUrl, sandboxApi]);
+  const [goneSandboxIds, setGoneSandboxIds] = useState(() => /* @__PURE__ */ new Set());
+  const markGone = useCallback((sandboxId) => {
+    setGoneSandboxIds((prev) => prev.has(sandboxId) ? prev : new Set(prev).add(sandboxId));
+  }, []);
   const keepAlive = useSandboxKeepAlive({
     run: token || activeChannelId ? sandboxCall : null,
     sandboxId: (_h = builds[0]) == null ? void 0 : _h.sandboxId,
-    live: activeChannelId !== null && Boolean((_i = builds[0]) == null ? void 0 : _i.sandboxId)
+    live: activeChannelId !== null && Boolean((_i = builds[0]) == null ? void 0 : _i.sandboxId),
+    onGone: markGone
   });
   const {
     noteActivity
@@ -647,36 +730,66 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
   useEffect(() => {
     if (activeChannelId) noteActivity();
   }, [activeChannelId, posts.length, chat.messages.length, noteActivity]);
-  const resumePreview = useCallback(async (build) => {
+  const savedCopies = useRef(/* @__PURE__ */ new Map());
+  const backupRepoRef = useRef(null);
+  const resumePreview = useCallback(async (build, opts) => {
+    var _a2;
     const channelId = activeChannelId;
     if (!build.sandboxId || !tokenRef.current && !channelId) return "failed";
     noteActivity();
-    const result = await restoreSandbox(sandboxCall, {
+    const result = await wakeSandbox(sandboxCall, {
       sandboxId: build.sandboxId,
       snapshotId: build.snapshotId,
+      backupRepo: backupRepoRef.current,
       templateId: build.templateId,
-      previewUrl: build.previewUrl
+      previewUrl: build.previewUrl,
+      asked: opts == null ? void 0 : opts.asked
     });
-    if (result.outcome === "ok" && result.sandboxId && result.previewUrl) {
-      rememberRestored(build.key, result.sandboxId, result.previewUrl);
-      if (activeProject && activeProject.previewUrl === build.previewUrl) {
-        upsertProject(__spreadProps(__spreadValues({}, activeProject), {
-          previewUrl: result.previewUrl
-        }));
-      }
-      if (channelId) {
-        sandboxApi.persist({
-          channelId,
-          sandboxId: result.sandboxId,
-          sandboxUrl: result.previewUrl.split("?")[0],
-          snapshotId: build.snapshotId,
-          previousSandboxId: build.sandboxId
-        }).then((saved) => {
-          if (!saved.ok) console.warn("[WebBuilderNative] persist sandbox identity failed", saved.message);
-        }).catch(() => void 0);
-      }
-    }
+    const named = (_a2 = result.savedCopy) == null ? void 0 : _a2.snapshotId;
+    if (named) savedCopies.current.set(build.key, named);
+    if (result.outcome === "gone") markGone(build.sandboxId);
     return result.outcome;
+  }, [activeChannelId, sandboxCall, noteActivity, markGone]);
+  const restorePreview = useCallback(async (build) => {
+    const channelId = activeChannelId;
+    if (!build.sandboxId || !tokenRef.current && !channelId) {
+      return {
+        ok: false,
+        reason: "Sign in again to restore this site."
+      };
+    }
+    noteActivity();
+    const snapshotId = savedCopies.current.get(build.key) || build.snapshotId;
+    const result = await restoreFromSavedCopy(sandboxCall, {
+      sandboxId: build.sandboxId,
+      snapshotId,
+      templateId: build.templateId
+    });
+    if (result.outcome !== "ok") return {
+      ok: false,
+      reason: result.reason,
+      snapshotId: snapshotId != null ? snapshotId : null
+    };
+    rememberRestored(build.key, result.sandboxId, result.previewUrl);
+    if (activeProject && activeProject.previewUrl === build.previewUrl) {
+      upsertProject(__spreadProps(__spreadValues({}, activeProject), {
+        previewUrl: result.previewUrl
+      }));
+    }
+    if (channelId && !directChannels.has(channelId)) {
+      sandboxApi.persist({
+        channelId,
+        sandboxId: result.sandboxId,
+        sandboxUrl: result.previewUrl.split("?")[0],
+        snapshotId: snapshotId != null ? snapshotId : null,
+        previousSandboxId: build.sandboxId
+      }).then((saved) => {
+        if (!saved.ok) console.warn("[WebBuilderNative] persist sandbox identity failed", saved.message);
+      }).catch(() => void 0);
+    }
+    return {
+      ok: true
+    };
   }, [activeChannelId, sandboxCall, noteActivity, sandboxApi, rememberRestored, activeProject, upsertProject]);
   const redoFor = useRef(/* @__PURE__ */ new Map());
   const undo = useCallback(async (build) => {
@@ -714,6 +827,144 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
     const result = await sandboxCall(WEBBUILDER_EXPORT_ZIP_ACTION, exportZipParams(build.sandboxId, build.title), 10 * 60 * 1e3).catch(() => null);
     return exportZipUrl(result);
   }, [sandboxCall, noteActivity]);
+  const apollo = useApolloClient();
+  const publishTools = useMemo(() => ({
+    run: (build, action, params, timeoutMs = 6e4) => sandboxCall(action, __spreadValues(__spreadValues({}, build.sandboxId ? {
+      sandbox_id: build.sandboxId
+    } : {}), params), timeoutMs).catch((err) => ({
+      successful: false,
+      error: err instanceof Error ? err.message : String(err)
+    })),
+    review: async (build) => {
+      var _a2, _b2;
+      if (!activeChannelId || !build.sandboxId) return null;
+      const result = await apollo.query({
+        query: PRE_PUBLISH_REVIEW_QUERY,
+        variables: {
+          channelId: activeChannelId,
+          sandboxId: build.sandboxId
+        },
+        fetchPolicy: "network-only"
+      });
+      return (_b2 = (_a2 = result.data) == null ? void 0 : _a2.aiCodePrePublishReview) != null ? _b2 : null;
+    }
+  }), [sandboxCall, apollo, activeChannelId]);
+  const projectTools = useMemo(() => ({
+    loadSettings: async (channelId) => {
+      var _a2, _b2;
+      const result = await apollo.query({
+        query: PROJECT_SETTINGS_QUERY,
+        variables: {
+          channelId
+        },
+        fetchPolicy: "network-only"
+      });
+      const raw = (_b2 = (_a2 = result.data) == null ? void 0 : _a2.aiCodeProjectSettings) != null ? _b2 : null;
+      const settings = parseProjectSettings(raw);
+      if (settings) instructionsByChannel.current.set(channelId, settings.instructions);
+      return raw;
+    },
+    updateSettings: async (channelId, patch) => {
+      var _a2;
+      const result = await apollo.mutate({
+        mutation: UPDATE_PROJECT_SETTINGS_MUTATION,
+        variables: {
+          channelId,
+          input: patch
+        }
+      });
+      const raw = (_a2 = result.data) == null ? void 0 : _a2.aiCodeUpdateProjectSettings;
+      if (readSettingsUpdateReply(raw).ok) {
+        if (typeof patch.instructions === "string") {
+          instructionsByChannel.current.set(channelId, patch.instructions);
+        }
+        const row = projects.find((p) => p.channelId === channelId);
+        if (typeof patch.title === "string" && row) upsertProject(__spreadProps(__spreadValues({}, row), {
+          title: patch.title
+        }));
+      }
+      return raw;
+    },
+    deleteProject: async (channelId) => {
+      var _a2, _b2;
+      try {
+        const result2 = await apollo.mutate({
+          mutation: DELETE_PROJECT_MUTATION,
+          variables: {
+            channelId
+          }
+        });
+        const reply = readDeleteReply((_a2 = result2.data) == null ? void 0 : _a2.aiCodeDeleteProject);
+        if (reply.ok) instructionsByChannel.current.delete(channelId);
+        return reply;
+      } catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        if (!OUTDATED_SERVER_RE.test(message)) return {
+          ok: false,
+          error: message
+        };
+      }
+      const result = await apollo.mutate({
+        mutation: DELETE_CHANNEL_MUTATION,
+        variables: {
+          id: channelId
+        }
+      });
+      const done = ((_b2 = result.data) == null ? void 0 : _b2.deleteChannel) !== false;
+      return done ? {
+        ok: true
+      } : {
+        ok: false,
+        error: "The project could not be deleted."
+      };
+    },
+    pushToGithub: async (build, repoName) => {
+      var _a2;
+      if (!build.sandboxId) return {
+        ok: false,
+        message: "Wake the preview to push to GitHub."
+      };
+      noteActivity();
+      try {
+        const result = await apollo.mutate({
+          mutation: PUSH_TO_GITHUB_MUTATION,
+          variables: {
+            sandboxId: build.sandboxId,
+            repoName,
+            channelId: activeChannelId
+          }
+        });
+        const payload = (_a2 = result.data) == null ? void 0 : _a2.aiCodePushToGithub;
+        return (payload == null ? void 0 : payload.success) ? {
+          ok: true,
+          message: payload.message || "Push started. Follow along in the chat."
+        } : {
+          ok: false,
+          message: "The push didn't start. Try again in a moment."
+        };
+      } catch (e) {
+        return {
+          ok: false,
+          message: "The push didn't start. Check your connection and try again."
+        };
+      }
+    },
+    integrations: async (search) => {
+      var _a2, _b2;
+      const result = await apollo.query({
+        query: INTEGRATION_CATALOG_QUERY,
+        variables: search ? {
+          search
+        } : {},
+        fetchPolicy: "network-only"
+      });
+      return (_b2 = (_a2 = result.data) == null ? void 0 : _a2.integrationConnectors) != null ? _b2 : [];
+    }
+  }), [apollo, projects, upsertProject, noteActivity, activeChannelId]);
+  useEffect(() => {
+    if (!activeChannelId) return;
+    projectTools.loadSettings(activeChannelId).catch(() => void 0);
+  }, [activeChannelId]);
   const codeApi = useMemo(() => createCodeWorkspace(sandboxCall, (edit) => {
     handEditsRef.current = [...handEditsRef.current.filter((e) => e.file !== edit.file), edit];
   }), [sandboxCall]);
@@ -735,6 +986,7 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
     };
   }, [activeChannelId, readBackupRepo, newestBuildAt]);
   const siteBackupRepo = (backupRepo == null ? void 0 : backupRepo.channelId) === activeChannelId ? backupRepo.repo : null;
+  backupRepoRef.current = siteBackupRepo;
   const code = useMemo(() => {
     const target = (build) => ({
       sandboxId: build.sandboxId,
@@ -762,7 +1014,7 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
     };
   }, [codeApi, siteLiveUrl, siteBackupRepo]);
   const [deployToCloudflare] = useAiCodeDeployToCloudflareMutation({});
-  const publish = useCallback(async (build, projectName) => {
+  const publish = useCallback(async (build, projectName, customDomain) => {
     var _a2;
     if (!build.sandboxId || !activeChannelId) {
       return {
@@ -771,18 +1023,24 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
       };
     }
     if (build.origin === "agent") {
-      sendAgentTurn(activeChannelId, agentPublishMessage(build, projectName));
+      sendAgentTurn(activeChannelId, agentPublishMessage(build, projectName, customDomain));
       return {
         ok: true
       };
     }
     try {
-      const result = await deployToCloudflare({
-        variables: {
-          sandboxId: build.sandboxId,
-          projectName,
-          channelId: activeChannelId
-        }
+      const variables = {
+        sandboxId: build.sandboxId,
+        projectName,
+        channelId: activeChannelId
+      };
+      const result = customDomain ? await apollo.mutate({
+        mutation: DEPLOY_WITH_DOMAIN_MUTATION,
+        variables: __spreadProps(__spreadValues({}, variables), {
+          customDomain
+        })
+      }) : await deployToCloudflare({
+        variables
       });
       const payload = (_a2 = result.data) == null ? void 0 : _a2.aiCodeDeployToCloudflare;
       if (payload == null ? void 0 : payload.success) return {
@@ -800,7 +1058,7 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
         message: "Publishing didn't start. Check your connection and try again."
       };
     }
-  }, [deployToCloudflare, activeChannelId, sendAgentTurn]);
+  }, [deployToCloudflare, activeChannelId, sendAgentTurn, apollo]);
   let startBlocked = null;
   if (!token && !cdecli.channelConnected) {
     startBlocked = session.ready && session.error ? session.error : "Connecting to Web Builder\u2026";
@@ -812,7 +1070,7 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
     projectsHasMore: gallery.hasMore,
     refreshProjects,
     loadMoreProjects: gallery.loadMore,
-    templates: catalog.templates,
+    templates: galleryTemplates,
     templatesLoading: catalog.loading,
     templatesError: catalog.error,
     reloadTemplates,
@@ -829,9 +1087,13 @@ ${agentEditNote(target)}` : ""}${editsNote}`;
     hostChat,
     publish,
     resumePreview,
+    restorePreview,
+    goneSandboxIds,
     undo,
     redo,
     downloadCode,
+    publishTools,
+    projectTools,
     code
   };
 }export{useWebBuilderHost};//# sourceMappingURL=useWebBuilderHost.js.map