probo/apps/console/src/hooks/graph/FrameworkGraph.ts

// Copyright (c) 2025-2026 Probo Inc <hello@getprobo.com>.
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

import { sprintf } from "@probo/helpers";
import { useTranslate } from "@probo/i18n";
import { useConfirm } from "@probo/ui";
import { useCallback } from "react";
import { graphql } from "relay-runtime";

import { useMutationWithToasts } from "../useMutationWithToasts";

/* eslint-disable relay/unused-fields, relay/must-colocate-fragment-spreads */

export const connectionListKey = "FrameworksListQuery_frameworks";

export const frameworksQuery = graphql`
  query FrameworkGraphListQuery($organizationId: ID!) {
    organization: node(id: $organizationId) {
      ... on Organization {
        id
        canCreateFramework: permission(action: "core:framework:create")
        frameworks(first: 100)
          @connection(key: "FrameworksListQuery_frameworks") {
          __id
          edges {
            node {
              id
              canUpdate: permission(action: "core:framework:update")
              canDelete: permission(action: "core:framework:delete")
              ...FrameworksPageCardFragment
            }
          }
        }
      }
    }
  }
`;

const deleteFrameworkMutation = graphql`
  mutation FrameworkGraphDeleteMutation(
    $input: DeleteFrameworkInput!
    $connections: [ID!]!
  ) {
    deleteFramework(input: $input) {
      deletedFrameworkId @deleteEdge(connections: $connections)
    }
  }
`;

export const useDeleteFrameworkMutation = (
  framework: { id: string; name: string },
  connectionId: string,
) => {
  const [commitDelete] = useMutationWithToasts(deleteFrameworkMutation, {
    errorMessage: "Failed to delete framework",
    successMessage: "Framework deleted successfully",
  });
  const confirm = useConfirm();
  const { __ } = useTranslate();

  return useCallback(
    (options?: { onSuccess?: () => void }) => {
      return confirm(
        () => {
          return commitDelete({
            variables: {
              input: {
                frameworkId: framework.id,
              },
              connections: [connectionId],
            },
            ...options,
          });
        },
        {
          message: sprintf(
            __(
              "This will permanently delete framework \"%s\". This action cannot be undone.",
            ),
            framework.name,
          ),
        },
      );
    },
    [framework, connectionId, commitDelete, confirm, __],
  );
};

export const frameworkNodeQuery = graphql`
  query FrameworkGraphNodeQuery($frameworkId: ID!) {
    node(id: $frameworkId) {
      ... on Framework {
        id
        name
        ...FrameworkDetailPageFragment
      }
    }
  }
`;

export const frameworkControlNodeQuery = graphql`
  query FrameworkGraphControlNodeQuery($controlId: ID!) {
    node(id: $controlId) {
      ... on Control {
        id
        name
        sectionTitle
        description
        bestPractice
        implemented
        notImplementedJustification
        canUpdate: permission(action: "core:control:update")
        canDelete: permission(action: "core:control:delete")
        canCreateMeasureMapping: permission(
          action: "core:control:create-measure-mapping"
        )
        canDeleteMeasureMapping: permission(
          action: "core:control:delete-measure-mapping"
        )
        canCreateDocumentMapping: permission(
          action: "core:control:create-document-mapping"
        )
        canDeleteDocumentMapping: permission(
          action: "core:control:delete-document-mapping"
        )
        canCreateAuditMapping: permission(
          action: "core:control:create-audit-mapping"
        )
        canDeleteAuditMapping: permission(
          action: "core:control:delete-audit-mapping"
        )
        canCreateSnapshotMapping: permission(
          action: "core:control:create-snapshot-mapping"
        )
        canDeleteSnapshotMapping: permission(
          action: "core:control:delete-snapshot-mapping"
        )
        canCreateObligationMapping: permission(
          action: "core:control:create-obligation-mapping"
        )
        canDeleteObligationMapping: permission(
          action: "core:control:delete-obligation-mapping"
        )
        ...FrameworkControlDialogFragment
        measures(first: 100)
          @connection(key: "FrameworkGraphControl_measures") {
          __id
          edges {
            node {
              id
              ...LinkedMeasuresCardFragment
            }
          }
        }
        documents(first: 100)
          @connection(key: "FrameworkGraphControl_documents") {
          __id
          edges {
            node {
              id
              ...LinkedDocumentsCardFragment
            }
          }
        }
        audits(first: 100) @connection(key: "FrameworkGraphControl_audits") {
          __id
          edges {
            node {
              id
              ...LinkedAuditsCardFragment
            }
          }
        }
        obligations(first: 100)
          @connection(key: "FrameworkGraphControl_obligations") {
          __id
          edges {
            node {
              id
              ...LinkedObligationsCardFragment
            }
          }
        }
        snapshots(first: 100)
          @connection(key: "FrameworkGraphControl_snapshots") {
          __id
          edges {
            node {
              id
              ...LinkedSnapshotsCardFragment
            }
          }
        }
      }
    }
  }
`;