@hed-hog/core 0.0.328 → 0.0.329

package/src/route/dto/create.dto.ts

@@ -1,14 +1,32 @@
-import { getLocaleText } from '@hed-hog/api-locale';
-import { IsIn, IsString } from 'class-validator';
-import { HttpMethod } from '../../types/http-method';
-
-export class CreateDTO {
-  @IsString({ message: (args) => getLocaleText('validation.urlMustBeString', args.value) })
-  url: string;
-
-  @IsString({ message: (args) => getLocaleText('validation.methodMustBeString', args.value) })
-  @IsIn(['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS', 'HEAD'], {
-    message: (args) => getLocaleText('validation.methodMustBeValid', args.value),
-  })
-  method: HttpMethod;
-}
+import { getLocaleText } from '@hed-hog/api-locale';
+import { IsIn, IsOptional, IsString } from 'class-validator';
+import { HttpMethod } from '../../types/http-method';
+
+export type RouteType = 'HTTP' | 'MCP';
+
+const HTTP_METHODS = ['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS', 'HEAD'];
+
+export class CreateDTO {
+  @IsOptional()
+  @IsString({ message: (args) => getLocaleText('validation.urlMustBeString', args.value) })
+  url?: string;
+
+  @IsOptional()
+  @IsString({ message: (args) => getLocaleText('validation.methodMustBeString', args.value) })
+  @IsIn(HTTP_METHODS, {
+    message: (args) => getLocaleText('validation.methodMustBeValid', args.value),
+  })
+  method?: HttpMethod;
+
+  @IsOptional()
+  @IsIn(['HTTP', 'MCP'])
+  type?: RouteType;
+
+  @IsOptional()
+  @IsString()
+  tool_name?: string;
+
+  @IsOptional()
+  @IsString()
+  name?: string;
+}

package/src/route/dto/update.dto.ts

@@ -1,16 +1,31 @@
-import { getLocaleText } from '@hed-hog/api-locale';
-import { IsIn, IsOptional, IsString } from 'class-validator';
-import { HttpMethod } from '../../types/http-method';
-
-export class UpdateDTO {
-  @IsString({ message: (args) => getLocaleText('validation.urlMustBeString', args.value) })
-  @IsOptional()
-  url?: string;
-
-  @IsString({ message: (args) => getLocaleText('validation.methodMustBeString', args.value) })
-  @IsOptional()
-  @IsIn(['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS', 'HEAD'], {
-    message: (args) => getLocaleText('validation.methodMustBeValid', args.value),
-  })
-  method?: HttpMethod;
-}
+import { getLocaleText } from '@hed-hog/api-locale';
+import { IsIn, IsOptional, IsString } from 'class-validator';
+import { HttpMethod } from '../../types/http-method';
+import { RouteType } from './create.dto';
+
+const HTTP_METHODS = ['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS', 'HEAD'];
+
+export class UpdateDTO {
+  @IsOptional()
+  @IsString({ message: (args) => getLocaleText('validation.urlMustBeString', args.value) })
+  url?: string;
+
+  @IsOptional()
+  @IsString({ message: (args) => getLocaleText('validation.methodMustBeString', args.value) })
+  @IsIn(HTTP_METHODS, {
+    message: (args) => getLocaleText('validation.methodMustBeValid', args.value),
+  })
+  method?: HttpMethod;
+
+  @IsOptional()
+  @IsIn(['HTTP', 'MCP'])
+  type?: RouteType;
+
+  @IsOptional()
+  @IsString()
+  tool_name?: string;
+
+  @IsOptional()
+  @IsString()
+  name?: string;
+}

package/src/route/route.service.ts

@@ -17,7 +17,7 @@ export class RouteService {
   ) {}
 
   async list(paginationParams: PaginationDTO) {
-    const fields = ['url', 'method'];
+    const fields = ['url', 'method', 'tool_name', 'name'];
 
     const OR: any[] = this.prismaService.createInsensitiveSearch(
       fields,
@@ -45,20 +45,31 @@ export class RouteService {
     return route;
   }
 
-  async create({ url, method }: CreateDTO, locale: string): Promise<any> {
-
-    const existingRoute = await this.prismaService.route.findFirst({
-      where: {
-        url,
-        method,
-      },
-    });
+  async create({ url, method, type = 'HTTP', tool_name, name }: CreateDTO, locale: string): Promise<any> {
+
+    if (type === 'HTTP' && url && method) {
+      const existingRoute = await this.prismaService.route.findFirst({
+        where: { url, method },
+      });
+      if (existingRoute) {
+        throw new BadRequestException(
+          getLocaleText('itemAlreadyExists', locale, `Route with URL ${url} and method ${method} already exists`).replace('{{item}}', 'Route'),
+        );
+      }
+    }
 
-    if (existingRoute) {
-      throw new BadRequestException(getLocaleText('itemAlreadyExists', locale, `Route with URL ${url} and method ${method} already exists`).replace('{{item}}', 'Route'));
+    if (type === 'MCP' && tool_name) {
+      const existingTool = await this.prismaService.route.findFirst({
+        where: { tool_name },
+      });
+      if (existingTool) {
+        throw new BadRequestException(
+          getLocaleText('itemAlreadyExists', locale, `MCP tool '${tool_name}' already exists`).replace('{{item}}', 'MCP tool'),
+        );
+      }
     }
 
-    return this.prismaService.route.create({ data: { url, method } });
+    return this.prismaService.route.create({ data: { url, method, type: type as any, tool_name, name } });
   }
 
   async update({ id, data }: { id: number; data: UpdateDTO }, locale: string): Promise<any> {