@haku-sci/utils/dist/microservice/tcp.service.js

utils from haku-sci. Library only

"use strict";
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
    return c > 3 && r && Object.defineProperty(target, key, r), r;
};
var __metadata = (this && this.__metadata) || function (k, v) {
    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.RestrictSender = RestrictSender;
const common_1 = require("@nestjs/common");
const core_1 = require("@nestjs/core");
const RESTRICT_SENDER_KEY = 'restrict_sender';
function RestrictSender(allowedSenders) {
    return (0, common_1.applyDecorators)((0, common_1.SetMetadata)(RESTRICT_SENDER_KEY, allowedSenders), (0, common_1.UseInterceptors)(RestrictSenderInterceptor));
}
let RestrictSenderInterceptor = class RestrictSenderInterceptor {
    constructor(reflector) {
        this.reflector = reflector;
    }
    intercept(context, next) {
        const ctx = context.switchToRpc();
        const data = ctx.getData();
        const handler = context.getHandler();
        const target = context.getClass();
        const allowedSenders = this.reflector.getAllAndOverride(RESTRICT_SENDER_KEY, [
            context.getHandler(),
            context.getClass(),
        ]) || [];
        const sender = data?.sender;
        if (!allowedSenders.includes(sender)) {
            throw new common_1.ForbiddenException(`Sender '${sender}' is not authorized`);
        }
        return next.handle();
    }
};
RestrictSenderInterceptor = __decorate([
    (0, common_1.Injectable)(),
    __metadata("design:paramtypes", [core_1.Reflector])
], RestrictSenderInterceptor);
//# sourceMappingURL=tcp.service.js.map