remake for latest biu

package.json

@@ -24,21 +24,30 @@
     "@nestjs/common": "^7.5.1",
     "@nestjs/core": "^7.5.1",
     "@nestjs/platform-express": "^7.5.1",
-    "@types/lodash": "^4.14.168",
+    "@nestjs/swagger": "^4.8.0",
+    "@nestjs/typeorm": "^7.1.5",
+    "@types/cookie": "^0.4.0",
     "axios": "^0.21.1",
-    "lodash": "^4.17.20",
+    "class-transformer": "^0.4.0",
+    "class-validator": "^0.13.1",
+    "cookie": "^0.4.1",
+    "lodash": "^4.17.21",
     "moment": "^2.29.1",
+    "mysql": "^2.18.1",
     "p-queue": "^6.6.2",
     "reflect-metadata": "^0.1.13",
     "rimraf": "^3.0.2",
-    "rxjs": "^6.6.3"
+    "rxjs": "^6.6.3",
+    "swagger-ui-express": "^4.1.6",
+    "typeorm": "^0.2.32"
   },
   "devDependencies": {
     "@nestjs/cli": "^7.5.1",
     "@nestjs/schematics": "^7.1.3",
     "@nestjs/testing": "^7.5.1",
-    "@types/express": "^4.17.8",
+    "@types/express": "^4.17.12",
     "@types/jest": "^26.0.15",
+    "@types/lodash": "^4.14.170",
     "@types/node": "^14.14.6",
     "@types/supertest": "^2.0.10",
     "@typescript-eslint/eslint-plugin": "^4.6.1",

src/app.controller.ts

 import { Controller, Get } from '@nestjs/common';
 import { AppService } from './app.service';
+import { ApiTags } from '@nestjs/swagger';
 
 @Controller()
 export class AppController {
   constructor(private readonly appService: AppService) {}
 
   @Get('tips.json')
+  @ApiTags('tips')
   async getTips(): Promise<string[]> {
     return await this.appService.getTips();
   }

src/app.service.ts

@@ -4,18 +4,32 @@ import moment from 'moment';
 import PQueue from 'p-queue';
 import axios from 'axios';
 import _ from 'lodash';
+import qs from 'qs';
+import cookie from 'cookie';
 
 const blacklistedWords = ['曲', '词', '曲', '詞', '[Vv]ocal', ':', '：'];
 const blacklistedRegex = blacklistedWords.map((word) => new RegExp(word));
 
+interface LoginInfo {
+  email: string;
+  password: string;
+}
+
 @Injectable()
 export class AppService {
   cache: string[];
   lastFetchTime: moment.Moment;
   queue: PQueue;
+  loginInfo: LoginInfo;
   constructor(private log: AppLogger) {
     this.log.setContext('ygopro-tips-biu-generator');
     this.queue = new PQueue({ concurrency: 1 });
+    if (process.env.BIU_EMAIL && process.env.BIU_PASSWORD) {
+      this.loginInfo = {
+        email: process.env.BIU_EMAIL,
+        password: process.env.BIU_PASSWORD,
+      };
+    }
   }
 
   async getTips(): Promise<string[]> {
@@ -68,11 +82,49 @@ export class AppService {
     }
   }
 
+  async getLoginCookie() {
+    if (!this.loginInfo) {
+      return undefined;
+    }
+    try {
+      this.log.log(`Logging in.`);
+      const ret = await axios.post(
+        'https://web.biu.moe/User/doLogin',
+        qs.stringify(this.loginInfo),
+        {
+          responseType: 'json',
+          timeout: 30000,
+        },
+      );
+      if (!ret.data.status) {
+        this.log.error(`Login failed: ${ret.data.message}`);
+        return undefined;
+      }
+      const setCookieContent = ret.headers['set-cookie'] as string[];
+      if (!setCookieContent) {
+        return undefined;
+      }
+      const contentHits = setCookieContent.find((c) => c.includes('biuInfo'));
+      if (!contentHits) {
+        return undefined;
+      }
+      const token = cookie.parse(contentHits).biuInfo as string;
+      this.log.log(`Login success: ${token}`);
+      return cookie.serialize('biuInfo', token);
+    } catch (e) {
+      this.log.error(`Login error: ${e.toString()}`);
+      return undefined;
+    }
+  }
+
   async fetchTipsTask(): Promise<string[]> {
     try {
       this.log.log(`Fetching song list.`);
       const { data } = await axios.get('https://web.biu.moe/Index/home', {
         responseType: 'document',
+        headers: {
+          Cookie: (await this.getLoginCookie()) || '',
+        },
         timeout: 30000,
       });
       const urls = (data as string).match(/href="\/s(\d+)"/g);

src/main.ts

 import { NestFactory } from '@nestjs/core';
+import { SwaggerModule, DocumentBuilder } from '@nestjs/swagger';
 import { AppModule } from './app.module';
 
 async function bootstrap() {
   const app = await NestFactory.create(AppModule);
   app.enableCors();
+
+  const documentConfig = new DocumentBuilder()
+    .setTitle('ygopro-tips-biu-generator')
+    .setDescription('YGOPro Tips 生成工具')
+    .setVersion('1.0')
+    .addTag('tips')
+    .build();
+
+  const document = SwaggerModule.createDocument(app, documentConfig);
+  SwaggerModule.setup('docs', app, document);
+
   await app.listen(3000);
 }
 bootstrap();