1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
import { useState, useEffect } from 'react';
import { useStdin } from 'ink';
export interface PipedInputState {
data: string | null; // Use null initially to distinguish from empty string
isLoading: boolean;
error: string | null;
isPiped: boolean; // Flag to indicate if input was piped
}
export function usePipedInput(): PipedInputState {
const { stdin, setRawMode, isRawModeSupported } = useStdin();
// Keep exit available if needed, e.g., for error handling, but maybe let consumer handle it
// const { exit } = useApp();
const [pipedData, setPipedData] = useState<string | null>(null);
const [isLoading, setIsLoading] = useState<boolean>(true); // Assume loading until checked
const [error, setError] = useState<string | null>(null);
const [isPiped, setIsPiped] = useState<boolean>(false);
useEffect(() => {
// Determine if input is piped ONLY ONCE
const checkIsPiped = !stdin || !stdin.isTTY;
setIsPiped(checkIsPiped);
if (checkIsPiped) {
// Piped input detected
if (isRawModeSupported) {
setRawMode(false); // Ensure raw mode is off for stream reading
}
// Ensure stdin is available (it should be if !isTTY)
if (!stdin) {
setError('Stdin stream is unavailable.');
setIsLoading(false);
return; // Cannot proceed
}
let data = '';
const handleData = (chunk: Buffer) => {
data += chunk.toString();
};
const handleError = (err: Error) => {
setError('Error reading from stdin: ' + err.message);
setIsLoading(false);
// Decide if the hook should trigger exit or just report the error
// exit();
};
const handleEnd = () => {
setPipedData(data);
setIsLoading(false);
// Don't exit here, let the component using the hook decide
};
stdin.on('data', handleData);
stdin.on('error', handleError);
stdin.on('end', handleEnd);
// Cleanup listeners
return () => {
stdin.removeListener('data', handleData);
stdin.removeListener('error', handleError);
stdin.removeListener('end', handleEnd);
};
} else {
// No piped input (running interactively)
setIsLoading(false);
// Optionally set an 'info' state or just let isLoading=false & isPiped=false suffice
// setError('No piped input detected.'); // Maybe don't treat this as an 'error'
}
// Intentionally run only once on mount or when stdin theoretically changes
}, [stdin, isRawModeSupported, setRawMode /*, exit */]);
return { data: pipedData, isLoading, error, isPiped };
}
|
