@russ-b/nestjs-common-tools/dist/grpc/interceptors/struct.interceptor.js

NestJS utility tools

"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);
};
var _a;
Object.defineProperty(exports, "__esModule", { value: true });
exports.StructInterceptor = void 0;
const common_1 = require("@nestjs/common");
const rxjs_1 = require("rxjs");
const transformers_1 = require("../transformers");
const package_definition_1 = require("../services/package-definition");
const constants_1 = require("@nestjs/microservices/constants");
const GOOGLE_PROTOBUF_STRUCT = 'google.protobuf.Struct';
let StructInterceptor = class StructInterceptor {
    constructor(packageDefinitionService) {
        this.packageDefinitionService = packageDefinitionService;
    }
    intercept(context, next) {
        const grpcContext = context.switchToRpc();
        const { service, rpc } = Reflect.getMetadata(constants_1.PATTERN_METADATA, context.getHandler())[0];
        const protoFields = this.packageDefinitionService
            .getManager()
            .getRequestFields(service, rpc);
        const fields = this.packageDefinitionService
            .getManager()
            .findFieldsByType(protoFields, GOOGLE_PROTOBUF_STRUCT);
        context.getArgs()[0] = this.traverseAndTransform(grpcContext.getData(), fields, transformers_1.StructTransformer.toObject);
        return next.handle().pipe((0, rxjs_1.map)((data) => {
            const protoFields = this.packageDefinitionService
                .getManager()
                .getResponseFields(service, rpc);
            const fields = this.packageDefinitionService
                .getManager()
                .findFieldsByType(protoFields, GOOGLE_PROTOBUF_STRUCT);
            return this.traverseAndTransform(data, fields, transformers_1.StructTransformer.toStruct);
        }));
    }
    traverseAndTransform(obj, fields, transformer) {
        if (!obj) {
            return obj;
        }
        if (Array.isArray(obj)) {
            return obj.map((item) => this.traverseAndTransform(item, fields, transformer));
        }
        const result = { ...obj };
        fields.forEach((f) => {
            const key = f.name;
            if (f.fields?.length) {
                result[key] = this.traverseAndTransform(result[key], f.fields, transformer);
            }
            else {
                result[key] = transformer(result[key]);
            }
        });
        return result;
    }
};
exports.StructInterceptor = StructInterceptor;
exports.StructInterceptor = StructInterceptor = __decorate([
    (0, common_1.Injectable)(),
    __metadata("design:paramtypes", [typeof (_a = typeof package_definition_1.PackageDefinitionService !== "undefined" && package_definition_1.PackageDefinitionService) === "function" ? _a : Object])
], StructInterceptor);
//# sourceMappingURL=struct.interceptor.js.map