feat: add playback get route (#1341)

src/helpers/docker_queue.ts

@@ -50,7 +50,7 @@ const upload_contest_files = async (sub_base_dir: string, queue_front: queue_ele
       const file_name = await fs_promises.readdir(`${sub_base_dir}/${queue_front.room_id}/output`);
       const upload_file_promises = file_name.map(filename => {
         console.log("filename: " + filename);
-        const key = `${contest_name}/arena/${queue_front.room_id}/${queue_front.room_id}.${filename}`;
+        const key = `${contest_name}/arena/${queue_front.room_id}/${filename}`;
         const localFilePath = `${sub_base_dir}/${queue_front.room_id}/output/${filename}`;
         return utils.uploadObject(localFilePath, key, cos, config)
           .then(() => {

src/helpers/hasura.ts

@@ -665,7 +665,28 @@ export const get_map_name: any = async (map_id: string) => {
   return query_map.contest_map[0].filename ?? null;
 }
 
-
+/**
+ * get contest_name by room_id
+ * @param {uuid} room_id
+ * @returns {string} contest_name
+ */
+export const get_contest_name_by_room: any = async (room_id: string) => {
+  const query_contest_name = await client.request(
+    gql`
+      query get_contest_name($room_id: uuid!) {
+        contest_room(where: {room_id: {_eq: $room_id}}) {
+          contest {
+            name
+          }
+        }
+      }
+    `,
+    {
+      room_id: room_id
+    }
+  );
+  return query_contest_name.contest_room[0]?.contest?.name ?? null;
+}
 
 /**
   ============================================================================

src/routes/arena.ts

@@ -501,5 +501,35 @@ router.post("/finish", async (req, res) => {
   }
 });
 
+/**
+ * @param {uuid} room_id
+ * 用于获取回放的路由，直接返回文件。
+ */
+router.get("/playback", async (req, res) => {
+  try {
+    const room_id = req.body.room_id;
+
+    const contest_name = await hasura.get_contest_name_by_room(room_id);
+    const base_directory = await utils.get_base_directory();
+    const playbackLocalPath = `${base_directory}/temp/${room_id}/playback.thuaipb`;
+    const playbackCOSPath = `${contest_name}/arena/${room_id}/playback.thuaipb`;
+
+    const cos = await utils.initCOS();
+    const config = await utils.getConfig();
+    await fs.mkdir(`${base_directory}/temp/${room_id}`, { recursive: true });
+    await utils.downloadObject(playbackCOSPath, playbackLocalPath, cos, config);
+
+    res.setHeader(
+      "Content-Disposition",
+      "attachment;filename=playback.thuaipb"
+    );
+    res.status(200).sendFile(playbackLocalPath, () => {
+      utils.deleteAllFilesInDir(`${base_directory}/temp/${room_id}`);
+    });
+  } catch (err) {
+    console.log(err);
+    return res.status(404).send("404 Not Found: Playback not found");
+  }
+});
 
 export default router;

src/routes/competition.ts

@@ -799,5 +799,36 @@ router.post("/finish-one", async (req, res) => {
   }
 });
 
+/**
+ * @param {uuid} room_id
+ * 用于获取回放的路由，直接返回文件。
+ */
+router.get("/playback", async (req, res) => {
+  try {
+    const room_id = req.body.room_id;
+
+    const contest_name = await hasura.get_contest_name_by_room(room_id);
+    const base_directory = await utils.get_base_directory();
+    const playbackLocalPath = `${base_directory}/temp/${room_id}/playback.thuaipb`;
+    const playbackCOSPath = `${contest_name}/competition/${room_id}/playback.thuaipb`;
+
+    const cos = await utils.initCOS();
+    const config = await utils.getConfig();
+    await fs.mkdir(`${base_directory}/temp/${room_id}`, { recursive: true });
+    await utils.downloadObject(playbackCOSPath, playbackLocalPath, cos, config);
+
+    res.setHeader(
+      "Content-Disposition",
+      "attachment;filename=playback.thuaipb"
+    );
+    res.status(200).sendFile(playbackLocalPath, () => {
+      utils.deleteAllFilesInDir(`${base_directory}/temp/${room_id}`);
+    });
+  } catch (err) {
+    console.log(err);
+    return res.status(404).send("404 Not Found: Playback not found");
+  }
+});
+
 
 export default router;