feat: auto-retry

src/dom.ts

@@ -1,9 +1,20 @@
 import fetch from "node-fetch";
 import { JSDOM } from "jsdom";
 
-export async function getDOM(endpoint: string) {
-  const {
-    window: { document },
-  } = new JSDOM(await fetch(endpoint).then((res) => res.text()));
-  return document;
+export async function getDOM(endpoint: string): Promise<Document | undefined> {
+  let retryCount = 3;
+  while (retryCount > 0) {
+    try {
+      const {
+        window: { document },
+      } = new JSDOM(await fetch(endpoint).then((res) => res.text()));
+      return document;
+    } catch (err) {
+      if (err.code === "ETIMEOUT") {
+        retryCount -= 1;
+      } else {
+        throw err;
+      }
+    }
+  }
 }

src/item.ts

@@ -39,6 +39,9 @@ export type Item = ImageItem | WritingItem | OtherMaterialItem;
 export async function getItemDetail(itemId: string): Promise<Item> {
   const viewUrl = `https://aryion.com/g4/view/${itemId}`;
   const document = await getDOM(viewUrl);
+  if (!document) {
+    throw new Error("Cannot fetch document");
+  }
 
   const authorAvatarUrl =
     document.querySelector<HTMLImageElement>(".avatar")?.src;
@@ -133,6 +136,9 @@ export async function getItemDetail(itemId: string): Promise<Item> {
 
       if (mimeType !== "application/pdf") {
         const res = await getDOM(contentUrl);
+        if (!res) {
+          throw new Error("Cannot fetch document");
+        }
         writing.content = {
           text: res.body.textContent?.trim() || "",
           html: res.body.innerHTML,

src/latest.ts

@@ -86,6 +86,10 @@ export async function getLatestUpdates(
   }
 
   const document = await getDOM(target);
+  if (!document) {
+    throw new Error("Cannot fetch document");
+  }
+
   const cursor = parsePageCursor(document)!;
 
   const updates = Array.from(document.querySelectorAll(".detail-item")).map(

src/user.ts

@@ -3,6 +3,9 @@ import { AryionUserNotFoundError } from "./error";
 
 export async function verifyUser(aryionUsername: string) {
   const document = await getDOM(`https://aryion.com/g4/user/${aryionUsername}`);
+  if (!document) {
+    throw new Error("Cannot fetch document");
+  }
 
   if (
     document.querySelector<HTMLSpanElement>(".g-box-title")!.textContent ===