From c47691b16afc54182682171dbfeb7ec2651685cb Mon Sep 17 00:00:00 2001
From: notnull <notnull@anarchyplanet.org>
Date: Sun, 21 Jul 2019 05:44:41 -0400
Subject: [PATCH] added chat functionality, namespaces

---
 package-lock.json      |   5 ++
 package.json           |   4 +-
 server/socket/index.js |  34 +++++++++++--
 src/App.js             | 110 +++++++++++++++++++++++++++++++++++------
 4 files changed, 133 insertions(+), 20 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index 908f46b..1af5023 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -8389,6 +8389,11 @@
         }
       }
     },
+    "moment": {
+      "version": "2.24.0",
+      "resolved": "https://registry.npmjs.org/moment/-/moment-2.24.0.tgz",
+      "integrity": "sha512-bV7f+6l2QigeBBZSM/6yTNq4P2fNpSWj/0e7jQcy87A8e7o2nAfP/34/2ky5Vw4B9S446EtIhodAzkFCcR4dQg=="
+    },
     "morgan": {
       "version": "1.9.1",
       "resolved": "https://registry.npmjs.org/morgan/-/morgan-1.9.1.tgz",
diff --git a/package.json b/package.json
index babb683..52f84ca 100644
--- a/package.json
+++ b/package.json
@@ -5,12 +5,14 @@
   "dependencies": {
     "bootstrap": "^4.3.1",
     "cors": "^2.8.5",
+    "moment": "^2.24.0",
     "morgan": "^1.9.1",
     "react": "^16.8.6",
     "react-dom": "^16.8.6",
     "react-scripts": "3.0.1",
     "socket.io": "^2.2.0",
-    "socket.io-client": "^2.2.0"
+    "socket.io-client": "^2.2.0",
+    "uuid": "^3.3.2"
   },
   "scripts": {
     "start": "react-scripts start",
diff --git a/server/socket/index.js b/server/socket/index.js
index b930ab7..b2d8285 100644
--- a/server/socket/index.js
+++ b/server/socket/index.js
@@ -1,13 +1,39 @@
 module.exports = io => {
   io.on('connection', async socket => {
     console.log(`A socket connection to the server has been made: ${socket.id}`)
+    socket.join('#chat')
+    io.emit(
+      'received rooms',
+      Object.keys(io.sockets.adapter.rooms)
+        .filter(r => r[0] === '#')
+        .map(k => ({
+          roomName: k,
+          sockets: Object.keys(io.sockets.adapter.rooms[k]['sockets']),
+        }))
+    )
+    socket.broadcast.emit('user connected', { socketId: socket.id })
 
     socket.on('message', message => {
-      socket.broadcast.emit('message', message)
+      io.to(message.room).emit(message)
+    })
+    socket.on('join', payload => {
+      console.log(payload.room)
+      socket.join({ room: payload.room })
+    })
+    socket.on('disconnect', async () => {
+      io.emit('user disconnected', { socketId: socket.id })
+      console.log(`${socket.id} has disconnected.`)
+      io.emit(
+        'received rooms',
+        Object.keys(io.sockets.adapter.rooms)
+          .filter(r => r[0] === '#')
+          .map(k => ({
+            roomName: k,
+            sockets: Object.keys(io.sockets.adapter.rooms[k]['sockets']),
+          }))
+      )
     })
 
-    socket.on('disconnect', async () =>
-      console.log(`${socket.id} has disconnected.`)
-    )
+    console.log(io.sockets.adapter.rooms['#chat'])
   })
 }
diff --git a/src/App.js b/src/App.js
index 2cabb61..3418b39 100644
--- a/src/App.js
+++ b/src/App.js
@@ -1,27 +1,67 @@
 import React from 'react'
 import socket from './socket'
 
-const initialState = { loading: true }
-const initialData = { message: 'Hello, world!' }
+import uuid from 'uuid'
+import moment from 'moment'
+
+const initialState = {
+  loading: true,
+  messages: [],
+  message: '',
+  rooms: [],
+  room: '/',
+}
 
 class App extends React.Component {
   constructor() {
     super()
     this.state = initialState
     this.socket = socket
+    this.handleChange = this.handleChange.bind(this)
+    this.handleSubmit = this.handleSubmit.bind(this)
+    this.joinRoom = this.joinRoom.bind(this)
   }
 
   async fetchData() {
-    try {
-      const data = await initialData
-      this.setState({ ...data })
-    } catch (e) {
-    } finally {
-      this.setState({ loading: false })
-    }
+    this.setState({ loading: false })
   }
-  componentWillMount() {
+
+  componentDidMount() {
     this.fetchData()
+    this.socket.on('message', msg => {
+      const messages = this.state.messages.concat(msg)
+      this.setState({ messages })
+    })
+
+    this.socket.on('user connected', payload => {
+      console.log('a new user connected!', payload)
+    })
+
+    this.socket.on('user disconnected', payload => {
+      console.log('a user disconnected.', payload)
+    })
+
+    this.socket.on('received rooms', rooms => {
+      console.log(rooms)
+      this.setState({ rooms })
+    })
+  }
+
+  handleChange(e) {
+    this.setState({ [e.target.name]: e.target.value })
+  }
+
+  handleSubmit(e) {
+    e.preventDefault()
+    const message = {
+      id: uuid(),
+      date: moment().format('MMMM D YYYY, h:mm a'),
+      socketId: this.socket.id,
+      text: this.state.message,
+      room: this.state.room,
+    }
+    this.sendMessage(message)
+    this.setState({ message: '' })
   }
   renderLoading() {
     return <div>loading...</div>
@@ -30,7 +70,12 @@ class App extends React.Component {
   renderError() {
     return <div>something went wrong.</div>
   }
-
+  joinRoom(room) {
+    this.socket.emit('join', { room })
+  }
+  sendMessage(msg) {
+    this.socket.emit('message', msg)
+  }
   renderApp() {
     return (
       <main role="main" className="d-flex h-100 flex-column">
@@ -39,10 +84,20 @@ class App extends React.Component {
         <div className="d-flex h-100">
           {/*Side Bar 2*/}
           <div
-            className="d-none d-sm-block col-sm-2 bg-dark text-light p-2"
+            className="d-flex flex-column col-sm-2 bg-dark text-light p-2"
             id="sidebar2"
           >
-            <h5 className="border-bottom pb-2">Channels</h5>
+            <h5 className=" border-bottom pb-2">Channels</h5>
+            {this.state.rooms.map(r => (
+              <button
+                key={r.roomName}
+                className="btn btn-dark d-flex"
+                onClick={() => this.joinRoom(r.roomName)}
+              >
+                <div className="mr-auto">{r.roomName}</div>
+                <div className="ml-auto">{r.sockets.length}</div>
+              </button>
+            ))}
           </div>
 
           {/*Main Section*/}
@@ -53,8 +108,33 @@ class App extends React.Component {
             </div>
             {/*Main Section Content*/}
             <div className="d-flex flex-column border border-secondary flex-grow-1">
-              <div className="bg-dark flex-grow-1 p-2">orange</div>
-              <div className="bg-light p-2">blue</div>
+              <div className="bg-dark flex-grow-1 p-2 text-light">
+                {this.state.messages
+                  .filter(m => m.room === this.state.room)
+                  .map(m => (
+                    <div className="d-flex justify-content-between" key={m.id}>
+                      <div>{m.date}</div>
+                      <div>{m.socketId}</div>
+                      <div>{m.text}</div>
+                    </div>
+                  ))}
+              </div>
+              <form onSubmit={this.handleSubmit}>
+                <input
+                  className="bg-light p-2 w-100"
+                  name="message"
+                  value={this.state.message}
+                  placeholder="Enter message"
+                  onChange={this.handleChange}
+                />
+                <button
+                  className="btn btn-dark"
+                  type="submit"
+                  onSubmit={this.handleSubmit}
+                >
+                  submit
+                </button>
+              </form>
             </div>
           </div>
         </div>