v4.5 (#403)

packages/support/outLink/auth.ts

@@ -63,5 +63,6 @@ export type AuthShareChatInitProps = {
 };
 
 export function authShareChatInit(data: AuthShareChatInitProps) {
+  if (!global.feConfigs?.isPlus) return;
   return POST('/support/outLink/authShareChatInit', data);
 }

packages/support/package.json

@@ -5,7 +5,8 @@
     "@fastgpt/common": "workspace:*",
     "cookie": "^0.5.0",
     "jsonwebtoken": "^9.0.2",
-    "axios": "^1.5.1"
+    "axios": "^1.5.1",
+    "next": "13.5.2"
   },
   "devDependencies": {
     "@types/cookie": "^0.5.2",

packages/support/user/auth.ts

@@ -1,8 +1,8 @@
+import type { NextApiResponse, NextApiRequest } from 'next';
 import Cookie from 'cookie';
-import { authJWT } from './tools';
+import jwt from 'jsonwebtoken';
 import { authOpenApiKey } from '../openapi/auth';
 import { authOutLinkId } from '../outLink/auth';
-
 import { MongoUser } from './schema';
 import type { UserModelSchema } from './type.d';
 import { ERROR_ENUM } from '@fastgpt/common/constant/errorCode';
@@ -39,7 +39,7 @@ export const authUser = async ({
   authBalance = false,
   authOutLink
 }: {
-  req: any;
+  req: NextApiRequest;
   authToken?: boolean;
   authRoot?: boolean;
   authApiKey?: boolean;
@@ -165,3 +165,42 @@ export const authUser = async ({
     apikey: openApiKey
   };
 };
+
+/* 生成 token */
+export function generateToken(userId: string) {
+  const key = process.env.TOKEN_KEY as string;
+  const token = jwt.sign(
+    {
+      userId,
+      exp: Math.floor(Date.now() / 1000) + 60 * 60 * 24 * 7
+    },
+    key
+  );
+  return token;
+}
+// auth token
+export function authJWT(token: string) {
+  return new Promise<string>((resolve, reject) => {
+    const key = process.env.TOKEN_KEY as string;
+
+    jwt.verify(token, key, function (err, decoded: any) {
+      if (err || !decoded?.userId) {
+        reject(ERROR_ENUM.unAuthorization);
+        return;
+      }
+      resolve(decoded.userId);
+    });
+  });
+}
+
+/* set cookie */
+export const setCookie = (res: NextApiResponse, token: string) => {
+  res.setHeader(
+    'Set-Cookie',
+    `token=${token}; Path=/; HttpOnly; Max-Age=604800; Samesite=None; Secure;`
+  );
+};
+/* clear cookie */
+export const clearCookie = (res: NextApiResponse) => {
+  res.setHeader('Set-Cookie', 'token=; Path=/; Max-Age=0');
+};

packages/support/user/tools.ts

@@ -1,28 +0,0 @@
-import jwt from 'jsonwebtoken';
-import { ERROR_ENUM } from '@fastgpt/common/constant/errorCode';
-
-/* 生成 token */
-export const generateToken = (userId: string) => {
-  const key = process.env.TOKEN_KEY as string;
-  const token = jwt.sign(
-    {
-      userId,
-      exp: Math.floor(Date.now() / 1000) + 60 * 60 * 24 * 7
-    },
-    key
-  );
-  return token;
-};
-// auth token
-export const authJWT = (token: string) =>
-  new Promise<string>((resolve, reject) => {
-    const key = process.env.TOKEN_KEY as string;
-
-    jwt.verify(token, key, function (err, decoded: any) {
-      if (err || !decoded?.userId) {
-        reject(ERROR_ENUM.unAuthorization);
-        return;
-      }
-      resolve(decoded.userId);
-    });
-  });