From 616f0a7d40f9951769815041909bfe379df9b92e Mon Sep 17 00:00:00 2001
From: Pavel Vostretsov <p.vostretsov@skbkontur.ru>
Date: Tue, 5 Mar 2024 15:58:05 +0500
Subject: [PATCH] add same logic to details cards in table

---
 .../Api/RtqMonitoringTaskMeta.cs              |  4 ++++
 .../Api/RtqMonitoringTaskModel.cs             |  4 ----
 .../TaskDetailsPage/TaskDetailsPage.tsx       | 20 +++++++++----------
 .../TaskTable/TaskDetails/TaskDetails.tsx     |  7 +++++--
 4 files changed, 18 insertions(+), 17 deletions(-)

diff --git a/Cassandra.DistributedTaskQueue.Monitoring/Api/RtqMonitoringTaskMeta.cs b/Cassandra.DistributedTaskQueue.Monitoring/Api/RtqMonitoringTaskMeta.cs
index 6514f3da..97635402 100644
--- a/Cassandra.DistributedTaskQueue.Monitoring/Api/RtqMonitoringTaskMeta.cs
+++ b/Cassandra.DistributedTaskQueue.Monitoring/Api/RtqMonitoringTaskMeta.cs
@@ -54,6 +54,10 @@ public class RtqMonitoringTaskMeta
         [JsonConverter(typeof(StringEnumConverter))]
         public TaskState State { get; set; }
 
+        [CanBeNull]
+        [JsonProperty("taskActions")]
+        public TaskActions TaskActions { get; set; }
+
         [JsonProperty("attempts")]
         public int Attempts { get; set; }
 
diff --git a/Cassandra.DistributedTaskQueue.Monitoring/Api/RtqMonitoringTaskModel.cs b/Cassandra.DistributedTaskQueue.Monitoring/Api/RtqMonitoringTaskModel.cs
index acaeb983..567a5c1b 100644
--- a/Cassandra.DistributedTaskQueue.Monitoring/Api/RtqMonitoringTaskModel.cs
+++ b/Cassandra.DistributedTaskQueue.Monitoring/Api/RtqMonitoringTaskModel.cs
@@ -27,9 +27,5 @@ public class RtqMonitoringTaskModel
         [NotNull, ItemNotNull]
         [JsonProperty("exceptionInfos")]
         public string[] ExceptionInfos { get; set; }
-
-        [CanBeNull]
-        [JsonProperty("taskActions")]
-        public TaskActions TaskActions { get; set; }
     }
 }
\ No newline at end of file
diff --git a/cassandra-distributed-task-queue-ui/src/components/TaskDetailsPage/TaskDetailsPage.tsx b/cassandra-distributed-task-queue-ui/src/components/TaskDetailsPage/TaskDetailsPage.tsx
index 56d6b7d5..1768222d 100644
--- a/cassandra-distributed-task-queue-ui/src/components/TaskDetailsPage/TaskDetailsPage.tsx
+++ b/cassandra-distributed-task-queue-ui/src/components/TaskDetailsPage/TaskDetailsPage.tsx
@@ -57,18 +57,16 @@ export function TaskDetailsPage({
             return null;
         }
 
-        const isCancelable =
-            taskDetails.taskActions == null
-                ? allowRerunOrCancel && cancelableStates.includes(taskDetails.taskMeta.state)
-                : taskDetails.taskActions.canCancel;
+        const canCancel = taskDetails.taskMeta.taskActions
+            ? taskDetails.taskMeta.taskActions.canCancel
+            : allowRerunOrCancel && cancelableStates.includes(taskDetails.taskMeta.state);
 
-        const isRerunable =
-            taskDetails.taskActions == null
-                ? allowRerunOrCancel && rerunableStates.includes(taskDetails.taskMeta.state)
-                : taskDetails.taskActions.canRerun;
+        const canRerun = taskDetails.taskMeta.taskActions
+            ? taskDetails.taskMeta.taskActions.canRerun
+            : allowRerunOrCancel && rerunableStates.includes(taskDetails.taskMeta.state);
 
         const relatedTasksRequest = customDetailRenderer.getRelatedTasksLocation(taskDetails);
-        if (!isCancelable && !isRerunable && relatedTasksRequest == null) {
+        if (!canCancel && !canRerun && relatedTasksRequest == null) {
             return null;
         }
 
@@ -86,14 +84,14 @@ export function TaskDetailsPage({
                         </RouterLink>
                     </Fit>
                 )}
-                {isCancelable && (
+                {canCancel && (
                     <Fit>
                         <Link icon={<XIcon16Regular />} use="danger" data-tid="CancelButton" onClick={cancel}>
                             Cancel task
                         </Link>
                     </Fit>
                 )}
-                {isRerunable && (
+                {canRerun && (
                     <Fit>
                         <Button
                             use="link"
diff --git a/cassandra-distributed-task-queue-ui/src/components/TaskTable/TaskDetails/TaskDetails.tsx b/cassandra-distributed-task-queue-ui/src/components/TaskTable/TaskDetails/TaskDetails.tsx
index 1a370c82..1f74af87 100644
--- a/cassandra-distributed-task-queue-ui/src/components/TaskTable/TaskDetails/TaskDetails.tsx
+++ b/cassandra-distributed-task-queue-ui/src/components/TaskTable/TaskDetails/TaskDetails.tsx
@@ -37,6 +37,9 @@ export function TaskDetails(props: TaskDetailsProps): JSX.Element {
     const theme = React.useContext(ThemeContext);
     const { customStateCaptions } = useCustomSettings();
 
+    const canCancel = taskInfo.taskActions ? taskInfo.taskActions.canCancel : cancelableStates.includes(taskInfo.state);
+    const canRerun = taskInfo.taskActions ? taskInfo.taskActions.canRerun : rerunableStates.includes(taskInfo.state);
+
     const renderTaskDate = (
         taskInfo: RtqMonitoringTaskMeta,
         caption: string,
@@ -93,7 +96,7 @@ export function TaskDetails(props: TaskDetailsProps): JSX.Element {
                                         <Fit>
                                             <Link
                                                 data-tid="Cancel"
-                                                disabled={!cancelableStates.includes(taskInfo.state)}
+                                                disabled={!canCancel}
                                                 onClick={onCancel}
                                                 icon={<XIcon16Regular />}>
                                                 Cancel
@@ -102,7 +105,7 @@ export function TaskDetails(props: TaskDetailsProps): JSX.Element {
                                         <Fit>
                                             <Link
                                                 data-tid="Rerun"
-                                                disabled={!rerunableStates.includes(taskInfo.state)}
+                                                disabled={!canRerun}
                                                 onClick={onRerun}
                                                 icon={<ArrowRoundTimeForwardIcon16Regular />}>
                                                 Rerun