@kdump/code-cli-any-llm/dist/middleware/timeout.middleware.js

> A unified gateway for the Gemini, opencode, crush, and Qwen Code AI CLIs

"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 TimeoutMiddleware_1;
Object.defineProperty(exports, "__esModule", { value: true });
exports.TimeoutMiddleware = void 0;
const common_1 = require("@nestjs/common");
const common_2 = require("@nestjs/common");
const config_1 = require("@nestjs/config");
let TimeoutMiddleware = TimeoutMiddleware_1 = class TimeoutMiddleware {
    configService;
    logger = new common_2.Logger(TimeoutMiddleware_1.name);
    timeout;
    constructor(configService) {
        this.configService = configService;
        const configuredTimeout = this.configService.get('gateway.requestTimeout');
        const envTimeout = process.env.CAL_REQUEST_TIMEOUT
            ? parseInt(process.env.CAL_REQUEST_TIMEOUT, 10)
            : undefined;
        const resolved = configuredTimeout ?? envTimeout ?? 3600000;
        this.timeout =
            Number.isFinite(resolved) && resolved > 0 ? resolved : 3600000;
    }
    use(req, res, next) {
        const timeoutId = setTimeout(() => {
            if (!res.headersSent) {
                this.logger.warn(`Request timeout for ${req.method} ${req.originalUrl}`);
                res.status(504).json({
                    statusCode: 504,
                    message: 'Gateway timeout',
                    error: 'Timeout',
                    timestamp: new Date().toISOString(),
                    path: req.originalUrl,
                });
            }
        }, this.timeout);
        const originalEnd = res.end.bind(res);
        res.end = function (chunk, encoding) {
            clearTimeout(timeoutId);
            return originalEnd.call(this, chunk, encoding);
        };
        req.socket.on('error', (error) => {
            clearTimeout(timeoutId);
            this.logger.error(`Socket error for ${req.method} ${req.originalUrl}: ${error.message}`);
        });
        next();
    }
};
exports.TimeoutMiddleware = TimeoutMiddleware;
exports.TimeoutMiddleware = TimeoutMiddleware = TimeoutMiddleware_1 = __decorate([
    (0, common_1.Injectable)(),
    __metadata("design:paramtypes", [config_1.ConfigService])
], TimeoutMiddleware);
//# sourceMappingURL=timeout.middleware.js.map