staff_data/apps/server/src/trpc/trpc.router.ts

import { INestApplication, Injectable, Logger } from '@nestjs/common';
import { DepartmentRouter } from '@server/models/department/department.router';
import { StaffRouter } from '@server/models/staff/staff.router';
import { TaxonomyRouter } from '@server/models/taxonomy/taxonomy.router';
import { TermRouter } from '@server/models/term/term.router';
import { TrpcService } from '@server/trpc/trpc.service';
import * as trpcExpress from '@trpc/server/adapters/express';
import ws, { WebSocketServer } from 'ws';
import { AppConfigRouter } from '@server/models/app-config/app-config.router';
import { MessageRouter } from '@server/models/message/message.router';
import { PostRouter } from '@server/models/post/post.router';
import { VisitRouter } from '@server/models/visit/visit.router';
import { RoleMapRouter } from '@server/models/rbac/rolemap.router';
import { TransformRouter } from '@server/models/transform/transform.router';
import { RoleRouter } from '@server/models/rbac/role.router';
import { ResourceRouter } from '../models/resource/resource.router';
import { TrainContentRouter } from '@server/models/train-content/trainContent.router';
import { TrainSituationRouter } from '@server/models/train-situation/trainSituation.router';
import { DailyTrainRouter } from '@server/models/daily-train/dailyTrain.router';
import { SystemLogRouter } from '@server/models/sys-logs/systemLog.router';

@Injectable()
export class TrpcRouter {
  logger = new Logger(TrpcRouter.name);
  constructor(
    private readonly trpc: TrpcService,
    private readonly post: PostRouter,
    private readonly department: DepartmentRouter,
    private readonly staff: StaffRouter,
    private readonly term: TermRouter,
    private readonly taxonomy: TaxonomyRouter,
    private readonly role: RoleRouter,
    private readonly rolemap: RoleMapRouter,
    private readonly transform: TransformRouter,
    private readonly app_config: AppConfigRouter,
    private readonly message: MessageRouter,
    private readonly visitor: VisitRouter,
    private readonly resource: ResourceRouter,
    private readonly trainContent: TrainContentRouter,
    private readonly trainSituation:TrainSituationRouter,
    private readonly dailyTrain:DailyTrainRouter,
    private readonly systemLogRouter: SystemLogRouter,
  ) {}
  getRouter() {
    return;
  }
  appRouter = this.trpc.router({
    transform: this.transform.router,
    post: this.post.router,
    department: this.department.router,
    staff: this.staff.router,
    term: this.term.router,
    taxonomy: this.taxonomy.router,
    role: this.role.router,
    rolemap: this.rolemap.router,
    message: this.message.router,
    app_config: this.app_config.router,
    visitor: this.visitor.router,
    resource: this.resource.router,
    trainContent:this.trainContent.router,
    trainSituation:this.trainSituation.router,
    dailyTrain:this.dailyTrain.router,
    systemLog: this.systemLogRouter.router,
  });
  wss: WebSocketServer = undefined;

  async applyMiddleware(app: INestApplication) {
    app.use(
      `/trpc`,
      trpcExpress.createExpressMiddleware({
        router: this.appRouter,
        createContext: this.trpc.createExpressContext,
        onError(opts) {
          const { error, type, path, input, ctx, req } = opts;
          // console.error('TRPC Error:', error);
        },
      }),
    );
    // applyWSSHandler({
    //   wss: this.websocketService.getWss("trpc"),
    //   router: this.appRouter,
    //   createContext: this.trpc.createWSSContext,
    // });
  }
}