Merge pull request #990 from trycompai/main

[comp] Production Deploy

Author: Marfuen

apps/app/src/app/(app)/setup/actions/create-organization-minimal.ts

@@ -3,8 +3,10 @@
 import { createStripeCustomer } from '@/actions/organization/lib/create-stripe-customer';
 import { initializeOrganization } from '@/actions/organization/lib/initialize-organization';
 import { authActionClientWithoutOrg } from '@/actions/safe-action';
+import { createFleetLabelForOrg } from '@/jobs/tasks/device/create-fleet-label-for-org';
 import { auth } from '@/utils/auth';
 import { db } from '@comp/db';
+import { tasks } from '@trigger.dev/sdk/v3';
 import { revalidatePath } from 'next/cache';
 import { headers } from 'next/headers';
 import { z } from 'zod';
@@ -110,6 +112,11 @@ export const createOrganizationMinimal = authActionClientWithoutOrg
       revalidatePath(`/${orgId}`);
       revalidatePath('/setup');
 
+      // Create Fleet Label.
+      await tasks.trigger<typeof createFleetLabelForOrg>('create-fleet-label-for-org', {
+        organizationId: orgId,
+      });
+
       return {
         success: true,
         organizationId: orgId,

apps/app/src/app/(app)/setup/actions/create-organization.ts

@@ -3,6 +3,7 @@
 import { createStripeCustomer } from '@/actions/organization/lib/create-stripe-customer';
 import { initializeOrganization } from '@/actions/organization/lib/initialize-organization';
 import { authActionClientWithoutOrg } from '@/actions/safe-action';
+import { createFleetLabelForOrg } from '@/jobs/tasks/device/create-fleet-label-for-org';
 import { onboardOrganization as onboardOrganizationTask } from '@/jobs/tasks/onboarding/onboard-organization';
 import { auth } from '@/utils/auth';
 import { db } from '@comp/db';
@@ -142,6 +143,11 @@ export const createOrganization = authActionClientWithoutOrg
 
       (await cookies()).set('publicAccessToken', handle.publicAccessToken);
 
+      // Create Fleet Label.
+      await tasks.trigger<typeof createFleetLabelForOrg>('create-fleet-label-for-org', {
+        organizationId: orgId,
+      });
+
       return {
         success: true,
         handle: handle.id,

apps/portal/public/osquery-agent.jpeg

@@ -0,0 +1,62 @@
+'use server';
+
+import { db } from '@comp/db';
+
+export async function acceptPolicy(policyId: string, memberId: string) {
+  try {
+    // Add the member ID to the signedBy array if not already present
+    const policy = await db.policy.findUnique({
+      where: { id: policyId },
+    });
+
+    if (!policy) {
+      throw new Error('Policy not found');
+    }
+
+    if (!policy.signedBy.includes(memberId)) {
+      await db.policy.update({
+        where: { id: policyId },
+        data: {
+          signedBy: {
+            push: memberId,
+          },
+        },
+      });
+    }
+
+    return { success: true };
+  } catch (error) {
+    console.error('Error accepting policy:', error);
+    return { success: false, error: 'Failed to accept policy' };
+  }
+}
+
+export async function acceptAllPolicies(policyIds: string[], memberId: string) {
+  try {
+    // Update all policies to include the member ID in signedBy array
+    const updatePromises = policyIds.map(async (policyId) => {
+      const policy = await db.policy.findUnique({
+        where: { id: policyId },
+      });
+
+      if (policy && !policy.signedBy.includes(memberId)) {
+        return db.policy.update({
+          where: { id: policyId },
+          data: {
+            signedBy: {
+              push: memberId,
+            },
+          },
+        });
+      }
+      return null;
+    });
+
+    await Promise.all(updatePromises);
+
+    return { success: true };
+  } catch (error) {
+    console.error('Error accepting all policies:', error);
+    return { success: false, error: 'Failed to accept all policies' };
+  }
+}