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);
};
import DatabaseService from "./DatabaseService";
import ObjectID from "../../Types/ObjectID";
import Version from "../../Types/Version";
import BadDataException from "../../Types/Exception/BadDataException";
import Model, { RunbookAgentConnectionStatus, } from "../../Models/DatabaseModels/RunbookAgent";
import OneUptimeDate from "../../Types/Date";
import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
export class Service extends DatabaseService {
    constructor() {
        super(Model);
    }
    async onBeforeCreate(createBy) {
        if (!createBy.data.key) {
            createBy.data.key = ObjectID.generate().toString();
        }
        if (!createBy.data.agentVersion) {
            createBy.data.agentVersion = new Version("1.0.0");
        }
        if (!createBy.data.connectionStatus) {
            createBy.data.connectionStatus =
                RunbookAgentConnectionStatus.Disconnected;
        }
        return { createBy, carryForward: [] };
    }
    async findByIdAndKey(data) {
        if (!data.agentId || !data.agentKey) {
            return null;
        }
        return this.findOneBy({
            query: {
                _id: data.agentId.toString(),
                key: data.agentKey,
            },
            select: {
                _id: true,
                projectId: true,
                name: true,
            },
            props: { isRoot: true },
        });
    }
    async heartbeat(data) {
        if (!data.agentId) {
            throw new BadDataException("agentId is required");
        }
        const update = {
            lastAlive: OneUptimeDate.getCurrentDate(),
            connectionStatus: RunbookAgentConnectionStatus.Connected,
        };
        if (data.agentVersion) {
            update["agentVersion"] = data.agentVersion;
        }
        if (data.hostInfo) {
            update["hostInfo"] = data.hostInfo;
        }
        await this.updateOneById({
            id: data.agentId,
            data: update,
            props: { isRoot: true },
        });
    }
}
__decorate([
    CaptureSpan(),
    __metadata("design:type", Function),
    __metadata("design:paramtypes", [Object]),
    __metadata("design:returntype", Promise)
], Service.prototype, "onBeforeCreate", null);
__decorate([
    CaptureSpan(),
    __metadata("design:type", Function),
    __metadata("design:paramtypes", [Object]),
    __metadata("design:returntype", Promise)
], Service.prototype, "findByIdAndKey", null);
__decorate([
    CaptureSpan(),
    __metadata("design:type", Function),
    __metadata("design:paramtypes", [Object]),
    __metadata("design:returntype", Promise)
], Service.prototype, "heartbeat", null);
export default new Service();
//# sourceMappingURL=RunbookAgentService.js.map