diff --git a/apps/console/src/pages/organizations/frameworks/FrameworkLayoutView.tsx b/apps/console/src/pages/organizations/frameworks/FrameworkLayoutView.tsx index f970f01d5..48450282d 100644 --- a/apps/console/src/pages/organizations/frameworks/FrameworkLayoutView.tsx +++ b/apps/console/src/pages/organizations/frameworks/FrameworkLayoutView.tsx @@ -23,6 +23,7 @@ import type { FrameworkLayoutViewDeleteMutation } from "./__generated__/Framewor import { PageTemplate } from "@/components/PageTemplate"; import { FrameworkLayoutViewSkeleton } from "./FrameworkLayout"; import { ControlList } from "./FrameworkLayoutView/ControlList"; +import { FrameworkLayoutViewExportAuditMutation } from "./__generated__/FrameworkLayoutViewExportAuditMutation.graphql"; const FrameworkLayoutViewQuery = graphql` query FrameworkLayoutViewQuery($frameworkId: ID!) { @@ -60,6 +61,14 @@ const DeleteFrameworkMutation = graphql` } `; +const exportAuditMutation = graphql` + mutation FrameworkLayoutViewExportAuditMutation($input: ExportAuditInput!) { + exportAudit(input: $input) { + success + } + } +`; + function FrameworkLayoutViewContent({ queryRef, }: { @@ -73,17 +82,28 @@ function FrameworkLayoutViewContent({ const [isDeleteDialogOpen, setIsDeleteDialogOpen] = useState(false); const [isDeleting, setIsDeleting] = useState(false); + const [exportAudit, _] = + useMutation(exportAuditMutation); + // Setup delete mutation const [commitDeleteMutation] = useMutation( - DeleteFrameworkMutation, + DeleteFrameworkMutation ); + const handleExportAudit = useCallback(() => { + exportAudit({ + variables: { + input: { frameworkId: framework.id }, + }, + }); + }, [exportAudit, framework.id]); + const handleDeleteFramework = useCallback(() => { setIsDeleting(true); const connectionId = ConnectionHandler.getConnectionID( organizationId!, - "FrameworkListView_frameworks", + "FrameworkListView_frameworks" ); commitDeleteMutation({ @@ -140,6 +160,9 @@ function FrameworkLayoutViewContent({ Edit Framework +