|
1 | | -import { useTasksApi } from "@repo/webview-shared/react"; |
| 1 | +import { |
| 2 | + getState, |
| 3 | + setState, |
| 4 | + type Task, |
| 5 | + type TasksPushMessage, |
| 6 | + type TaskTemplate, |
| 7 | +} from "@repo/webview-shared"; |
| 8 | +import { useMessage, useTasksApi } from "@repo/webview-shared/react"; |
2 | 9 | import { VscodeProgressRing } from "@vscode-elements/react-elements"; |
3 | | -import { useEffect, useState } from "react"; |
| 10 | +import { useCallback, useEffect, useState, useRef } from "react"; |
4 | 11 |
|
5 | | -import type { Task, TaskTemplate } from "@repo/webview-shared"; |
| 12 | +import { |
| 13 | + CollapsibleSection, |
| 14 | + CreateTaskSection, |
| 15 | + ErrorState, |
| 16 | + NoTemplateState, |
| 17 | + NotSupportedState, |
| 18 | + TaskList, |
| 19 | +} from "./components"; |
| 20 | +import { POLLING_CONFIG } from "./config"; |
| 21 | +import { taskArraysEqual, templateArraysEqual } from "./utils"; |
| 22 | + |
| 23 | +interface PersistedState { |
| 24 | + tasks: Task[]; |
| 25 | + templates: TaskTemplate[]; |
| 26 | + createExpanded: boolean; |
| 27 | + historyExpanded: boolean; |
| 28 | + tasksSupported: boolean; |
| 29 | +} |
6 | 30 |
|
7 | 31 | export default function App() { |
8 | 32 | const api = useTasksApi(); |
9 | | - const [loading, setLoading] = useState(true); |
10 | | - const [error, setError] = useState<string | null>(null); |
11 | | - const [tasks, setTasks] = useState<Task[]>([]); |
12 | | - const [templates, setTemplates] = useState<TaskTemplate[]>([]); |
13 | | - const [tasksSupported, setTasksSupported] = useState(true); |
| 33 | + |
| 34 | + const persistedState = useRef(getState<PersistedState>()); |
| 35 | + const restored = persistedState.current; |
| 36 | + |
| 37 | + const [initialized, setInitialized] = useState(!!restored?.tasks?.length); |
| 38 | + const [tasks, setTasks] = useState<Task[]>(restored?.tasks ?? []); |
| 39 | + const [templates, setTemplates] = useState<TaskTemplate[]>( |
| 40 | + restored?.templates ?? [], |
| 41 | + ); |
| 42 | + const [tasksSupported, setTasksSupported] = useState( |
| 43 | + restored?.tasksSupported ?? true, |
| 44 | + ); |
| 45 | + |
| 46 | + const [createExpanded, setCreateExpanded] = useState( |
| 47 | + restored?.createExpanded ?? true, |
| 48 | + ); |
| 49 | + const [historyExpanded, setHistoryExpanded] = useState( |
| 50 | + restored?.historyExpanded ?? true, |
| 51 | + ); |
| 52 | + |
| 53 | + useEffect(() => { |
| 54 | + setState<PersistedState>({ |
| 55 | + tasks, |
| 56 | + templates, |
| 57 | + createExpanded, |
| 58 | + historyExpanded, |
| 59 | + tasksSupported, |
| 60 | + }); |
| 61 | + }, [tasks, templates, createExpanded, historyExpanded, tasksSupported]); |
| 62 | + |
| 63 | + const [initLoading, setInitLoading] = useState(!restored?.tasks?.length); |
| 64 | + const [initError, setInitError] = useState<string | null>(null); |
14 | 65 |
|
15 | 66 | useEffect(() => { |
| 67 | + let cancelled = false; |
| 68 | + |
| 69 | + async function initialize() { |
| 70 | + try { |
| 71 | + const data = await api.init(); |
| 72 | + if (cancelled) return; |
| 73 | + |
| 74 | + setTasks(data.tasks); |
| 75 | + setTemplates(data.templates); |
| 76 | + setTasksSupported(data.tasksSupported); |
| 77 | + setInitialized(true); |
| 78 | + setInitError(null); |
| 79 | + } catch (err) { |
| 80 | + if (cancelled) return; |
| 81 | + setInitError( |
| 82 | + err instanceof Error ? err.message : "Failed to initialize", |
| 83 | + ); |
| 84 | + } finally { |
| 85 | + if (!cancelled) { |
| 86 | + setInitLoading(false); |
| 87 | + } |
| 88 | + } |
| 89 | + } |
| 90 | + |
| 91 | + void initialize(); |
| 92 | + return () => { |
| 93 | + cancelled = true; |
| 94 | + }; |
| 95 | + }, [api]); |
| 96 | + |
| 97 | + const tasksRef = useRef<Task[]>(tasks); |
| 98 | + tasksRef.current = tasks; |
| 99 | + |
| 100 | + const templatesRef = useRef<TaskTemplate[]>(templates); |
| 101 | + templatesRef.current = templates; |
| 102 | + |
| 103 | + // Poll for task list updates |
| 104 | + useEffect(() => { |
| 105 | + if (!initialized) return; |
| 106 | + |
| 107 | + let cancelled = false; |
| 108 | + const pollInterval = setInterval(() => { |
| 109 | + api |
| 110 | + .getTasks() |
| 111 | + .then((updatedTasks) => { |
| 112 | + if (cancelled) return; |
| 113 | + if (!taskArraysEqual(tasksRef.current, updatedTasks)) { |
| 114 | + setTasks(updatedTasks); |
| 115 | + } |
| 116 | + }) |
| 117 | + .catch(() => undefined); |
| 118 | + }, POLLING_CONFIG.TASK_LIST_INTERVAL_MS); |
| 119 | + |
| 120 | + return () => { |
| 121 | + cancelled = true; |
| 122 | + clearInterval(pollInterval); |
| 123 | + }; |
| 124 | + }, [api, initialized]); |
| 125 | + |
| 126 | + const handleRetry = useCallback(() => { |
| 127 | + setInitLoading(true); |
| 128 | + setInitError(null); |
| 129 | + |
16 | 130 | api |
17 | 131 | .init() |
18 | 132 | .then((data) => { |
19 | 133 | setTasks(data.tasks); |
20 | 134 | setTemplates(data.templates); |
21 | 135 | setTasksSupported(data.tasksSupported); |
22 | | - setLoading(false); |
| 136 | + setInitialized(true); |
| 137 | + }) |
| 138 | + .catch((err: unknown) => { |
| 139 | + setInitError( |
| 140 | + err instanceof Error ? err.message : "Failed to initialize", |
| 141 | + ); |
23 | 142 | }) |
24 | | - .catch((err) => { |
25 | | - setError(err instanceof Error ? err.message : "Failed to initialize"); |
26 | | - setLoading(false); |
| 143 | + .finally(() => { |
| 144 | + setInitLoading(false); |
27 | 145 | }); |
28 | 146 | }, [api]); |
29 | 147 |
|
30 | | - if (loading) { |
| 148 | + useMessage<TasksPushMessage>((msg) => { |
| 149 | + switch (msg.type) { |
| 150 | + case "tasksUpdated": |
| 151 | + setTasks(msg.data); |
| 152 | + break; |
| 153 | + |
| 154 | + case "taskUpdated": { |
| 155 | + const updatedTask = msg.data; |
| 156 | + setTasks((prev) => |
| 157 | + prev.map((t) => (t.id === updatedTask.id ? updatedTask : t)), |
| 158 | + ); |
| 159 | + break; |
| 160 | + } |
| 161 | + |
| 162 | + case "refresh": { |
| 163 | + api |
| 164 | + .getTasks() |
| 165 | + .then((updatedTasks) => { |
| 166 | + if (!taskArraysEqual(tasksRef.current, updatedTasks)) { |
| 167 | + setTasks(updatedTasks); |
| 168 | + } |
| 169 | + }) |
| 170 | + .catch(() => undefined); |
| 171 | + api |
| 172 | + .getTemplates() |
| 173 | + .then((updatedTemplates) => { |
| 174 | + if (!templateArraysEqual(templatesRef.current, updatedTemplates)) { |
| 175 | + setTemplates(updatedTemplates); |
| 176 | + } |
| 177 | + }) |
| 178 | + .catch(() => undefined); |
| 179 | + break; |
| 180 | + } |
| 181 | + |
| 182 | + case "showCreateForm": |
| 183 | + setCreateExpanded(true); |
| 184 | + break; |
| 185 | + |
| 186 | + case "logsAppend": |
| 187 | + // Task detail view will handle this in next PR |
| 188 | + break; |
| 189 | + } |
| 190 | + }); |
| 191 | + |
| 192 | + const handleSelectTask = useCallback((_taskId: string) => { |
| 193 | + // Task detail view will be added in next PR |
| 194 | + }, []); |
| 195 | + |
| 196 | + if (initLoading) { |
31 | 197 | return ( |
32 | 198 | <div className="loading-container"> |
33 | 199 | <VscodeProgressRing /> |
34 | 200 | </div> |
35 | 201 | ); |
36 | 202 | } |
37 | 203 |
|
38 | | - if (error) { |
39 | | - return ( |
40 | | - <div className="error-container"> |
41 | | - <p>Error: {error}</p> |
42 | | - </div> |
43 | | - ); |
| 204 | + if (initError && tasks.length === 0) { |
| 205 | + return <ErrorState message={initError} onRetry={handleRetry} />; |
44 | 206 | } |
45 | 207 |
|
46 | | - if (!tasksSupported) { |
47 | | - return ( |
48 | | - <div className="not-supported"> |
49 | | - <p>Tasks are not supported on this Coder server.</p> |
50 | | - </div> |
51 | | - ); |
| 208 | + if (initialized && !tasksSupported) { |
| 209 | + return <NotSupportedState />; |
| 210 | + } |
| 211 | + |
| 212 | + if (initialized && templates.length === 0) { |
| 213 | + return <NoTemplateState />; |
52 | 214 | } |
53 | 215 |
|
54 | 216 | return ( |
55 | 217 | <div className="tasks-panel"> |
56 | | - <h3>Tasks Infrastructure Ready</h3> |
57 | | - <p>Templates: {templates.length}</p> |
58 | | - <p>Tasks: {tasks.length}</p> |
59 | | - <p>Full UI coming in next PR...</p> |
| 218 | + <CollapsibleSection |
| 219 | + title="Create new task" |
| 220 | + expanded={createExpanded} |
| 221 | + onToggle={() => setCreateExpanded(!createExpanded)} |
| 222 | + > |
| 223 | + <CreateTaskSection templates={templates} /> |
| 224 | + </CollapsibleSection> |
| 225 | + |
| 226 | + <CollapsibleSection |
| 227 | + title="Task History" |
| 228 | + expanded={historyExpanded} |
| 229 | + onToggle={() => setHistoryExpanded(!historyExpanded)} |
| 230 | + > |
| 231 | + <TaskList tasks={tasks} onSelectTask={handleSelectTask} /> |
| 232 | + </CollapsibleSection> |
60 | 233 | </div> |
61 | 234 | ); |
62 | 235 | } |
0 commit comments