feat(movie): episodeCode + seasons

Author: bartholomej

src/dto/movie.ts

@@ -20,6 +20,8 @@ export interface CSFDMovie extends CSFDScreening {
   seasons: CSFDSeason[] | null;
   episodes: CSFDSeason[] | null;
   parent: CSFDParent | null;
+  episodeCode: string | null;
+  seasonName: string | null;
 }
 
 export interface CSFDParent {

src/helpers/movie.helper.ts

@@ -84,6 +84,24 @@ export const getMovieId = (el: HTMLElement): number => {
   return parseIdFromUrl(url);
 };
 
+export const getSerieasAndSeasonTitle = (el: HTMLElement): { seriesName: string; seasonName: string | null } => {
+  const titleElement = el.querySelector('h1');
+  if (!titleElement) {
+    return { seriesName: null, seasonName: null };
+  }
+
+  const fullText = titleElement.innerText.trim();
+
+  // Check if there's a series part indicated by ' - '
+  if (fullText.includes(' - ')) {
+    const [seriesName, seasonName] = fullText.split(' - ').map(part => part.trim());
+    return { seriesName, seasonName };
+  }
+
+  // If no series part found, return just the name
+  return { seriesName: fullText, seasonName: null };
+};
+
 export const getMovieTitle = (el: HTMLElement): string => {
   return el.querySelector('h1').innerText.split(`(`)[0].trim();
 };
@@ -243,7 +261,7 @@ const parseMoviePeople = (el: HTMLElement): CSFDMovieCreator[] => {
   );
 };
 
-export const getSeasonsOrEpisodes = (el: HTMLElement): CSFDSeason[] | null => {
+export const getSeasonsOrEpisodes = (el: HTMLElement, serie?: { id: number; title: string; }): CSFDSeason[] | null => {
   const childrenList = el.querySelector('.film-episodes-list');
   if (!childrenList) return null;
 
@@ -263,20 +281,45 @@ export const getSeasonsOrEpisodes = (el: HTMLElement): CSFDSeason[] | null => {
   });
 }
 
-export const getParent = (el: HTMLElement): CSFDParent | null => {
+export const getEpisodeCode = (el: HTMLElement): string | null => {
+  const filmHeaderName = el.querySelector('.film-header-name h1');
+  if (!filmHeaderName) return null;
+
+  const text = filmHeaderName.textContent?.trim() || '';
+  const match = text.match(/\(([^)]+)\)/);
+  const code = match ? match[1] : null;
+
+  return code;
+}
+
+export const detectSeasonOrEpisodeListType = (el: HTMLElement) => {
+  const headerText = el.querySelector('.box-header h3')?.innerText.trim() ?? '';
+
+  if (headerText.includes('Série')) return 'seasons';
+  if (headerText.startsWith('Epizody')) return 'episodes';
+  return null;
+}
+
+export const getSeasonorEpisodeParent = (el: HTMLElement, serie?: { id: number; name: string; }): CSFDParent | null => {
   const parents = el.querySelectorAll('.film-header h2 a');
-  if (parents.length === 0) return null;
+  if (parents.length === 0) {
+    if (!serie) return null;
+    return { series: serie, season: 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
-    }
-  };
+  const seriesId = parseIdFromUrl(parentSeries?.getAttribute('href'));
+  const seasonId = parseIdFromUrl(parentSeason?.getAttribute('href'));
+  const seriesName = parentSeries?.textContent?.trim() || null;
+  const seasonName = parentSeason?.textContent?.trim() || null;
+
+  const series = seriesId && seriesName ? { id: seriesId, name: seriesName } : null;
+  const season = seasonId && seasonName ? { id: seasonId, name: seasonName } : null;
+
+  if (!series && !season) return null;
+
+  return { series, season };
 }
 
 export const getMovieGroup = (el: HTMLElement, group: CSFDCreatorGroups | CSFDCreatorGroupsEnglish | CSFDCreatorGroupsSlovak): CSFDMovieCreator[] => {

src/services/movie.service.ts

@@ -3,6 +3,8 @@ import { CSFDFilmTypes } from '../dto/global';
 import { CSFDMovie } from '../dto/movie';
 import { fetchPage } from '../fetchers';
 import {
+  detectSeasonOrEpisodeListType,
+  getEpisodeCode,
   getLocalizedCreatorLabel,
   getMovieBoxMovies,
   getMovieColorRating,
@@ -23,8 +25,9 @@ import {
   getMovieType,
   getMovieVods,
   getMovieYear,
-  getParent,
-  getSeasonsOrEpisodes
+  getSeasonorEpisodeParent,
+  getSeasonsOrEpisodes,
+  getSerieasAndSeasonTitle
 } from '../helpers/movie.helper';
 import { CSFDOptions } from '../types';
 import { movieUrl } from '../vars';
@@ -56,9 +59,13 @@ export class MovieScraper {
     options: CSFDOptions
   ) {
     const type = getMovieType(el) as CSFDFilmTypes;
+    const { seriesName = null, seasonName = null } = type === 'série' ? getSerieasAndSeasonTitle(el) : {};
+    const seasonOrEpisodeListType = detectSeasonOrEpisodeListType(el);
+
+    const title = type === 'série' && seriesName ? seriesName : getMovieTitle(el);
     return {
       id: movieId,
-      title: getMovieTitle(el),
+      title,
       year: getMovieYear(jsonLd),
       duration: getMovieDuration(jsonLd, el),
       descriptions: getMovieDescriptions(el),
@@ -90,9 +97,11 @@ export class MovieScraper {
       premieres: getMoviePremieres(asideEl),
       related: getMovieBoxMovies(asideEl, 'Související'),
       similar: getMovieBoxMovies(asideEl, 'Podobné'),
-      seasons: type === 'seriál' ? getSeasonsOrEpisodes(el) : null,
-      episodes: type === 'série' ? getSeasonsOrEpisodes(el) : null,
-      parent: getParent(el)
+      seasons: seasonOrEpisodeListType === 'seasons' ? getSeasonsOrEpisodes(el) : null,
+      episodes: seasonOrEpisodeListType === 'episodes' ? getSeasonsOrEpisodes(el) : null,
+      parent: (type === 'seriál') ? null : getSeasonorEpisodeParent(el, { id: movieId, name: title }),
+      episodeCode: type === 'epizoda' ? getEpisodeCode(el) : null,
+      seasonName,
     };
   }
 }