1 files changed, 30 insertions, 0 deletions
diff --git a/app/assets/javascripts/ci/artifacts/graphql/cache_update.js b/app/assets/javascripts/ci/artifacts/graphql/cache_update.js
new file mode 100644
index 00000000000..9fa6114c7d4
--- /dev/null
+++ b/app/assets/javascripts/ci/artifacts/graphql/cache_update.js
@@ -0,0 +1,30 @@
+import produce from 'immer';
+
+export const hasErrors = ({ errors = [] }) => errors?.length;
+
+export function removeArtifactFromStore(store, deletedArtifactId, query, variables) {
+  if (hasErrors(deletedArtifactId)) return;
+
+  const sourceData = store.readQuery({
+    query,
+    variables,
+  });
+
+  const data = produce(sourceData, (draftData) => {
+    draftData.project.jobs.nodes = draftData.project.jobs.nodes.map((jobNode) => {
+      return {
+        ...jobNode,
+        artifacts: {
+          ...jobNode.artifacts,
+          nodes: jobNode.artifacts.nodes.filter(({ id }) => id !== deletedArtifactId),
+        },
+      };
+    });
+  });
+
+  store.writeQuery({
+    query,
+    variables,
+    data,
+  });
+}