tidy up union types so we can reuse the MissingHeadReportSchema typename

codecov · Jan 6, 2025 · 2b112ad · 2b112ad
1 parent 307a1a9
commit 2b112ad
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/pages/PullRequestPage/queries/PullPageDataQueryOpts.tsx b/src/pages/PullRequestPage/queries/PullPageDataQueryOpts.tsx
@@ -22,6 +22,11 @@ const BundleAnalysisReportSchema = z.object({
   isCached: z.boolean(),
 })
 
+const BundleAnalysisReportUnion = z.discriminatedUnion('__typename', [
+  BundleAnalysisReportSchema,
+  z.object({ __typename: MissingHeadReportSchema.shape.__typename }),
+])
+
 const BundleAnalysisComparisonResult = z.union([
   z.literal('BundleAnalysisComparison'),
   FirstPullRequestSchema.shape.__typename,
@@ -53,12 +58,7 @@ const RepositorySchema = z.object({
           commitid: z.string(),
           bundleAnalysis: z
             .object({
-              bundleAnalysisReport: z
-                .discriminatedUnion('__typename', [
-                  BundleAnalysisReportSchema,
-                  z.object({ __typename: z.literal('MissingHeadReport') }),
-                ])
-                .nullable(),
+              bundleAnalysisReport: BundleAnalysisReportUnion.nullable(),
             })
             .nullable(),
         })