summaryrefslogtreecommitdiff
path: root/packages/frontend/src/stream.ts
blob: 1e2d31480cbfa171a01e5e5da3db457d9f75344c (plain)
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
/*
 * SPDX-FileCopyrightText: syuilo and other misskey contributors
 * SPDX-License-Identifier: AGPL-3.0-only
 */

import * as Misskey from 'misskey-js';
import { markRaw } from 'vue';
import { $i } from '@/account.js';
import { url } from '@/config.js';

let stream: Misskey.Stream | null = null;
let timeoutHeartBeat: number | null = null;

export let isReloading: boolean = false;

export function useStream(): Misskey.Stream {
	if (stream) return stream;

	stream = markRaw(new Misskey.Stream(url, $i ? {
		token: $i.token,
	} : null));

	timeoutHeartBeat = window.setTimeout(heartbeat, 1000 * 60);

	return stream;
}

export function reloadStream() {
	if (!stream) return useStream();
	if (timeoutHeartBeat) window.clearTimeout(timeoutHeartBeat);
	isReloading = true;

	stream.close();
	stream.once('_connected_', () => isReloading = false);
	stream.stream.reconnect();
	timeoutHeartBeat = window.setTimeout(heartbeat, 1000 * 60);

	return stream;
}

function heartbeat(): void {
	if (stream != null && document.visibilityState === 'visible') {
		stream.heartbeat();
	}
	timeoutHeartBeat = window.setTimeout(heartbeat, 1000 * 60);
}