web2api-server/src/ws/handler.ts

server part of web2api service(1/2)

import type { WSContext } from "hono/ws";
import type {
  RxErrorType,
  RxRegisterType,
  RxStreamErrorType,
  RxStreamType,
  RxType,
  TxErrorType,
} from "./type.js";
import { globalClientManager } from "./client-manage.js";
import consola from "consola/basic";

/*
 * Handlers for handling the message through ws from client
 */

export function handleRegister(data: RxRegisterType, ws: WSContext) {
  const { support, version } = data.content;
  consola.debug(
    `[client] registered, clientVersion:${version}, support:${JSON.stringify(
      support
    )}`
  );
  globalClientManager.setClientVersion(version);
  globalClientManager.setProvideModels(support);
}

export function handleStream(data: RxStreamType, ws: WSContext) {
  const { content, id } = data;
  const { textPart, done } = content;

  if (done) {
    globalClientManager.enqueueChatStream(id, "", "close");
  } else if (textPart)
    globalClientManager.enqueueChatStream(id, content.textPart);
}

export function handleStreamError(data: RxStreamErrorType, ws: WSContext) {
  const { content, id } = data;
  globalClientManager.enqueueChatStream(id, content, "error");
}

export function handleClientError(data: RxErrorType, ws: WSContext) {
  console.error("received client error.", data);
}
export function handleError(data: unknown, ws: WSContext) {
  console.error("received unknown data:", data);
}