ani-client 2.3.0 → 2.5.0

package/dist/index.js

@@ -1,5 +1,80 @@
 'use strict';
 
+var __defProp = Object.defineProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+
+// src/utils/dataloader.ts
+var BatchLoader = class {
+  constructor(batchFetch, maxWaitMs = 50, maxBatchSize = 50) {
+    this.batchFetch = batchFetch;
+    this.maxWaitMs = maxWaitMs;
+    this.maxBatchSize = maxBatchSize;
+  }
+  queue = /* @__PURE__ */ new Map();
+  timeout = null;
+  /**
+   * Queue an ID to be fetched in the next batch.
+   * Returns a Promise that resolves when the batch request completes.
+   */
+  async load(id) {
+    return new Promise((resolve, reject) => {
+      let callbacks = this.queue.get(id);
+      if (!callbacks) {
+        callbacks = [];
+        this.queue.set(id, callbacks);
+      }
+      callbacks.push({ resolve, reject });
+      if (this.queue.size >= this.maxBatchSize) {
+        if (this.timeout) {
+          clearTimeout(this.timeout);
+          this.timeout = null;
+        }
+        void this.dispatch();
+        return;
+      }
+      if (this.timeout === null) {
+        this.timeout = setTimeout(() => {
+          void this.dispatch();
+        }, this.maxWaitMs);
+        const timer = this.timeout;
+        if (typeof timer.unref === "function") {
+          timer.unref();
+        }
+      }
+    });
+  }
+  async dispatch() {
+    this.timeout = null;
+    if (this.queue.size === 0) return;
+    const currentQueue = this.queue;
+    this.queue = /* @__PURE__ */ new Map();
+    const ids = Array.from(currentQueue.keys());
+    try {
+      const results = await this.batchFetch(ids);
+      const resultMap = /* @__PURE__ */ new Map();
+      for (const item of results) {
+        resultMap.set(item.id, item);
+      }
+      for (const [id, callbacks] of currentQueue.entries()) {
+        const result = resultMap.get(id);
+        if (result) {
+          for (const cb of callbacks) cb.resolve(result);
+        } else {
+          const err = new Error(`Item with ID ${id} not found in batch response`);
+          for (const cb of callbacks) cb.reject(err);
+        }
+      }
+    } catch (err) {
+      for (const callbacks of currentQueue.values()) {
+        for (const cb of callbacks) cb.reject(err);
+      }
+    }
+  }
+};
+
 // src/utils/markdown.ts
 function isSafeUrl(url) {
   return /^https?:\/\//i.test(url);
@@ -575,6 +650,25 @@ var AniListError = class _AniListError extends Error {
 };
 
 // src/queries/fragments.ts
+var fragments_exports = {};
+__export(fragments_exports, {
+  CHARACTER_FIELDS: () => CHARACTER_FIELDS,
+  CHARACTER_FIELDS_COMPACT: () => CHARACTER_FIELDS_COMPACT,
+  CHARACTER_FIELDS_WITH_VA: () => CHARACTER_FIELDS_WITH_VA,
+  MEDIA_FIELDS: () => MEDIA_FIELDS,
+  MEDIA_FIELDS_BASE: () => MEDIA_FIELDS_BASE,
+  MEDIA_FIELDS_LIGHT: () => MEDIA_FIELDS_LIGHT,
+  MEDIA_LIST_FIELDS: () => MEDIA_LIST_FIELDS,
+  MEDIA_RECOMMENDATION_FIELDS: () => MEDIA_RECOMMENDATION_FIELDS,
+  RELATIONS_FIELDS: () => RELATIONS_FIELDS,
+  STAFF_FIELDS: () => STAFF_FIELDS,
+  STAFF_MEDIA_FIELDS: () => STAFF_MEDIA_FIELDS,
+  STUDIO_FIELDS: () => STUDIO_FIELDS,
+  THREAD_FIELDS: () => THREAD_FIELDS,
+  USER_FAVORITES_FIELDS: () => USER_FAVORITES_FIELDS,
+  USER_FIELDS: () => USER_FIELDS,
+  VOICE_ACTOR_FIELDS_COMPACT: () => VOICE_ACTOR_FIELDS_COMPACT
+});
 var MEDIA_FIELDS_LIGHT = `
   __typename
   id
@@ -1265,6 +1359,9 @@ function buildMediaIncludeQuery(include) {
       statusDistribution { status amount }
     }`);
   }
+  if (include.customFields) {
+    extra.push(include.customFields);
+  }
   return extra;
 }
 function buildMediaByIdQuery(include) {
@@ -1492,6 +1589,14 @@ query ($id: Int!) {
     ${CHARACTER_FIELDS_WITH_VA}
   }
 }`;
+function buildCharacterByIdQuery(include) {
+  if (!include?.customFields) {
+    return include?.voiceActors ? QUERY_CHARACTER_BY_ID_WITH_VA : QUERY_CHARACTER_BY_ID;
+  }
+  const fields = include.voiceActors ? CHARACTER_FIELDS_WITH_VA : CHARACTER_FIELDS;
+  return `query ($id: Int!) { Character(id: $id) { ${fields}
+${include.customFields} } }`;
+}
 var QUERY_CHARACTER_SEARCH = `
 query ($search: String, $sort: [CharacterSort], $page: Int, $perPage: Int) {
   Page(page: $page, perPage: $perPage) {
@@ -1610,6 +1715,15 @@ query ($id: Int!, $perPage: Int) {
     ${STAFF_MEDIA_FIELDS}
   }
 }`;
+function buildStaffByIdQuery(include) {
+  if (!include?.customFields) {
+    return include?.characterRoles || include?.media ? QUERY_STAFF_BY_ID_WITH_MEDIA : QUERY_STAFF_BY_ID;
+  }
+  const extra = include?.characterRoles || include?.media ? `
+    ${STAFF_MEDIA_FIELDS}` : "";
+  return `query ($id: Int!, $perPage: Int) { Staff(id: $id) { ${STAFF_FIELDS}${extra}
+${include.customFields} } }`;
+}
 var QUERY_STAFF_SEARCH = `
 query ($search: String, $sort: [StaffSort], $page: Int, $perPage: Int) {
   Page(page: $page, perPage: $perPage) {
@@ -1627,9 +1741,15 @@ query ($id: Int!) {
     ${STUDIO_FIELDS}
   }
 }`;
-function buildStudioByIdQuery(mediaPerPage) {
-  if (mediaPerPage === void 0) return QUERY_STUDIO_BY_ID;
-  const pp = clampPerPage(mediaPerPage);
+function buildStudioByIdQuery(include) {
+  const custom = include?.customFields ? `
+    ${include.customFields}` : "";
+  if (!include?.media) {
+    if (!custom) return QUERY_STUDIO_BY_ID;
+    return `query ($id: Int!) { Studio(id: $id) { ${STUDIO_FIELDS}${custom} } }`;
+  }
+  const mediaOpts = typeof include.media === "object" ? include.media : {};
+  const pp = clampPerPage(mediaOpts.perPage ?? 25);
   return `
 query ($id: Int!) {
   Studio(id: $id) {
@@ -1643,7 +1763,7 @@ query ($id: Int!) {
       nodes {
         ${MEDIA_FIELDS_LIGHT}
       }
-    }
+    }${custom}
   }
 }`;
 }
@@ -1926,7 +2046,7 @@ function isNetworkError(err) {
 // src/client/character.ts
 async function getCharacter(client, id, include) {
   validateId(id, "characterId");
-  const query = include?.voiceActors ? QUERY_CHARACTER_BY_ID_WITH_VA : QUERY_CHARACTER_BY_ID;
+  const query = buildCharacterByIdQuery(include);
   const data = await client.request(query, { id });
   return data.Character;
 }
@@ -2403,12 +2523,13 @@ async function searchReviews(client, options = {}) {
 // src/client/staff.ts
 async function getStaff(client, id, include) {
   validateId(id, "staffId");
-  if (include?.media) {
+  const query = buildStaffByIdQuery(include);
+  if (include?.characterRoles || include?.media) {
     const perPage = typeof include.media === "object" ? include.media.perPage ?? 25 : 25;
-    const data2 = await client.request(QUERY_STAFF_BY_ID_WITH_MEDIA, { id, perPage });
+    const data2 = await client.request(query, { id, perPage });
     return data2.Staff;
   }
-  const data = await client.request(QUERY_STAFF_BY_ID, { id });
+  const data = await client.request(query, { id });
   return data.Staff;
 }
 async function searchStaff(client, options = {}) {
@@ -2423,13 +2544,8 @@ async function searchStaff(client, options = {}) {
 // src/client/studio.ts
 async function getStudio(client, id, include) {
   validateId(id, "studioId");
-  if (include?.media) {
-    const perPage = typeof include.media === "object" ? include.media.perPage : void 0;
-    const query = buildStudioByIdQuery(perPage);
-    const data2 = await client.request(query, { id });
-    return data2.Studio;
-  }
-  const data = await client.request(QUERY_STUDIO_BY_ID, { id });
+  const query = buildStudioByIdQuery(include);
+  const data = await client.request(query, { id });
   return data.Studio;
 }
 async function searchStudios(client, options = {}) {
@@ -2531,7 +2647,7 @@ function mapFavorites(fav) {
 
 // src/client/index.ts
 var DEFAULT_API_URL = "https://graphql.anilist.co";
-var LIB_VERSION = "2.3.0" ;
+var LIB_VERSION = "2.5.0" ;
 var AniListClient = class {
   apiUrl;
   headers;
@@ -2543,6 +2659,10 @@ var AniListClient = class {
   inFlight = /* @__PURE__ */ new Map();
   _rateLimitInfo;
   _lastRequestMeta;
+  mediaLoader;
+  characterLoader;
+  staffLoader;
+  batchingEnabled;
   constructor(options = {}) {
     this.apiUrl = options.apiUrl ?? DEFAULT_API_URL;
     this.headers = {
@@ -2558,6 +2678,12 @@ var AniListClient = class {
     this.hooks = options.hooks ?? {};
     this.logger = options.logger;
     this.signal = options.signal;
+    this.batchingEnabled = options.batching?.enabled ?? true;
+    const windowMs = options.batching?.windowMs ?? 50;
+    const maxBatchSize = options.batching?.maxBatchSize ?? 50;
+    this.mediaLoader = new BatchLoader((ids) => this.getMediaBatch(ids), windowMs, maxBatchSize);
+    this.characterLoader = new BatchLoader((ids) => this.getCharacterBatch(ids), windowMs, maxBatchSize);
+    this.staffLoader = new BatchLoader((ids) => this.getStaffBatch(ids), windowMs, maxBatchSize);
   }
   /**
    * The current rate limit information from the last API response.
@@ -2660,6 +2786,17 @@ var AniListClient = class {
         remaining: Number.parseInt(rlRemaining, 10),
         reset: Number.parseInt(rlReset, 10)
       };
+      const remainingPercent = this._rateLimitInfo.remaining / this._rateLimitInfo.limit;
+      if (remainingPercent <= 0.15) {
+        const resetIn = this._rateLimitInfo.reset - Math.floor(Date.now() / 1e3);
+        const resetInSeconds = resetIn > 0 ? resetIn : 0;
+        this.logger?.warn("Smart Alert: AniList API rate limit is critically low", {
+          remaining: this._rateLimitInfo.remaining,
+          limit: this._rateLimitInfo.limit,
+          resetInSeconds
+        });
+        this.hooks.onRateLimitAlert?.(this._rateLimitInfo.remaining, resetInSeconds);
+      }
     }
     const durationMs = Date.now() - start;
     const data = json.data;
@@ -2688,7 +2825,10 @@ var AniListClient = class {
    * @param include - Optional related data to include
    */
   async getMedia(id, include) {
-    return getMedia(this, id, include);
+    if (!this.batchingEnabled || include) {
+      return getMedia(this, id, include);
+    }
+    return this.mediaLoader.load(id);
   }
   async getMediaCharacters(mediaId, options = {}) {
     return getMediaCharacters(this, mediaId, options);
@@ -2756,7 +2896,10 @@ var AniListClient = class {
   }
   /** Fetch a character by AniList ID. Pass `{ voiceActors: true }` to include VA data. */
   async getCharacter(id, include) {
-    return getCharacter(this, id, include);
+    if (!this.batchingEnabled || include) {
+      return getCharacter(this, id, include);
+    }
+    return this.characterLoader.load(id);
   }
   /** Search for characters by name. */
   async searchCharacters(options = {}) {
@@ -2764,7 +2907,10 @@ var AniListClient = class {
   }
   /** Fetch a staff member by AniList ID. Pass `{ media: true }` or `{ media: { perPage } }` for media credits. */
   async getStaff(id, include) {
-    return getStaff(this, id, include);
+    if (!this.batchingEnabled || include) {
+      return getStaff(this, id, include);
+    }
+    return this.staffLoader.load(id);
   }
   /** Search for staff (voice actors, directors, etc.). */
   async searchStaff(options = {}) {
@@ -2878,12 +3024,25 @@ var AniListClient = class {
       page++;
     }
   }
+  /**
+   * Utility to fetch all pages of a paginated request and return them as a single array.
+   *
+   * @param fetchPage - A function that takes a page number and returns a `PagedResult<T>`
+   * @param maxPages - Maximum number of pages to fetch (default: Infinity)
+   */
+  async fetchAll(fetchPage, maxPages = Number.POSITIVE_INFINITY) {
+    const results = [];
+    for await (const item of this.paginate(fetchPage, maxPages)) {
+      results.push(item);
+    }
+    return results;
+  }
   /** Fetch multiple media entries in a single API request. */
   async getMediaBatch(ids) {
     if (ids.length === 0) return [];
     validateIds(ids, "mediaId");
     const [singleMediaId] = ids;
-    if (ids.length === 1 && singleMediaId !== void 0) return [await this.getMedia(singleMediaId)];
+    if (ids.length === 1 && singleMediaId !== void 0) return [await getMedia(this, singleMediaId)];
     return this.executeBatch(ids, buildBatchMediaQuery, "m");
   }
   /** Fetch multiple characters in a single API request. */
@@ -2891,7 +3050,8 @@ var AniListClient = class {
     if (ids.length === 0) return [];
     validateIds(ids, "characterId");
     const [singleCharId] = ids;
-    if (ids.length === 1 && singleCharId !== void 0) return [await this.getCharacter(singleCharId)];
+    if (ids.length === 1 && singleCharId !== void 0)
+      return [await getCharacter(this, singleCharId)];
     return this.executeBatch(ids, buildBatchCharacterQuery, "c");
   }
   /** Fetch multiple staff members in a single API request. */
@@ -2899,7 +3059,7 @@ var AniListClient = class {
     if (ids.length === 0) return [];
     validateIds(ids, "staffId");
     const [singleStaffId] = ids;
-    if (ids.length === 1 && singleStaffId !== void 0) return [await this.getStaff(singleStaffId)];
+    if (ids.length === 1 && singleStaffId !== void 0) return [await getStaff(this, singleStaffId)];
     return this.executeBatch(ids, buildBatchStaffQuery, "s");
   }
   /** @internal */
@@ -2969,6 +3129,7 @@ exports.AniListClient = AniListClient;
 exports.AniListError = AniListError;
 exports.CharacterRole = CharacterRole;
 exports.CharacterSort = CharacterSort;
+exports.Fragments = fragments_exports;
 exports.MediaFormat = MediaFormat;
 exports.MediaListSort = MediaListSort;
 exports.MediaListStatus = MediaListStatus;