-
Notifications
You must be signed in to change notification settings - Fork 0
/
bot-socket.ts
73 lines (56 loc) · 2.2 KB
/
bot-socket.ts
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
/*
This file is part of botten-nappet -- a Twitch bot and streaming tool.
<https://joelpurra.com/projects/botten-nappet/>
Copyright (c) 2018 Joel Purra <https://joelpurra.com/>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
import Rx from "rxjs";
import io from "socket.io-client";
import ConsoleLog from "./console-log";
export default class BotSocket {
public url: string;
private socket: SocketIOClient.Socket | null;
private sharedSocketObservable: Rx.Observable<any> | null;
private socketObservable: Rx.Observable<any> | null;
private logger: ConsoleLog;
constructor(
logger: ConsoleLog,
url: string,
) {
this.logger = logger;
this.url = url;
this.socket = null;
this.sharedSocketObservable = null;
this.socketObservable = null;
}
public async connect(): Promise<void> {
// TODO: disconnect.
this.socket = io(this.url);
this.socketObservable = Rx.Observable.fromEvent<any>(this.socket, "message");
this.sharedSocketObservable = this.socketObservable.share();
this.socket.on("connect", () => {
this.logger.log("connect");
});
this.socket.on("message", (data: any) => {
this.logger.log("message", data);
});
this.socket.on("disconnect", () => {
this.logger.log("disconnect");
});
}
public get dataObservable(): Rx.Observable<any> {
// assert.hasLength(arguments, 0);
// assert.not.null(this.sharedSocketObservable);
// TODO: better null handling.
return this.sharedSocketObservable!;
}
}