feat: completed license server integration

backend-pg/src/ee/services/audit-log/audit-log-queue.ts

@@ -1,18 +1,24 @@
 import { QueueJobs, QueueName, TQueueServiceFactory } from "@app/queue";
+import { TProjectDalFactory } from "@app/services/project/project-dal";
 
+import { TLicenseServiceFactory } from "../license/license-service";
 import { TAuditLogDalFactory } from "./audit-log-dal";
 import { TCreateAuditLogDTO } from "./audit-log-types";
 
 type TAuditLogQueueServiceFactoryDep = {
   auditLogDal: TAuditLogDalFactory;
   queueService: TQueueServiceFactory;
+  projectDal: Pick<TProjectDalFactory, "findById">;
+  licenseService: Pick<TLicenseServiceFactory, "getPlan">;
 };
 
 export type TAuditLogQueueServiceFactory = ReturnType<typeof auditLogQueueServiceFactory>;
 
 export const auditLogQueueServiceFactory = ({
   auditLogDal,
-  queueService
+  queueService,
+  projectDal,
+  licenseService
 }: TAuditLogQueueServiceFactoryDep) => {
   const pushToLog = async (data: TCreateAuditLogDTO) => {
     await queueService.queue(QueueName.AuditLog, QueueJobs.AuditLog, data, {
@@ -24,8 +30,19 @@ export const auditLogQueueServiceFactory = ({
   };
 
   queueService.start(QueueName.AuditLog, async (job) => {
-    const { actor, orgId, event, ipAddress, projectId, userAgent, userAgentType } = job.data;
+    const { actor, event, ipAddress, projectId, userAgent, userAgentType } = job.data;
+    let { orgId } = job.data;
     const MS_IN_DAY = 24 * 60 * 60 * 1000;
+
+    if (!orgId) {
+      // it will never be undefined for both org and project id
+      // TODO(akhilmhdh): use caching here in dal to avoid db calls
+      const project = await projectDal.findById(projectId as string);
+      orgId = project.orgId;
+    }
+
+    const plan = await licenseService.getPlan(orgId);
+    const ttl = plan.auditLogsRetentionDays * MS_IN_DAY;
     await auditLogDal.create({
       actor: actor.type,
       actorMetadata: actor.metadata,
@@ -34,7 +51,7 @@ export const auditLogQueueServiceFactory = ({
       ipAddress,
       orgId,
       eventType: event.type,
-      expiresAt: new Date(Date.now() + 30 * MS_IN_DAY),
+      expiresAt: new Date(Date.now() + ttl),
       eventMetadata: event.metadata,
       userAgentType
     });