30 lines
1.2 KiB
TypeScript
30 lines
1.2 KiB
TypeScript
|
import { create } from 'zustand';
|
||
|
|
|
||
|
|
// Narrow store for the "next message starts a goal" flag. Armed by the
|
||
|
|
// composer target button (works for existing sessions AND session drafts)
|
||
|
|
// and by the run-as-goal flows (fork dialog, plan send); consumed by
|
||
|
|
// sendMessage in session-ui-store, which turns the sent prompt into the
|
||
|
|
// goal objective — unless the arming flow supplied a richer objective
|
||
|
|
// override (e.g. the plan content instead of "Implement this plan: X").
|
||
|
|
interface SessionGoalArmStore {
|
||
|
|
armed: boolean;
|
||
|
|
objectiveOverride: string | null;
|
||
|
|
setArmed: (armed: boolean, objectiveOverride?: string | null) => void;
|
||
|
|
/** Read-and-clear in one step at send time. */
|
||
|
|
consume: () => { armed: boolean; objectiveOverride: string | null };
|
||
|
|
}
|
||
|
|
|
||
|
|
export const useSessionGoalArmStore = create<SessionGoalArmStore>((set, get) => ({
|
||
|
|
armed: false,
|
||
|
|
objectiveOverride: null,
|
||
|
|
setArmed: (armed, objectiveOverride = null) => set({
|
||
|
|
armed,
|
||
|
|
objectiveOverride: armed ? objectiveOverride : null,
|
||
|
|
}),
|
||
|
|
consume: () => {
|
||
|
|
const { armed, objectiveOverride } = get();
|
||
|
|
if (armed) set({ armed: false, objectiveOverride: null });
|
||
|
|
return { armed, objectiveOverride };
|
||
|
|
},
|
||
|
|
}));
|