hrms-api-org/src/controllers/SocketController.ts

import { Body, Controller, Post, Request, Route, Security } from "tsoa";
import { sendWebSocket } from "../services/webSocket";
import { RequestWithUser } from "../middlewares/user";

@Route("/api/v1/org/through-socket")
export class SocketController extends Controller {
  @Post("notify")
  async notify(
    @Body()
    payload: {
      message: string;
      userId?: string | string[];
      roles?: string | string[];
      error?: boolean;
    },
  ) {
    sendWebSocket(
      "socket-notification",
      { success: !payload.error, message: payload.message },
      {
        roles: payload.roles || [],
        userId: payload.userId || [],
      },
    );
  }

  @Post("notify-from-token")
  @Security("bearerAuth")
  async notifyFromToken(
    @Body()
    payload: {
      message: string;
      targetUserId?: string | string[];
      roles?: string | string[];
      error?: boolean;
    },
    @Request() req: RequestWithUser,
  ) {
    sendWebSocket(
      "socket-notification",
      { success: !payload.error, message: payload.message },
      {
        roles: payload.roles || req.user.role || [],
        userId: payload.targetUserId || req.user.sub || [],
      },
    );
  }
}