✨ feat: add builtin search with SearXNG

lobehub · Oct 12, 2024 · 6779a39 · 6779a39
1 parent c59d8aa
commit 6779a39
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 5 deletions.
diff --git a/src/config/tools.ts b/src/config/tools.ts
@@ -4,11 +4,11 @@ import { z } from 'zod';
 export const getToolsConfig = () => {
  return createEnv({
  runtimeEnv: {
- SEARXNG_URL: process.env.SEARXNG_URL || 'https://searx.tiekoetter.com/',
+ SEARXNG_URL: process.env.SEARXNG_URL,
  },
 
  server: {
- SEARXNG_URL: z.string().url(),
+ SEARXNG_URL: z.string().url().optional(),
  },
  });
 };

diff --git a/src/libs/trpc/client/tools.ts b/src/libs/trpc/client/tools.ts
@@ -6,6 +6,13 @@ import type { ToolsRouter } from '@/server/routers/tools';
 export const toolsClient = createTRPCClient<ToolsRouter>({
  links: [
  httpBatchLink({
+ headers: async () => {
+ // dynamic import to avoid circular dependency
+ const { createHeaderWithAuth } = await import('@/services/_auth');
+
+ return createHeaderWithAuth();
+ },
+ maxURLLength: 2083,
  transformer: superjson,
  url: '/trpc/tools',
  }),

diff --git a/src/server/modules/SearXNG.ts b/src/server/modules/SearXNG.ts
@@ -1,4 +1,5 @@
 import qs from 'query-string';
+import urlJoin from 'url-join';
 
 import { SearchResponse } from '@/types/tool/search';
 
@@ -17,7 +18,7 @@ export class SearXNGClient {
  q: query,
  });
 
- const response = await fetch(`${this.baseUrl}/search?${searchParams}`);
+ const response = await fetch(urlJoin(this.baseUrl, `/search?${searchParams}`));
 
  return await response.json();
  } catch (error) {

diff --git a/src/server/routers/tools/search.ts b/src/server/routers/tools/search.ts
@@ -1,18 +1,26 @@
+import { TRPCError } from '@trpc/server';
 import { z } from 'zod';
 
 import { toolsEnv } from '@/config/tools';
-import { authedProcedure, router } from '@/libs/trpc';
+import { isServerMode } from '@/const/version';
+import { authedProcedure, passwordProcedure, router } from '@/libs/trpc';
 import { SearXNGClient } from '@/server/modules/SearXNG';
 
+const searchProcedure = isServerMode ? authedProcedure : passwordProcedure;
+
 export const searchRouter = router({
- query: authedProcedure
+ query: searchProcedure
  .input(
  z.object({
  query: z.string(),
  searchEngine: z.array(z.string()).optional(),
  }),
  )
  .query(async ({ input }) => {
+ if (!toolsEnv.SEARXNG_URL) {
+ throw new TRPCError({ code: 'SERVICE_UNAVAILABLE', message: 'SearXNG is not configured' });
+ }
+
  const client = new SearXNGClient(toolsEnv.SEARXNG_URL);
 
  return await client.search(input.query, input.searchEngine);