import { useTranslate } from "@probo/i18n";
import { graphql } from "relay-runtime";
import { useMutationWithToasts } from "../useMutationWithToasts";
import type { MeasureGraphDeleteMutation } from "./__generated__/MeasureGraphDeleteMutation.graphql";

export const measuresQuery = graphql`
  query MeasureGraphListQuery($organizationId: ID!) {
    organization: node(id: $organizationId) {
      id
      ...MeasuresPageFragment
    }
  }
`;

const deleteMeasureMutation = graphql`
  mutation MeasureGraphDeleteMutation(
    $input: DeleteMeasureInput!
    $connections: [ID!]!
  ) {
    deleteMeasure(input: $input) {
      deletedMeasureId @deleteEdge(connections: $connections)
    }
  }
`;

export const MeasureConnectionKey = "MeasuresGraphListQuery__measures";

export function useDeleteMeasureMutation() {
  const { __ } = useTranslate();

  return useMutationWithToasts<MeasureGraphDeleteMutation>(
    deleteMeasureMutation,
    {
      successMessage: __("Measure deleted successfully."),
      errorMessage: __("Failed to delete measure. Please try again."),
    }
  );
}

export const measureNodeQuery = graphql`
  query MeasureGraphNodeQuery($measureId: ID!) {
    node(id: $measureId) {
      ... on Measure {
        id
        name
        description
        state
        category
        evidencesInfos: evidences(first: 0) {
          totalCount
        }
        risksInfos: risks(first: 0) {
          totalCount
        }
        tasksInfos: tasks(first: 0) {
          totalCount
        }
        controlsInfos: controls(first: 0) {
          totalCount
        }
        ...MeasureRisksTabFragment
        ...MeasureTasksTabFragment
        ...MeasureControlsTabFragment
        ...MeasureFormDialogMeasureFragment
        ...MeasureEvidencesTabFragment
      }
    }
  }
`;

const measureUpdateMutation = graphql`
  mutation MeasureGraphUpdateMutation($input: UpdateMeasureInput!) {
    updateMeasure(input: $input) {
      measure {
        ...MeasureFormDialogMeasureFragment
      }
    }
  }
`;

export const useUpdateMeasure = () => {
  const { __ } = useTranslate();

  return useMutationWithToasts(measureUpdateMutation, {
    successMessage: __("Measure updated successfully."),
    errorMessage: __("Failed to update measure. Please try again."),
  });
};