From 490c4c302333148d63180557a457fe58e751e73f Mon Sep 17 00:00:00 2001
From: dominik-zeglen <flesz3@o2.pl>
Date: Thu, 5 Mar 2020 18:48:54 +0100
Subject: [PATCH] Check shop errors too

---
 src/collections/containers/CollectionOperations.tsx | 6 +++++-
 src/collections/views/CollectionDetails.tsx         | 9 +++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/collections/containers/CollectionOperations.tsx b/src/collections/containers/CollectionOperations.tsx
index fd40ff642..9a65c37f1 100644
--- a/src/collections/containers/CollectionOperations.tsx
+++ b/src/collections/containers/CollectionOperations.tsx
@@ -54,6 +54,7 @@ interface CollectionUpdateOperationsProps {
     >;
   }) => React.ReactNode;
   onUpdate: (data: CollectionUpdate) => void;
+  onUpdateWithCollection: (data: CollectionUpdateWithHomepage) => void;
   onProductAssign: (data: CollectionAssignProduct) => void;
   onProductUnassign: (data: UnassignCollectionProduct) => void;
   onRemove: (data: RemoveCollection) => void;
@@ -62,6 +63,7 @@ interface CollectionUpdateOperationsProps {
 const CollectionOperations: React.FC<CollectionUpdateOperationsProps> = ({
   children,
   onUpdate,
+  onUpdateWithCollection,
   onProductAssign,
   onProductUnassign,
   onRemove
@@ -72,7 +74,9 @@ const CollectionOperations: React.FC<CollectionUpdateOperationsProps> = ({
         {(...removeCollection) => (
           <TypedCollectionAssignProductMutation onCompleted={onProductAssign}>
             {(...assignProduct) => (
-              <TypedCollectionUpdateWithHomepageMutation onCompleted={onUpdate}>
+              <TypedCollectionUpdateWithHomepageMutation
+                onCompleted={onUpdateWithCollection}
+              >
                 {(...updateWithHomepage) => (
                   <TypedUnassignCollectionProductMutation
                     onCompleted={onProductUnassign}
diff --git a/src/collections/views/CollectionDetails.tsx b/src/collections/views/CollectionDetails.tsx
index 74075e85e..f5cbe6b93 100644
--- a/src/collections/views/CollectionDetails.tsx
+++ b/src/collections/views/CollectionDetails.tsx
@@ -35,6 +35,7 @@ import {
   CollectionUrlQueryParams,
   CollectionUrlDialog
 } from "../urls";
+import { CollectionUpdateWithHomepage } from "../types/CollectionUpdateWithHomepage";
 
 interface CollectionDetailsProps {
   id: string;
@@ -93,6 +94,13 @@ export const CollectionDetails: React.FC<CollectionDetailsProps> = ({
             }
           }
         };
+        const handleCollectioUpdateWithHomepage = (
+          data: CollectionUpdateWithHomepage
+        ) => {
+          if (data.homepageCollectionUpdate.errors.length === 0) {
+            handleCollectionUpdate(data);
+          }
+        };
 
         const handleProductAssign = (data: CollectionAssignProduct) => {
           if (data.collectionAddProducts.errors.length === 0) {
@@ -130,6 +138,7 @@ export const CollectionDetails: React.FC<CollectionDetailsProps> = ({
         return (
           <CollectionOperations
             onUpdate={handleCollectionUpdate}
+            onUpdateWithCollection={handleCollectioUpdateWithHomepage}
             onProductAssign={handleProductAssign}
             onProductUnassign={handleProductUnassign}
             onRemove={handleCollectionRemove}