refactor: remove source/language/host steps, existing-endpoint is the only flow

Author: aidandaly24

src/cli/tui/screens/mcp/AddGatewayTargetScreen.tsx

@@ -4,14 +4,8 @@ import type { SelectableItem } from '../../components';
 import { HELP_TEXT } from '../../constants';
 import { useListNavigation } from '../../hooks';
 import { generateUniqueName } from '../../utils';
-import type { AddGatewayTargetConfig, ComputeHost, TargetLanguage } from './types';
-import {
-  COMPUTE_HOST_OPTIONS,
-  MCP_TOOL_STEP_LABELS,
-  SKIP_FOR_NOW,
-  SOURCE_OPTIONS,
-  TARGET_LANGUAGE_OPTIONS,
-} from './types';
+import type { AddGatewayTargetConfig } from './types';
+import { MCP_TOOL_STEP_LABELS, SKIP_FOR_NOW } from './types';
 import { useAddGatewayTargetWizard } from './useAddGatewayTargetWizard';
 import { Box, Text } from 'ink';
 import React, { useMemo } from 'react';
@@ -31,16 +25,6 @@ export function AddGatewayTargetScreen({
 }: AddGatewayTargetScreenProps) {
   const wizard = useAddGatewayTargetWizard(existingGateways);
 
-  const sourceItems: SelectableItem[] = useMemo(
-    () => SOURCE_OPTIONS.map(o => ({ id: o.id, title: o.title, description: o.description })),
-    []
-  );
-
-  const languageItems: SelectableItem[] = useMemo(
-    () => TARGET_LANGUAGE_OPTIONS.map(o => ({ id: o.id, title: o.title, description: o.description })),
-    []
-  );
-
   const gatewayItems: SelectableItem[] = useMemo(
     () => [
       ...existingGateways.map(g => ({ id: g, title: g })),
@@ -49,47 +33,18 @@ export function AddGatewayTargetScreen({
     [existingGateways]
   );
 
-  const hostItems: SelectableItem[] = useMemo(
-    () => COMPUTE_HOST_OPTIONS.map(o => ({ id: o.id, title: o.title, description: o.description })),
-    []
-  );
-
-  const isSourceStep = wizard.step === 'source';
-  const isLanguageStep = wizard.step === 'language';
   const isGatewayStep = wizard.step === 'gateway';
-  const isHostStep = wizard.step === 'host';
   const isTextStep = wizard.step === 'name' || wizard.step === 'endpoint';
   const isConfirmStep = wizard.step === 'confirm';
   const noGatewaysAvailable = isGatewayStep && existingGateways.length === 0;
 
-  const sourceNav = useListNavigation({
-    items: sourceItems,
-    onSelect: item => wizard.setSource(item.id as 'existing-endpoint' | 'create-new'),
-    onExit: () => wizard.goBack(),
-    isActive: isSourceStep,
-  });
-
-  const languageNav = useListNavigation({
-    items: languageItems,
-    onSelect: item => wizard.setLanguage(item.id as TargetLanguage),
-    onExit: () => onExit(),
-    isActive: isLanguageStep,
-  });
-
   const gatewayNav = useListNavigation({
     items: gatewayItems,
     onSelect: item => wizard.setGateway(item.id),
     onExit: () => wizard.goBack(),
     isActive: isGatewayStep && !noGatewaysAvailable,
   });
 
-  const hostNav = useListNavigation({
-    items: hostItems,
-    onSelect: item => wizard.setHost(item.id as ComputeHost),
-    onExit: () => wizard.goBack(),
-    isActive: isHostStep,
-  });
-
   useListNavigation({
     items: [{ id: 'confirm', title: 'Confirm' }],
     onSelect: () => onComplete(wizard.config),
@@ -108,19 +63,6 @@ export function AddGatewayTargetScreen({
   return (
     <Screen title="Add MCP Tool" onExit={onExit} helpText={helpText} headerContent={headerContent}>
       <Panel>
-        {isSourceStep && (
-          <WizardSelect
-            title="Select source"
-            description="How would you like to create this MCP tool?"
-            items={sourceItems}
-            selectedIndex={sourceNav.selectedIndex}
-          />
-        )}
-
-        {isLanguageStep && (
-          <WizardSelect title="Select language" items={languageItems} selectedIndex={languageNav.selectedIndex} />
-        )}
-
         {isGatewayStep && !noGatewaysAvailable && (
           <WizardSelect
             title="Select gateway"
@@ -132,15 +74,6 @@ export function AddGatewayTargetScreen({
 
         {noGatewaysAvailable && <NoGatewaysMessage />}
 
-        {isHostStep && (
-          <WizardSelect
-            title="Select compute host"
-            description="Where will this tool run?"
-            items={hostItems}
-            selectedIndex={hostNav.selectedIndex}
-          />
-        )}
-
         {isTextStep && (
           <TextInput
             key={wizard.step}

src/cli/tui/screens/mcp/useAddGatewayTargetWizard.ts

@@ -1,19 +1,15 @@
 import { APP_DIR, MCP_APP_SUBDIR } from '../../../../lib';
 import type { ToolDefinition } from '../../../../schema';
-import type { AddGatewayTargetConfig, AddGatewayTargetStep, ComputeHost, TargetLanguage } from './types';
+import type { AddGatewayTargetConfig, AddGatewayTargetStep } from './types';
 import { SKIP_FOR_NOW } from './types';
 import { useCallback, useMemo, useState } from 'react';
 
 /**
- * Dynamic steps based on source.
- * - Existing endpoint: name → source → endpoint → gateway → confirm
- * - Create new: name → source → language → gateway → confirm
+ * Steps for adding a gateway target (existing endpoint only).
+ * name → endpoint → gateway → confirm
  */
-function getSteps(source?: 'existing-endpoint' | 'create-new'): AddGatewayTargetStep[] {
-  if (source === 'existing-endpoint') {
-    return ['name', 'source', 'endpoint', 'gateway', 'confirm'];
-  }
-  return ['name', 'source', 'language', 'gateway', 'confirm'];
+function getSteps(): AddGatewayTargetStep[] {
+  return ['name', 'endpoint', 'gateway', 'confirm'];
 }
 
 function deriveToolDefinition(name: string): ToolDefinition {
@@ -29,6 +25,7 @@ function getDefaultConfig(): AddGatewayTargetConfig {
     name: '',
     description: '',
     sourcePath: '',
+    source: 'existing-endpoint',
     language: 'Python',
     host: 'Lambda',
     toolDefinition: deriveToolDefinition(''),
@@ -39,16 +36,15 @@ export function useAddGatewayTargetWizard(existingGateways: string[] = []) {
   const [config, setConfig] = useState<AddGatewayTargetConfig>(getDefaultConfig);
   const [step, setStep] = useState<AddGatewayTargetStep>('name');
 
-  const steps = useMemo(() => getSteps(config.source), [config.source]);
+  const steps = useMemo(() => getSteps(), []);
   const currentIndex = steps.indexOf(step);
 
   const goBack = useCallback(() => {
-    // Recalculate steps in case source changed
-    const currentSteps = getSteps(config.source);
+    const currentSteps = getSteps();
     const idx = currentSteps.indexOf(step);
     const prevStep = currentSteps[idx - 1];
     if (prevStep) setStep(prevStep);
-  }, [config.source, step]);
+  }, [step]);
 
   const setName = useCallback((name: string) => {
     setConfig(c => ({
@@ -58,19 +54,7 @@ export function useAddGatewayTargetWizard(existingGateways: string[] = []) {
       sourcePath: `${APP_DIR}/${MCP_APP_SUBDIR}/${name}`,
       toolDefinition: deriveToolDefinition(name),
     }));
-    setStep('source');
-  }, []);
-
-  const setSource = useCallback((source: 'existing-endpoint' | 'create-new') => {
-    setConfig(c => ({
-      ...c,
-      source,
-    }));
-    if (source === 'existing-endpoint') {
-      setStep('endpoint');
-    } else {
-      setStep('language');
-    }
+    setStep('endpoint');
   }, []);
 
   const setEndpoint = useCallback((endpoint: string) => {
@@ -81,14 +65,6 @@ export function useAddGatewayTargetWizard(existingGateways: string[] = []) {
     setStep('gateway');
   }, []);
 
-  const setLanguage = useCallback((language: TargetLanguage) => {
-    setConfig(c => ({
-      ...c,
-      language,
-    }));
-    setStep('gateway');
-  }, []);
-
   const setGateway = useCallback((gateway: string) => {
     setConfig(c => {
       const isSkipped = gateway === SKIP_FOR_NOW;
@@ -97,14 +73,6 @@ export function useAddGatewayTargetWizard(existingGateways: string[] = []) {
     });
   }, []);
 
-  const setHost = useCallback((host: ComputeHost) => {
-    setConfig(c => ({
-      ...c,
-      host,
-    }));
-    setStep('confirm');
-  }, []);
-
   const reset = useCallback(() => {
     setConfig(getDefaultConfig());
     setStep('name');
@@ -118,11 +86,8 @@ export function useAddGatewayTargetWizard(existingGateways: string[] = []) {
     existingGateways,
     goBack,
     setName,
-    setSource,
     setEndpoint,
-    setLanguage,
     setGateway,
-    setHost,
     reset,
   };
 }