99 lines
2.3 KiB
TypeScript
99 lines
2.3 KiB
TypeScript
|
import { get, writable } from 'svelte/store';
|
||
|
|
import { addNotification, notiStore } from '../stores/noti';
|
||
|
|
import {
|
||
|
|
recipeData,
|
||
|
|
recipeDataError,
|
||
|
|
recipeLoading,
|
||
|
|
recipeOverviewData,
|
||
|
|
recipeStreamMeta
|
||
|
|
} from '../stores/recipeStore';
|
||
|
|
|
||
|
|
export const messages = writable<string[]>([]);
|
||
|
|
|
||
|
|
type WSMessage = { type: string; payload: any };
|
||
|
|
|
||
|
|
const handlers: Record<string, (payload: any) => void> = {
|
||
|
|
chat: (p) => messages.update((m) => [...m, p]),
|
||
|
|
ping: (p) => console.log('ping from server'),
|
||
|
|
recipeResponse: (p) => {
|
||
|
|
let recipe_result = p.result;
|
||
|
|
let recipe_request = p.request;
|
||
|
|
|
||
|
|
if (recipe_result) {
|
||
|
|
addNotification('INFO:Start fetch recipe!');
|
||
|
|
}
|
||
|
|
},
|
||
|
|
stream_data_start: (p) => {
|
||
|
|
let stream_id = p.stream_id;
|
||
|
|
let total_size = p.total_size;
|
||
|
|
let chunk_size = p.chunk_size;
|
||
|
|
|
||
|
|
if (stream_id) {
|
||
|
|
addNotification('INFO:Start streaming data');
|
||
|
|
recipeLoading.set(true);
|
||
|
|
recipeStreamMeta.set({
|
||
|
|
id: stream_id,
|
||
|
|
total_size: total_size,
|
||
|
|
chunk_size: chunk_size,
|
||
|
|
progress: 0
|
||
|
|
});
|
||
|
|
recipeData.set([]);
|
||
|
|
recipeOverviewData.set([]);
|
||
|
|
}
|
||
|
|
},
|
||
|
|
stream_data_error: (p) => {
|
||
|
|
recipeLoading.set(false);
|
||
|
|
recipeDataError.set(p);
|
||
|
|
|
||
|
|
setTimeout(() => {
|
||
|
|
addNotification(`ERROR:${p.error}`);
|
||
|
|
}, 2000);
|
||
|
|
},
|
||
|
|
stream_data_chunk: (p) => {
|
||
|
|
let current_meta = get(recipeStreamMeta);
|
||
|
|
if (current_meta) {
|
||
|
|
let stream_id = current_meta.id;
|
||
|
|
|
||
|
|
let progress_response_id = p.stream_id;
|
||
|
|
if (stream_id === progress_response_id) {
|
||
|
|
let current_response_end = p.start_idx + current_meta.chunk_size;
|
||
|
|
let percent = (current_response_end / current_meta.total_size) * 100;
|
||
|
|
if (percent > 100) {
|
||
|
|
percent = 100;
|
||
|
|
}
|
||
|
|
let data = p.data;
|
||
|
|
let currentData = get(recipeData);
|
||
|
|
for (let rp of data) {
|
||
|
|
currentData.push(rp);
|
||
|
|
}
|
||
|
|
recipeData.set(currentData);
|
||
|
|
recipeStreamMeta.set({
|
||
|
|
...current_meta,
|
||
|
|
progress: percent
|
||
|
|
});
|
||
|
|
|
||
|
|
// build overview
|
||
|
|
|
||
|
|
if (percent == 100) {
|
||
|
|
addNotification(`INFO:Current progress ${percent}%`);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
},
|
||
|
|
stream_data_end: (p) => {
|
||
|
|
recipeLoading.set(false);
|
||
|
|
},
|
||
|
|
stream_patch_update: (p) => {}
|
||
|
|
};
|
||
|
|
|
||
|
|
export function handleIncomingMessages(raw: string) {
|
||
|
|
const msg: WSMessage = JSON.parse(raw);
|
||
|
|
console.log(`${new Date().toLocaleTimeString()}:ws msg`, msg);
|
||
|
|
if (msg == null) {
|
||
|
|
// error response
|
||
|
|
addNotification('ERR:No response from server');
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
handlers[msg.type]?.(msg.payload);
|
||
|
|
}
|