feat: integrate ts-rest (#5741)

* feat: integrate ts-rest

* chore: classify core contract and pro contract

* chore: update lockfile

* chore: tweak dir structure

* chore: tweak dir structure

Author: xqvvu

packages/global/common/tsRest/fastgpt/client.ts

@@ -0,0 +1,172 @@
+import { contract } from './contracts';
+import { initClient, tsRestFetchApi } from '@ts-rest/core';
+import { TOKEN_ERROR_CODE } from '../../error/errorCode';
+import { getNanoid } from '../../string/tools';
+import { type ApiFetcherArgs } from '@ts-rest/core';
+import { AnyResponseSchema } from '../../type';
+import { ZodError } from 'zod';
+import { getWebReqUrl } from '../../../../web/common/system/utils';
+
+export const client = initClient(contract, {
+  baseUrl: getWebReqUrl('/api'),
+  throwOnUnknownStatus: true,
+  validateResponse: false,
+  credentials: 'include',
+  baseHeaders: {
+    'Content-Type': 'application/json;charset=utf-8'
+  },
+  api: async (args: BeforeFetchOptions) => {
+    const prepare = beforeFetch(args);
+    const response = await tsRestFetchApi(args);
+    return afterFetch(response, prepare);
+  }
+});
+
+const WHITE_LIST = ['/chat/share', '/chat', '/login'];
+async function isTokenExpired() {
+  if (WHITE_LIST.includes(window.location.pathname)) return;
+
+  await client.support.user.account.logout();
+  const lastRoute = encodeURIComponent(location.pathname + location.search);
+  window.location.replace(getWebReqUrl(`/login?lastRoute=${lastRoute}`));
+}
+
+export function checkBusinessCode(code: number) {
+  if (code in TOKEN_ERROR_CODE) {
+    isTokenExpired();
+    return;
+  }
+}
+
+type Item = { id: string; controller: AbortController };
+const queue = new Map<string, Item[]>();
+function checkMaxRequestLimitation(options: { url: string; max: number }): {
+  id: string;
+  signal: AbortSignal;
+  release: () => void;
+} {
+  const { url, max } = options;
+  const id = getNanoid();
+  const controller = new AbortController();
+  const item = queue.get(url);
+
+  const current = item ?? [];
+  if (current.length >= max) {
+    const first = current.shift()!;
+    first.controller.abort();
+  }
+  current.push({ id, controller });
+  if (!item) queue.set(url, current);
+
+  const release = () => {
+    const item = queue.get(url);
+    if (!item) return;
+
+    const index = item.findIndex((item) => item.id === id);
+    if (index !== -1) {
+      item.splice(index, 1);
+    }
+
+    if (item.length <= 0) {
+      queue.delete(url);
+    }
+  };
+
+  return { id, signal: controller.signal, release };
+}
+
+function checkHttpStatus(status: number): status is 200 {
+  if (status !== 200) return false;
+  return true;
+}
+
+type BeforeFetchOptions = ApiFetcherArgs & { max?: number };
+function beforeFetch(options: BeforeFetchOptions):
+  | {
+      limit: { id: string; url: string; release: () => void };
+    }
+  | undefined {
+  const { max, ...args } = options;
+  if (!max || max <= 0) return;
+
+  const { id, signal, release } = checkMaxRequestLimitation({ url: args.path, max });
+  args.fetchOptions ??= {};
+  args.fetchOptions.signal = signal;
+
+  return {
+    limit: { id, url: args.path, release }
+  };
+}
+
+function afterFetch(
+  response: Awaited<ReturnType<typeof tsRestFetchApi>>,
+  prepare?: ReturnType<typeof beforeFetch>
+) {
+  if (checkHttpStatus(response.status)) {
+    try {
+      const body = AnyResponseSchema.parse(response.body);
+
+      response.body = body.data;
+
+      if (prepare?.limit) {
+        prepare.limit.release();
+      }
+
+      return response;
+    } catch (error) {
+      if (error instanceof ZodError) {
+        throw new Error(error.message);
+      }
+
+      throw new Error('Unknown error while intercept response');
+    }
+  } else {
+    throw new Error(`HTTP error, status: ${response.status}`);
+  }
+}
+
+type Client = typeof client;
+type U<T> = { [K in keyof T]: T[K] extends (...args: any[]) => any ? T[K] : U<T[K]> }[keyof T];
+export type Endpoints = U<Client>;
+type _Options<T extends Endpoints> = NonNullable<Parameters<T>[0]>;
+type ExtractBodySchema<T extends Endpoints> = 'body' extends keyof _Options<T>
+  ? _Options<T>['body']
+  : never;
+type ExtractQuerySchema<T extends Endpoints> = 'query' extends keyof _Options<T>
+  ? _Options<T>['query']
+  : never;
+export type Params<T extends Endpoints> = (ExtractBodySchema<T> extends never
+  ? {}
+  : ExtractBodySchema<T>) &
+  (ExtractQuerySchema<T> extends never ? {} : ExtractQuerySchema<T>);
+export type Options<T extends Endpoints> = Omit<_Options<T>, 'body' | 'query'>;
+type Body<T extends Endpoints> = Extract<Awaited<ReturnType<T>>, { status: 200 }>['body'];
+type RestAPIResult<T extends Endpoints> = Body<T>;
+
+const call = async <T extends Endpoints>(
+  api: T,
+  options: _Options<T>
+): Promise<RestAPIResult<T>> => {
+  const res = await api(options as any);
+
+  if (res.status !== 200) {
+    throw new Error(`Unexpected status: ${res.status}`);
+  }
+
+  return res.body as RestAPIResult<T>;
+};
+
+export const RestAPI = <T extends Endpoints>(
+  endpoint: T,
+  transform?: (params: Params<T>) => {
+    body?: ExtractBodySchema<T> extends never ? any : ExtractBodySchema<T>;
+    query?: ExtractQuerySchema<T> extends never ? any : ExtractQuerySchema<T>;
+  }
+) => {
+  return (params?: Params<T>, options?: Options<T>) => {
+    const transformedData = params && transform ? transform(params) : {};
+    const finalOptions = { ...options, ...transformedData } as _Options<T>;
+
+    return call(endpoint, finalOptions);
+  };
+};

packages/global/common/tsRest/fastgpt/contracts/chat/index.ts

@@ -0,0 +1,6 @@
+import { settingContract } from './setting';
+import { c } from '../../../init';
+
+export const chatContract = c.router({
+  setting: settingContract
+});

packages/global/common/tsRest/fastgpt/contracts/chat/setting/favourite/index.ts

@@ -0,0 +1,94 @@
+import { ObjectIdSchema } from '../../../../../../type';
+import {
+  ChatFavouriteAppResponseItemSchema,
+  ChatFavouriteAppUpdateSchema
+} from '../../../../../../../core/chat/favouriteApp/type';
+import { c } from '../../../../../init';
+import { z } from 'zod';
+
+export const favouriteContract = c.router({
+  list: {
+    path: '/proApi/core/chat/setting/favourite/list',
+    method: 'GET',
+    query: z.object({
+      name: z.string().optional().openapi({ example: 'FastGPT' }),
+      tag: z.string().optional().openapi({ example: 'i7Ege2W2' })
+    }),
+    responses: {
+      200: z.array(ChatFavouriteAppResponseItemSchema)
+    },
+    metadata: {
+      tags: ['chat']
+    },
+    description: '获取精选应用列表',
+    summary: '获取精选应用列表'
+  },
+
+  update: {
+    path: '/proApi/core/chat/setting/favourite/update',
+    method: 'PUT',
+    body: ChatFavouriteAppUpdateSchema,
+    responses: {
+      200: c.type<void>()
+    },
+    metadata: {
+      tags: ['chat']
+    },
+    description: '更新精选应用',
+    summary: '更新精选应用'
+  },
+
+  delete: {
+    path: '/proApi/core/chat/setting/favourite/delete',
+    method: 'DELETE',
+    query: z.object({
+      id: ObjectIdSchema
+    }),
+    responses: {
+      200: c.type<void>()
+    },
+    metadata: {
+      tags: ['chat']
+    },
+    description: '删除精选应用',
+    summary: '删除精选应用'
+  },
+
+  order: {
+    path: '/proApi/core/chat/setting/favourite/order',
+    method: 'PUT',
+    body: z.array(
+      z.object({
+        id: ObjectIdSchema,
+        order: z.number()
+      })
+    ),
+    responses: {
+      200: c.type<void>()
+    },
+    metadata: {
+      tags: ['chat']
+    },
+    description: '更新精选应用顺序',
+    summary: '更新精选应用顺序'
+  },
+
+  tags: {
+    path: '/proApi/core/chat/setting/favourite/tags',
+    method: 'PUT',
+    body: z.array(
+      z.object({
+        id: z.string(),
+        tags: z.array(z.string())
+      })
+    ),
+    responses: {
+      200: c.type<void>()
+    },
+    metadata: {
+      tags: ['chat']
+    },
+    description: '更新精选应用标签',
+    summary: '更新精选应用标签'
+  }
+});

packages/global/common/tsRest/fastgpt/contracts/chat/setting/index.ts

@@ -0,0 +1,37 @@
+import {
+  ChatSettingResponseSchema,
+  ChatSettingSchema
+} from '../../../../../../core/chat/setting/type';
+import { c } from '../../../../init';
+import { favouriteContract } from './favourite';
+
+export const settingContract = c.router({
+  favourite: favouriteContract,
+
+  detail: {
+    path: '/proApi/core/chat/setting/detail',
+    method: 'GET',
+    responses: {
+      200: ChatSettingResponseSchema
+    },
+    metadata: {
+      tags: ['chat']
+    },
+    description: '获取聊天设置',
+    summary: '获取聊天设置'
+  },
+
+  update: {
+    path: '/proApi/core/chat/setting/update',
+    method: 'PUT',
+    body: ChatSettingSchema.partial(),
+    responses: {
+      200: c.type<void>()
+    },
+    metadata: {
+      tags: ['chat']
+    },
+    description: '更新聊天设置',
+    summary: '更新聊天设置'
+  }
+});

packages/global/common/tsRest/fastgpt/contracts/index.ts

@@ -0,0 +1,14 @@
+import { supportContract } from './support';
+import { chatContract } from './chat';
+import { c } from '../../init';
+
+// 前端使用的完整合约（开源 + Pro）
+// FastGPT 后端使用的合约
+export const contract = c.router({
+  chat: {
+    ...chatContract
+  },
+  support: {
+    ...supportContract
+  }
+});

packages/global/common/tsRest/fastgpt/contracts/support/index.ts

@@ -0,0 +1,6 @@
+import { userContract } from './user';
+import { c } from '../../../init';
+
+export const supportContract = c.router({
+  user: userContract
+});

packages/global/common/tsRest/fastgpt/contracts/support/user/account/index.ts

@@ -0,0 +1,17 @@
+import { c } from '../../../../../init';
+
+export const accountContract = c.router({
+  logout: {
+    path: '/support/user/account/login',
+    method: 'POST',
+    body: c.type<undefined>(),
+    responses: {
+      200: c.type<void>()
+    },
+    metadata: {
+      tags: ['support']
+    },
+    description: '退出登录',
+    summary: '退出登录'
+  }
+});

packages/global/common/tsRest/fastgpt/contracts/support/user/index.ts

@@ -0,0 +1,6 @@
+import { accountContract } from '../../../../fastgpt/contracts/support/user/account';
+import { c } from '../../../../init';
+
+export const userContract = c.router({
+  account: accountContract
+});

packages/global/common/tsRest/fastgpt/server.ts

@@ -0,0 +1,21 @@
+import { createNextRouter } from '@ts-rest/next';
+import { createNextRoute } from '@ts-rest/next';
+import { contract } from './contracts';
+
+/**
+ * 创建 FastGPT 单个路由
+ */
+export function createServerRoute(
+  implementation: Parameters<typeof createNextRoute<typeof contract>>[1]
+) {
+  return createNextRoute(contract, implementation);
+}
+
+/**
+ * 创建 FastGPT 路由器
+ */
+export function createServerRouter(
+  router: Parameters<typeof createNextRouter<typeof contract>>[1]
+) {
+  return createNextRouter(contract, router);
+}