refactor(publishing): consolidate social media API error handling and retry strategy in publishing (#188)

Author: Yuuki-Sakura

project/backend/apps/aitoearn-channel/src/common/guards/sk-key-auth.guard.ts

@@ -0,0 +1,45 @@
+import {
+  CanActivate,
+  createParamDecorator,
+  ExecutionContext,
+  Injectable,
+  Logger,
+  UnauthorizedException,
+} from '@nestjs/common'
+import { Reflector } from '@nestjs/core'
+import { SkKeyService } from '../../core/sk-key/sk-key.service'
+
+export const GetSkKey = createParamDecorator(
+  (data: string, ctx: ExecutionContext) => {
+    const req = ctx.switchToHttp().getRequest()
+    return req['skKey']
+  },
+)
+
+@Injectable()
+export class SkKeyAuthGuard implements CanActivate {
+  private readonly logger = new Logger(SkKeyAuthGuard.name)
+  constructor(
+    private reflector: Reflector,
+    private readonly skKeyService: SkKeyService,
+  ) {}
+
+  async canActivate(context: ExecutionContext): Promise<boolean> {
+    const request = context.switchToHttp().getRequest()
+    // 获取令牌sk
+    const key = request.headers['sk-key'] || request.query['sk-key']
+
+    try {
+      const keyInfo = await this.skKeyService.getInfo(key)
+      if (!keyInfo) {
+        throw new UnauthorizedException('令牌验证失败')
+      }
+      request['skKey'] = keyInfo
+    }
+    catch (error) {
+      this.logger.error(error)
+      throw new UnauthorizedException('令牌验证失败')
+    }
+    return true
+  }
+}

project/backend/apps/aitoearn-channel/src/common/guards/skKeyAuth.guard.ts

@@ -7,7 +7,7 @@ import {
   UnauthorizedException,
 } from '@nestjs/common'
 import { Reflector } from '@nestjs/core'
-import { SkKeyService } from '../../core/skKey/skKey.service'
+import { SkKeyService } from '../../core/sk-key/sk-key.service'
 
 export const GetSkKey = createParamDecorator(
   (data: string, ctx: ExecutionContext) => {

project/backend/apps/aitoearn-channel/src/core/account/account.module.ts

@@ -2,7 +2,7 @@ import { Global, Module } from '@nestjs/common'
 import { MongooseModule } from '@nestjs/mongoose'
 import { Account, AccountSchema } from '../../libs/database/schema/account.schema'
 import { AccountService } from './account.service'
-import { PublishRecordService } from './publishRecord.service'
+import { PublishRecordService } from './publish-record.service'
 
 @Global()
 @Module({

…rc/core/account/publishRecord.service.ts …c/core/account/publish-record.service.tsproject/backend/apps/aitoearn-channel/src/core/account/publishRecord.service.ts renamed to project/backend/apps/aitoearn-channel/src/core/account/publish-record.service.ts project/backend/apps/aitoearn-channel/src/core/account/publishRecord.service.ts renamed to project/backend/apps/aitoearn-channel/src/core/account/publish-record.service.ts

@@ -1,34 +1,25 @@
 import { Module } from '@nestjs/common'
-import { McpModule as TheMcpModule } from '@rekog/mcp-nest'
-import { SkKeyAuthGuard } from '../common/guards/skKeyAuth.guard'
 import { AliGreenModule } from '../core/ali-green/ali-green.module'
-import { PublishModule } from '../core/publish/publish.module'
 import { AccountModule } from './account/account.module'
-import { DataCubeModule } from './dataCube/dataCube.module'
+import { DataCubeModule } from './data-cube/data-cube.module'
 import { EngagementModule } from './engagement/engagement.module'
 import { FileModule } from './file/file.module'
 import { InteracteModule } from './interact/interact.module'
-import { McpModule } from './mcp/mcp.module'
-import { MetaModule } from './plat/meta/meta.module'
-import { PlatModule } from './plat/plat.module'
-import { TiktokModule } from './plat/tiktok/tiktok.module'
-import { TwitterModule } from './plat/twitter/twitter.module'
-import { WxPlatModule } from './plat/wxPlat/wxPlat.module'
-import { YoutubeModule } from './plat/youtube/youtube.module'
-import { SkKeyModule } from './skKey/skKey.module'
+import { MetaModule } from './platforms/meta/meta.module'
+import { PlatModule } from './platforms/platforms.module'
+import { TiktokModule } from './platforms/tiktok/tiktok.module'
+import { TwitterModule } from './platforms/twitter/twitter.module'
+import { WxPlatModule } from './platforms/wx-plat/wx-plat.module'
+import { YoutubeModule } from './platforms/youtube/youtube.module'
+import { PublishModule } from './publishing/publishing.module'
+import { SkKeyModule } from './sk-key/sk-key.module'
 import { TestModule } from './test/test.module'
 
 @Module({
   imports: [
     TestModule,
     FileModule,
     SkKeyModule,
-    McpModule,
-    TheMcpModule.forRoot({
-      name: 'channel-mcp-server',
-      version: '1.0.0',
-      guards: [SkKeyAuthGuard],
-    }),
     AccountModule,
     PublishModule,
     TwitterModule,

project/backend/apps/aitoearn-channel/src/core/core.module.ts

@@ -8,7 +8,7 @@
 import { Injectable, Logger } from '@nestjs/common'
 import { OnEvent } from '@nestjs/event-emitter'
 import { AccountType, AitoearnServerClientService } from '@yikart/aitoearn-server-client'
-import { BilibiliService } from '../plat/bilibili/bilibili.service'
+import { BilibiliService } from '../platforms/bilibili/bilibili.service'
 import { DataCubeBase } from './data.base'
 
 @Injectable()

…rc/core/dataCube/bilibiliData.service.ts …/core/data-cube/bilibili-data.service.tsproject/backend/apps/aitoearn-channel/src/core/dataCube/bilibiliData.service.ts renamed to project/backend/apps/aitoearn-channel/src/core/data-cube/bilibili-data.service.ts project/backend/apps/aitoearn-channel/src/core/dataCube/bilibiliData.service.ts renamed to project/backend/apps/aitoearn-channel/src/core/data-cube/bilibili-data.service.ts

@@ -9,11 +9,11 @@ import { Body, Controller, Post } from '@nestjs/common'
 import { AccountType } from '@yikart/aitoearn-server-client'
 import { AppException, ResponseCode } from '@yikart/common'
 import { AccountService } from '../account/account.service'
-import { BilibiliDataService } from './bilibiliData.service'
+import { BilibiliDataService } from './bilibili-data.service'
 import { DataCubeBase } from './data.base'
-import { AccountDto, ArcDto } from './dto/dataCube.dto'
-import { WxGzhDataService } from './wxGzhData.service'
-import { YoutubeDataService } from './youtubeData.service'
+import { AccountDto, ArcDto } from './dto/data-cube.dto'
+import { WxGzhDataService } from './wx-gzh-data.service'
+import { YoutubeDataService } from './youtube-data.service'
 
 @Controller()
 export class DataCubeController {

…src/core/dataCube/dataCube.controller.ts …c/core/data-cube/data-cube.controller.tsproject/backend/apps/aitoearn-channel/src/core/dataCube/dataCube.controller.ts renamed to project/backend/apps/aitoearn-channel/src/core/data-cube/data-cube.controller.ts project/backend/apps/aitoearn-channel/src/core/dataCube/dataCube.controller.ts renamed to project/backend/apps/aitoearn-channel/src/core/data-cube/data-cube.controller.ts

@@ -0,0 +1,23 @@
+import { Module } from '@nestjs/common'
+import { PinterestDataService } from '../../core/data-cube/pinterest-data.service'
+import { PinterestModule } from '../../core/platforms/pinterest/pinterest.module'
+import { AccountModule } from '../account/account.module'
+import { BilibiliModule } from '../platforms/bilibili/bilibili.module'
+import { MetaModule } from '../platforms/meta/meta.module'
+import { WxPlatModule } from '../platforms/wx-plat/wx-plat.module'
+import { YoutubeModule } from '../platforms/youtube/youtube.module'
+import { BilibiliDataService } from './bilibili-data.service'
+import { DataCubeController } from './data-cube.controller'
+import { FacebookDataService } from './facebook-data.service'
+import { InstagramDataService } from './instagram.service'
+import { ThreadsDataService } from './threads.service'
+import { WxGzhDataService } from './wx-gzh-data.service'
+import { YoutubeDataService } from './youtube-data.service'
+
+@Module({
+  imports: [BilibiliModule, MetaModule, YoutubeModule, WxPlatModule, PinterestModule, AccountModule],
+  controllers: [DataCubeController],
+  providers: [BilibiliDataService, FacebookDataService, InstagramDataService, ThreadsDataService, YoutubeDataService, WxGzhDataService, PinterestDataService],
+  exports: [BilibiliDataService, FacebookDataService, InstagramDataService, ThreadsDataService, YoutubeDataService, WxGzhDataService, PinterestDataService],
+})
+export class DataCubeModule {}

project/backend/apps/aitoearn-channel/src/core/data-cube/data-cube.module.ts

@@ -3,7 +3,7 @@ import {
   ChannelAccountDataCube,
   ChannelArcDataBulk,
   ChannelArcDataCube,
-} from '../plat/common'
+} from '../platforms/common'
 
 export abstract class DataCubeBase {
   /**