@oneuptime/common 11.0.2 → 11.0.3

package/Server/Services/LogAggregationService.ts

@@ -109,6 +109,7 @@ export class LogAggregationService {
     new Map([
       ["hostId", ServiceType.Host],
       ["dockerHostId", ServiceType.DockerHost],
+      ["podmanHostId", ServiceType.PodmanHost],
       ["kubernetesClusterId", ServiceType.KubernetesCluster],
       ["proxmoxClusterId", ServiceType.ProxmoxCluster],
       ["cephClusterId", ServiceType.CephCluster],

package/Server/Services/MetricAggregationService.ts

@@ -48,6 +48,7 @@ export class MetricAggregationService {
     new Map([
       ["hostId", ServiceType.Host],
       ["dockerHostId", ServiceType.DockerHost],
+      ["podmanHostId", ServiceType.PodmanHost],
       ["kubernetesClusterId", ServiceType.KubernetesCluster],
       ["proxmoxClusterId", ServiceType.ProxmoxCluster],
       ["cephClusterId", ServiceType.CephCluster],

package/Server/Services/OpenTelemetryIngestService.ts

@@ -20,17 +20,21 @@ import TelemetryRetentionConfig from "../../Types/Telemetry/TelemetryRetentionCo
 import ServiceType from "../../Types/Telemetry/ServiceType";
 import Host from "../../Models/DatabaseModels/Host";
 import DockerHost from "../../Models/DatabaseModels/DockerHost";
+import PodmanHost from "../../Models/DatabaseModels/PodmanHost";
 import KubernetesCluster from "../../Models/DatabaseModels/KubernetesCluster";
 import ProxmoxCluster from "../../Models/DatabaseModels/ProxmoxCluster";
 import CephCluster from "../../Models/DatabaseModels/CephCluster";
+import DockerSwarmCluster from "../../Models/DatabaseModels/DockerSwarmCluster";
 import ServerlessFunction from "../../Models/DatabaseModels/ServerlessFunction";
 import CloudResource from "../../Models/DatabaseModels/CloudResource";
 import RumApplication from "../../Models/DatabaseModels/RumApplication";
 import HostService from "./HostService";
 import DockerHostService from "./DockerHostService";
+import PodmanHostService from "./PodmanHostService";
 import KubernetesClusterService from "./KubernetesClusterService";
 import ProxmoxClusterService from "./ProxmoxClusterService";
 import CephClusterService from "./CephClusterService";
+import DockerSwarmClusterService from "./DockerSwarmClusterService";
 import ServerlessFunctionService from "./ServerlessFunctionService";
 import CloudResourceService from "./CloudResourceService";
 import RumApplicationService from "./RumApplicationService";
@@ -675,6 +679,23 @@ export default class OTelIngestService {
             dockerHost?.telemetryRetentionConfig ?? null,
         };
       }
+      if (primaryEntityType === ServiceType.PodmanHost) {
+        const podmanHost: PodmanHost | null =
+          await PodmanHostService.findOneById({
+            id: resourceId,
+            select: {
+              retainTelemetryDataForDays: true,
+              telemetryRetentionConfig: true,
+            },
+            props: { isRoot: true },
+          });
+        return {
+          retainTelemetryDataForDays:
+            podmanHost?.retainTelemetryDataForDays ?? null,
+          telemetryRetentionConfig:
+            podmanHost?.telemetryRetentionConfig ?? null,
+        };
+      }
       if (primaryEntityType === ServiceType.KubernetesCluster) {
         const cluster: KubernetesCluster | null =
           await KubernetesClusterService.findOneById({
@@ -723,6 +744,22 @@ export default class OTelIngestService {
           telemetryRetentionConfig: cluster?.telemetryRetentionConfig ?? null,
         };
       }
+      if (primaryEntityType === ServiceType.DockerSwarmCluster) {
+        const cluster: DockerSwarmCluster | null =
+          await DockerSwarmClusterService.findOneById({
+            id: resourceId,
+            select: {
+              retainTelemetryDataForDays: true,
+              telemetryRetentionConfig: true,
+            },
+            props: { isRoot: true },
+          });
+        return {
+          retainTelemetryDataForDays:
+            cluster?.retainTelemetryDataForDays ?? null,
+          telemetryRetentionConfig: cluster?.telemetryRetentionConfig ?? null,
+        };
+      }
       if (primaryEntityType === ServiceType.ServerlessFunction) {
         const serverlessFunction: ServerlessFunction | null =
           await ServerlessFunctionService.findOneById({

package/Server/Services/PodmanHostLabelRuleEngineService.ts

@@ -0,0 +1,198 @@
+import Label from "../../Models/DatabaseModels/Label";
+import PodmanHost from "../../Models/DatabaseModels/PodmanHost";
+import PodmanHostLabelRule from "../../Models/DatabaseModels/PodmanHostLabelRule";
+import PodmanHostLabelRuleService from "./PodmanHostLabelRuleService";
+import PodmanHostService from "./PodmanHostService";
+import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import logger, { LogAttributes } from "../Utils/Logger";
+
+class PodmanHostLabelRuleEngineServiceClass {
+  /**
+   * Evaluates PodmanHostLabelRule rows for the given Podman host and attaches matched
+   * labels to it. The union is deduped against labels already on the Podman host
+   * before insert to avoid PK conflicts on the join table.
+   */
+  @CaptureSpan()
+  public async applyRulesToPodmanHost(podmanHost: PodmanHost): Promise<void> {
+    if (!podmanHost.id || !podmanHost.projectId) {
+      return;
+    }
+
+    try {
+      const rules: Array<PodmanHostLabelRule> =
+        await PodmanHostLabelRuleService.findBy({
+          query: {
+            projectId: podmanHost.projectId,
+            isEnabled: true,
+          },
+          props: { isRoot: true },
+          select: {
+            _id: true,
+            name: true,
+            podmanHostLabels: { _id: true },
+            podmanHostNamePattern: true,
+            podmanHostDescriptionPattern: true,
+            labelsToAdd: { _id: true },
+          },
+          limit: 100,
+          skip: 0,
+        });
+
+      if (rules.length === 0) {
+        return;
+      }
+
+      const podmanHostWithDetails: PodmanHost | null =
+        await PodmanHostService.findOneById({
+          id: podmanHost.id,
+          select: {
+            name: true,
+            description: true,
+            labels: { _id: true },
+          },
+          props: { isRoot: true },
+        });
+
+      if (!podmanHostWithDetails) {
+        return;
+      }
+
+      const labelIdsToAdd: Set<string> = new Set();
+
+      for (const rule of rules) {
+        const matches: boolean = this.doesPodmanHostMatchRule(
+          podmanHostWithDetails,
+          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: Set<string> = new Set(
+        (podmanHostWithDetails.labels || [])
+          .map((l: Label) => {
+            return l.id?.toString() || "";
+          })
+          .filter((id: string) => {
+            return id !== "";
+          }),
+      );
+
+      const newLabelIds: Array<string> = Array.from(labelIdsToAdd).filter(
+        (id: string) => {
+          return !existingLabelIds.has(id);
+        },
+      );
+      if (newLabelIds.length === 0) {
+        return;
+      }
+
+      await PodmanHostService.getRepository()
+        .createQueryBuilder()
+        .relation(PodmanHost, "labels")
+        .of(podmanHost.id.toString())
+        .add(newLabelIds);
+
+      /*
+       * Sync in-memory podmanHost.labels so a downstream owner-rule engine in
+       * the same onCreateSuccess chain can match on rule-added labels.
+       */
+      const mergedLabelIds: Set<string> = new Set([
+        ...existingLabelIds,
+        ...newLabelIds,
+      ]);
+      podmanHost.labels = Array.from(mergedLabelIds).map((id: string) => {
+        const label: Label = new Label();
+        label.id = new ObjectID(id);
+        return label;
+      });
+
+      logger.debug(
+        `PodmanHostLabelRuleEngine attached ${newLabelIds.length} labels to Podman host ${podmanHost.id}`,
+        { projectId: podmanHost.projectId.toString() } as LogAttributes,
+      );
+    } catch (error) {
+      logger.error(`Error applying Podman host label rules: ${error}`, {
+        projectId: podmanHost.projectId?.toString(),
+        podmanHostId: podmanHost.id?.toString(),
+      } as LogAttributes);
+    }
+  }
+
+  private doesPodmanHostMatchRule(
+    podmanHost: PodmanHost,
+    rule: PodmanHostLabelRule,
+  ): boolean {
+    if (rule.podmanHostLabels && rule.podmanHostLabels.length > 0) {
+      if (!podmanHost.labels || podmanHost.labels.length === 0) {
+        return false;
+      }
+      const ruleLabelIds: Array<string> = rule.podmanHostLabels.map(
+        (l: Label) => {
+          return l.id?.toString() || "";
+        },
+      );
+      const labelIds: Array<string> = podmanHost.labels.map((l: Label) => {
+        return l.id?.toString() || "";
+      });
+      if (
+        !ruleLabelIds.some((id: string) => {
+          return labelIds.includes(id);
+        })
+      ) {
+        return false;
+      }
+    }
+
+    if (
+      rule.podmanHostNamePattern &&
+      (!podmanHost.name ||
+        !this.testRegex(rule.podmanHostNamePattern, podmanHost.name, rule))
+    ) {
+      return false;
+    }
+
+    if (
+      rule.podmanHostDescriptionPattern &&
+      (!podmanHost.description ||
+        !this.testRegex(
+          rule.podmanHostDescriptionPattern,
+          podmanHost.description,
+          rule,
+        ))
+    ) {
+      return false;
+    }
+
+    return true;
+  }
+
+  private testRegex(
+    pattern: string,
+    value: string,
+    rule: PodmanHostLabelRule,
+  ): boolean {
+    try {
+      const regex: RegExp = new RegExp(pattern, "i");
+      return regex.test(value);
+    } catch {
+      logger.warn(
+        `Invalid regex in Podman host label rule ${rule.id}: ${pattern}`,
+      );
+      return false;
+    }
+  }
+}
+
+export default new PodmanHostLabelRuleEngineServiceClass();

package/Server/Services/PodmanHostLabelRuleService.ts

@@ -0,0 +1,14 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/PodmanHostLabelRule";
+import { IsBillingEnabled } from "../EnvironmentConfig";
+
+export class Service extends DatabaseService<Model> {
+  public constructor() {
+    super(Model);
+    if (IsBillingEnabled) {
+      this.hardDeleteItemsOlderThanInDays("createdAt", 3 * 365);
+    }
+  }
+}
+
+export default new Service();

package/Server/Services/PodmanHostOwnerRuleEngineService.ts

@@ -0,0 +1,216 @@
+import Label from "../../Models/DatabaseModels/Label";
+import PodmanHost from "../../Models/DatabaseModels/PodmanHost";
+import PodmanHostOwnerRule from "../../Models/DatabaseModels/PodmanHostOwnerRule";
+import PodmanHostOwnerUser from "../../Models/DatabaseModels/PodmanHostOwnerUser";
+import PodmanHostOwnerTeam from "../../Models/DatabaseModels/PodmanHostOwnerTeam";
+import PodmanHostOwnerRuleService from "./PodmanHostOwnerRuleService";
+import PodmanHostOwnerUserService from "./PodmanHostOwnerUserService";
+import PodmanHostOwnerTeamService from "./PodmanHostOwnerTeamService";
+import PodmanHostService from "./PodmanHostService";
+import ObjectID from "../../Types/ObjectID";
+import CaptureSpan from "../Utils/Telemetry/CaptureSpan";
+import logger, { LogAttributes } from "../Utils/Logger";
+
+class PodmanHostOwnerRuleEngineServiceClass {
+  /**
+   * Evaluates PodmanHostOwnerRule rows for the given Podman host and adds matched
+   * owner users / teams via PodmanHostOwnerUserService / PodmanHostOwnerTeamService. Rules
+   * with notifyOwners set notify the added owners; rules with notifyOwners off
+   * add silently.
+   */
+  @CaptureSpan()
+  public async applyRulesToPodmanHost(podmanHost: PodmanHost): Promise<void> {
+    if (!podmanHost.id || !podmanHost.projectId) {
+      return;
+    }
+
+    try {
+      const rules: Array<PodmanHostOwnerRule> =
+        await PodmanHostOwnerRuleService.findBy({
+          query: {
+            projectId: podmanHost.projectId,
+            isEnabled: true,
+          },
+          props: { isRoot: true },
+          select: {
+            _id: true,
+            name: true,
+            notifyOwners: true,
+            podmanHostLabels: { _id: true },
+            podmanHostNamePattern: true,
+            podmanHostDescriptionPattern: true,
+            ownerUsers: { _id: true },
+            ownerTeams: { _id: true },
+          },
+          limit: 100,
+          skip: 0,
+        });
+
+      if (rules.length === 0) {
+        return;
+      }
+
+      const podmanHostWithDetails: PodmanHost | null =
+        await PodmanHostService.findOneById({
+          id: podmanHost.id,
+          select: {
+            name: true,
+            description: true,
+            labels: { _id: true },
+          },
+          props: { isRoot: true },
+        });
+
+      if (!podmanHostWithDetails) {
+        return;
+      }
+
+      const usersByNotify: Map<boolean, Set<string>> = new Map([
+        [true, new Set()],
+        [false, new Set()],
+      ]);
+      const teamsByNotify: Map<boolean, Set<string>> = new Map([
+        [true, new Set()],
+        [false, new Set()],
+      ]);
+
+      const matchedRules: Array<PodmanHostOwnerRule> = [];
+
+      for (const rule of rules) {
+        const matches: boolean = this.doesPodmanHostMatchRule(
+          podmanHostWithDetails,
+          rule,
+        );
+        if (!matches) {
+          continue;
+        }
+        let ruleAddedAny: boolean = false;
+        const notify: boolean = 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: Set<string> = usersByNotify.get(notify)!;
+        const teamIds: Set<string> = teamsByNotify.get(notify)!;
+
+        for (const userId of userIds) {
+          const owner: PodmanHostOwnerUser = new PodmanHostOwnerUser();
+          owner.podmanHostId = podmanHost.id;
+          owner.projectId = podmanHost.projectId;
+          owner.userId = new ObjectID(userId);
+          owner.isOwnerNotified = !notify;
+          await PodmanHostOwnerUserService.create({
+            data: owner,
+            props: { isRoot: true },
+          });
+        }
+
+        for (const teamId of teamIds) {
+          const owner: PodmanHostOwnerTeam = new PodmanHostOwnerTeam();
+          owner.podmanHostId = podmanHost.id;
+          owner.projectId = podmanHost.projectId;
+          owner.teamId = new ObjectID(teamId);
+          owner.isOwnerNotified = !notify;
+          await PodmanHostOwnerTeamService.create({
+            data: owner,
+            props: { isRoot: true },
+          });
+        }
+      }
+
+      logger.debug(
+        `PodmanHostOwnerRuleEngine added owners to Podman host ${podmanHost.id}`,
+        { projectId: podmanHost.projectId.toString() } as LogAttributes,
+      );
+    } catch (error) {
+      logger.error(`Error applying Podman host owner rules: ${error}`, {
+        projectId: podmanHost.projectId?.toString(),
+        podmanHostId: podmanHost.id?.toString(),
+      } as LogAttributes);
+    }
+  }
+
+  private doesPodmanHostMatchRule(
+    podmanHost: PodmanHost,
+    rule: PodmanHostOwnerRule,
+  ): boolean {
+    if (rule.podmanHostLabels && rule.podmanHostLabels.length > 0) {
+      if (!podmanHost.labels || podmanHost.labels.length === 0) {
+        return false;
+      }
+      const ruleLabelIds: Array<string> = rule.podmanHostLabels.map(
+        (l: Label) => {
+          return l.id?.toString() || "";
+        },
+      );
+      const labelIds: Array<string> = podmanHost.labels.map((l: Label) => {
+        return l.id?.toString() || "";
+      });
+      if (
+        !ruleLabelIds.some((id: string) => {
+          return labelIds.includes(id);
+        })
+      ) {
+        return false;
+      }
+    }
+
+    if (
+      rule.podmanHostNamePattern &&
+      (!podmanHost.name ||
+        !this.testRegex(rule.podmanHostNamePattern, podmanHost.name, rule))
+    ) {
+      return false;
+    }
+
+    if (
+      rule.podmanHostDescriptionPattern &&
+      (!podmanHost.description ||
+        !this.testRegex(
+          rule.podmanHostDescriptionPattern,
+          podmanHost.description,
+          rule,
+        ))
+    ) {
+      return false;
+    }
+
+    return true;
+  }
+
+  private testRegex(
+    pattern: string,
+    value: string,
+    rule: PodmanHostOwnerRule,
+  ): boolean {
+    try {
+      const regex: RegExp = new RegExp(pattern, "i");
+      return regex.test(value);
+    } catch {
+      logger.warn(
+        `Invalid regex in Podman host owner rule ${rule.id}: ${pattern}`,
+      );
+      return false;
+    }
+  }
+}
+
+export default new PodmanHostOwnerRuleEngineServiceClass();

package/Server/Services/PodmanHostOwnerRuleService.ts

@@ -0,0 +1,14 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/PodmanHostOwnerRule";
+import { IsBillingEnabled } from "../EnvironmentConfig";
+
+export class Service extends DatabaseService<Model> {
+  public constructor() {
+    super(Model);
+    if (IsBillingEnabled) {
+      this.hardDeleteItemsOlderThanInDays("createdAt", 3 * 365);
+    }
+  }
+}
+
+export default new Service();

package/Server/Services/PodmanHostOwnerTeamService.ts

@@ -0,0 +1,10 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/PodmanHostOwnerTeam";
+
+export class Service extends DatabaseService<Model> {
+  public constructor() {
+    super(Model);
+  }
+}
+
+export default new Service();

package/Server/Services/PodmanHostOwnerUserService.ts

@@ -0,0 +1,10 @@
+import DatabaseService from "./DatabaseService";
+import Model from "../../Models/DatabaseModels/PodmanHostOwnerUser";
+
+export class Service extends DatabaseService<Model> {
+  public constructor() {
+    super(Model);
+  }
+}
+
+export default new Service();