"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.JISInfotextsRepository = void 0;
const ropid_gtfs_1 = require("../../ropid-gtfs");
const const_1 = require("../../../schema-definitions/const");
const JISInfotextsModel_1 = require("../../../schema-definitions/jis/models/JISInfotextsModel");
const AbstractValidatableRepository_1 = require("@golemio/core/dist/helpers/data-access/postgres/repositories/AbstractValidatableRepository");
const CoreToken_1 = require("@golemio/core/dist/helpers/ioc/CoreToken");
const golemio_errors_1 = require("@golemio/core/dist/shared/golemio-errors");
const golemio_validator_1 = require("@golemio/core/dist/shared/golemio-validator");
const sequelize_1 = require("@golemio/core/dist/shared/sequelize");
const tsyringe_1 = require("@golemio/core/dist/shared/tsyringe");
const RepositoryUtils_1 = require("../../shared/RepositoryUtils");
const JISContainerToken_1 = require("../ioc/JISContainerToken");
const JISInfotextsRopidGTFSStopsRepository_1 = require("./JISInfotextsRopidGTFSStopsRepository");
let JISInfotextsRepository = class JISInfotextsRepository extends AbstractValidatableRepository_1.AbstractValidatableRepository {
    constructor(connector, logger, infotextStopRepository, gtfsStopRepository) {
        super(connector, logger);
        this.logger = logger;
        this.infotextStopRepository = infotextStopRepository;
        this.gtfsStopRepository = gtfsStopRepository;
        this.schema = const_1.PG_SCHEMA;
        this.tableName = JISInfotextsModel_1.JISInfotextsModel.tableName;
        this.validator = new golemio_validator_1.JSONSchemaValidator("JISInfotextsRepository", JISInfotextsModel_1.JISInfotextsModel.jsonSchema);
        this.sequelizeModel = connector
            .getConnection()
            .define(this.tableName, JISInfotextsModel_1.JISInfotextsModel.attributeModel, { schema: this.schema });
        this.sequelizeModel.belongsToMany(this.gtfsStopRepository.sequelizeModel, {
            through: this.infotextStopRepository["sequelizeModel"],
            foreignKey: "infotext_id",
            otherKey: "stop_id",
            as: "stops",
        });
        this.gtfsStopRepository.sequelizeModel.belongsToMany(this.sequelizeModel, {
            through: this.infotextStopRepository["sequelizeModel"],
            foreignKey: "stop_id",
            otherKey: "infotext_id",
        });
        this.infotextStopRepository["sequelizeModel"].belongsTo(this.sequelizeModel, {
            targetKey: "id",
            foreignKey: "infotext_id",
        });
        this.infotextStopRepository["sequelizeModel"].belongsTo(this.gtfsStopRepository.sequelizeModel, {
            targetKey: "stop_id",
            foreignKey: "stop_id",
        });
        this.sequelizeModel.hasMany(this.infotextStopRepository["sequelizeModel"], {
            sourceKey: "id",
            foreignKey: "infotext_id",
        });
        this.gtfsStopRepository.sequelizeModel.hasMany(this.infotextStopRepository["sequelizeModel"], {
            sourceKey: "stop_id",
            foreignKey: "stop_id",
        });
    }
    /**
     * Refresh all data from VYMI and delete old data
     *
     * @param data The data to be upserted
     * @param options Options for the operation
     */
    async refreshData(data, options) {
        try {
            const currentInfotexts = await this.sequelizeModel.findAll({ transaction: options?.transaction });
            const currentInfotextsMap = new Map();
            for (const infotext of currentInfotexts) {
                currentInfotextsMap.set(infotext.id, infotext);
            }
            const toBeUpserted = [];
            for (const infotext of data) {
                const info = currentInfotextsMap.get(infotext.id);
                if (!info ||
                    info.updated_at <= infotext.updated_timestamp ||
                    info.active_period_end?.getTime() !== infotext.active_period_end?.getTime() ||
                    info.active_period_start.getTime() !== infotext.active_period_start.getTime()) {
                    toBeUpserted.push(infotext);
                }
                currentInfotextsMap.delete(infotext.id);
            }
            const toBeDeleted = [];
            for (const { id } of currentInfotextsMap.values()) {
                toBeDeleted.push(id);
            }
            const [upserted] = await Promise.all([
                this.sequelizeModel.bulkCreate(toBeUpserted, {
                    updateOnDuplicate: this.getUpdateAttributes(),
                    transaction: options?.transaction,
                }),
                this.sequelizeModel.destroy({
                    where: {
                        id: { [sequelize_1.Op.in]: toBeDeleted },
                    },
                    transaction: options?.transaction,
                }),
            ]);
            return upserted;
        }
        catch (err) {
            RepositoryUtils_1.RepositoryUtils.mapSequelizeError(err, this.constructor.name, "refreshData");
        }
    }
    getUpdateAttributes() {
        return Object.keys(JISInfotextsModel_1.JISInfotextsModel.attributeModel).filter((attribute) => !["created_at"].includes(attribute));
    }
    /**
     * @param options
     * @returns The number of deleted items
     */
    async deleteAll(options) {
        try {
            return await this.sequelizeModel.destroy({ where: {}, transaction: options?.transaction });
        }
        catch (err) {
            throw new golemio_errors_1.GeneralError("Error in deleteAll", this.constructor.name, err);
        }
    }
    async findAll() {
        return await this.sequelizeModel.findAll({
            include: [
                {
                    model: this.gtfsStopRepository.sequelizeModel,
                    as: "stops",
                    attributes: ["stop_id"],
                    through: { attributes: [] },
                    required: true,
                },
            ],
        });
    }
};
exports.JISInfotextsRepository = JISInfotextsRepository;
exports.JISInfotextsRepository = JISInfotextsRepository = __decorate([
    (0, tsyringe_1.injectable)(),
    __param(0, (0, tsyringe_1.inject)(CoreToken_1.CoreToken.PostgresConnector)),
    __param(1, (0, tsyringe_1.inject)(CoreToken_1.CoreToken.Logger)),
    __param(2, (0, tsyringe_1.inject)(JISContainerToken_1.JISContainerToken.JISInfotextsRopidGTFSStopsRepository)),
    __param(3, (0, tsyringe_1.inject)(JISContainerToken_1.JISContainerToken.RopidGTFSStopsRepository)),
    __metadata("design:paramtypes", [Object, Object, JISInfotextsRopidGTFSStopsRepository_1.JISInfotextsRopidGTFSStopsRepository,
        ropid_gtfs_1.RopidGTFSStopsModel])
], JISInfotextsRepository);
//# sourceMappingURL=JISInfotextsRepository.js.map