4ca5fa500b
Add an in-memory DraftContext that persists textarea content when the AgentChat component unmounts due to route navigation. The draft is restored when the user returns to the chat view. The store is session-scoped (not localStorage) so drafts are cleared on page reload. Closes #3129
46 lines
1.2 KiB
TypeScript
46 lines
1.2 KiB
TypeScript
import { createContext, useContext, useCallback, useRef } from 'react';
|
|
|
|
/**
|
|
* In-memory draft store that survives component unmounts but not page reloads.
|
|
* Keyed by an arbitrary string (e.g. route path or conversation id).
|
|
*/
|
|
|
|
export interface DraftContextType {
|
|
getDraft: (key: string) => string;
|
|
setDraft: (key: string, value: string) => void;
|
|
clearDraft: (key: string) => void;
|
|
}
|
|
|
|
export const DraftContext = createContext<DraftContextType>({
|
|
getDraft: () => '',
|
|
setDraft: () => {},
|
|
clearDraft: () => {},
|
|
});
|
|
|
|
export function useDraftStore(): DraftContextType {
|
|
const store = useRef<Map<string, string>>(new Map());
|
|
|
|
const getDraft = useCallback((key: string): string => {
|
|
return store.current.get(key) ?? '';
|
|
}, []);
|
|
|
|
const setDraft = useCallback((key: string, value: string): void => {
|
|
store.current.set(key, value);
|
|
}, []);
|
|
|
|
const clearDraft = useCallback((key: string): void => {
|
|
store.current.delete(key);
|
|
}, []);
|
|
|
|
return { getDraft, setDraft, clearDraft };
|
|
}
|
|
|
|
export function useDraft(key: string) {
|
|
const { getDraft, setDraft, clearDraft } = useContext(DraftContext);
|
|
return {
|
|
draft: getDraft(key),
|
|
saveDraft: (value: string) => setDraft(key, value),
|
|
clearDraft: () => clearDraft(key),
|
|
};
|
|
}
|