✨ feat: can replace index url

lobehub · Aug 20, 2023 · dc3d0cc · dc3d0cc
1 parent 2e78c39
commit dc3d0cc
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 20 deletions.
diff --git a/.env.example b/.env.example
@@ -0,0 +1,2 @@
+# relpace url with your internal plugins index url
+PLUGINS_INDEX_URL=https://example.com/index.json
diff --git a/README.md b/README.md
@@ -6,4 +6,4 @@
 [coverage]: https://codecov.io/gh/lobehub/chat-plugins-gateway/branch/master/graph/badge.svg
 [codecov-url]: https://codecov.io/gh/lobehub/chat-plugins-gateway/branch/master
 
-Lobe Chat 插件市场列表
+Lobe Chat 插件网关
diff --git a/api/v1/_validator.ts b/api/v1/_validator.ts
diff --git a/api/v1/runner.ts b/api/v1/runner.ts
@@ -5,19 +5,21 @@ import {
   ErrorType,
   LobeChatPlugin,
   LobeChatPluginsMarketIndex,
+  PluginRequestPayload,
   createErrorResponse,
   marketIndexSchema,
   pluginManifestSchema,
   pluginMetaSchema,
+  pluginRequestPayloadSchema,
 } from '@lobehub/chat-plugin-sdk';
 
-import { PluginPayload, payloadSchema } from './_validator';
-
 export const config = {
   runtime: 'edge',
 };
 
-const INDEX_URL = `https://registry.npmmirror.com/@lobehub/lobe-chat-plugins/latest/files`;
+const DEFAULT_INDEX_URL =
+  process.env.PLUGINS_INDEX_URL ??
+  'https://registry.npmmirror.com/@lobehub/lobe-chat-plugins/latest/files';
 
 export default async (req: Request) => {
   // ==========  1. 校验请求方法 ========== //
@@ -27,22 +29,23 @@ export default async (req: Request) => {
     });
 
   // ==========  2. 校验请求入参基础格式 ========== //
-  const requestPayload = (await req.json()) as PluginPayload;
+  const requestPayload = (await req.json()) as PluginRequestPayload;
 
-  const payloadParseResult = payloadSchema.safeParse(requestPayload);
+  const payloadParseResult = pluginRequestPayloadSchema.safeParse(requestPayload);
 
   if (!payloadParseResult.success)
     return createErrorResponse(ErrorType.BadRequest, payloadParseResult.error);
 
-  const { name, arguments: args } = requestPayload;
+  const { name, arguments: args, indexUrl } = requestPayload;
 
   console.info(`plugin call: ${name}`);
 
+  const marketIndexUrl = indexUrl ?? DEFAULT_INDEX_URL;
   // ==========  3. 获取插件市场索引 ========== //
 
   let marketIndex: LobeChatPluginsMarketIndex | undefined;
   try {
-    const indexRes = await fetch(INDEX_URL);
+    const indexRes = await fetch(marketIndexUrl);
     marketIndex = await indexRes.json();
   } catch (error) {
     console.error(error);
@@ -52,7 +55,7 @@ export default async (req: Request) => {
   // 插件市场索引不存在
   if (!marketIndex)
     return createErrorResponse(ErrorType.PluginMarketIndexNotFound, {
-      indexUrl: INDEX_URL,
+      indexUrl,
       message: '[gateway] plugin market index not found',
     });
 
@@ -62,7 +65,7 @@ export default async (req: Request) => {
   if (!indexParseResult.success)
     return createErrorResponse(ErrorType.PluginMarketIndexInvalid, {
       error: indexParseResult.error,
-      indexUrl: INDEX_URL,
+      indexUrl,
       marketIndex,
       message: '[gateway] plugin market index is invalid',
     });
@@ -89,7 +92,7 @@ export default async (req: Request) => {
   // 校验插件是否存在
   if (!pluginMeta)
     return createErrorResponse(ErrorType.PluginMetaNotFound, {
-      message: `[gateway] plugin '${name}' is not found，please check the plugin list in ${INDEX_URL}, or create an issue to [lobe-chat-plugins](https://github.com/lobehub/lobe-chat-plugins/issues)`,
+      message: `[gateway] plugin '${name}' is not found，please check the plugin list in ${indexUrl}, or create an issue to [lobe-chat-plugins](https://github.com/lobehub/lobe-chat-plugins/issues)`,
       name,
     });
 

diff --git a/package.json b/package.json
@@ -47,7 +47,7 @@
     "prettier": "^2",
     "semantic-release": "^21",
     "typescript": "^5",
-    "vercel": "^29",
+    "vercel": "^28.20.0",
     "vitest": "latest"
   }
 }