From 0078507ea40c40e8fc1b004f0413453129d7348d Mon Sep 17 00:00:00 2001
From: zhl <zhl010101@gmail.com>
Date: Fri, 18 Dec 2020 19:36:24 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E4=B8=80=E4=B8=AA=E6=9F=A5?=
 =?UTF-8?q?=E6=89=BE=E5=AF=B9=E9=9D=A2=E7=8E=A9=E5=AE=B6=E7=9A=84=E6=96=B9?=
 =?UTF-8?q?=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/global.d.ts          |  5 +++++
 src/rooms/GeneralRoom.ts | 16 ++++++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/src/global.d.ts b/src/global.d.ts
index ee8da43..621ab06 100644
--- a/src/global.d.ts
+++ b/src/global.d.ts
@@ -260,6 +260,11 @@ declare module "colyseus" {
          */
         getAssistClient(sessionId: string): RobotClient;
 
+        /**
+         * 获取对位玩家
+         * @param srcPlayer
+         */
+        getOppositePlayer(srcPlayer: string|Player): Player;
     }
 }
 
diff --git a/src/rooms/GeneralRoom.ts b/src/rooms/GeneralRoom.ts
index 76ee8b2..724dfae 100644
--- a/src/rooms/GeneralRoom.ts
+++ b/src/rooms/GeneralRoom.ts
@@ -282,4 +282,20 @@ export class GeneralRoom extends Room {
   getAssistClient(sessionId: string): RobotClient {
     return this.assistMap.get(sessionId);
   }
+
+  getOppositePlayer(srcPlayer: string | Player): Player {
+    let playerId;
+    if (typeof srcPlayer === 'string') {
+      playerId = srcPlayer as string;
+    } else {
+      playerId = srcPlayer.id;
+    }
+    if (!this.state.players.has(playerId)) {
+      return null;
+    }
+    const sessionIds = [...this.state.players.keys()];
+    let idx = sessionIds.indexOf(playerId);
+    let playerId2 = sessionIds[(2 + idx) % 4];
+    return this.state.players.get(playerId2);
+  }
 }