Files
ALwrity/frontend/src/stores/workflowStore.ts
2026-03-10 17:01:36 +05:30

572 lines
21 KiB
TypeScript

import { create } from 'zustand';
import { persist } from 'zustand/middleware';
import {
TodayTask,
DailyWorkflow,
WorkflowProgress,
UserWorkflowPreferences,
NavigationState,
WorkflowStatus,
WorkflowError,
TodayWorkflowScheduleStatus
} from '../types/workflow';
import { taskWorkflowOrchestrator } from '../services/TaskWorkflowOrchestrator';
import { apiClient } from '../api/client';
const isServerWorkflowId = (workflowId: string) => workflowId.startsWith('daily-');
const normalizeDependencies = (dependencies: unknown): string[] => {
if (Array.isArray(dependencies)) {
return dependencies.map(dep => String(dep).trim()).filter(Boolean);
}
if (typeof dependencies === 'string') {
const raw = dependencies.trim();
if (!raw) return [];
try {
const parsed = JSON.parse(raw);
if (Array.isArray(parsed)) {
return parsed.map(dep => String(dep).trim()).filter(Boolean);
}
} catch {}
return [raw];
}
return [];
};
const normalizeServerWorkflow = (workflow: DailyWorkflow): DailyWorkflow => ({
...workflow,
tasks: Array.isArray(workflow.tasks)
? workflow.tasks.map(task => ({
...task,
dependencies: normalizeDependencies(task.dependencies),
}))
: [],
});
const toWorkflowError = (error: unknown, fallbackMessage: string): WorkflowError => {
if (error instanceof WorkflowError) return error;
const message = error instanceof Error ? error.message : fallbackMessage;
return new WorkflowError({
code: 'WORKFLOW_ERROR',
message,
timestamp: new Date(),
recoverable: false,
});
};
const computeProgressAndNavigation = (workflow: DailyWorkflow): { progress: WorkflowProgress; navigation: NavigationState } => {
const tasks = Array.isArray(workflow.tasks) ? workflow.tasks : [];
const totalTasks = tasks.length;
const completedTasks = tasks.filter(t => t.status === 'completed' || t.status === 'skipped').length;
const completionPercentage = totalTasks > 0 ? (completedTasks / totalTasks) * 100 : 0;
const currentIndex = (() => {
if (typeof workflow.currentTaskIndex === 'number' && workflow.currentTaskIndex >= 0 && workflow.currentTaskIndex < totalTasks) {
return workflow.currentTaskIndex;
}
const idx = tasks.findIndex(t => t.status !== 'completed' && t.status !== 'skipped');
return idx >= 0 ? idx : Math.max(0, totalTasks - 1);
})();
const currentTask = tasks[currentIndex] || null;
const nextTask = tasks.slice(currentIndex + 1).find(t => t.status !== 'completed' && t.status !== 'skipped') || null;
const previousTask = currentIndex > 0 ? tasks[currentIndex - 1] : null;
const estimatedTimeRemaining = tasks
.filter(t => t.status !== 'completed' && t.status !== 'skipped')
.reduce((sum, t) => sum + (t.estimatedTime || 0), 0);
return {
progress: {
completedTasks,
totalTasks,
completionPercentage,
currentTask: currentTask || undefined,
nextTask: nextTask || undefined,
estimatedTimeRemaining,
actualTimeSpent: workflow.actualTimeSpent || 0,
},
navigation: {
currentTask,
previousTask,
nextTask,
canGoBack: currentIndex > 0,
canGoForward: Boolean(nextTask),
}
};
};
interface WorkflowState {
// Current workflow state
currentWorkflow: DailyWorkflow | null;
workflowProgress: WorkflowProgress | null;
navigationState: NavigationState | null;
scheduleStatus: TodayWorkflowScheduleStatus | null;
// User preferences
userPreferences: UserWorkflowPreferences | null;
// UI state
isWorkflowModalOpen: boolean;
isLoading: boolean;
error: WorkflowError | null;
isDegradedMode: boolean;
degradedModeReason: string | null;
// Actions
loadTodayWorkflow: (date?: string) => Promise<void>;
refreshScheduleStatus: (date?: string) => Promise<void>;
generateDailyWorkflow: (userId: string, date?: string) => Promise<void>;
startWorkflow: (workflowId: string) => Promise<void>;
pauseWorkflow: (workflowId: string) => Promise<void>;
stopWorkflow: (workflowId: string) => Promise<void>;
completeTask: (taskId: string, completionData?: any) => Promise<void>;
skipTask: (taskId: string) => Promise<void>;
moveToNextTask: () => Promise<void>;
moveToPreviousTask: () => Promise<void>;
// UI actions
openWorkflowModal: () => void;
closeWorkflowModal: () => void;
setError: (error: WorkflowError | null) => void;
clearError: () => void;
// Preferences
updateUserPreferences: (preferences: Partial<UserWorkflowPreferences>) => void;
// Utility actions
refreshWorkflowProgress: () => void;
getCurrentTask: () => TodayTask | null;
getNextTask: () => TodayTask | null;
isWorkflowComplete: () => boolean;
getCompletionPercentage: () => number;
}
export const useWorkflowStore = create<WorkflowState>()(
persist(
(set, get) => ({
// Initial state
currentWorkflow: null,
workflowProgress: null,
navigationState: null,
scheduleStatus: null,
userPreferences: null,
isWorkflowModalOpen: false,
isLoading: false,
error: null,
isDegradedMode: false,
degradedModeReason: null,
loadTodayWorkflow: async (date?: string) => {
set({ isLoading: true, error: null });
try {
const resp = await apiClient.get('/api/today-workflow', { params: date ? { date } : {} });
const serverWorkflow = resp?.data?.data?.workflow as DailyWorkflow | undefined;
const planSummary = resp?.data?.data?.plan?.provenance_summary;
const scheduleStatus = resp?.data?.data?.schedule_status as TodayWorkflowScheduleStatus | undefined;
if (serverWorkflow && Array.isArray(serverWorkflow.tasks)) {
if (planSummary && !serverWorkflow.provenanceSummary) {
serverWorkflow.provenanceSummary = planSummary;
}
const normalizedWorkflow = normalizeServerWorkflow(serverWorkflow);
const derived = computeProgressAndNavigation(normalizedWorkflow);
set({
currentWorkflow: normalizedWorkflow,
workflowProgress: derived.progress,
navigationState: derived.navigation,
scheduleStatus: scheduleStatus || null,
isLoading: false,
isDegradedMode: false,
degradedModeReason: null,
});
return;
}
throw new WorkflowError({
code: 'WORKFLOW_SCHEMA_INVALID',
message: 'Server workflow response is missing a valid tasks array.',
timestamp: new Date(),
recoverable: false,
suggestedAction: 'Refresh and try again. If this persists, contact support.'
});
} catch (error: any) {
if (error?.response?.status === 404) {
set({
currentWorkflow: null,
workflowProgress: null,
navigationState: null,
isLoading: false,
isDegradedMode: false,
degradedModeReason: null,
});
await get().refreshScheduleStatus(date);
return;
}
set({
error: toWorkflowError(error, 'Failed to load workflow from server.'),
isLoading: false,
isDegradedMode: false,
degradedModeReason: null,
});
}
},
refreshScheduleStatus: async (date?: string) => {
try {
const resp = await apiClient.get('/api/today-workflow/status', { params: date ? { date } : {} });
const scheduleStatus = resp?.data?.data as TodayWorkflowScheduleStatus | undefined;
set({ scheduleStatus: scheduleStatus || null });
} catch {
set({ scheduleStatus: null });
}
},
// Generate daily workflow
generateDailyWorkflow: async (userId: string, date?: string) => {
set({ isLoading: true, error: null });
try {
const resp = await apiClient.post('/api/today-workflow/generate', null, { params: date ? { date } : {} });
const serverWorkflow = resp?.data?.data?.workflow as DailyWorkflow | undefined;
const planSummary = resp?.data?.data?.plan?.provenance_summary;
const scheduleStatus = resp?.data?.data?.schedule_status as TodayWorkflowScheduleStatus | undefined;
if (serverWorkflow && Array.isArray(serverWorkflow.tasks)) {
if (planSummary && !serverWorkflow.provenanceSummary) {
serverWorkflow.provenanceSummary = planSummary;
}
const normalizedWorkflow = normalizeServerWorkflow(serverWorkflow);
const derived = computeProgressAndNavigation(normalizedWorkflow);
set({
currentWorkflow: normalizedWorkflow,
workflowProgress: derived.progress,
navigationState: derived.navigation,
scheduleStatus: scheduleStatus || null,
isLoading: false,
isDegradedMode: false,
degradedModeReason: null,
});
return;
}
throw new WorkflowError({
code: 'WORKFLOW_SCHEMA_INVALID',
message: 'Server workflow response is missing a valid tasks array.',
timestamp: new Date(),
recoverable: false,
suggestedAction: 'Refresh and try again. If this persists, contact support.'
});
} catch (error) {
set({
error: toWorkflowError(error, 'Failed to generate workflow from server.'),
isLoading: false,
isDegradedMode: false,
degradedModeReason: null,
});
}
},
// Start workflow
startWorkflow: async (workflowId: string) => {
set({ isLoading: true, error: null });
try {
const current = get().currentWorkflow;
if (current && current.id === workflowId && isServerWorkflowId(workflowId)) {
const tasks = current.tasks.map((t, idx) => {
if (idx === current.currentTaskIndex && t.status === 'pending') {
return { ...t, status: 'in_progress' as const, startedAt: new Date() };
}
return t;
});
const updated: DailyWorkflow = { ...current, workflowStatus: 'in_progress', startedAt: new Date(), tasks };
const derived = computeProgressAndNavigation(updated);
set({ currentWorkflow: updated, workflowProgress: derived.progress, navigationState: derived.navigation, isLoading: false });
return;
}
const workflow = await taskWorkflowOrchestrator.startWorkflow(workflowId);
const progress = taskWorkflowOrchestrator.getWorkflowProgress(workflow.id);
const navigation = taskWorkflowOrchestrator.getNavigationState(workflow.id);
set({ currentWorkflow: workflow, workflowProgress: progress, navigationState: navigation, isLoading: false });
} catch (error) {
const workflowError = error as WorkflowError;
set({
error: workflowError,
isLoading: false
});
}
},
// Pause workflow
pauseWorkflow: async (workflowId: string) => {
set({ isLoading: true, error: null });
try {
// For now, we'll just update the workflow status to paused
// In a real implementation, this would call the orchestrator
const currentWorkflow = get().currentWorkflow;
if (currentWorkflow && currentWorkflow.id === workflowId) {
const pausedWorkflow = {
...currentWorkflow,
workflowStatus: 'paused' as const,
pausedAt: new Date()
};
set({
currentWorkflow: pausedWorkflow,
isLoading: false
});
}
} catch (error) {
const workflowError = error as WorkflowError;
set({
error: workflowError,
isLoading: false
});
}
},
// Stop workflow
stopWorkflow: async (workflowId: string) => {
set({ isLoading: true, error: null });
try {
// For now, we'll just update the workflow status to stopped
// In a real implementation, this would call the orchestrator
const currentWorkflow = get().currentWorkflow;
if (currentWorkflow && currentWorkflow.id === workflowId) {
const stoppedWorkflow = {
...currentWorkflow,
workflowStatus: 'stopped' as const,
completedAt: new Date()
};
set({
currentWorkflow: stoppedWorkflow,
isLoading: false
});
}
} catch (error) {
const workflowError = error as WorkflowError;
set({
error: workflowError,
isLoading: false
});
}
},
// Complete task
completeTask: async (taskId: string, completionData?: any) => {
const { currentWorkflow } = get();
if (!currentWorkflow) return;
set({ isLoading: true, error: null });
try {
if (isServerWorkflowId(currentWorkflow.id)) {
await apiClient.post(`/api/today-workflow/tasks/${taskId}/status`, {
status: 'completed',
completion_notes: completionData?.userNotes
});
const tasks = currentWorkflow.tasks.map(t => (t.id === taskId ? { ...t, status: 'completed' as const, completedAt: new Date() } : t));
const completedTasks = tasks.filter(t => t.status === 'completed' || t.status === 'skipped').length;
const totalTasks = tasks.length;
const workflowStatus: WorkflowStatus = totalTasks > 0 && completedTasks === totalTasks ? 'completed' : 'in_progress';
const updated: DailyWorkflow = {
...currentWorkflow,
tasks,
completedTasks,
totalTasks,
workflowStatus,
completedAt: workflowStatus === 'completed' ? new Date() : currentWorkflow.completedAt,
};
const derived = computeProgressAndNavigation(updated);
set({ currentWorkflow: updated, workflowProgress: derived.progress, navigationState: derived.navigation, isLoading: false });
return;
}
const progress = await taskWorkflowOrchestrator.completeTask(currentWorkflow.id, taskId, completionData);
const navigation = taskWorkflowOrchestrator.getNavigationState(currentWorkflow.id);
const updatedWorkflow = taskWorkflowOrchestrator.getWorkflow(currentWorkflow.userId, currentWorkflow.date);
set({ currentWorkflow: updatedWorkflow, workflowProgress: progress, navigationState: navigation, isLoading: false });
} catch (error) {
const workflowError = error as WorkflowError;
set({
error: workflowError,
isLoading: false
});
}
},
// Skip task
skipTask: async (taskId: string) => {
const { currentWorkflow } = get();
if (!currentWorkflow) return;
set({ isLoading: true, error: null });
try {
if (isServerWorkflowId(currentWorkflow.id)) {
await apiClient.post(`/api/today-workflow/tasks/${taskId}/status`, { status: 'skipped' });
const tasks = currentWorkflow.tasks.map(t => (t.id === taskId ? { ...t, status: 'skipped' as const, completedAt: new Date() } : t));
const completedTasks = tasks.filter(t => t.status === 'completed' || t.status === 'skipped').length;
const totalTasks = tasks.length;
const workflowStatus: WorkflowStatus = totalTasks > 0 && completedTasks === totalTasks ? 'completed' : currentWorkflow.workflowStatus;
const updated: DailyWorkflow = { ...currentWorkflow, tasks, completedTasks, totalTasks, workflowStatus };
const derived = computeProgressAndNavigation(updated);
set({ currentWorkflow: updated, workflowProgress: derived.progress, navigationState: derived.navigation, isLoading: false });
return;
}
const progress = await taskWorkflowOrchestrator.skipTask(currentWorkflow.id, taskId);
const navigation = taskWorkflowOrchestrator.getNavigationState(currentWorkflow.id);
const updatedWorkflow = taskWorkflowOrchestrator.getWorkflow(currentWorkflow.userId, currentWorkflow.date);
set({ currentWorkflow: updatedWorkflow, workflowProgress: progress, navigationState: navigation, isLoading: false });
} catch (error) {
const workflowError = error as WorkflowError;
set({
error: workflowError,
isLoading: false
});
}
},
// Move to next task
moveToNextTask: async () => {
const { currentWorkflow } = get();
if (!currentWorkflow) return;
set({ isLoading: true, error: null });
try {
if (isServerWorkflowId(currentWorkflow.id)) {
const tasks = currentWorkflow.tasks;
const nextIndex = tasks.findIndex((t, idx) => idx > currentWorkflow.currentTaskIndex && t.status !== 'completed' && t.status !== 'skipped');
const updated: DailyWorkflow = {
...currentWorkflow,
currentTaskIndex: nextIndex >= 0 ? nextIndex : currentWorkflow.currentTaskIndex,
};
const derived = computeProgressAndNavigation(updated);
set({ currentWorkflow: updated, workflowProgress: derived.progress, navigationState: derived.navigation, isLoading: false });
return;
}
await taskWorkflowOrchestrator.moveToNextTask(currentWorkflow.id);
const progress = taskWorkflowOrchestrator.getWorkflowProgress(currentWorkflow.id);
const navigation = taskWorkflowOrchestrator.getNavigationState(currentWorkflow.id);
const updatedWorkflow = taskWorkflowOrchestrator.getWorkflow(currentWorkflow.userId, currentWorkflow.date);
set({ currentWorkflow: updatedWorkflow, workflowProgress: progress, navigationState: navigation, isLoading: false });
} catch (error) {
const workflowError = error as WorkflowError;
set({
error: workflowError,
isLoading: false
});
}
},
// Move to previous task
moveToPreviousTask: async () => {
const { currentWorkflow } = get();
if (!currentWorkflow) return;
set({ isLoading: true, error: null });
try {
// This would need to be implemented in the orchestrator
// For now, we'll just refresh the navigation state
const navigation = taskWorkflowOrchestrator.getNavigationState(currentWorkflow.id);
set({
navigationState: navigation,
isLoading: false
});
} catch (error) {
const workflowError = error as WorkflowError;
set({
error: workflowError,
isLoading: false
});
}
},
// UI actions
openWorkflowModal: () => set({ isWorkflowModalOpen: true }),
closeWorkflowModal: () => set({ isWorkflowModalOpen: false }),
setError: (error: WorkflowError | null) => set({ error }),
clearError: () => set({ error: null }),
// Update user preferences
updateUserPreferences: (preferences: Partial<UserWorkflowPreferences>) => {
const { userPreferences } = get();
set({
userPreferences: {
...userPreferences,
...preferences
} as UserWorkflowPreferences
});
},
// Utility actions
refreshWorkflowProgress: () => {
const { currentWorkflow } = get();
if (!currentWorkflow) return;
try {
if (isServerWorkflowId(currentWorkflow.id)) {
const derived = computeProgressAndNavigation(currentWorkflow);
set({ workflowProgress: derived.progress, navigationState: derived.navigation });
return;
}
const progress = taskWorkflowOrchestrator.getWorkflowProgress(currentWorkflow.id);
const navigation = taskWorkflowOrchestrator.getNavigationState(currentWorkflow.id);
set({ workflowProgress: progress, navigationState: navigation });
} catch (error) {
console.warn('Failed to refresh workflow progress:', error);
}
},
getCurrentTask: () => {
const { navigationState } = get();
return navigationState?.currentTask || null;
},
getNextTask: () => {
const { navigationState } = get();
return navigationState?.nextTask || null;
},
isWorkflowComplete: () => {
const { workflowProgress } = get();
return workflowProgress ? workflowProgress.completedTasks === workflowProgress.totalTasks : false;
},
getCompletionPercentage: () => {
const { workflowProgress } = get();
return workflowProgress?.completionPercentage || 0;
}
}),
{
name: 'workflow-store',
partialize: (state) => ({
userPreferences: state.userPreferences,
currentWorkflow: state.currentWorkflow
})
}
)
);
export default useWorkflowStore;