Add redirection logic if no artworks or cms field is false

artsy · May 24, 2021 · ed5f831 · ed5f831
1 parent 4e7ef2f
commit ed5f831
Show file tree

Hide file tree

Showing 2 changed files with 106 additions and 40 deletions.
diff --git a/src/v2/Apps/Partner/partnerRoutes.tsx b/src/v2/Apps/Partner/partnerRoutes.tsx
@@ -180,9 +180,38 @@ export const partnerRoutes: AppRouteConfig[] = [
           ) {
             partner(id: $partnerId) @principalField {
               ...Works_partner @arguments(input: $input)
+              displayWorksSection
+              counts {
+                eligibleArtworks
+              }
             }
           }
         `,
+        render: ({ Component, props, match }) => {
+          if (!(Component && props)) {
+            return
+          }
+
+          const { partner } = props as any
+
+          if (!partner) {
+            return
+          }
+
+          const {
+            displayWorksSection,
+            counts: { eligibleArtworks },
+          } = partner
+
+          if (!displayWorksSection || eligibleArtworks <= 0) {
+            throw new RedirectException(
+              `/partner2/${match.params.partnerId}`,
+              302
+            )
+          }
+
+          return <Component {...props} />
+        },
       },
       {
         getComponent: () => ArtistsRoute,

diff --git a/src/v2/__generated__/partnerRoutes_WorksQuery.graphql.ts b/src/v2/__generated__/partnerRoutes_WorksQuery.graphql.ts