feat: add pagination to fetchByCountry at flixhq

consumet · Feb 22, 2024 · f54bc7f · f54bc7f
1 parent ba9e25b
commit f54bc7f
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 11 deletions.
diff --git a/src/models/types.ts b/src/models/types.ts
@@ -32,7 +32,7 @@ export interface IAnimeResult {
   [x: string]: any; // other fields
 }
 
-export interface ISearch<T> {
+export interface Pagination<T> {
   currentPage?: number;
   hasNextPage?: boolean;
   totalPages?: number;
@@ -43,6 +43,10 @@ export interface ISearch<T> {
   results: T[];
 }
 
+export interface ISearch<T> extends Pagination<T> {}
+
+export interface IByCountry<T> extends Pagination<T> {}
+
 export interface Trailer {
   id: string;
   site?: string;

diff --git a/src/providers/movies/flixhq.ts b/src/providers/movies/flixhq.ts
@@ -12,6 +12,7 @@ import {
   ISearch,
 } from '../../models';
 import { MixDrop, VidCloud } from '../../extractors';
+import { IByCountry } from '../../models/types';
 
 class FlixHQ extends MovieParser {
   override readonly name = 'FlixHQ';
@@ -384,14 +385,24 @@ class FlixHQ extends MovieParser {
     }
   };
 
-  fetchByCountry = async (country: string): Promise<IMovieResult[]> => {
+  fetchByCountry = async (country: string, page: number = 1): Promise<IByCountry<IMovieResult>> => {
+    const result: IByCountry<IMovieResult> = {
+      currentPage: page,
+      hasNextPage: false,
+      results: [],
+    };
+    const navSelector = 'div.pre-pagination:nth-child(3) > nav:nth-child(1) > ul:nth-child(1)';
+
     try {
-      const { data } = await this.client.get(`${this.baseUrl}/country/${country}`);
+      const { data } = await this.client.get(`${this.baseUrl}/country/${country}/?page=${page}`);
       const $ = load(data);
 
-      const movies = $('div.container > section.block_area > div.block_area-content > div.film_list-wrap > div.flw-item')
-        .map((i, el) => {
-          const movie = {
+      result.hasNextPage =
+        $(navSelector).length > 0 ? !$(navSelector).children().last().hasClass('active') : false;
+
+      $('div.container > section.block_area > div.block_area-content > div.film_list-wrap > div.flw-item')
+        .each((i, el) => {
+          result.results.push({
             id: $(el).find('div.film-poster > a').attr('href')?.slice(1)!,
             title: $(el).find('div.film-detail > h2.film-name > a').attr('title')!,
             url: `${this.baseUrl}${$(el).find('div.film-poster > a').attr('href')}`,
@@ -403,23 +414,22 @@ class FlixHQ extends MovieParser {
                 ? TvType.MOVIE
                 : TvType.TVSERIES,
 
-          }
-          return movie
+          })
         })
         .get();
-      return movies;
+      return result;
     } catch (err) {
       throw new Error((err as Error).message);
     }
   };
 }
 
 // (async () => {
-//   const movie = new FlixHQ();
+// const movie = new FlixHQ();
 //   const search = await movie.search('the flash');
 //   // const movieInfo = await movie.fetchEpisodeSources('1168337', 'tv/watch-vincenzo-67955');
 //   // const recentTv = await movie.fetchTrendingTvShows();
-//   console.log(search);
+// console.log(search);
 // })();
 
 export default FlixHQ;