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 KubernetesCluster from "../../Models/DatabaseModels/KubernetesCluster";
import KubernetesClusterOwnerUser from "../../Models/DatabaseModels/KubernetesClusterOwnerUser";
import KubernetesClusterOwnerTeam from "../../Models/DatabaseModels/KubernetesClusterOwnerTeam";
import KubernetesClusterOwnerRuleService from "./KubernetesClusterOwnerRuleService";
import KubernetesClusterOwnerUserService from "./KubernetesClusterOwnerUserService";
import KubernetesClusterOwnerTeamService from "./KubernetesClusterOwnerTeamService";
import KubernetesClusterService from "./KubernetesClusterService";
import ObjectID from "../../Types/ObjectID";
import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
import logger from "../Utils/Logger";
class KubernetesClusterOwnerRuleEngineServiceClass {
    /**
     * Evaluates KubernetesClusterOwnerRule rows for the given Kubernetes cluster and adds matched
     * owner users / teams via KubernetesClusterOwnerUserService / KubernetesClusterOwnerTeamService. Rules
     * with notifyOwners set notify the added owners; rules with notifyOwners off
     * add silently.
     */
    async applyRulesToKubernetesCluster(kubernetesCluster) {
        var _a, _b;
        if (!kubernetesCluster.id || !kubernetesCluster.projectId) {
            return;
        }
        try {
            const rules = await KubernetesClusterOwnerRuleService.findBy({
                query: {
                    projectId: kubernetesCluster.projectId,
                    isEnabled: true,
                },
                props: { isRoot: true },
                select: {
                    _id: true,
                    name: true,
                    notifyOwners: true,
                    kubernetesClusterLabels: { _id: true },
                    kubernetesClusterNamePattern: true,
                    kubernetesClusterDescriptionPattern: true,
                    ownerUsers: { _id: true },
                    ownerTeams: { _id: true },
                },
                limit: 100,
                skip: 0,
            });
            if (rules.length === 0) {
                return;
            }
            const kubernetesClusterWithDetails = await KubernetesClusterService.findOneById({
                id: kubernetesCluster.id,
                select: {
                    name: true,
                    description: true,
                    labels: { _id: true },
                },
                props: { isRoot: true },
            });
            if (!kubernetesClusterWithDetails) {
                return;
            }
            const usersByNotify = new Map([
                [true, new Set()],
                [false, new Set()],
            ]);
            const teamsByNotify = new Map([
                [true, new Set()],
                [false, new Set()],
            ]);
            const matchedRules = [];
            for (const rule of rules) {
                const matches = this.doesKubernetesClusterMatchRule(kubernetesClusterWithDetails, rule);
                if (!matches) {
                    continue;
                }
                let ruleAddedAny = false;
                const notify = rule.notifyOwners !== false;
                for (const user of rule.ownerUsers || []) {
                    if (user.id) {
                        usersByNotify.get(notify).add(user.id.toString());
                        ruleAddedAny = true;
                    }
                }
                for (const team of rule.ownerTeams || []) {
                    if (team.id) {
                        teamsByNotify.get(notify).add(team.id.toString());
                        ruleAddedAny = true;
                    }
                }
                if (ruleAddedAny) {
                    matchedRules.push(rule);
                }
            }
            if (matchedRules.length === 0) {
                return;
            }
            for (const notify of [true, false]) {
                const userIds = usersByNotify.get(notify);
                const teamIds = teamsByNotify.get(notify);
                for (const userId of userIds) {
                    const owner = new KubernetesClusterOwnerUser();
                    owner.kubernetesClusterId = kubernetesCluster.id;
                    owner.projectId = kubernetesCluster.projectId;
                    owner.userId = new ObjectID(userId);
                    owner.isOwnerNotified = !notify;
                    await KubernetesClusterOwnerUserService.create({
                        data: owner,
                        props: { isRoot: true },
                    });
                }
                for (const teamId of teamIds) {
                    const owner = new KubernetesClusterOwnerTeam();
                    owner.kubernetesClusterId = kubernetesCluster.id;
                    owner.projectId = kubernetesCluster.projectId;
                    owner.teamId = new ObjectID(teamId);
                    owner.isOwnerNotified = !notify;
                    await KubernetesClusterOwnerTeamService.create({
                        data: owner,
                        props: { isRoot: true },
                    });
                }
            }
            logger.debug(`KubernetesClusterOwnerRuleEngine added owners to Kubernetes cluster ${kubernetesCluster.id}`, { projectId: kubernetesCluster.projectId.toString() });
        }
        catch (error) {
            logger.error(`Error applying Kubernetes cluster owner rules: ${error}`, {
                projectId: (_a = kubernetesCluster.projectId) === null || _a === void 0 ? void 0 : _a.toString(),
                kubernetesClusterId: (_b = kubernetesCluster.id) === null || _b === void 0 ? void 0 : _b.toString(),
            });
        }
    }
    doesKubernetesClusterMatchRule(kubernetesCluster, rule) {
        if (rule.kubernetesClusterLabels &&
            rule.kubernetesClusterLabels.length > 0) {
            if (!kubernetesCluster.labels || kubernetesCluster.labels.length === 0) {
                return false;
            }
            const ruleLabelIds = rule.kubernetesClusterLabels.map((l) => {
                var _a;
                return ((_a = l.id) === null || _a === void 0 ? void 0 : _a.toString()) || "";
            });
            const labelIds = kubernetesCluster.labels.map((l) => {
                var _a;
                return ((_a = l.id) === null || _a === void 0 ? void 0 : _a.toString()) || "";
            });
            if (!ruleLabelIds.some((id) => {
                return labelIds.includes(id);
            })) {
                return false;
            }
        }
        if (rule.kubernetesClusterNamePattern &&
            (!kubernetesCluster.name ||
                !this.testRegex(rule.kubernetesClusterNamePattern, kubernetesCluster.name, rule))) {
            return false;
        }
        if (rule.kubernetesClusterDescriptionPattern &&
            (!kubernetesCluster.description ||
                !this.testRegex(rule.kubernetesClusterDescriptionPattern, kubernetesCluster.description, rule))) {
            return false;
        }
        return true;
    }
    testRegex(pattern, value, rule) {
        try {
            const regex = new RegExp(pattern, "i");
            return regex.test(value);
        }
        catch (_a) {
            logger.warn(`Invalid regex in Kubernetes cluster owner rule ${rule.id}: ${pattern}`);
            return false;
        }
    }
}
__decorate([
    CaptureSpan(),
    __metadata("design:type", Function),
    __metadata("design:paramtypes", [KubernetesCluster]),
    __metadata("design:returntype", Promise)
], KubernetesClusterOwnerRuleEngineServiceClass.prototype, "applyRulesToKubernetesCluster", null);
export default new KubernetesClusterOwnerRuleEngineServiceClass();
//# sourceMappingURL=KubernetesClusterOwnerRuleEngineService.js.map