From 2a221f180ebf6f86e3709401804108fcda65184f Mon Sep 17 00:00:00 2001
From: marvin-borner@live.com
Date: Wed, 11 Apr 2018 18:36:44 +0200
Subject: Started implementing chat encryption

---
 assets/php/Chatserver/src/ChatProcessor.php | 118 ++++++++++++++++++++++++++++
 1 file changed, 118 insertions(+)
 create mode 100644 assets/php/Chatserver/src/ChatProcessor.php

(limited to 'assets/php/Chatserver/src/ChatProcessor.php')

diff --git a/assets/php/Chatserver/src/ChatProcessor.php b/assets/php/Chatserver/src/ChatProcessor.php
new file mode 100644
index 0000000..da78c9b
--- /dev/null
+++ b/assets/php/Chatserver/src/ChatProcessor.php
@@ -0,0 +1,118 @@
+<?php
+
+namespace Websocket;
+
+//use Ratchet\MessageComponentInterface;
+use Ratchet\ConnectionInterface;
+use Ratchet\WebSocket\MessageComponentInterface;
+use Ratchet\RFC6455\Messaging\MessageInterface;
+use Nubs\RandomNameGenerator\Alliteration;
+
+class ChatProcessor implements MessageComponentInterface
+{
+    protected $clients;
+    private $subscriptions;
+    private $users;
+    private $connectedUsersNames;
+
+    public function __construct() {
+        $this->clients = new \SplObjectStorage;
+        $this->subscriptions = [];
+        $this->users = [];
+        $this->connectedUsersNames = [];
+    }
+
+    public function onOpen(ConnectionInterface $conn) {
+        $generator = new Alliteration();
+        $this->clients->attach($conn);
+        $this->users[$conn->resourceId] = $conn;
+        $this->connectedUsersNames[$conn->resourceId] = $generator->getName();
+    }
+
+    /*public function onMessage(ConnectionInterface $from, $msg) {
+        $numRecv = count($this->clients) - 1;
+        echo sprintf('Connection %d sending message "%s" to %d other connection%s' . "\n"
+            , $from->resourceId, $msg, $numRecv, $numRecv == 1 ? '' : 's');
+
+        foreach ($this->clients as $client) {
+            if ($from === $client) {
+                $client->send("<b>You</b> - " . $msg);
+            } else {
+                $client->send("<b>" . $from->resourceId . "</b> - " . $msg);
+            }
+        }
+    }
+    */
+
+    public function onMessage(ConnectionInterface $conn, MessageInterface $msg) {
+        $data = json_decode($msg);
+        switch ($data->ClientMessageType) {
+            case "Subscribe":
+                $this->subscriptions[$conn->resourceId] = $data->Channel;
+                foreach ($this->subscriptions as $id => $channel) {
+                    if ($this->subscriptions[$conn->resourceId] == $channel) {
+                        $MessageObject = new \stdClass();
+                        $MessageObject->ServerMessage = true;
+                        $MessageObject->ServerMessageType = "GroupJoin";
+                        $MessageObject->GroupName = $channel;
+                        $MessageObject->Username = $this->connectedUsersNames[$conn->resourceId];
+                        if ($id === $conn->resourceId) {
+                            $MessageObject->WasHimself = true;
+                        } else {
+                            $MessageObject->WasHimself = false;
+                        }
+                        $MessageJson = json_encode($MessageObject, true);
+                        $this->users[$id]->send($MessageJson);
+                    }
+                }
+                break;
+            case "Message":
+                if (isset($this->subscriptions[$conn->resourceId])) {
+                    $target = $this->subscriptions[$conn->resourceId];
+                    foreach ($this->subscriptions as $id => $channel) {
+                        if ($channel == $target) {
+                            $MessageObject = new \stdClass();
+                            $MessageObject->ServerMessage = false;
+                            $MessageObject->Username = $this->connectedUsersNames[$conn->resourceId];
+                            $MessageObject->Message = htmlspecialchars($data->Message);
+                            if ($id === $conn->resourceId) {
+                                $MessageObject->WasHimself = true;
+                            } else {
+                                $MessageObject->WasHimself = false;
+                            }
+                            $MessageJson = json_encode($MessageObject, true);
+                            $this->users[$id]->send($MessageJson);
+                        }
+                    }
+                }
+        }
+    }
+
+    public function onClose(ConnectionInterface $conn) {
+        $this->clients->detach($conn);
+        foreach ($this->clients as $client) {
+            if (isset($this->subscriptions[$conn->resourceId])) {
+                $target = $this->subscriptions[$conn->resourceId];
+                foreach ($this->subscriptions as $id => $channel) {
+                    if ($channel == $target) {
+                        $MessageObject = new \stdClass();
+                        $MessageObject->ServerMessage = true;
+                        $MessageObject->ServerMessageType = "UserDisconnect";
+                        $MessageObject->Username = $this->connectedUsersNames[$conn->resourceId];
+                        $MessageJson = json_encode($MessageObject, true);
+                        $this->users[$id]->send($MessageJson);
+                    }
+                }
+            }
+        }
+        unset($this->users[$conn->resourceId]);
+        unset($this->subscriptions[$conn->resourceId]);
+        unset($this->connectedUsersNames[$conn->resourceId]);
+    }
+
+    public function onError(ConnectionInterface $conn, \Exception $e) {
+        echo "An error has occurred: {$e->getMessage()}\n";
+
+        $conn->close();
+    }
+}
\ No newline at end of file
-- 
cgit v1.2.3