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);
}