claude-code/src/components/TeleportStash.tsx
2026-03-31 23:03:47 +08:00

116 lines
3.7 KiB
TypeScript

import figures from 'figures';
import React, { useEffect, useState } from 'react';
import { Box, Text } from '../ink.js';
import { logForDebugging } from '../utils/debug.js';
import type { GitFileStatus } from '../utils/git.js';
import { getFileStatus, stashToCleanState } from '../utils/git.js';
import { Select } from './CustomSelect/index.js';
import { Dialog } from './design-system/Dialog.js';
import { Spinner } from './Spinner.js';
type TeleportStashProps = {
onStashAndContinue: () => void;
onCancel: () => void;
};
export function TeleportStash({
onStashAndContinue,
onCancel
}: TeleportStashProps): React.ReactNode {
const [gitFileStatus, setGitFileStatus] = useState<GitFileStatus | null>(null);
const changedFiles = gitFileStatus !== null ? [...gitFileStatus.tracked, ...gitFileStatus.untracked] : [];
const [loading, setLoading] = useState(true);
const [stashing, setStashing] = useState(false);
const [error, setError] = useState<string | null>(null);
// Load changed files on mount
useEffect(() => {
const loadChangedFiles = async () => {
try {
const fileStatus = await getFileStatus();
setGitFileStatus(fileStatus);
} catch (err) {
const errorMessage = err instanceof Error ? err.message : String(err);
logForDebugging(`Error getting changed files: ${errorMessage}`, {
level: 'error'
});
setError('Failed to get changed files');
} finally {
setLoading(false);
}
};
void loadChangedFiles();
}, []);
const handleStash = async () => {
setStashing(true);
try {
logForDebugging('Stashing changes before teleport...');
const success = await stashToCleanState('Teleport auto-stash');
if (success) {
logForDebugging('Successfully stashed changes');
onStashAndContinue();
} else {
setError('Failed to stash changes');
}
} catch (err_0) {
const errorMessage_0 = err_0 instanceof Error ? err_0.message : String(err_0);
logForDebugging(`Error stashing changes: ${errorMessage_0}`, {
level: 'error'
});
setError('Failed to stash changes');
} finally {
setStashing(false);
}
};
const handleSelectChange = (value: string) => {
if (value === 'stash') {
void handleStash();
} else {
onCancel();
}
};
if (loading) {
return <Box flexDirection="column" padding={1}>
<Box marginBottom={1}>
<Spinner />
<Text> Checking git status{figures.ellipsis}</Text>
</Box>
</Box>;
}
if (error) {
return <Box flexDirection="column" padding={1}>
<Text bold color="error">
Error: {error}
</Text>
<Box marginTop={1}>
<Text dimColor>Press </Text>
<Text bold>Escape</Text>
<Text dimColor> to cancel</Text>
</Box>
</Box>;
}
const showFileCount = changedFiles.length > 8;
return <Dialog title="Working Directory Has Changes" onCancel={onCancel}>
<Text>
Teleport will switch git branches. The following changes were found:
</Text>
<Box flexDirection="column" paddingLeft={2}>
{changedFiles.length > 0 ? showFileCount ? <Text>{changedFiles.length} files changed</Text> : changedFiles.map((file: string, index: number) => <Text key={index}>{file}</Text>) : <Text dimColor>No changes detected</Text>}
</Box>
<Text>
Would you like to stash these changes and continue with teleport?
</Text>
{stashing ? <Box>
<Spinner />
<Text> Stashing changes...</Text>
</Box> : <Select options={[{
label: 'Stash changes and continue',
value: 'stash'
}, {
label: 'Exit',
value: 'exit'
}]} onChange={handleSelectChange} />}
</Dialog>;
}