fix: parse episode overwrite

yjl9903 · Jun 2, 2023 · caa281b · caa281b
1 parent cc3b89e
commit caa281b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 20 deletions.
diff --git a/packages/core/src/space/schema.ts b/packages/core/src/space/schema.ts
@@ -84,7 +84,20 @@ export const AnimePlanSchema = z
     date: z.coerce.date().optional(),
     rewrite: z
       .object({
-        episode: z.number().optional()
+        episode: z
+          .union([
+            z.coerce.number().transform(n => ({
+              offset: n,
+              gte: Number.MIN_SAFE_INTEGER,
+              lte: Number.MAX_SAFE_INTEGER
+            })),
+            z.object({
+              offset: z.coerce.number(),
+              gte: z.coerce.number().default(Number.MIN_SAFE_INTEGER),
+              lte: z.coerce.number().default(Number.MAX_SAFE_INTEGER)
+            })
+          ])
+          .optional()
       })
       .passthrough()
       .optional(),
@@ -143,7 +156,7 @@ export interface AnimePlan {
    * Rewrite the inferred things
    */
   readonly rewrite?: {
-    readonly episode?: number | { offset: number; gte?: number; lte?: number };
+    readonly episode?: { offset: number; gte: number; lte: number };
   };
 }
 

diff --git a/packages/core/src/system/anime.ts b/packages/core/src/system/anime.ts
@@ -199,24 +199,8 @@ export class Anime {
     if (episode !== undefined) {
       const overwrite = this.plan.rewrite?.episode;
       if (overwrite !== undefined) {
-        const parsed = z
-          .union([
-            z.coerce.number().transform(n => ({
-              offset: n,
-              gte: Number.MIN_SAFE_INTEGER,
-              lte: Number.MAX_SAFE_INTEGER
-            })),
-            z.object({
-              offset: z.coerce.number(),
-              gte: z.coerce.number().default(Number.MIN_SAFE_INTEGER),
-              lte: z.coerce.number().default(Number.MAX_SAFE_INTEGER)
-            })
-          ])
-          .safeParse(overwrite);
-        if (parsed.success) {
-          if (parsed.data.gte <= episode && episode <= parsed.data.lte) {
-            return episode + parsed.data.offset;
-          }
+        if (overwrite.gte <= episode && episode <= overwrite.lte) {
+          return episode + overwrite.offset;
         }
       }
       return episode;