69 lines
2.1 KiB
TypeScript
69 lines
2.1 KiB
TypeScript
import toast from "react-hot-toast";
|
|
import { WS_BACKEND_URL } from "./config";
|
|
import {
|
|
APP_ERROR_WEB_SOCKET_CODE,
|
|
USER_CLOSE_WEB_SOCKET_CODE,
|
|
} from "./constants";
|
|
import { FullGenerationSettings } from "./types";
|
|
|
|
const ERROR_MESSAGE =
|
|
"Error generating code. Check the Developer Console AND the backend logs for details. Feel free to open a Github issue.";
|
|
|
|
const CANCEL_MESSAGE = "Code generation cancelled";
|
|
|
|
export function generateCode(
|
|
wsRef: React.MutableRefObject<WebSocket | null>,
|
|
params: FullGenerationSettings,
|
|
onChange: (chunk: string) => void,
|
|
onSetCode: (code: string) => void,
|
|
onStatusUpdate: (status: string) => void,
|
|
onCancel: () => void,
|
|
onComplete: () => void
|
|
) {
|
|
const wsUrl = `${WS_BACKEND_URL}/generate-code`;
|
|
console.log("Connecting to backend @ ", wsUrl);
|
|
|
|
const ws = new WebSocket(wsUrl);
|
|
wsRef.current = ws;
|
|
|
|
ws.addEventListener("open", () => {
|
|
ws.send(JSON.stringify(params));
|
|
});
|
|
|
|
ws.addEventListener("message", async (event: MessageEvent) => {
|
|
const response = JSON.parse(event.data);
|
|
if (response.type === "chunk") {
|
|
onChange(response.value);
|
|
} else if (response.type === "status") {
|
|
onStatusUpdate(response.value);
|
|
} else if (response.type === "setCode") {
|
|
onSetCode(response.value);
|
|
} else if (response.type === "error") {
|
|
console.error("Error generating code", response.value);
|
|
toast.error(response.value);
|
|
}
|
|
});
|
|
|
|
ws.addEventListener("close", (event) => {
|
|
console.log("Connection closed", event.code, event.reason);
|
|
if (event.code === USER_CLOSE_WEB_SOCKET_CODE) {
|
|
toast.success(CANCEL_MESSAGE);
|
|
onCancel();
|
|
} else if (event.code === APP_ERROR_WEB_SOCKET_CODE) {
|
|
console.error("Known server error", event);
|
|
onCancel();
|
|
} else if (event.code !== 1000) {
|
|
console.error("Unknown server or connection error", event);
|
|
toast.error(ERROR_MESSAGE);
|
|
onCancel();
|
|
} else {
|
|
onComplete();
|
|
}
|
|
});
|
|
|
|
ws.addEventListener("error", (error) => {
|
|
console.error("WebSocket error", error);
|
|
toast.error(ERROR_MESSAGE);
|
|
});
|
|
}
|