From 3ee32b3ef211b8a4ef4d5465ffa9608f5a015117 Mon Sep 17 00:00:00 2001
From: ntr <ntr@smokestack.io>
Date: Tue, 12 Nov 2019 16:32:19 +1100
Subject: [PATCH] pvp leave queue implemented

---
 client/src/actions.jsx         |  1 +
 client/src/components/play.jsx | 43 +++++++++++++++++++++++++++-------
 client/src/events.jsx          |  6 +++++
 client/src/reducers.jsx        |  1 +
 client/src/socket.jsx          | 19 ++++++++++++---
 server/src/events.rs           | 14 +++++++++++
 server/src/rpc.rs              |  7 +++++-
 7 files changed, 79 insertions(+), 12 deletions(-)

diff --git a/client/src/actions.jsx b/client/src/actions.jsx
index 59959393..ebec075d 100644
--- a/client/src/actions.jsx
+++ b/client/src/actions.jsx
@@ -41,6 +41,7 @@ export const setReclaiming = value => ({ type: 'SET_RECLAIMING', value });
 export const setShowLog = value => ({ type: 'SET_SHOW_LOG', value });
 export const setShop = value => ({ type: 'SET_SHOP', value });
 export const setSubscription = value => ({ type: 'SET_SUBSCRIPTION', value });
+export const setPvp = value => ({ type: 'SET_PVP', value });
 
 export const setTeam = value => ({ type: 'SET_TEAM', value: Array.from(value) });
 export const setTeamPage = value => ({ type: 'SET_TEAM_PAGE', value });
diff --git a/client/src/components/play.jsx b/client/src/components/play.jsx
index 2059f774..2c97b77f 100644
--- a/client/src/components/play.jsx
+++ b/client/src/components/play.jsx
@@ -14,6 +14,7 @@ const addState = connect(
             account,
             instances,
             invite,
+            pvp,
         } = state;
 
         function sendInstanceState(id) {
@@ -32,15 +33,21 @@ const addState = connect(
             ws.sendInstanceInvite();
         }
 
+        function sendInstanceLeave() {
+            ws.sendInstanceLeave();
+        }
+
         return {
             account,
             instances,
             invite,
+            pvp,
 
             sendInstanceState,
             sendInstanceQueue,
             sendInstancePractice,
             sendInstanceInvite,
+            sendInstanceLeave,
         };
     },
 
@@ -67,11 +74,13 @@ function Play(args) {
         account,
         instances,
         invite,
+        pvp,
 
         sendInstanceState,
         sendInstanceQueue,
         sendInstancePractice,
         sendInstanceInvite,
+        sendInstanceLeave,
 
         setNav,
     } = args;
@@ -125,6 +134,31 @@ function Play(args) {
         );
     };
 
+    const pvpBtn = () => {
+        if (pvp) return (
+            <figure>
+                <button
+                    class="ready"
+                    onClick={() => sendInstanceLeave()}
+                    >
+                    Cancel
+                </button>
+                <figcaption>Finding Opponent</figcaption>
+            </figure>
+        );
+
+        return (
+            <figure>
+                <button
+                    class="ready"
+                    onClick={() => sendInstanceQueue()}>
+                    PVP
+                </button>
+                <figcaption>Matchmaking</figcaption>
+            </figure>
+        );
+    }
+
     const subscription = account.subscribed
         ? <button
             class="yellow-btn"
@@ -142,14 +176,7 @@ function Play(args) {
         if (!instances.length) {
             return (
                 <div class='list play'>
-                    <figure>
-                        <button
-                            class="ready"
-                            onClick={() => sendInstanceQueue()}>
-                            PVP
-                        </button>
-                        <figcaption>Matchmaking</figcaption>
-                    </figure>
+                    {pvpBtn()}
                     {inviteBtn()}
                     <figure>
                         <button
diff --git a/client/src/events.jsx b/client/src/events.jsx
index 898f1ef6..9c7a149b 100644
--- a/client/src/events.jsx
+++ b/client/src/events.jsx
@@ -30,6 +30,10 @@ function registerEvents(store) {
         store.dispatch(actions.setPing(ping));
     }
 
+    function setPvp(v) {
+        store.dispatch(actions.setPvp(v));
+    }
+
     function setNav(v) {
         store.dispatch(actions.setNav(v));
     }
@@ -210,6 +214,7 @@ function registerEvents(store) {
         const { account, instance, ws, tutorial } = store.getState();
         if (v) {
             setInvite(null);
+            setPvp(false);
             const player = v.players.find(p => p.id === account.id);
             store.dispatch(actions.setPlayer(player));
 
@@ -370,6 +375,7 @@ function registerEvents(store) {
         setItemInfo,
         setInvite,
         setPing,
+        setPvp,
         setShop,
         setTeam,
         setSubscription,
diff --git a/client/src/reducers.jsx b/client/src/reducers.jsx
index 0ddb68b6..c9479f56 100644
--- a/client/src/reducers.jsx
+++ b/client/src/reducers.jsx
@@ -48,6 +48,7 @@ module.exports = {
     player:             createReducer(null, 'SET_PLAYER'),
     reclaiming:         createReducer(false, 'SET_RECLAIMING'),
     shop:               createReducer(false, 'SET_SHOP'),
+    pvp:                createReducer(null, 'SET_PVP'),
 
     subscription:       createReducer(null, 'SET_SUBSCRIPTION'),
 
diff --git a/client/src/socket.jsx b/client/src/socket.jsx
index 9ab92870..959bfc30 100644
--- a/client/src/socket.jsx
+++ b/client/src/socket.jsx
@@ -151,6 +151,10 @@ function createSocket(events) {
         send(['InstanceQueue', {}]);
     }
 
+    function sendInstanceLeave() {
+        send(['InstanceLeave', {}]);
+    }
+
     function sendInstanceInvite() {
         send(['InstanceInvite', {}]);
     }
@@ -274,8 +278,16 @@ function createSocket(events) {
         Pong: onPong,
         Demo: onDemo,
 
-        QueueRequested: () => events.notify('PVP queue request received.'),
-        QueueJoined: () => events.notify('You have joined the PVP queue.'),
+        // QueueRequested: () => events.notify('PVP queue request received.'),
+        QueueRequested: () => true,
+        QueueJoined: () => {
+            events.notify('You have joined the PVP queue.');
+            events.setPvp(true);
+        },
+        QueueLeft: () => {
+            events.notify('You have left the PVP queue.');
+            events.setPvp(false);
+        },
         QueueFound: () => events.notify('Your PVP game has started.'),
         InviteRequested: () => events.notify('PVP invite request received.'),
         Invite: code => events.setInvite(code),
@@ -343,7 +355,7 @@ function createSocket(events) {
             message: 'disconnected',
             position: 'topRight',
         });
-        return setTimeout(connect, 2000);
+        return setTimeout(connect, 5000);
     }
 
     function connect() {
@@ -390,6 +402,7 @@ function createSocket(events) {
         sendInstanceInvite,
         sendInstanceJoin,
         sendInstanceChat,
+        sendInstanceLeave,
 
         sendVboxAccept,
         sendVboxAcceptEquip,
diff --git a/server/src/events.rs b/server/src/events.rs
index 294cfb03..fef6d71c 100644
--- a/server/src/events.rs
+++ b/server/src/events.rs
@@ -60,6 +60,7 @@ pub enum Event {
     Invite(Id),
     Join(Id, String),
     Joined(Id),
+    Leave(Id),
 
     Chat(Id, Uuid, String),
     ChatClear(Id, Uuid),
@@ -249,6 +250,7 @@ impl Events {
                 // or flag the requester as pvp ready
                 let requester = self.clients.get_mut(&id).unwrap();
                 requester.pvp = true;
+                requester.tx.send(RpcMessage::QueueJoined(()))?;
                 info!("joined game queue id={:?} account={:?}", requester.id, requester.account);
                 return Ok(());
             },
@@ -305,6 +307,18 @@ impl Events {
                 return Ok(());
             },
 
+            Event::Leave(id) => {
+                // check whether request is valid
+                let c = self.clients.get_mut(&id)
+                    .ok_or(format_err!("connection not found id={:?}", id))?;
+
+                c.pvp = false;
+                c.tx.send(RpcMessage::QueueLeft(()))?;
+                info!("left game queue id={:?} account={:?}", c.id, c.account);
+                return Ok(());
+            },
+
+
             Event::Chat(id, instance, msg) => {
                 // set the chat state of this connection
                 {
diff --git a/server/src/rpc.rs b/server/src/rpc.rs
index fbd8a2f8..0e5fa28d 100644
--- a/server/src/rpc.rs
+++ b/server/src/rpc.rs
@@ -63,7 +63,7 @@ pub enum RpcMessage {
 
     QueueRequested(()),
     QueueJoined(()),
-    QueueCancelled(()),
+    QueueLeft(()),
     QueueFound(()),
 
     InviteRequested(()),
@@ -105,6 +105,7 @@ pub enum RpcRequest {
     InstanceInvite {},
     InstanceJoin { code: String },
     InstanceQueue {},
+    InstanceLeave {},
     InstancePractice {},
     InstanceAbandon { instance_id: Uuid },
     InstanceReady { instance_id: Uuid },
@@ -168,6 +169,10 @@ impl Connection {
                         self.events.send(Event::Join(self.id, code))?;
                         Ok(RpcMessage::Joining(()))
                     },
+                    RpcRequest::InstanceLeave {} => {
+                        self.events.send(Event::Leave(self.id))?;
+                        Ok(RpcMessage::Processing(()))
+                    },
 
                     RpcRequest::InstanceChat { instance_id, index } => {
                         if !account.subscribed {