57 lines
1.6 KiB
TypeScript
57 lines
1.6 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import * as Misskey from 'misskey-js';
|
|
import { markRaw } from 'vue';
|
|
import { $i } from '@/account.js';
|
|
import { wsOrigin } from '@/config.js';
|
|
import { StreamMock } from '@/scripts/stream-mock.js';
|
|
import { isEmbedPage } from '@/scripts/embed-page.js';
|
|
import { defaultStore } from '@/store.js';
|
|
|
|
// heart beat interval in ms
|
|
const HEART_BEAT_INTERVAL = 1000 * 60;
|
|
|
|
let stream: Misskey.Stream | null = null;
|
|
let timeoutHeartBeat: ReturnType<typeof setTimeout> | null = null;
|
|
let lastHeartbeatCall = 0;
|
|
|
|
export function useStream(): Misskey.Stream {
|
|
if (stream) return stream;
|
|
|
|
if (isEmbedPage() || defaultStore.state.disableWebsocket === true) {
|
|
stream = markRaw(new StreamMock(wsOrigin, null) as unknown as Misskey.Stream);
|
|
return stream;
|
|
} else {
|
|
stream = markRaw(new Misskey.Stream(wsOrigin, $i ? {
|
|
token: $i.token,
|
|
} : null));
|
|
}
|
|
|
|
if (timeoutHeartBeat) window.clearTimeout(timeoutHeartBeat);
|
|
timeoutHeartBeat = window.setTimeout(heartbeat, HEART_BEAT_INTERVAL);
|
|
|
|
// send heartbeat right now when last send time is over HEART_BEAT_INTERVAL
|
|
document.addEventListener('visibilitychange', () => {
|
|
if (
|
|
!stream
|
|
|| document.visibilityState !== 'visible'
|
|
|| Date.now() - lastHeartbeatCall < HEART_BEAT_INTERVAL
|
|
) return;
|
|
heartbeat();
|
|
});
|
|
|
|
return stream;
|
|
}
|
|
|
|
function heartbeat(): void {
|
|
if (stream != null && document.visibilityState === 'visible') {
|
|
stream.heartbeat();
|
|
}
|
|
lastHeartbeatCall = Date.now();
|
|
if (timeoutHeartBeat) window.clearTimeout(timeoutHeartBeat);
|
|
timeoutHeartBeat = window.setTimeout(heartbeat, HEART_BEAT_INTERVAL);
|
|
}
|