From b85dade35167eca71931cb852dcc853ba94bf783 Mon Sep 17 00:00:00 2001
From: ntr <ntr@smokestack.io>
Date: Sun, 15 Sep 2019 15:42:47 +1000
Subject: [PATCH] game invites

---
 client/assets/styles/controls.less  |   2 +-
 client/assets/styles/styles.less    |   6 ++
 client/src/actions.jsx              |   1 +
 client/src/components/play.ctrl.jsx |  54 ++++++++++
 client/src/events.jsx               |  26 +++++
 client/src/reducers.jsx             |   1 +
 client/src/socket.jsx               |  20 +++-
 server/src/events.rs                |  60 ++++++++++-
 server/src/rpc.rs                   | 149 +++++++++++++++-------------
 server/src/warden.rs                |   4 +
 10 files changed, 250 insertions(+), 73 deletions(-)

diff --git a/client/assets/styles/controls.less b/client/assets/styles/controls.less
index eebcc6c6..58bc0b9f 100644
--- a/client/assets/styles/controls.less
+++ b/client/assets/styles/controls.less
@@ -53,7 +53,7 @@ aside {
             border-color: forestgreen;
         }
 
-        &:active, &:focus {
+        &:active, &:focus, &.enabled {
             background: forestgreen;
             color: black;
             border-color: forestgreen;
diff --git a/client/assets/styles/styles.less b/client/assets/styles/styles.less
index 66bdb2cd..83ab40d9 100644
--- a/client/assets/styles/styles.less
+++ b/client/assets/styles/styles.less
@@ -286,3 +286,9 @@ li {
     background-repeat: no-repeat;
     background-position: center;
 }
+
+#clipboard {
+  width: 1px;
+  height: 1px;
+  padding: 0px;
+}
\ No newline at end of file
diff --git a/client/src/actions.jsx b/client/src/actions.jsx
index d4fe0dc8..ccfadcaa 100644
--- a/client/src/actions.jsx
+++ b/client/src/actions.jsx
@@ -19,6 +19,7 @@ export const setConstructRename = value => ({ type: 'SET_CONSTRUCT_RENAME', valu
 export const setGame = value => ({ type: 'SET_GAME', value });
 export const setInfo = value => ({ type: 'SET_INFO', value });
 export const setEmail = value => ({ type: 'SET_EMAIL', value });
+export const setInvite = value => ({ type: 'SET_INVITE', value });
 export const setInstance = value => ({ type: 'SET_INSTANCE', value });
 export const setInstances = value => ({ type: 'SET_INSTANCES', value });
 export const setItemEquip = value => ({ type: 'SET_ITEM_EQUIP', value });
diff --git a/client/src/components/play.ctrl.jsx b/client/src/components/play.ctrl.jsx
index e87e01e5..0df4f4d0 100644
--- a/client/src/components/play.ctrl.jsx
+++ b/client/src/components/play.ctrl.jsx
@@ -1,11 +1,14 @@
 const preact = require('preact');
 const { connect } = require('preact-redux');
 
+const { errorToast, infoToast } = require('../utils');
+
 const addState = connect(
     function receiveState(state) {
         const {
             ws,
             instances,
+            invite,
         } = state;
 
         function sendInstanceState(id) {
@@ -20,12 +23,18 @@ const addState = connect(
             ws.sendInstanceQueue();
         }
 
+        function sendInstanceInvite() {
+            ws.sendInstanceInvite();
+        }
+
         return {
             instances,
+            invite,
 
             sendInstanceState,
             sendInstanceQueue,
             sendInstancePractice,
+            sendInstanceInvite,
         };
     }
 );
@@ -33,10 +42,12 @@ const addState = connect(
 function JoinButtons(args) {
     const {
         instances,
+        invite,
 
         sendInstanceState,
         sendInstanceQueue,
         sendInstancePractice,
+        sendInstanceInvite,
     } = args;
 
     if (instances.length) {
@@ -55,6 +66,48 @@ function JoinButtons(args) {
         );
     }
 
+    const inviteBtn = () => {
+        if (!invite) {
+            return (
+                <button
+                    class='pvp ready'
+                    onClick={() => sendInstanceInvite()}
+                    type="submit">
+                    Invite
+                </button>
+            );
+        }
+
+        function copyClick(e) {
+            const link = `${document.location.origin}#join=${invite}`;
+            const textArea = document.createElement('textarea', { id: '#clipboard' });
+            textArea.value = link;
+            document.body.appendChild(textArea);
+            textArea.focus();
+            textArea.select();
+
+            try {
+                document.execCommand('copy');
+                infoToast('Invite link copied.');
+            } catch (err) {
+                console.error('link copy error', err);
+                errorToast('Invite link copy error.');
+            }
+
+            document.body.removeChild(textArea);
+            return true;
+        }
+
+        return (
+            <button
+                class='pvp ready enabled'
+                onClick={copyClick}
+                type="submit">
+                Copy Link
+            </button>
+        );
+    };
+
     return (
         <aside class='play-ctrl'>
             <div class="timer-container"></div>
@@ -65,6 +118,7 @@ function JoinButtons(args) {
                     type="submit">
                     PVP
                 </button>
+                {inviteBtn()}
                 <button
                     class='practice ready'
                     onClick={() => sendInstancePractice()}
diff --git a/client/src/events.jsx b/client/src/events.jsx
index 242e2888..79b6eed6 100644
--- a/client/src/events.jsx
+++ b/client/src/events.jsx
@@ -1,3 +1,5 @@
+const querystring = require('query-string');
+
 const eachSeries = require('async/eachSeries');
 const sample = require('lodash/sample');
 
@@ -168,9 +170,20 @@ function registerEvents(store) {
         return store.dispatch(actions.setInstances(v));
     }
 
+    function setInvite(code) {
+        if (!code) return store.dispatch(actions.setInvite(null));
+
+        navigator.clipboard.writeText(code).then(() => {
+            notify(`your invite code ${code} was copied to the clipboard.`);
+        }, () => {});
+
+        return store.dispatch(actions.setInvite(code));
+    }
+
     function setInstance(v) {
         const { account, instance, ws } = store.getState();
         if (v) {
+            setInvite(null);
             const player = v.players.find(p => p.id === account.id);
             store.dispatch(actions.setPlayer(player));
 
@@ -272,6 +285,16 @@ function registerEvents(store) {
     } */
     // setup / localstorage
 
+    function urlHashChange() {
+        const { ws } = store.getState();
+        const cmds = querystring.parse(location.hash);
+
+        if (cmds.join) ws.sendInstanceJoin(cmds.join);
+        return true;
+    }
+
+    window.addEventListener('hashchange', urlHashChange, false);
+
     return {
         clearCombiner,
         clearConstructRename,
@@ -289,12 +312,15 @@ function registerEvents(store) {
         setEmail,
         setInstance,
         setItemInfo,
+        setInvite,
         setPing,
         setShop,
         setTeam,
         setSubscription,
         setWs,
 
+        urlHashChange,
+
         notify,
     };
 }
diff --git a/client/src/reducers.jsx b/client/src/reducers.jsx
index cfc6dad5..349d6e91 100644
--- a/client/src/reducers.jsx
+++ b/client/src/reducers.jsx
@@ -31,6 +31,7 @@ module.exports = {
     constructRename:    createReducer(null, 'SET_CONSTRUCT_RENAME'),
     game:               createReducer(null, 'SET_GAME'),
     email:              createReducer(null, 'SET_EMAIL'),
+    invite:             createReducer(null, 'SET_INVITE'),
     info:               createReducer(null, 'SET_INFO'),
     instance:           createReducer(null, 'SET_INSTANCE'),
     instances:          createReducer([], 'SET_INSTANCES'),
diff --git a/client/src/socket.jsx b/client/src/socket.jsx
index e5c11d5c..1c6f6f37 100644
--- a/client/src/socket.jsx
+++ b/client/src/socket.jsx
@@ -1,5 +1,3 @@
-const querystring = require('query-string');
-
 const toast = require('izitoast');
 const cbor = require('borc');
 
@@ -128,6 +126,14 @@ function createSocket(events) {
         send(['InstanceQueue', {}]);
     }
 
+    function sendInstanceInvite() {
+        send(['InstanceInvite', {}]);
+    }
+
+    function sendInstanceJoin(code) {
+        send(['InstanceJoin', { code }]);
+    }
+
     function sendInstanceReady(instanceId) {
         send(['InstanceReady', { instance_id: instanceId }]);
     }
@@ -241,6 +247,9 @@ function createSocket(events) {
 
         QueueRequested: () => events.notify('pvp queue request received'),
         QueueJoined: () => events.notify('you have joined the pvp queue'),
+        InviteRequested: () => events.notify('pvp queue request received'),
+        Invite: code => events.setInvite(code),
+        Joining: () => events.notify('searching for instance...'),
 
         Error: errHandler,
     };
@@ -286,6 +295,8 @@ function createSocket(events) {
         sendPing();
         sendItemInfo();
 
+        events.urlHashChange();
+
         return true;
     }
 
@@ -312,8 +323,6 @@ function createSocket(events) {
             ws = null;
         }
 
-        console.log(querystring.parse(location.hash));
-
         ws = new WebSocket(SOCKET_URL);
         ws.binaryType = 'arraybuffer';
 
@@ -322,6 +331,7 @@ function createSocket(events) {
         ws.addEventListener('message', onMessage);
         ws.addEventListener('error', onError);
         ws.addEventListener('close', onClose);
+
         return ws;
     }
 
@@ -342,6 +352,8 @@ function createSocket(events) {
         sendInstancePractice,
         sendInstanceQueue,
         sendInstanceState,
+        sendInstanceInvite,
+        sendInstanceJoin,
 
         sendVboxAccept,
         sendVboxApply,
diff --git a/server/src/events.rs b/server/src/events.rs
index da9fe692..ad3d898f 100644
--- a/server/src/events.rs
+++ b/server/src/events.rs
@@ -12,6 +12,7 @@ use account;
 use account::Account;
 use game;
 use instance;
+use names;
 
 use pg::{Db, PgPool};
 use rpc::RpcMessage;
@@ -54,6 +55,9 @@ pub enum Event {
 
     // client events
     Queue(Id),
+    Invite(Id),
+    Join(Id, String),
+    Joined(Id),
 }
 
 struct WsClient {
@@ -62,6 +66,7 @@ struct WsClient {
     tx: Sender<RpcMessage>,
     subs: HashSet<Uuid>,
     pvp: bool,
+    invite: Option<String>,
 }
 
 impl Events {
@@ -115,7 +120,7 @@ impl Events {
                     None => None,
                 };
 
-                let client = WsClient { id, tx, account: account_id, subs: HashSet::new(), pvp: false };
+                let client = WsClient { id, tx, account: account_id, subs: HashSet::new(), pvp: false, invite: None };
                 self.clients.insert(id, client);
 
                 info!("clients={:?}", self.clients.len());
@@ -223,6 +228,59 @@ impl Events {
                 info!("joined game queue id={:?} account={:?}", requester.id, requester.account);
                 return Ok(());
             },
+
+            Event::Invite(id) => {
+                // check whether request is valid
+                let c = self.clients.get_mut(&id)
+                    .ok_or(format_err!("connection not found id={:?}", id))?;
+
+                if let None = c.account {
+                    return Err(err_msg("cannot join pvp queue anonymously"));
+                }
+
+                let code = names::name().split_whitespace().collect::<Vec<&str>>().join("-");
+                info!("pvp invite request id={:?} account={:?} code={:?}", c.id, c.account, code);
+                c.invite = Some(code.clone());
+                c.tx.send(RpcMessage::Invite(code))?;
+                return Ok(());
+            },
+
+            Event::Join(id, code) => {
+                // check whether request is valid
+                let c = self.clients.get(&id)
+                    .ok_or(format_err!("connection not found id={:?}", id))?;
+
+                if let None = c.account {
+                    return Err(err_msg("cannot join pvp queue anonymously"));
+                }
+
+                info!("pvp join request id={:?} account={:?} code={:?}", c.id, c.account, code);
+
+                let inv = self.clients.iter()
+                    .filter(|(_id, c)| c.invite.is_some())
+                    .find(|(_id, c)| match c.invite {
+                        Some(ref c) => *c == code,
+                        None => false,
+                    })
+                    .map(|(_id, c)| PvpRequest { id: c.id, account: c.account.unwrap(), tx: c.tx.clone() })
+                    .ok_or(format_err!("invite not found code={:?}", code))?;
+
+                let join = PvpRequest { id: c.id, account: c.account.unwrap(), tx: c.tx.clone() };
+
+                self.warden.send(GameEvent::Match((join, inv)))?;
+                return Ok(());
+            },
+
+            Event::Joined(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.invite = None;
+                return Ok(());
+            },
+
         }
     }
 }
diff --git a/server/src/rpc.rs b/server/src/rpc.rs
index d3955988..3439a6c4 100644
--- a/server/src/rpc.rs
+++ b/server/src/rpc.rs
@@ -62,6 +62,10 @@ pub enum RpcMessage {
     QueueJoined(()),
     QueueCancelled(()),
 
+    InviteRequested(()),
+    Invite(String),
+    Joining(()),
+
     Error(String),
 }
 
@@ -91,6 +95,8 @@ pub enum RpcRequest {
     SubscriptionState {},
     EmailState {},
 
+    InstanceInvite {},
+    InstanceJoin { code: String },
     InstanceQueue {},
     InstancePractice {},
     InstanceAbandon { instance_id: Uuid },
@@ -136,99 +142,108 @@ impl Connection {
                     None => return Err(err_msg("auth required")),
                 };
 
-                // evented but authorization required
-                match v {
-                    RpcRequest::InstanceQueue {} => {
-                        self.events.send(Event::Queue(self.id))?;
-                        return Ok(RpcMessage::QueueRequested(()));
-                    },
-                    _ => (),
-                };
-
-                // all good, let's make a tx and process
-                let mut tx = db.transaction()?;
-
                 let request = v.clone();
 
                 let response = match v {
-                    RpcRequest::AccountState {} =>
-                        Ok(RpcMessage::AccountState(account.clone())),
-                    RpcRequest::AccountConstructs {} =>
-                        Ok(RpcMessage::AccountConstructs(account::constructs(&mut tx, &account)?)),
-                    RpcRequest::AccountInstances {} =>
-                        Ok(RpcMessage::AccountInstances(account::account_instances(&mut tx, account)?)),
-                    RpcRequest::AccountSetTeam { ids } =>
-                        Ok(RpcMessage::AccountTeam(account::set_team(&mut tx, &account, ids)?)),
+                    // evented but authorization required
+                    RpcRequest::InstanceQueue {} => {
+                        self.events.send(Event::Queue(self.id))?;
+                        Ok(RpcMessage::QueueRequested(()))
+                    },
+                    RpcRequest::InstanceInvite {} => {
+                        self.events.send(Event::Invite(self.id))?;
+                        Ok(RpcMessage::InviteRequested(()))
+                    },
+                    RpcRequest::InstanceJoin { code } => {
+                        self.events.send(Event::Join(self.id, code))?;
+                        Ok(RpcMessage::Joining(()))
+                    },
+                    _ => {
+                        // all good, let's make a tx and process
+                        let mut tx = db.transaction()?;
 
-                    RpcRequest::EmailState {} =>
-                        Ok(RpcMessage::EmailState(mail::select_account(&db, account.id)?)),
+                        let res = match v {
+                            RpcRequest::AccountState {} =>
+                                Ok(RpcMessage::AccountState(account.clone())),
+                            RpcRequest::AccountConstructs {} =>
+                                Ok(RpcMessage::AccountConstructs(account::constructs(&mut tx, &account)?)),
+                            RpcRequest::AccountInstances {} =>
+                                Ok(RpcMessage::AccountInstances(account::account_instances(&mut tx, account)?)),
+                            RpcRequest::AccountSetTeam { ids } =>
+                                Ok(RpcMessage::AccountTeam(account::set_team(&mut tx, &account, ids)?)),
 
-                    RpcRequest::SubscriptionState {} =>
-                        Ok(RpcMessage::SubscriptionState(payments::account_subscription(&db, &self.stripe, &account)?)),
+                            RpcRequest::EmailState {} =>
+                                Ok(RpcMessage::EmailState(mail::select_account(&db, account.id)?)),
 
-                    // RpcRequest::AccountShop {} =>
-                    //     Ok(RpcMessage::AccountShop(mtx::account_shop(&mut tx, &account)?)),
+                            RpcRequest::SubscriptionState {} =>
+                                Ok(RpcMessage::SubscriptionState(payments::account_subscription(&db, &self.stripe, &account)?)),
 
-                    // RpcRequest::ConstructDelete" => handle_construct_delete(data, &mut tx, account),
+                            // RpcRequest::AccountShop {} =>
+                            //     Ok(RpcMessage::AccountShop(mtx::account_shop(&mut tx, &account)?)),
 
-                    RpcRequest::GameState { id } =>
-                        Ok(RpcMessage::GameState(game_state(&mut tx, account, id)?)),
+                            // RpcRequest::ConstructDelete" => handle_construct_delete(data, &mut tx, account),
 
-                    RpcRequest::GameSkill { game_id, construct_id, target_construct_id, skill } =>
-                        Ok(RpcMessage::GameState(game_skill(&mut tx, account, game_id, construct_id, target_construct_id, skill)?)),
+                            RpcRequest::GameState { id } =>
+                                Ok(RpcMessage::GameState(game_state(&mut tx, account, id)?)),
 
-                    RpcRequest::GameSkillClear { game_id } =>
-                        Ok(RpcMessage::GameState(game_skill_clear(&mut tx, account, game_id)?)),
+                            RpcRequest::GameSkill { game_id, construct_id, target_construct_id, skill } =>
+                                Ok(RpcMessage::GameState(game_skill(&mut tx, account, game_id, construct_id, target_construct_id, skill)?)),
 
-                    RpcRequest::GameReady { id } =>
-                        Ok(RpcMessage::GameState(game_ready(&mut tx, account, id)?)),
+                            RpcRequest::GameSkillClear { game_id } =>
+                                Ok(RpcMessage::GameState(game_skill_clear(&mut tx, account, game_id)?)),
 
-                    RpcRequest::InstancePractice {} =>
-                        Ok(RpcMessage::InstanceState(instance_practice(&mut tx, account)?)),
+                            RpcRequest::GameReady { id } =>
+                                Ok(RpcMessage::GameState(game_ready(&mut tx, account, id)?)),
 
-                    // these two can return GameState or InstanceState
-                    RpcRequest::InstanceReady { instance_id } =>
-                        Ok(instance_ready(&mut tx, account, instance_id)?),
-                    RpcRequest::InstanceState { instance_id } =>
-                        Ok(instance_state(&mut tx, instance_id)?),
-                    RpcRequest::InstanceAbandon { instance_id } =>
-                        Ok(instance_abandon(&mut tx, account, instance_id)?),
+                            RpcRequest::InstancePractice {} =>
+                                Ok(RpcMessage::InstanceState(instance_practice(&mut tx, account)?)),
 
-                    RpcRequest::VboxAccept { instance_id, group, index } =>
-                        Ok(RpcMessage::InstanceState(vbox_accept(&mut tx, account, instance_id, group, index)?)),
+                            // these two can return GameState or InstanceState
+                            RpcRequest::InstanceReady { instance_id } =>
+                                Ok(instance_ready(&mut tx, account, instance_id)?),
+                            RpcRequest::InstanceState { instance_id } =>
+                                Ok(instance_state(&mut tx, instance_id)?),
+                            RpcRequest::InstanceAbandon { instance_id } =>
+                                Ok(instance_abandon(&mut tx, account, instance_id)?),
 
-                    RpcRequest::VboxApply { instance_id, construct_id, index } =>
-                        Ok(RpcMessage::InstanceState(vbox_apply(&mut tx, account, instance_id, construct_id, index)?)),
+                            RpcRequest::VboxAccept { instance_id, group, index } =>
+                                Ok(RpcMessage::InstanceState(vbox_accept(&mut tx, account, instance_id, group, index)?)),
 
-                    RpcRequest::VboxCombine { instance_id, indices } =>
-                        Ok(RpcMessage::InstanceState(vbox_combine(&mut tx, account, instance_id, indices)?)),
+                            RpcRequest::VboxApply { instance_id, construct_id, index } =>
+                                Ok(RpcMessage::InstanceState(vbox_apply(&mut tx, account, instance_id, construct_id, index)?)),
 
-                    RpcRequest::VboxDiscard { instance_id } =>
-                        Ok(RpcMessage::InstanceState(vbox_discard(&mut tx, account, instance_id)?)),
+                            RpcRequest::VboxCombine { instance_id, indices } =>
+                                Ok(RpcMessage::InstanceState(vbox_combine(&mut tx, account, instance_id, indices)?)),
 
-                    RpcRequest::VboxReclaim { instance_id, index } =>
-                        Ok(RpcMessage::InstanceState(vbox_reclaim(&mut tx, account, instance_id, index)?)),
+                            RpcRequest::VboxDiscard { instance_id } =>
+                                Ok(RpcMessage::InstanceState(vbox_discard(&mut tx, account, instance_id)?)),
 
-                    RpcRequest::VboxUnequip { instance_id, construct_id, target } =>
-                        Ok(RpcMessage::InstanceState(vbox_unequip(&mut tx, account, instance_id, construct_id, target)?)),
+                            RpcRequest::VboxReclaim { instance_id, index } =>
+                                Ok(RpcMessage::InstanceState(vbox_reclaim(&mut tx, account, instance_id, index)?)),
 
-                    RpcRequest::MtxConstructSpawn {} =>
-                        Ok(RpcMessage::ConstructSpawn(mtx::new_construct(&mut tx, account)?)),
+                            RpcRequest::VboxUnequip { instance_id, construct_id, target } =>
+                                Ok(RpcMessage::InstanceState(vbox_unequip(&mut tx, account, instance_id, construct_id, target)?)),
 
-                    RpcRequest::MtxConstructApply { mtx, construct_id, name } =>
-                        Ok(RpcMessage::AccountTeam(mtx::apply(&mut tx, account, mtx, construct_id, name)?)),
+                            RpcRequest::MtxConstructSpawn {} =>
+                                Ok(RpcMessage::ConstructSpawn(mtx::new_construct(&mut tx, account)?)),
 
-                    RpcRequest::MtxBuy { mtx } =>
-                        Ok(RpcMessage::AccountShop(mtx::buy(&mut tx, account, mtx)?)),
+                            RpcRequest::MtxConstructApply { mtx, construct_id, name } =>
+                                Ok(RpcMessage::AccountTeam(mtx::apply(&mut tx, account, mtx, construct_id, name)?)),
 
-                    RpcRequest::SubscriptionEnding { ending } =>
-                        Ok(RpcMessage::SubscriptionState(payments::subscription_ending(&mut tx, &self.stripe, account, ending)?)),
+                            RpcRequest::MtxBuy { mtx } =>
+                                Ok(RpcMessage::AccountShop(mtx::buy(&mut tx, account, mtx)?)),
 
-                    _ => Err(format_err!("unknown request request={:?}", request)),
+                            RpcRequest::SubscriptionEnding { ending } =>
+                                Ok(RpcMessage::SubscriptionState(payments::subscription_ending(&mut tx, &self.stripe, account, ending)?)),
+
+                            _ => Err(format_err!("unknown request request={:?}", request)),
+                        };
+
+                        tx.commit()?;
+                        res
+                    }
                 };
 
-                tx.commit()?;
-
                 info!("request={:?} account={:?} duration={:?}", request, account.name, begin.elapsed());
 
                 return response;
diff --git a/server/src/warden.rs b/server/src/warden.rs
index 9d58e678..262f6187 100644
--- a/server/src/warden.rs
+++ b/server/src/warden.rs
@@ -90,6 +90,10 @@ impl Warden {
     fn on_match(&mut self, pair: Pair) -> Result<(), Error> {
         info!("received pair={:?}", pair);
 
+        // clear pvp status
+        self.events.send(Event::Joined(pair.0.id))?;
+        self.events.send(Event::Joined(pair.1.id))?;
+
         let db = self.pool.get()?;
         let mut tx = db.transaction()?;