FastGPT/client/src/service/response.ts

import { sseResponseEventEnum } from '@/constants/chat';
import { NextApiResponse } from 'next';
import {
  openaiError,
  openaiAccountError,
  proxyError,
  ERROR_RESPONSE,
  ERROR_ENUM
} from './errorCode';
import { clearCookie, sseResponse, addLog } from './utils/tools';

export interface ResponseType<T = any> {
  code: number;
  message: string;
  data: T;
}

export const jsonRes = <T = any>(
  res: NextApiResponse,
  props?: {
    code?: number;
    message?: string;
    data?: T;
    error?: any;
  }
) => {
  const { code = 200, message = '', data = null, error } = props || {};

  const errResponseKey = typeof error === 'string' ? error : error?.message;
  // Specified error
  if (ERROR_RESPONSE[errResponseKey]) {
    // login is expired
    if (errResponseKey === ERROR_ENUM.unAuthorization) {
      clearCookie(res);
    }

    return res.json(ERROR_RESPONSE[errResponseKey]);
  }

  // another error
  let msg = message || error?.message;
  if ((code < 200 || code >= 400) && !message) {
    msg = error?.message || '请求错误';
    if (typeof error === 'string') {
      msg = error;
    } else if (proxyError[error?.code]) {
      msg = '接口连接异常';
    } else if (error?.response?.data?.error?.message) {
      msg = error?.response?.data?.error?.message;
    } else if (openaiAccountError[error?.response?.data?.error?.code]) {
      msg = openaiAccountError[error?.response?.data?.error?.code];
    } else if (openaiError[error?.response?.statusText]) {
      msg = openaiError[error.response.statusText];
    }

    addLog.error(msg, {
      message: error.message,
      stack: error.stack,
      ...(error.config && {
        config: {
          headers: error.config.headers,
          url: error.config.url,
          data: error.config.data
        }
      }),
      ...(error.response && {
        response: {
          status: error.response.status,
          statusText: error.response.statusText
        }
      })
    });
  }

  res.status(code).json({
    code,
    statusText: '',
    message: msg,
    data: data !== undefined ? data : null
  });
};

export const sseErrRes = (res: NextApiResponse, error: any) => {
  const errResponseKey = typeof error === 'string' ? error : error?.message;

  // Specified error
  if (ERROR_RESPONSE[errResponseKey]) {
    // login is expired
    if (errResponseKey === ERROR_ENUM.unAuthorization) {
      clearCookie(res);
    }

    return sseResponse({
      res,
      event: sseResponseEventEnum.error,
      data: JSON.stringify(ERROR_RESPONSE[errResponseKey])
    });
  }

  let msg = error?.message || '请求错误';
  if (typeof error === 'string') {
    msg = error;
  } else if (proxyError[error?.code]) {
    msg = '接口连接异常';
  } else if (error?.response?.data?.error?.message) {
    msg = error?.response?.data?.error?.message;
  } else if (openaiAccountError[error?.response?.data?.error?.code]) {
    msg = openaiAccountError[error?.response?.data?.error?.code];
  } else if (openaiError[error?.response?.statusText]) {
    msg = openaiError[error.response.statusText];
  }

  addLog.error(`sse error: ${msg}`, {
    message: error.message,
    stack: error.stack,
    ...(error.config && {
      config: {
        headers: error.config.headers,
        url: error.config.url,
        data: error.config.data
      }
    }),
    ...(error.response && {
      response: {
        status: error.response.status,
        statusText: error.response.statusText
      }
    })
  });

  sseResponse({
    res,
    event: sseResponseEventEnum.error,
    data: JSON.stringify({ message: msg })
  });
};