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 Dashboard from "../../Models/DatabaseModels/Dashboard";
import Label from "../../Models/DatabaseModels/Label";
import DashboardLabelRuleService from "./DashboardLabelRuleService";
import DashboardService from "./DashboardService";
import ObjectID from "../../Types/ObjectID";
import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
import logger from "../Utils/Logger";
class DashboardLabelRuleEngineServiceClass {
    /**
     * Evaluates DashboardLabelRule rows for the given dashboard and attaches
     * matched labels to it. The union is deduped against labels already on the
     * dashboard before insert to avoid PK conflicts on the join table.
     */
    async applyRulesToDashboard(dashboard) {
        var _a, _b;
        if (!dashboard.id || !dashboard.projectId) {
            return;
        }
        try {
            const rules = await DashboardLabelRuleService.findBy({
                query: {
                    projectId: dashboard.projectId,
                    isEnabled: true,
                },
                props: { isRoot: true },
                select: {
                    _id: true,
                    name: true,
                    dashboardLabels: { _id: true },
                    dashboardNamePattern: true,
                    dashboardDescriptionPattern: true,
                    labelsToAdd: { _id: true },
                },
                limit: 100,
                skip: 0,
            });
            if (rules.length === 0) {
                return;
            }
            const dashboardWithDetails = await DashboardService.findOneById({
                id: dashboard.id,
                select: {
                    name: true,
                    description: true,
                    labels: { _id: true },
                },
                props: { isRoot: true },
            });
            if (!dashboardWithDetails) {
                return;
            }
            const labelIdsToAdd = new Set();
            for (const rule of rules) {
                const matches = this.doesDashboardMatchRule(dashboardWithDetails, rule);
                if (!matches) {
                    continue;
                }
                for (const label of rule.labelsToAdd || []) {
                    if (label.id) {
                        labelIdsToAdd.add(label.id.toString());
                    }
                }
            }
            if (labelIdsToAdd.size === 0) {
                return;
            }
            const existingLabelIds = new Set((dashboardWithDetails.labels || [])
                .map((l) => {
                var _a;
                return ((_a = l.id) === null || _a === void 0 ? void 0 : _a.toString()) || "";
            })
                .filter((id) => {
                return id !== "";
            }));
            const newLabelIds = Array.from(labelIdsToAdd).filter((id) => {
                return !existingLabelIds.has(id);
            });
            if (newLabelIds.length === 0) {
                return;
            }
            await DashboardService.getRepository()
                .createQueryBuilder()
                .relation(Dashboard, "labels")
                .of(dashboard.id.toString())
                .add(newLabelIds);
            /*
             * Sync in-memory dashboard.labels so a downstream owner-rule engine in
             * the same onCreateSuccess chain can match on rule-added labels.
             */
            const mergedLabelIds = new Set([
                ...existingLabelIds,
                ...newLabelIds,
            ]);
            dashboard.labels = Array.from(mergedLabelIds).map((id) => {
                const label = new Label();
                label.id = new ObjectID(id);
                return label;
            });
            logger.debug(`DashboardLabelRuleEngine attached ${newLabelIds.length} labels to dashboard ${dashboard.id}`, { projectId: dashboard.projectId.toString() });
        }
        catch (error) {
            logger.error(`Error applying dashboard label rules: ${error}`, {
                projectId: (_a = dashboard.projectId) === null || _a === void 0 ? void 0 : _a.toString(),
                dashboardId: (_b = dashboard.id) === null || _b === void 0 ? void 0 : _b.toString(),
            });
        }
    }
    doesDashboardMatchRule(dashboard, rule) {
        if (rule.dashboardLabels && rule.dashboardLabels.length > 0) {
            if (!dashboard.labels || dashboard.labels.length === 0) {
                return false;
            }
            const ruleLabelIds = rule.dashboardLabels.map((l) => {
                var _a;
                return ((_a = l.id) === null || _a === void 0 ? void 0 : _a.toString()) || "";
            });
            const labelIds = dashboard.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.dashboardNamePattern &&
            (!dashboard.name ||
                !this.testRegex(rule.dashboardNamePattern, dashboard.name, rule))) {
            return false;
        }
        if (rule.dashboardDescriptionPattern &&
            (!dashboard.description ||
                !this.testRegex(rule.dashboardDescriptionPattern, dashboard.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 dashboard label rule ${rule.id}: ${pattern}`);
            return false;
        }
    }
}
__decorate([
    CaptureSpan(),
    __metadata("design:type", Function),
    __metadata("design:paramtypes", [Dashboard]),
    __metadata("design:returntype", Promise)
], DashboardLabelRuleEngineServiceClass.prototype, "applyRulesToDashboard", null);
export default new DashboardLabelRuleEngineServiceClass();
//# sourceMappingURL=DashboardLabelRuleEngineService.js.map