"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 __param = (this && this.__param) || function (paramIndex, decorator) {
    return function (target, key) { decorator(target, key, paramIndex); }
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.TramMessageFilter = void 0;
const NotPublicVehiclesRedisRepository_1 = require("../../../../ropid-gtfs/data-access/cache/NotPublicVehiclesRedisRepository");
const CoreToken_1 = require("@golemio/core/dist/helpers/ioc/CoreToken");
const tsyringe_1 = require("@golemio/core/dist/shared/tsyringe");
const VPContainerToken_1 = require("../../../ioc/VPContainerToken");
const TimestampValidator_1 = require("./TimestampValidator");
const golemio_errors_1 = require("@golemio/core/dist/shared/golemio-errors");
const TRAM_SERVICING_ROUTE_ID_MIN = 50;
const TRAM_SERVICING_ROUTE_ID_MAX = 89;
const TRAM_UNUSUAL_RUN_NUMBER_MIN = 80;
let TramMessageFilter = class TramMessageFilter {
    constructor(logger, timestampValidator, notPublicVehiclesRepository) {
        this.logger = logger;
        this.timestampValidator = timestampValidator;
        this.notPublicVehiclesRepository = notPublicVehiclesRepository;
    }
    /**
     * Yield tram messages that are valid for processing
     *   - route id is not considered a servicing route (50-89, note: >89 is a night tram route)
     *   - run number is less than 80
     *   - OR registration number is in not-public vehicles whitelist (e.g., mazačka maintenance trams)
     *   - timestamp is in acceptable range
     *
     * Sets isNotPublic flag on messages to avoid duplicate whitelist checks in processor
     */
    async *yieldFilteredMessages(messages, timestamp) {
        // Load config once for the entire batch to avoid multiple Redis calls
        const { tram: { registrationNumbers }, routeIds, } = await this.notPublicVehiclesRepository.getConfig();
        const whitelistedRegistrationNumbers = new Set(registrationNumbers);
        const whitelistedRouteNames = new Set(routeIds);
        for (const message of messages) {
            const { route_id, run_number, registration_number, line_short_name } = message.run;
            if (!this.timestampValidator.isTimestampValid(timestamp, message.run_message.actual_stop_timestamp_real.toString())) {
                this.logger.error(new golemio_errors_1.GeneralError("Message timestamp 'takt' of value" +
                    ` '${message.run_message.actual_stop_timestamp_real.toISOString()}'` +
                    ` is not valid for line ${message.run.line_short_name}, run ${message.run.run_number}`, this.constructor.name, undefined, undefined, "pid"));
                continue;
            }
            // Check if this is a not-public vehicle
            // - Whitelisted registration number (e.g., mazačka maintenance tram)
            // - Whitelisted route name (e.g., ad-hoc routes like X1, X2, NAD)
            const isNotPublicByRegistration = whitelistedRegistrationNumbers.has(registration_number);
            const isNotPublicByRouteName = whitelistedRouteNames.has(line_short_name);
            const isNotPublicVehicle = isNotPublicByRegistration || isNotPublicByRouteName;
            // Allow if:
            // - Whitelisted by registration number OR route name (not-public vehicles)
            // - OR valid route_id and run_number (scheduled vehicles)
            if (isNotPublicVehicle || (this.isRouteIdValid(route_id) && this.isRunNumberValid(run_number))) {
                message.isNotPublic = isNotPublicVehicle; // ← Set flag
                yield message;
            }
            else {
                this.logger.info(`${this.constructor.name}: route id ${route_id} or run number ${run_number} is invalid`);
                this.logger.debug(message);
            }
        }
    }
    isRouteIdValid(routeId) {
        const parsedRouteId = Number.parseInt(routeId);
        return (!Number.isNaN(parsedRouteId) &&
            (parsedRouteId < TRAM_SERVICING_ROUTE_ID_MIN || parsedRouteId > TRAM_SERVICING_ROUTE_ID_MAX));
    }
    isRunNumberValid(runNumber) {
        return runNumber < TRAM_UNUSUAL_RUN_NUMBER_MIN;
    }
};
exports.TramMessageFilter = TramMessageFilter;
exports.TramMessageFilter = TramMessageFilter = __decorate([
    (0, tsyringe_1.injectable)(),
    __param(0, (0, tsyringe_1.inject)(CoreToken_1.CoreToken.Logger)),
    __param(1, (0, tsyringe_1.inject)(VPContainerToken_1.VPContainerToken.TimestampValidator)),
    __param(2, (0, tsyringe_1.inject)(VPContainerToken_1.VPContainerToken.NotPublicVehiclesRedisRepository)),
    __metadata("design:paramtypes", [Object, TimestampValidator_1.TimestampValidator,
        NotPublicVehiclesRedisRepository_1.NotPublicVehiclesRedisRepository])
], TramMessageFilter);
//# sourceMappingURL=TramMessageFilter.js.map