claude-code/src/components/agents/new-agent-creation/wizard-steps/ConfirmStepWrapper.tsx

import chalk from 'chalk';
import { type ReactNode, useCallback, useState } from 'react';
import {
  type AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
  logEvent,
} from 'src/services/analytics/index.js';
import { useSetAppState } from 'src/state/AppState.js';
import type { Tools } from '../../../../Tool.js';
import type { AgentDefinition } from '@claude-code-best/builtin-tools/tools/AgentTool/loadAgentsDir.js';
import { getActiveAgentsFromList } from '@claude-code-best/builtin-tools/tools/AgentTool/loadAgentsDir.js';
import { clearAgentDefinitionsCache } from '@claude-code-best/builtin-tools/tools/AgentTool/loadAgentsDir.js';
import { editFileInEditor } from '../../../../utils/promptEditor.js';
import { useWizard } from '../../../wizard/index.js';
import { getNewAgentFilePath, saveAgentToFile } from '../../agentFileUtils.js';
import type { AgentWizardData } from '../types.js';
import { ConfirmStep } from './ConfirmStep.js';

type Props = {
  tools: Tools;
  existingAgents: AgentDefinition[];
  onComplete: (message: string) => void;
};

export function ConfirmStepWrapper({ tools, existingAgents, onComplete }: Props): ReactNode {
  const { wizardData } = useWizard<AgentWizardData>();
  const [saveError, setSaveError] = useState<string | null>(null);
  const setAppState = useSetAppState();

  const saveAgent = useCallback(
    async (openInEditor: boolean): Promise<void> => {
      if (!wizardData?.finalAgent) return;

      try {
        await saveAgentToFile(
          wizardData.location!,
          wizardData.finalAgent.agentType,
          wizardData.finalAgent.whenToUse,
          wizardData.finalAgent.tools,
          wizardData.finalAgent.getSystemPrompt(),
          true,
          wizardData.finalAgent.color,
          wizardData.finalAgent.model,
          wizardData.finalAgent.memory,
        );

        setAppState(state => {
          if (!wizardData.finalAgent) return state;

          const allAgents = state.agentDefinitions.allAgents.concat(wizardData.finalAgent);
          return {
            ...state,
            agentDefinitions: {
              ...state.agentDefinitions,
              activeAgents: getActiveAgentsFromList(allAgents),
              allAgents,
            },
          };
        });

        clearAgentDefinitionsCache();

        if (openInEditor) {
          const filePath = getNewAgentFilePath({
            source: wizardData.location!,
            agentType: wizardData.finalAgent.agentType,
          });
          await editFileInEditor(filePath);
        }

        logEvent('tengu_agent_created', {
          agent_type: wizardData.finalAgent.agentType,
          generation_method: wizardData.wasGenerated ? 'generated' : 'manual',
          source: wizardData.location!,
          tool_count: wizardData.finalAgent.tools?.length ?? 'all',
          has_custom_model: !!wizardData.finalAgent.model,
          has_custom_color: !!wizardData.finalAgent.color,
          has_memory: !!wizardData.finalAgent.memory,
          memory_scope: wizardData.finalAgent.memory ?? 'none',
          ...(openInEditor ? { opened_in_editor: true } : {}),
        } as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS);

        const message = openInEditor
          ? `Created agent: ${chalk.bold(wizardData.finalAgent.agentType)} and opened in editor. ` +
            `If you made edits, restart to load the latest version.`
          : `Created agent: ${chalk.bold(wizardData.finalAgent.agentType)}`;
        onComplete(message);
      } catch (err) {
        setSaveError(err instanceof Error ? err.message : 'Failed to save agent');
      }
    },
    [wizardData, onComplete, setAppState],
  );

  const handleSave = useCallback(() => saveAgent(false), [saveAgent]);

  const handleSaveAndEdit = useCallback(() => saveAgent(true), [saveAgent]);

  return (
    <ConfirmStep
      tools={tools}
      existingAgents={existingAgents}
      onSave={handleSave}
      onSaveAndEdit={handleSaveAndEdit}
      error={saveError}
    />
  );
}