feat: show loading screen for active task if scheduled/running

Signed-off-by: Edward Ly <[email protected]>

Author: edward-ly

src/assistant.js

@@ -137,11 +137,55 @@ export async function openAssistantForm({
 		view.$on('load-task', (task) => {
 			if (!view.loading) {
 				console.debug('[assistant] loading task', task)
+				cancelTaskPolling()
+
 				view.selectedTaskTypeId = task.type
 				view.inputs = task.input
 				view.outputs = task.status === TASK_STATUS_STRING.successful ? task.output : null
 				view.selectedTaskId = task.id
 				lastTask = task
+
+				if ([TASK_STATUS_STRING.scheduled, TASK_STATUS_STRING.running].includes(task?.status)) {
+					getTask(task.id).then(response => {
+						const updatedTask = response.data?.ocs?.data?.task
+
+						if (![TASK_STATUS_STRING.scheduled, TASK_STATUS_STRING.running].includes(updatedTask?.status)) {
+							view.selectedTaskTypeId = updatedTask.type
+							view.inputs = updatedTask.input
+							view.outputs = updatedTask.status === TASK_STATUS_STRING.successful ? updatedTask.output : null
+							view.selectedTaskId = updatedTask.id
+							lastTask = updatedTask
+							return
+						}
+
+						view.loading = true
+						view.showSyncTaskRunning = true
+						view.progress = null
+						view.expectedRuntime = (updatedTask?.completionExpectedAt - updatedTask?.scheduledAt) || null
+
+						const setProgress = (progress) => {
+							view.progress = progress
+						}
+						pollTask(updatedTask.id, setProgress).then(finishedTask => {
+							console.debug('pollTask.then', finishedTask)
+							if (finishedTask.status === TASK_STATUS_STRING.successful) {
+								view.outputs = finishedTask?.output
+								view.selectedTaskId = finishedTask?.id
+							} else if (finishedTask.status === TASK_STATUS_STRING.failed) {
+								showError(t('assistant', 'Your task with ID {id} has failed', { id: finishedTask.id }))
+								console.error('[assistant] Task failed', finishedTask)
+								view.outputs = null
+							}
+							// resolve(finishedTask)
+							view.loading = false
+							view.showSyncTaskRunning = false
+						}).catch(error => {
+							console.debug('[assistant] poll error', error)
+						})
+					}).catch(error => {
+						console.error(error)
+					})
+				}
 			}
 		})
 		view.$on('new-task', () => {
@@ -172,6 +216,7 @@ export async function openAssistantForm({
 			view.$destroy()
 		})
 		view.$on('back-to-assistant', () => {
+			cancelTaskPolling()
 			view.showScheduleConfirmation = false
 			view.showSyncTaskRunning = false
 			view.loading = false
@@ -436,11 +481,55 @@ export async function openAssistantTask(task, { isInsideViewer = undefined, acti
 	})
 	view.$on('load-task', (task) => {
 		if (!view.loading) {
+			cancelTaskPolling()
+
 			view.selectedTaskTypeId = task.type
 			view.inputs = task.input
 			view.outputs = task.status === TASK_STATUS_STRING.successful ? task.output : null
 			view.selectedTaskId = task.id
 			lastTask = task
+
+			if ([TASK_STATUS_STRING.scheduled, TASK_STATUS_STRING.running].includes(task?.status)) {
+				getTask(task.id).then(response => {
+					const updatedTask = response.data?.ocs?.data?.task
+
+					if (![TASK_STATUS_STRING.scheduled, TASK_STATUS_STRING.running].includes(updatedTask?.status)) {
+						view.selectedTaskTypeId = updatedTask.type
+						view.inputs = updatedTask.input
+						view.outputs = updatedTask.status === TASK_STATUS_STRING.successful ? updatedTask.output : null
+						view.selectedTaskId = updatedTask.id
+						lastTask = updatedTask
+						return
+					}
+
+					view.loading = true
+					view.showSyncTaskRunning = true
+					view.progress = null
+					view.expectedRuntime = (updatedTask?.completionExpectedAt - updatedTask?.scheduledAt) || null
+
+					const setProgress = (progress) => {
+						view.progress = progress
+					}
+					pollTask(updatedTask.id, setProgress).then(finishedTask => {
+						console.debug('pollTask.then', finishedTask)
+						if (finishedTask.status === TASK_STATUS_STRING.successful) {
+							view.outputs = finishedTask?.output
+							view.selectedTaskId = finishedTask?.id
+						} else if (finishedTask.status === TASK_STATUS_STRING.failed) {
+							showError(t('assistant', 'Your task with ID {id} has failed', { id: finishedTask.id }))
+							console.error('[assistant] Task failed', finishedTask)
+							view.outputs = null
+						}
+						// resolve(finishedTask)
+						view.loading = false
+						view.showSyncTaskRunning = false
+					}).catch(error => {
+						console.debug('[assistant] poll error', error)
+					})
+				}).catch(error => {
+					console.error(error)
+				})
+			}
 		}
 	})
 	view.$on('new-task', () => {
@@ -469,6 +558,7 @@ export async function openAssistantTask(task, { isInsideViewer = undefined, acti
 		view.$destroy()
 	})
 	view.$on('back-to-assistant', () => {
+		cancelTaskPolling()
 		view.showScheduleConfirmation = false
 		view.showSyncTaskRunning = false
 		view.loading = false