ENG-39 Create new job on trigger for a full policy regen (#1701)

* feat(app): add option to regenerate all policies

* fix(app): rename 'Regenerate full policies' to 'Regenerate all policies'

* fix(app): rename description on policy regen dialog by removing 'mark it for review'

---------

Co-authored-by: chasprowebdev <chasgarciaprowebdev@gmail.com>

Author: github-actions[bot]

apps/app/src/app/(app)/[orgId]/policies/[policyId]/components/PolicyOverview.tsx

@@ -231,7 +231,7 @@ export function PolicyOverview({
               toast.info('Regeneration started');
             }}
             title="Regenerate Policy"
-            description="This will regenerate the policy content and mark it for review. Continue?"
+            description="This will regenerate the policy content. Continue?"
             confirmText="Regenerate"
             confirmIcon={<Icons.AI className="h-4 w-4" />}
           />

apps/app/src/app/(app)/[orgId]/policies/all/actions/regenerate-full-policies.ts

@@ -0,0 +1,30 @@
+'use server';
+
+import { authActionClient } from '@/actions/safe-action';
+import { generateFullPolicies } from '@/jobs/tasks/onboarding/generate-full-policies';
+import { tasks } from '@trigger.dev/sdk';
+import { z } from 'zod';
+
+export const regenerateFullPoliciesAction = authActionClient
+  .inputSchema(z.object({}))
+  .metadata({
+    name: 'regenerate-full-policies',
+    track: {
+      event: 'regenerate-full-policies',
+      channel: 'server',
+    },
+  })
+  .action(async ({ ctx }) => {
+    const { session } = ctx;
+
+    if (!session?.activeOrganizationId) {
+      throw new Error('No active organization');
+    }
+
+    await tasks.trigger<typeof generateFullPolicies>('generate-full-policies', {
+      organizationId: session.activeOrganizationId,
+    });
+
+    // Revalidation handled by safe-action middleware using x-pathname header
+    return { success: true };
+  });

apps/app/src/app/(app)/[orgId]/policies/all/components/FullPolicyHeaderActions.tsx

@@ -0,0 +1,82 @@
+'use client';
+
+import { Button } from '@comp/ui/button';
+import {
+  Dialog,
+  DialogContent,
+  DialogDescription,
+  DialogFooter,
+  DialogHeader,
+  DialogTitle,
+} from '@comp/ui/dialog';
+import {
+  DropdownMenu,
+  DropdownMenuContent,
+  DropdownMenuItem,
+  DropdownMenuTrigger,
+} from '@comp/ui/dropdown-menu';
+import { Icons } from '@comp/ui/icons';
+import { useAction } from 'next-safe-action/hooks';
+import { useState } from 'react';
+import { toast } from 'sonner';
+import { regenerateFullPoliciesAction } from '../actions/regenerate-full-policies';
+
+export function FullPolicyHeaderActions() {
+  const [isRegenerateConfirmOpen, setRegenerateConfirmOpen] = useState(false);
+
+  const regenerate = useAction(regenerateFullPoliciesAction, {
+    onSuccess: () => {
+      toast.success('Policy regeneration started. This may take a few minutes.');
+      setRegenerateConfirmOpen(false);
+    },
+    onError: (error) => {
+      toast.error(error.error.serverError || 'Failed to regenerate policies');
+    },
+  });
+
+  const handleRegenerate = async () => {
+    await regenerate.execute({});
+  };
+
+  return (
+    <>
+      <DropdownMenu>
+        <DropdownMenuTrigger asChild>
+          <Button size="icon" variant="ghost" className="m-0 size-auto p-2">
+            <Icons.Settings className="h-4 w-4" />
+          </Button>
+        </DropdownMenuTrigger>
+        <DropdownMenuContent align="end">
+          <DropdownMenuItem onClick={() => setRegenerateConfirmOpen(true)}>
+            <Icons.AI className="mr-2 h-4 w-4" /> Regenerate all policies
+          </DropdownMenuItem>
+        </DropdownMenuContent>
+      </DropdownMenu>
+
+      {/* Regenerate Confirmation Dialog */}
+      <Dialog open={isRegenerateConfirmOpen} onOpenChange={setRegenerateConfirmOpen}>
+        <DialogContent>
+          <DialogHeader>
+            <DialogTitle>Regenerate All Policies</DialogTitle>
+            <DialogDescription>
+              This will generate new policy content for all policies using your org context and
+              frameworks. Continue?
+            </DialogDescription>
+          </DialogHeader>
+          <DialogFooter>
+            <Button
+              variant="outline"
+              onClick={() => setRegenerateConfirmOpen(false)}
+              disabled={regenerate.status === 'executing'}
+            >
+              Cancel
+            </Button>
+            <Button onClick={handleRegenerate} disabled={regenerate.status === 'executing'}>
+              {regenerate.status === 'executing' ? 'Working…' : 'Confirm'}
+            </Button>
+          </DialogFooter>
+        </DialogContent>
+      </Dialog>
+    </>
+  );
+}

apps/app/src/app/(app)/[orgId]/policies/all/page.tsx

@@ -2,6 +2,7 @@ import PageWithBreadcrumb from '@/components/pages/PageWithBreadcrumb';
 import { getValidFilters } from '@/lib/data-table';
 import type { SearchParams } from '@/types';
 import type { Metadata } from 'next';
+import { FullPolicyHeaderActions } from './components/FullPolicyHeaderActions';
 import { PoliciesTable } from './components/policies-table';
 import { getPolicies } from './data/queries';
 import { searchParamsCache } from './data/validations';
@@ -23,7 +24,10 @@ export default async function PoliciesPage({ ...props }: PolicyTableProps) {
   ]);
 
   return (
-    <PageWithBreadcrumb breadcrumbs={[{ label: 'Policies', current: true }]}>
+    <PageWithBreadcrumb
+      breadcrumbs={[{ label: 'Policies', current: true }]}
+      headerRight={<FullPolicyHeaderActions />}
+    >
       <PoliciesTable promises={promises} />
     </PageWithBreadcrumb>
   );

apps/app/src/jobs/tasks/onboarding/generate-full-policies.ts

@@ -0,0 +1,52 @@
+import { db } from '@db';
+import { logger, queue, task } from '@trigger.dev/sdk';
+import { getOrganizationContext, triggerPolicyUpdates } from './onboard-organization-helpers';
+
+// v4 queues must be declared in advance
+const generateFullPoliciesQueue = queue({
+  name: 'generate-full-policies',
+  concurrencyLimit: 100,
+});
+
+export const generateFullPolicies = task({
+  id: 'generate-full-policies',
+  queue: generateFullPoliciesQueue,
+  retry: {
+    maxAttempts: 3,
+  },
+  run: async (payload: { organizationId: string }) => {
+    logger.info(`Starting full policy generation for organization ${payload.organizationId}`);
+
+    try {
+      // Get organization context
+      const { questionsAndAnswers } = await getOrganizationContext(payload.organizationId);
+
+      // Get frameworks
+      const frameworkInstances = await db.frameworkInstance.findMany({
+        where: {
+          organizationId: payload.organizationId,
+        },
+      });
+
+      const frameworks = await db.frameworkEditorFramework.findMany({
+        where: {
+          id: {
+            in: frameworkInstances.map((instance) => instance.frameworkId),
+          },
+        },
+      });
+
+      // Trigger policy updates for all policies
+      await triggerPolicyUpdates(payload.organizationId, questionsAndAnswers, frameworks);
+
+      logger.info(
+        `Successfully triggered policy updates for organization ${payload.organizationId}`,
+      );
+    } catch (error) {
+      logger.error(`Error during policy generation for organization ${payload.organizationId}:`, {
+        error: error instanceof Error ? error.message : String(error),
+      });
+      throw error;
+    }
+  },
+});