UI.tsx

import * as React from 'react';
import { Box, Text } from '@anthropic/ink';
import type { ToolProgressData } from 'src/Tool.js';
import type { ProgressMessage } from 'src/types/message.js';
import type { ThemeName } from 'src/utils/theme.js';
import type { Output } from './EnterWorktreeTool.js';

export function renderToolUseMessage(): React.ReactNode {
  return 'Creating worktree…';
}

export function renderToolResultMessage(
  output: Output,
  _progressMessagesForMessage: ProgressMessage<ToolProgressData>[],
  _options: { theme: ThemeName },
): React.ReactNode {
  if (!output) return null;
  return (
    <Box flexDirection="column">
      <Text>
        Switched to worktree
        {output.worktreeBranch ? (
          <>
            {' '}
            on branch <Text bold>{output.worktreeBranch}</Text>
          </>
        ) : null}
      </Text>
      <Text dimColor>{output.worktreePath}</Text>
    </Box>
  );
}