From 0d5a7597aba3c3926e4d6b1ea16cf0404bd1e397 Mon Sep 17 00:00:00 2001
From: undefined <i@undefined.moe>
Date: Fri, 7 Jun 2024 07:16:21 +0800
Subject: [PATCH] ui: testdata generate callback

---
 packages/ui-default/pages/problem_files.page.tsx | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/packages/ui-default/pages/problem_files.page.tsx b/packages/ui-default/pages/problem_files.page.tsx
index 030877c6a..d6c7d2fc5 100644
--- a/packages/ui-default/pages/problem_files.page.tsx
+++ b/packages/ui-default/pages/problem_files.page.tsx
@@ -348,7 +348,17 @@ const page = new NamedPage('problem_files', () => {
             </div>`,
           width: `${window.innerWidth - 200}px`,
           height: `${window.innerHeight - 100}px`,
-        }).open();
+        });
+        const callback = (data: MessageEvent<any>) => {
+          if (data.data.status === STATUS.STATUS_ACCEPTED) {
+            dialog.close();
+            Notification.success('Testdata generated successfully.');
+          }
+        };
+        window.addEventListener('message', callback, false);
+        await dialog.open();
+        window.removeEventListener('message', callback, false);
+        await pjax.request({ push: false });
       }
     } catch (error) {
       Notification.error([error.message, ...error.params].join(' '));