add kemono booru type compatibility

src/booru/booru.service.spec.ts

@@ -3,7 +3,7 @@ import { ConfigService } from '@nestjs/config'
 import { BooruTypesStringEnum } from '@alejandroakbal/universal-booru-wrapper'
 import { BooruService } from './booru.service'
 import { booruQueriesDTO } from './dto/booru-queries.dto'
-import { BooruEndpointParamsDTO } from './dto/request-booru.dto'
+import { BooruEndpointParamsDTO, SupportedBooruType } from './dto/request-booru.dto'
 import { BooruAuthManagerService } from './services/booru-auth-manager.service'
 
 describe('BooruService', () => {
@@ -123,4 +123,30 @@ describe('BooruService', () => {
 
 
   })
+
+  describe('Booru type resolution', () => {
+    it('should resolve kemono when supported and fail clearly otherwise', () => {
+      const maybeKemonoClass = (require('@alejandroakbal/universal-booru-wrapper') as any).Kemono
+
+      const params: BooruEndpointParamsDTO = {
+        booruType: 'kemono' as SupportedBooruType
+      }
+
+      const queries = {
+        baseEndpoint: 'kemono.cr'
+      } as booruQueriesDTO
+
+      if (maybeKemonoClass) {
+        const api = service.buildApiClass(params, queries)
+
+        expect(api).toBeInstanceOf(maybeKemonoClass)
+
+        return
+      }
+
+      expect(() => service.buildApiClass(params, queries)).toThrow(
+        'Kemono booru type requires @alejandroakbal/universal-booru-wrapper version that exports Kemono'
+      )
+    })
+  })
 })

src/booru/booru.service.ts

@@ -16,8 +16,9 @@ import {
   Rule34PahealNet,
   Rule34Xxx
 } from '@alejandroakbal/universal-booru-wrapper'
+import * as UBW from '@alejandroakbal/universal-booru-wrapper'
 import { booruQueriesDTO } from './dto/booru-queries.dto'
-import { BooruEndpointParamsDTO } from './dto/request-booru.dto'
+import { BooruEndpointParamsDTO, SupportedBooruType } from './dto/request-booru.dto'
 import { BooruAuthManagerService } from './services/booru-auth-manager.service'
 
 @Injectable()
@@ -112,7 +113,7 @@ export class BooruService {
     return {}
   }
 
-  private getApiClassByType(booruType: BooruTypesStringEnum) {
+  private getApiClassByType(booruType: SupportedBooruType) {
     switch (booruType) {
       case BooruTypesStringEnum.DANBOORU:
         return Danbooru
@@ -140,6 +141,18 @@ export class BooruService {
 
       case BooruTypesStringEnum.REALBOORU_COM:
         return RealBooruCom
+
+      case 'kemono': {
+        const maybeKemonoClass = (UBW as any).Kemono
+
+        if (!maybeKemonoClass) {
+          throw new Error(
+            'Kemono booru type requires @alejandroakbal/universal-booru-wrapper version that exports Kemono'
+          )
+        }
+
+        return maybeKemonoClass
+      }
     }
   }
 }

src/booru/dto/request-booru.dto.ts

@@ -1,12 +1,16 @@
 import { BooruTypesStringEnum } from '@alejandroakbal/universal-booru-wrapper'
 import { IsDefined, IsIn, IsNotEmpty, IsString } from 'class-validator'
 
-const BooruTypesToArray = Object.values(BooruTypesStringEnum)
+const AdditionalBooruTypes = ['kemono'] as const
+
+export type SupportedBooruType = BooruTypesStringEnum | (typeof AdditionalBooruTypes)[number]
+
+const BooruTypesToArray = [...Object.values(BooruTypesStringEnum), ...AdditionalBooruTypes]
 
 export class BooruEndpointParamsDTO {
   @IsDefined()
   @IsNotEmpty()
   @IsString()
   @IsIn(BooruTypesToArray)
-  readonly booruType: BooruTypesStringEnum
+  readonly booruType: SupportedBooruType
 }