feat(tv-series): seasons, episodes [kickoff]

Author: bartholomej

demo.ts

@@ -4,13 +4,13 @@ import { csfd } from './src';
 // csfd.setOptions({ optionsRequest: { credentials: 'include' } });
 
 // Parse movie
-csfd.movie(10135).then((movie) => console.log(movie));
+csfd.movie(621073).then((movie) => console.log(movie));
 
 // csfd.search('matrix').then((search) => console.log(search));
 // csfd.cinema(1, 'today').then((cinema) => console.log(cinema));
 
 // Parse creator
-csfd.creator(2120).then((creator) => console.log(creator));
+// csfd.creator(2120).then((creator) => console.log(creator));
 
 /**
  * USER RATINGS

src/dto/movie.ts

@@ -17,6 +17,14 @@ export interface CSFDMovie extends CSFDScreening {
   premieres: CSFDPremiere[];
   related: CSFDMovieListItem[];
   similar: CSFDMovieListItem[];
+  seasons: CSFDSeason[] | null;
+  episodes: CSFDSeason[] | null;
+  parent: CSFDParent | null;
+}
+
+export interface CSFDParent {
+  season: { id: number; name: string; };
+  series: { id: number; name: string; };
 }
 
 export type CSFDVodService =
@@ -86,6 +94,8 @@ export interface CSFDMovieListItem {
   url: string;
 }
 
+
+
 export type CSFDGenres =
   | 'Akční'
   | 'Animovaný'
@@ -181,3 +191,10 @@ export interface CSFDPremiere {
 }
 
 export type CSFDBoxContent = 'Související' | 'Podobné';
+
+export interface CSFDSeason {
+  id: number;
+  name: string;
+  url: string;
+  info: string | null;
+}

src/helpers/global.helper.ts

@@ -11,6 +11,16 @@ export const parseIdFromUrl = (url: string): number => {
   }
 };
 
+export const parseLastIdFromUrl = (url: string): number => {
+  if (url) {
+    const idSlug = url?.split('/')[3];
+    const id = idSlug?.split('-')[0];
+    return +id || null;
+  } else {
+    return null;
+  }
+};
+
 export const getColor = (cls: string): CSFDColorRating => {
   switch (cls) {
     case 'page-lightgrey':

src/helpers/movie.helper.ts

@@ -8,12 +8,14 @@ import {
   CSFDGenres,
   CSFDMovieCreator,
   CSFDMovieListItem,
+  CSFDParent,
   CSFDPremiere,
+  CSFDSeason,
   CSFDTitlesOther,
   CSFDVod,
   CSFDVodService
 } from '../dto/movie';
-import { addProtocol, getColor, parseISO8601Duration, parseIdFromUrl } from './global.helper';
+import { addProtocol, getColor, parseISO8601Duration, parseIdFromUrl, parseLastIdFromUrl } from './global.helper';
 
 /**
  * Maps language-specific movie creator group labels.
@@ -241,6 +243,42 @@ const parseMoviePeople = (el: HTMLElement): CSFDMovieCreator[] => {
   );
 };
 
+export const getSeasonsOrEpisodes = (el: HTMLElement): CSFDSeason[] | null => {
+  const childrenList = el.querySelector('.film-episodes-list');
+  if (!childrenList) return null;
+
+  const childrenNodes = childrenList.querySelectorAll('.film-title');
+  if (!childrenNodes?.length) return [];
+
+  return childrenNodes.map((season) => {
+    const nameContainer = season.querySelector('.film-title-name');
+    const infoContainer = season.querySelector('.info');
+
+    return {
+      id: parseLastIdFromUrl(nameContainer?.getAttribute('href') || ''),
+      name: nameContainer?.textContent?.trim() || null,
+      url: nameContainer?.getAttribute('href') || null,
+      info: infoContainer?.textContent?.replace(/[{()}]/g, '').trim() || null,
+    };
+  });
+}
+
+export const getParent = (el: HTMLElement): CSFDParent | null => {
+  const parents = el.querySelectorAll('.film-header h2 a');
+  if (parents.length === 0) return null;
+
+  const [parentSeries, parentSeason] = parents;
+
+  return {
+    series: {
+      id: parseIdFromUrl(parentSeries?.getAttribute('href') || ''), name: parentSeries?.textContent?.trim() || null
+    },
+    season: {
+      id: parseIdFromUrl(parentSeason?.getAttribute('href') || ''), name: parentSeason?.textContent?.trim() || null
+    }
+  };
+}
+
 export const getMovieGroup = (el: HTMLElement, group: CSFDCreatorGroups | CSFDCreatorGroupsEnglish | CSFDCreatorGroupsSlovak): CSFDMovieCreator[] => {
   const creators = el.querySelectorAll('.creators h4');
   const element = creators.filter((elem) => elem.textContent.trim().includes(group))[0];

src/services/movie.service.ts

@@ -22,7 +22,9 @@ import {
   getMovieTrivia,
   getMovieType,
   getMovieVods,
-  getMovieYear
+  getMovieYear,
+  getParent,
+  getSeasonsOrEpisodes
 } from '../helpers/movie.helper';
 import { CSFDOptions } from '../types';
 import { movieUrl } from '../vars';
@@ -52,15 +54,16 @@ export class MovieScraper {
     pageClasses: string[],
     jsonLd: string,
     options: CSFDOptions
-  ): CSFDMovie {
+  ) {
+    const type = getMovieType(el) as CSFDFilmTypes;
     return {
       id: movieId,
       title: getMovieTitle(el),
       year: getMovieYear(jsonLd),
       duration: getMovieDuration(jsonLd, el),
       descriptions: getMovieDescriptions(el),
       genres: getMovieGenres(el),
-      type: getMovieType(el) as CSFDFilmTypes,
+      type,
       url: movieUrl(movieId, { language: options?.language }),
       origins: getMovieOrigins(el),
       colorRating: getMovieColorRating(pageClasses),
@@ -86,7 +89,10 @@ export class MovieScraper {
       tags: getMovieTags(asideEl),
       premieres: getMoviePremieres(asideEl),
       related: getMovieBoxMovies(asideEl, 'Související'),
-      similar: getMovieBoxMovies(asideEl, 'Podobné')
+      similar: getMovieBoxMovies(asideEl, 'Podobné'),
+      seasons: type === 'seriál' ? getSeasonsOrEpisodes(el) : null,
+      episodes: type === 'série' ? getSeasonsOrEpisodes(el) : null,
+      parent: getParent(el)
     };
   }
 }