Unstaging (#121)

* only allow unstaging for completed staging * magic-mock debugging log * more status displayed * actually fix
c3g · Nov 19, 2024 · d8f6760 · d8f6760
1 parent 398e4ab
commit d8f6760
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/magic-mock/index.ts b/magic-mock/index.ts
@@ -48,4 +48,8 @@ void new Promise((resolve, reject) => {
         reject(err)
     })
     server.on("listening", () => console.log(`listening on port ${PORT}`))
+    server.on("request", (req, res) => {
+        console.log(req.method, req.url, req.headers)
+        console.log(res.statusCode, res.statusMessage)
+    })
 })
diff --git a/triton-client/src/components/DatasetCard.tsx b/triton-client/src/components/DatasetCard.tsx
@@ -126,7 +126,7 @@ function DatasetCard({ datasetID }: DatasetCardProps) {
     const requestDetails = useMemo(() => {
         return SUPPORTED_DOWNLOAD_TYPES.map((type) => {
             const req = requestByType[type]
-            if (req && !req.should_delete && req.status !== "FAILED") {
+            if (req && !req.should_delete && req.status === "SUCCESS") {
                 const { type, status } = req
                 const actions: ActionDropdownProps["actions"] = [
                     {
@@ -167,12 +167,6 @@ function DatasetCard({ datasetID }: DatasetCardProps) {
                     },
                 ]
 
-                let statusDescription: ReactNode
-                if (status === "SUCCESS") {
-                    statusDescription = "DOWNLOAD"
-                } else {
-                    statusDescription = "QUEUED"
-                }
                 const buttonStagingActive = (
                     <Button
                         key={type}
@@ -196,7 +190,7 @@ function DatasetCard({ datasetID }: DatasetCardProps) {
                         <Space>
                             {type}
                             {"|"}
-                            {statusDescription}
+                            {"DOWNLOAD"}
                         </Space>
                     </Button>
                 )
@@ -212,8 +206,8 @@ function DatasetCard({ datasetID }: DatasetCardProps) {
                 let statusDescription: ReactNode
                 if (req && req.should_delete) {
                     statusDescription = "UNSTAGING"
-                } else if (req?.status === "FAILED") {
-                    statusDescription = "FAILED"
+                } else if (req?.status) {
+                    statusDescription = req.status
                 } else {
                     statusDescription = "STAGE"
                 }