All files / ee/app/assets/javascripts/escalation_policies/graphql cache_updates.js

61.36% Statements 27/44
33.33% Branches 5/15
61.53% Functions 8/13
60.46% Lines 26/43

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117          4x       4x     4x 1x 1x       1x         1x 1x                         4x                                                   4x 1x   1x       1x         1x 1x 1x       1x             4x   4x         4x 1x 1x       4x               4x 1x     1x      
import produce from 'immer';
import { createAlert } from '~/alert';
 
import { s__ } from '~/locale';
 
const DELETE_ESCALATION_POLICY_ERROR = s__(
  'EscalationPolicies|The escalation policy could not be deleted. Please try again.',
);
 
const UPDATE_ESCALATION_POLICY_ERROR = s__(
  'EscalationPolicies|The escalation policy could not be updated. Please try again',
);
const addEscalationPolicyToStore = (store, query, { escalationPolicyCreate }, variables) => {
  const policy = escalationPolicyCreate?.escalationPolicy;
  Iif (!policy) {
    return;
  }
 
  const sourceData = store.readQuery({
    query,
    variables,
  });
 
  const data = produce(sourceData, (draftData) => {
    draftData.project.incidentManagementEscalationPolicies.nodes = [
      ...draftData.project.incidentManagementEscalationPolicies.nodes,
      policy,
    ];
  });
 
  store.writeQuery({
    query,
    variables,
    data,
  });
};
 
const updateEscalationPolicyInStore = (store, query, { escalationPolicyUpdate }, variables) => {
  const policy = escalationPolicyUpdate?.escalationPolicy;
  if (!policy) {
    return;
  }
 
  const sourceData = store.readQuery({
    query,
    variables,
  });
 
  const data = produce(sourceData, (draftData) => {
    draftData.project.incidentManagementEscalationPolicies.nodes = draftData.project.incidentManagementEscalationPolicies.nodes.map(
      (policyToUpdate) => {
        return policyToUpdate.id === policy.id ? policy : policyToUpdate;
      },
    );
  });
 
  store.writeQuery({
    query,
    variables,
    data,
  });
};
 
const deleteEscalationPolicFromStore = (store, query, { escalationPolicyDestroy }, variables) => {
  const escalationPolicy = escalationPolicyDestroy?.escalationPolicy;
 
  Iif (!escalationPolicy) {
    return;
  }
 
  const sourceData = store.readQuery({
    query,
    variables,
  });
 
  const data = produce(sourceData, (draftData) => {
    draftData.project.incidentManagementEscalationPolicies.nodes = draftData.project.incidentManagementEscalationPolicies.nodes.filter(
      ({ id }) => id !== escalationPolicy.id,
    );
  });
 
  store.writeQuery({
    query,
    variables,
    data,
  });
};
 
export const hasErrors = ({ errors = [] }) => errors?.length;
 
const onError = (data, message) => {
  createAlert({ message });
  throw new Error(data.errors);
};
 
export const updateStoreOnEscalationPolicyCreate = (store, query, data, variables) => {
  Eif (!hasErrors(data)) {
    addEscalationPolicyToStore(store, query, data, variables);
  }
};
 
export const updateStoreOnEscalationPolicyUpdate = (store, query, data, variables) => {
  if (hasErrors(data)) {
    onError(data, UPDATE_ESCALATION_POLICY_ERROR);
  } else {
    updateEscalationPolicyInStore(store, query, data, variables);
  }
};
 
export const updateStoreAfterEscalationPolicyDelete = (store, query, data, variables) => {
  Iif (hasErrors(data)) {
    onError(data, DELETE_ESCALATION_POLICY_ERROR);
  } else {
    deleteEscalationPolicFromStore(store, query, data, variables);
  }
};