47 lines
1.1 KiB
TypeScript
47 lines
1.1 KiB
TypeScript
/*
|
|
* 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);
|
|
}
|