movie.service.ts

import { HTMLElement, parse } from 'node-html-parser';
import { fetchMovie } from '../fetchers';
import {
  getColorRating,
  getDescriptions,
  getDirectors,
  getGenres,
  getGroup,
  getId,
  getOrigins,
  getOtherTitles,
  getPoster,
  getRating,
  getTags,
  getTitle,
  getType,
  getYear
} from '../helpers/movie.helper';
import { CSFDFilmTypes } from '../interfaces/global';
import { CSFDMovie } from '../interfaces/movie.interface';
import { movieUrl } from '../vars';

export class MovieScraper {
  private film: CSFDMovie;

  public async movie(movie: string | number): Promise<CSFDMovie> {
    const response = await fetchMovie(movie);

    const movieHtml = parse(response);

    const pageClasses = movieHtml.querySelector('.page-content').classNames;
    const asideNode = movieHtml.querySelector('.box-rating-container');
    const movieNode = movieHtml.querySelector('.main-movie-profile');
    this.buildMovie(movie, movieNode, asideNode, pageClasses);
    return this.film;
  }

  private buildMovie(
    movie: string | number,
    el: HTMLElement,
    elAside: HTMLElement,
    pageClasses: string[]
  ) {
    this.film = {
      id: getId(el),
      title: getTitle(el),
      year: getYear(el),
      descriptions: getDescriptions(el),
      genres: getGenres(el),
      type: getType(el) as CSFDFilmTypes,
      url: movieUrl(movie),
      origins: getOrigins(el),
      colorRating: getColorRating(pageClasses),
      rating: getRating(el),
      otherTitles: getOtherTitles(el),
      poster: getPoster(el),
      creators: {
        directors: getDirectors(el),
        writers: getGroup(el, 'Scénář'),
        cinematography: getGroup(el, 'Kamera'),
        music: getGroup(el, 'Hudba'),
        actors: getGroup(el, 'Hrají'),
        basedOn: getGroup(el, 'Předloha'),
        producers: getGroup(el, 'Produkce'),
        filmEditing: getGroup(el, 'Střih'),
        costumeDesign: getGroup(el, 'Kostýmy'),
        productionDesign: getGroup(el, 'Scénografie')
      },
      tags: getTags(elAside)
    };
  }
}