support direct load from ygocdb data

.dockerignore

@@ -36,6 +36,7 @@ lerna-debug.log*
 /data
 /output
 /config.yaml
+/ygocdb-data
 
 .git*
 Dockerfile

.gitignore

@@ -35,4 +35,5 @@ lerna-debug.log*
 
 /data
 /output
-/config.yaml
+/config.yaml
\ No newline at end of file
+/ygocdb-data
\ No newline at end of file

src/app.controller.ts

@@ -15,12 +15,22 @@ import {
 import { FileInterceptor } from '@nestjs/platform-express';
 import { FileUploadDto } from './dto/file-upload.dto';
 import YGOProDeck from 'ygopro-deck-encode';
-import { ApiError, BlankReturnMessageDto, DataQuery } from 'nesties';
+import {
+  ApiBlankResponse,
+  ApiError,
+  BlankReturnMessageDto,
+  DataQuery,
+  RequireToken,
+} from 'nesties';
 import { FillOptionsDto } from './dto/fill-options.dto';
+import { CardDataService } from './card-data/card-data.service';
 
 @Controller()
 export class AppController {
-  constructor(private readonly appService: AppService) {}
+  constructor(
+    private readonly appService: AppService,
+    private readonly cardDataService: CardDataService,
+  ) {}
 
   @Post('fill')
   @ApiOperation({
@@ -53,4 +63,15 @@ export class AppController {
     }
     return this.appService.fillDeckForm(ydk, dto);
   }
+
+  @Post('reload-ygocdb')
+  @ApiOperation({
+    summary: 'Reload YGOCDB Data',
+  })
+  @ApiBlankResponse()
+  @RequireToken()
+  async reloadYGOCDBData() {
+    await this.cardDataService.loadYGOCDBData();
+    return new BlankReturnMessageDto(200, 'success');
+  }
 }

src/app.service.ts

@@ -29,6 +29,7 @@ export class AppService extends ConsoleLogger {
     const doc = await PDFDocument.load(
       await fs.promises.readFile('./resources/deck_cn.pdf'),
     );
+    this.log(`Loaded PDF template for deck export.`);
     const directDrawFields = ['name', 'event', 'lastInitial'] as const;
     for (const field of directDrawFields) {
       if (!dto[field]) continue;
@@ -38,6 +39,7 @@ export class AppService extends ConsoleLogger {
         Coordinates[field],
       );
     }
+    this.log(`Filled direct draw fields.`);
     if (dto.date) {
       const date = moment(dto.date);
       // we need a dd-mm-yy format
@@ -61,6 +63,7 @@ export class AppService extends ConsoleLogger {
           moveRight(Coordinates.dateFirst, i),
         );
       }
+      this.log(`Filled date field.`);
     }
 
     const totalCounts = [ydk.main.length, ydk.extra.length, ydk.side.length];
@@ -73,6 +76,8 @@ export class AppService extends ConsoleLogger {
       );
     }
 
+    this.log(`Filled total counts.`);
+
     const copyFirstPageToBottom = async () => {
       const [newPage] = await doc.copyPages(doc, [0]);
       doc.addPage(newPage);
@@ -87,6 +92,7 @@ export class AppService extends ConsoleLogger {
       const addCount = pc - pageCount;
       for (let i = 0; i < addCount; ++i) {
         await copyFirstPageToBottom();
+        this.log(`Added page ${pageCount + i + 1} to PDF for deck export.`);
       }
       pageCount = pc;
     };
@@ -138,14 +144,22 @@ export class AppService extends ConsoleLogger {
     const filterType = (type: number) =>
       ydk.main.filter((id) => cardDataMap[id].type === type);
     await drawCards(filterType(1), Coordinates.main.monsters);
+    this.log(`Filled monster cards.`);
     await drawCards(filterType(2), Coordinates.main.spells);
+    this.log(`Filled spell cards.`);
     await drawCards(filterType(4), Coordinates.main.traps);
+    this.log(`Filled trap cards.`);
     await drawCards(ydk.extra, Coordinates.extra, 15);
+    this.log(`Filled extra deck cards.`);
     await drawCards(ydk.side, Coordinates.side, 15);
+    this.log(`Filled side deck cards.`);
 
     doc.removePage(0);
 
-    return new StreamableFile(Buffer.from(await doc.save()), {
+    const resBuf = Buffer.from(await doc.save());
+    this.log(`Generated PDF buffer for deck export.`);
+
+    return new StreamableFile(resBuf, {
       type: 'application/pdf',
       disposition: `attachment; filename="deck_${Date.now()}.pdf"`,
     });

src/card-data/card-data.service.ts

@@ -6,6 +6,7 @@ import { CacheKey, CacheTTL } from 'aragami';
 import { lastValueFrom } from 'rxjs';
 import BetterLock from 'better-lock';
 import { GenericReturnMessageDto } from 'nesties';
+import * as fs from 'node:fs';
 
 @CacheTTL(86400 * 30)
 export class CardData {
@@ -19,6 +20,13 @@ export class CardData {
   type: number; // 0x1 for monster, 0x2 for spell, 0x4 for trap
 }
 
+type YGOCDBData = {
+  id: number;
+  sc_name: string;
+  cn_name: string;
+  data: { type: number };
+};
+
 @Injectable()
 export class CardDataService {
   constructor(
@@ -28,15 +36,20 @@ export class CardDataService {
 
   private logger = new ConsoleLogger('CardDataService');
 
-  private async fetchCardData(id: number): Promise<CardData> {
+  private parseYgoCardType(card: YGOCDBData) {
+    // console.log(card);
     const data = new CardData();
-    data.id = id;
+    if (!card.id || !card.data) return;
+    data.id = card.id;
+    data.name = card.sc_name || card.cn_name || '未知卡片';
+    data.type = card.data.type & 0x7;
+    return data;
+  }
+
+  private async fetchCardData(id: number): Promise<CardData> {
     this.logger.log(`Fetching card data for ID ${id}`);
     const { data: resp, status } = await lastValueFrom(
-      this.http.get<{
-        sc_name: string;
-        data: { type: number };
-      }>('https://ygocdb.com/api/v0/card/' + id, {
+      this.http.get<YGOCDBData>('https://ygocdb.com/api/v0/card/' + id, {
         timeout: 30000,
         params: {
           show: 'all',
@@ -51,8 +64,7 @@ export class CardDataService {
         `Card ID ${id} not found`,
       ).toException();
     }
-    data.name = resp.sc_name;
-    data.type = resp.data.type & 0x7;
+    const data = this.parseYgoCardType(resp);
     this.logger.log(
       `Fetched card data for ID ${id}: ${data.type} ${data.name}`,
     );
@@ -61,7 +73,36 @@ export class CardDataService {
 
   private cardIdLock = new BetterLock();
 
+  private ygocdbExisting = new Map<number, CardData>();
+
+  async loadYGOCDBData() {
+    const content = await fs.promises.readFile('./ygocdb-data/cards.json');
+    const cards: YGOCDBData[] = Object.values(
+      JSON.parse(content.toString('utf-8')),
+    );
+    this.logger.log(`Loading ${cards.length} cards from YGOCDB data`);
+    this.ygocdbExisting.clear();
+    for (const card of cards) {
+      const data = this.parseYgoCardType(card);
+      this.ygocdbExisting.set(card.id, data);
+    }
+  }
+
+  async onApplicationBootstrap() {
+    try {
+      await this.loadYGOCDBData();
+      this.logger.log(
+        `Loaded ${this.ygocdbExisting.size} cards from YGOCDB data`,
+      );
+    } catch (e) {
+      this.logger.warn(`Failed to load YGOCDB data: ${e.message}`);
+    }
+  }
+
   async getCardData(id: number): Promise<CardData> {
+    if (this.ygocdbExisting.has(id)) {
+      return this.ygocdbExisting.get(id);
+    }
     return this.cardIdLock.acquire(id.toString(), () =>
       this.aragami.cache(CardData, id.toString(), () => this.fetchCardData(id)),
     );