-
Notifications
You must be signed in to change notification settings - Fork 85
/
index.js
106 lines (88 loc) · 3.14 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
const azure = require("@azure/web-pubsub-socket.io");
const { AzureCliCredential } = require("@azure/identity");
// Setup basic express server
const express = require('express');
const app = express();
const path = require('path');
const server = require('http').createServer(app);
const parse = require('url').parse;
// option 1: use connection string
const wpsOptions = {
hub: "eio_hub",
connectionString: process.argv[2] || process.env.WebPubSubConnectionString,
};
// option 2: use AAD auth
// step 1: in service portal , access control tab, add your account as "Web PubSub Service Owner"
// step 2: choose the credential you want to use, for example, you can call `az login` before running the app and use AzureCliCredential
/*
const wpsOptions = {
hub: "eio_hub",
endpoint: "https://sio.webpubsub.azure.com",
credential: new AzureCliCredential()
};
*/
function authentiacte(username) {
// Add your own authentication logic here
// if (...) return false;
return true;
}
async function main() {
const configureNegotiateOptions = (req) => {
const query = parse(req.url || "", true).query
const username = query["username"] ?? "annoyomous";
if (!authentiacte(username)) {
throw new Error(`Authentication Failed for username = ${username}`);
}
return {
userId: username,
};
}
const io = require('socket.io')(server);
azure.useAzureSocketIO(io, wpsOptions);
app.get("/negotiate", azure.negotiate(io, configureNegotiateOptions))
app.use(express.static(path.join(__dirname, 'public')));
// Chatroom
let numUsers = 0;
io.on('connection', socket => {
let addedUser = false;
// when the client emits 'new message', this listens and executes
socket.on('new message', (data) => {
// we tell the client to execute 'new message'
socket.broadcast.emit('new message', {
username: socket.username,
message: data
});
});
// when the client emits 'add user', this listens and executes
socket.on('add user', (username) => {
if (addedUser) return;
// we store the username in the socket session for this client
socket.username = username;
++numUsers;
addedUser = true;
socket.emit('login', {
numUsers: numUsers
});
// echo globally (all clients) that a person has connected
socket.broadcast.emit('user joined', {
username: socket.username,
numUsers: numUsers
});
});
// when the user disconnects.. perform this
socket.on('disconnect', () => {
if (addedUser) {
--numUsers;
// echo globally that this client has left
socket.broadcast.emit('user left', {
username: socket.username,
numUsers: numUsers
});
}
});
});
io.httpServer.listen(3000, () => {
console.log('Visit http://localhost:%d', 3000);
});
}
main();