Merge branch 'main' of https://github.com/trycompai/comp into chas/show-domain-status

Author: chasprowebdev

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/actions/task-automation-actions.ts

@@ -193,7 +193,7 @@ export async function executeAutomationScript(data: {
       body: data,
     });
 
-    await revalidateCurrentPath();
+    // Don't revalidate - causes page refresh. Test results are handled via polling/state.
     return { success: true, data: result };
   } catch (error) {
     const typedError = error as EnterpriseApiError;

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/chat.tsx

@@ -31,23 +31,29 @@ interface Props {
 
 export function Chat({ className, orgId, taskId, taskName, automationId }: Props) {
   const [input, setInput] = useState('');
-  const { chat, updateAutomationId } = useSharedChatContext();
+  const { chat, updateAutomationId, automationIdRef } = useSharedChatContext();
   const { messages, sendMessage, status } = useChat<ChatUIMessage>({
     chat,
   });
   const { setChatStatus, scriptUrl } = useTaskAutomationStore();
   const inputRef = useRef<HTMLInputElement | null>(null);
   const { automation } = useTaskAutomation();
 
+  // Update shared ref when automation is loaded from hook
+  if (automation?.id && automationIdRef.current === 'new') {
+    automationIdRef.current = automation.id;
+  }
+
   // Ephemeral mode - automation not created yet
-  const isEphemeral = automationId === 'new';
+  // Check the shared ref, not the URL param
+  const isEphemeral = automationIdRef.current === 'new';
 
   const { validateAndSubmitMessage, handleSecretAdded, handleInfoProvided } = useChatHandlers({
     sendMessage,
     setInput,
     orgId,
     taskId,
-    automationId,
+    automationId: automationIdRef.current,
     isEphemeral,
     updateAutomationId,
   });
@@ -85,7 +91,7 @@ export function Chat({ className, orgId, taskId, taskName, automationId }: Props
             orgId={orgId}
             taskId={taskId}
             taskName={taskName}
-            automationId={automationId}
+            automationId={automationIdRef.current}
             automationName={automation?.name}
             isEphemeral={isEphemeral}
           />

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/components/AutomationSettingsDialogs.tsx

@@ -21,25 +21,32 @@ import { useTaskAutomation } from '../hooks/use-task-automation';
 interface EditNameDialogProps {
   open: boolean;
   onOpenChange: (open: boolean) => void;
+  onSuccess?: () => void;
 }
 
 interface EditDescriptionDialogProps {
   open: boolean;
   onOpenChange: (open: boolean) => void;
+  onSuccess?: () => void;
 }
 
 interface DeleteDialogProps {
   open: boolean;
   onOpenChange: (open: boolean) => void;
+  onSuccess?: () => void;
 }
 
-export function EditNameDialog({ open, onOpenChange }: EditNameDialogProps) {
-  const { automation, mutate } = useTaskAutomation();
+export function EditNameDialog({ open, onOpenChange, onSuccess }: EditNameDialogProps) {
+  const { automation, mutate: mutateLocal } = useTaskAutomation();
   const { orgId, taskId, automationId } = useParams<{
     orgId: string;
     taskId: string;
     automationId: string;
   }>();
+
+  // Use real automation ID when available
+  const realAutomationId = automation?.id || automationId;
+
   const [name, setName] = useState(automation?.name || '');
   const [isSaving, setIsSaving] = useState(false);
 
@@ -57,7 +64,7 @@ export function EditNameDialog({ open, onOpenChange }: EditNameDialogProps) {
     setIsSaving(true);
     try {
       const response = await api.patch(
-        `/v1/tasks/${taskId}/automations/${automationId}`,
+        `/v1/tasks/${taskId}/automations/${realAutomationId}`,
         { name: name.trim() },
         orgId,
       );
@@ -66,7 +73,8 @@ export function EditNameDialog({ open, onOpenChange }: EditNameDialogProps) {
         throw new Error(response.error);
       }
 
-      await mutate(); // Refresh automation data
+      await mutateLocal(); // Refresh automation data in hook
+      await onSuccess?.(); // Notify parent to refresh (e.g., overview page)
       onOpenChange(false);
       toast.success('Automation name updated');
     } catch (error) {
@@ -111,8 +119,12 @@ export function EditNameDialog({ open, onOpenChange }: EditNameDialogProps) {
   );
 }
 
-export function EditDescriptionDialog({ open, onOpenChange }: EditDescriptionDialogProps) {
-  const { automation, mutate } = useTaskAutomation();
+export function EditDescriptionDialog({
+  open,
+  onOpenChange,
+  onSuccess,
+}: EditDescriptionDialogProps) {
+  const { automation, mutate: mutateLocal } = useTaskAutomation();
   const { orgId, taskId, automationId } = useParams<{
     orgId: string;
     taskId: string;
@@ -139,7 +151,8 @@ export function EditDescriptionDialog({ open, onOpenChange }: EditDescriptionDia
         throw new Error(response.error);
       }
 
-      await mutate(); // Refresh automation data
+      await mutateLocal(); // Refresh automation data in hook
+      await onSuccess?.(); // Notify parent to refresh (e.g., overview page)
       onOpenChange(false);
       toast.success('Automation description updated');
     } catch (error) {
@@ -185,7 +198,7 @@ export function EditDescriptionDialog({ open, onOpenChange }: EditDescriptionDia
   );
 }
 
-export function DeleteAutomationDialog({ open, onOpenChange }: DeleteDialogProps) {
+export function DeleteAutomationDialog({ open, onOpenChange, onSuccess }: DeleteDialogProps) {
   const { automation } = useTaskAutomation();
   const { orgId, taskId, automationId } = useParams<{
     orgId: string;

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/components/PublishDialog.tsx

@@ -17,18 +17,20 @@ import { useState } from 'react';
 import { toast } from 'sonner';
 import { publishAutomation } from '../actions/task-automation-actions';
 import { useAutomationVersions } from '../hooks/use-automation-versions';
+import { useSharedChatContext } from '../lib/chat-context';
 
 interface PublishDialogProps {
   open: boolean;
   onOpenChange: (open: boolean) => void;
 }
 
 export function PublishDialog({ open, onOpenChange }: PublishDialogProps) {
-  const { orgId, taskId, automationId } = useParams<{
+  const { orgId, taskId } = useParams<{
     orgId: string;
     taskId: string;
     automationId: string;
   }>();
+  const { automationIdRef } = useSharedChatContext();
   const [changelog, setChangelog] = useState('');
   const [isPublishing, setIsPublishing] = useState(false);
   const { mutate } = useAutomationVersions();
@@ -37,7 +39,12 @@ export function PublishDialog({ open, onOpenChange }: PublishDialogProps) {
     setIsPublishing(true);
 
     try {
-      const result = await publishAutomation(orgId, taskId, automationId, changelog || undefined);
+      const result = await publishAutomation(
+        orgId,
+        taskId,
+        automationIdRef.current,
+        changelog || undefined,
+      );
 
       if (!result.success) {
         throw new Error(result.error || 'Failed to publish');

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/components/chat/EmptyState.tsx

@@ -20,15 +20,15 @@ export function EmptyState({
 }: EmptyStateProps) {
   return (
     <div className="flex-1 min-h-0 overflow-y-auto h-full z-20">
-      <div className="w-full h-full flex flex-col items-center py-48">
+      <div className="w-full h-full flex flex-col items-center py-48 px-4">
         <div className="w-full max-w-3xl text-center space-y-8 mb-16">
           <p className="text-2xl font-medium text-primary tracking-wide z-20">
             What evidence do you want to collect?
           </p>
           <Input
             ref={inputRef}
             placeholder="Check if GitHub dependabot is enabled and tell me the result"
-            className="w-full max-w-3xl transition-all duration-200 hover:shadow-md hover:shadow-primary/5 hover:scale-[1.01] focus:shadow-lg focus:shadow-primary/10 focus:scale-[1.02] focus:ring-2 focus:ring-primary/30"
+            className="w-full max-w-3xl transition-all duration-200 hover:shadow-md hover:shadow-primary/5 focus:shadow-lg focus:shadow-primary/10 focus:ring-2 focus:ring-primary/30"
             value={input}
             onChange={(e) => onInputChange(e.target.value)}
             disabled={status === 'streaming' || status === 'submitted'}

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/components/chat/message-part/reasoning.tsx

@@ -61,7 +61,11 @@ export function Reasoning({ part, partIndex }: { part: ReasoningUIPart; partInde
         className="text-left w-full flex items-center gap-2.5 transition-all duration-200 group/btn"
         onClick={handleClick}
       >
-        <span className="text-xs text-muted-foreground truncate block">{getReasoningLabel()}</span>
+        <span
+          className={`text-xs truncate block ${isStreaming ? 'thinking-text' : 'text-muted-foreground'}`}
+        >
+          {getReasoningLabel()}
+        </span>
       </button>
     </div>
   );

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/components/chat/message.tsx

@@ -113,7 +113,7 @@ export const Message = memo(function Message({
         if (thisSessionIsThinking) {
           result.push(
             <div key={`thinking-${sessionId}-active`} className="flex items-center gap-2 py-1">
-              <span className="text-xs text-muted-foreground">Thinking...</span>
+              <span className="text-xs thinking-text">Thinking...</span>
             </div>,
           );
         } else {
@@ -191,7 +191,7 @@ export const Message = memo(function Message({
       if (finalThinkingIsActive) {
         result.push(
           <div key={`thinking-${sessionId}-active`} className="flex items-center gap-2 py-1">
-            <span className="text-xs text-muted-foreground">Thinking...</span>
+            <span className="text-xs thinking-text">Thinking...</span>
           </div>,
         );
       } else {

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/components/workflow/components/UnifiedWorkflowCard.tsx

@@ -143,7 +143,12 @@ export function UnifiedWorkflowCard({ steps, title, onTest, integrationsUsed }:
           </div>
         ) : (
           <button
-            onClick={onTest}
+            type="button"
+            onClick={(e) => {
+              e.preventDefault();
+              e.stopPropagation();
+              onTest?.();
+            }}
             className="w-full flex items-center justify-center gap-2 px-4 py-2 rounded-lg bg-primary text-primary-foreground font-medium hover:bg-primary/90 transition-colors animate-in fade-in duration-500"
           >
             <Play className="w-4 h-4" />

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/components/workflow/workflow-visualizer-simple.tsx

@@ -26,6 +26,7 @@ import { useEffect, useMemo, useState } from 'react';
 import { toast } from 'sonner';
 import { restoreVersion } from '../../actions/task-automation-actions';
 import {
+  useTaskAutomation,
   useTaskAutomationAnalyze,
   useTaskAutomationExecution,
   useTaskAutomationScript,
@@ -57,29 +58,35 @@ export function WorkflowVisualizerSimple({ className }: Props) {
     taskId: string;
     automationId: string;
   }>();
-  const { chat } = useSharedChatContext();
+  const { chat, automationIdRef } = useSharedChatContext();
   const { sendMessage } = useChat<ChatUIMessage>({ chat });
   const [publishDialogOpen, setPublishDialogOpen] = useState(false);
   const [isRestoring, setIsRestoring] = useState(false);
   const [confirmRestore, setConfirmRestore] = useState<EvidenceAutomationVersion | null>(null);
+  const { automation } = useTaskAutomation();
   const { versions } = useAutomationVersions();
 
+  // Update shared ref when automation is loaded from hook
+  if (automation?.id && automationIdRef.current === 'new') {
+    automationIdRef.current = automation.id;
+  }
+
   const {
     script,
     isLoading: isLoadingScript,
     refresh,
   } = useTaskAutomationScript({
     orgId: orgId,
     taskId: taskId,
-    automationId: automationId,
-    enabled: !!orgId && !!taskId && !!automationId,
+    automationId: automationIdRef.current,
+    enabled: !!orgId && !!taskId && automationIdRef.current !== 'new',
   });
 
   const handleRestoreVersion = async (version: EvidenceAutomationVersion) => {
     setIsRestoring(true);
 
     try {
-      const result = await restoreVersion(orgId, taskId, automationId, version.version);
+      const result = await restoreVersion(orgId, taskId, automationIdRef.current, version.version);
 
       if (!result.success) {
         throw new Error(result.error || 'Failed to restore version');
@@ -155,7 +162,10 @@ export function WorkflowVisualizerSimple({ className }: Props) {
   }, [executionResult, executionError]);
 
   const handleTest = async () => {
-    if (!orgId || !taskId) return;
+    if (!orgId || !taskId || automationIdRef.current === 'new') {
+      console.warn('Cannot test ephemeral automation');
+      return;
+    }
     await execute();
   };
 

apps/app/src/app/(app)/[orgId]/tasks/[taskId]/automation/[automationId]/hooks/index.ts

@@ -1,3 +1,4 @@
+export { useTaskAutomation } from './use-task-automation';
 export { useTaskAutomationAnalyze } from './use-task-automation-analyze';
 export { useTaskAutomationExecution } from './use-task-automation-execution';
 export { useTaskAutomationScript } from './use-task-automation-script';