lnreader-plugins/scripts/multisrc/rulate/template.ts

import { fetchFile, fetchApi } from '@libs/fetch';
import { Filters, FilterTypes } from '@libs/filterInputs';
import { Plugin } from '@typings/plugin';
import { NovelStatus } from '@libs/novelStatus';
import { load as parseHTML } from 'cheerio';
import dayjs from 'dayjs';

export interface RulateMetadata {
  id: string;
  sourceSite: string;
  sourceName: string;
  filters?: Filters;
}

class RulatePlugin implements Plugin.PluginBase {
  id: string;
  name: string;
  icon: string;
  site: string;
  version: string;
  filters?: Filters | undefined;

  constructor(metadata: RulateMetadata) {
    this.id = metadata.id;
    this.name = metadata.sourceName;
    this.icon = `multisrc/rulate/${metadata.id.toLowerCase()}/icon.png`;
    this.site = metadata.sourceSite;
    this.version = '1.0.1';
    this.filters = metadata.filters;
  }

  async popularNovels(
    pageNo: number,
    { filters, showLatestNovels }: Plugin.PopularNovelsOptions,
  ): Promise<Plugin.NovelItem[]> {
    const novels: Plugin.NovelItem[] = [];
    let url = this.site + '/search?t=';
    url += '&cat=' + (filters?.cat?.value || '0');
    url += '&s_lang=' + (filters?.s_lang?.value || '0');
    url += '&t_lang=' + (filters?.t_lang?.value || '0');
    url += '&type=' + (filters?.type?.value || '0');
    url += '&sort=' + (showLatestNovels ? '4' : filters?.sort?.value || '6');
    url += '&atmosphere=' + (filters?.atmosphere?.value || '0');
    url += '&adult=' + (filters?.adult?.value || '0');

    Object.entries(filters || {}).forEach(([type, { value }]) => {
      if (value instanceof Array && value.length) {
        url +=
          '&' +
          value
            .map(val => (type == 'extra' ? val + '=1' : type + '[]=' + val))
            .join('&');
      }
    });

    url += '&Book_page=' + pageNo;

    const body = await fetchApi(url).then(res => res.text());
    const loadedCheerio = parseHTML(body);

    loadedCheerio(
      'ul[class="search-results"] > li:not([class="ad_type_catalog"])',
    ).each((index, element) => {
      loadedCheerio(element).find('p > a').text();
      const name = loadedCheerio(element).find('p > a').text();
      const cover = loadedCheerio(element).find('img').attr('src');
      const path = loadedCheerio(element).find('p > a').attr('href');
      if (!name || !path) return;

      novels.push({ name, cover: this.site + cover, path });
    });

    return novels;
  }

  async parseNovel(novelPath: string): Promise<Plugin.SourceNovel> {
    let result = await fetchApi(this.site + novelPath);
    if (result.url.includes('mature?path=')) {
      const formData = new FormData();
      formData.append('path', novelPath);
      formData.append('ok', 'Да');

      result = await fetchApi(result.url, {
        method: 'POST',
        body: formData,
      });
    }
    const body = await result.text();
    const loadedCheerio = parseHTML(body);

    const novel: Plugin.SourceNovel = {
      path: novelPath,
      name: loadedCheerio(
        'div[class="container"] > div > div > h1, div.span8:nth-child(1) > h1:nth-child(2)',
      )
        .text()
        .trim(),
    };
    if (novel.name?.includes?.('[')) {
      novel.name = novel.name.split('[')[0].trim();
    }
    novel.cover =
      this.site + loadedCheerio('div[class="images"] > div img').attr('src');
    novel.summary = loadedCheerio('#Info > div:nth-child(3), .book-description')
      .text()
      .trim();
    novel.author = loadedCheerio(
      '.book-stats-icons_author > span:nth-child(2) > a:nth-child(1)',
    ).text();
    const genres: string[] = [];

    loadedCheerio('div.span5 > p, .span5 > div:nth-child(2) > p').each(
      function () {
        switch (loadedCheerio(this).find('strong').text()) {
          case 'Автор:':
            novel.author = loadedCheerio(this).find('em > a').text().trim();
            break;
          case 'Выпуск:':
            novel.status =
              loadedCheerio(this).find('em').text().trim() === 'продолжается'
                ? NovelStatus.Ongoing
                : NovelStatus.Completed;
            break;
          case 'Тэги:':
            loadedCheerio(this)
              .find('em > a')
              .each(function () {
                genres.push(loadedCheerio(this).text());
              });
            break;
          case 'Жанры:':
            loadedCheerio(this)
              .find('em > a')
              .each(function () {
                genres.push(loadedCheerio(this).text());
              });
            break;
        }
      },
    );

    if (genres.length) {
      novel.genres = genres.reverse().join(',');
    }

    const chapters: Plugin.ChapterItem[] = [];
    loadedCheerio('table > tbody > tr.chapter_row').each(
      (chapterIndex, element) => {
        const chapterName = loadedCheerio(element)
          .find('td[class="t"] > a')
          .text()
          .trim();
        const releaseDate = loadedCheerio(element)
          .find('td > span')
          .attr('title')
          ?.trim();
        const chapterUrl = loadedCheerio(element)
          .find('td[class="t"] > a')
          .attr('href');

        if (
          !loadedCheerio(element).find('td > span[class="disabled"]').length &&
          releaseDate &&
          chapterUrl
        ) {
          chapters.push({
            name: chapterName,
            path: chapterUrl,
            releaseTime: this.parseDate(releaseDate),
            chapterNumber: chapterIndex + 1,
          });
        }
      },
    );

    novel.chapters = chapters;
    return novel;
  }

  async parseChapter(chapterPath: string): Promise<string> {
    let result = await fetchApi(this.site + chapterPath);
    if (result.url.includes('mature?path=')) {
      const formData = new FormData();
      formData.append('path', chapterPath.split('/').slice(0, 3).join('/'));
      formData.append('ok', 'Да');

      result = await fetchApi(result.url, {
        method: 'POST',
        body: formData,
      });
    }
    const body = await result.text();
    const loadedCheerio = parseHTML(body);

    loadedCheerio('.content-text img').each((index, element) => {
      if (!loadedCheerio(element).attr('src')?.startsWith('http')) {
        const src = loadedCheerio(element).attr('src');
        loadedCheerio(element).attr('src', this.site + src);
      }
    });

    const chapterText = loadedCheerio('.content-text').html();
    return chapterText || '';
  }

  async searchNovels(searchTerm: string): Promise<Plugin.NovelItem[]> {
    const novels: Plugin.NovelItem[] = [];
    const result: response[] = await fetchApi(
      this.site + '/search/autocomplete?query=' + searchTerm,
    ).then(res => res.json());

    result.forEach(novel => {
      const name = novel.title_one + ' / ' + novel.title_two;
      if (!novel.url) return;

      novels.push({
        name,
        cover: this.site + novel.img,
        path: novel.url,
      });
    });

    return novels;
  }

  parseDate = (dateString: string | undefined = '') => {
    const months: { [key: string]: number } = {
      'янв.': 1,
      'февр.': 2,
      'мар.': 3,
      'апр.': 4,
      мая: 5,
      'июн.': 6,
      'июл.': 7,
      'авг.': 8,
      'сент.': 9,
      'окт.': 10,
      'нояб.': 11,
      'дек.': 12,
    };
    const [day, month, year, , time] = dateString.split(' ');
    if (day && months[month] && year && time) {
      return dayjs(year + '-' + months[month] + '-' + day + ' ' + time).format(
        'LLL',
      );
    }
    return dateString || null;
  };

  fetchImage = fetchFile;
}

interface response {
  id: number;
  title_one: string;
  title_two: string;
  url: string;
  img: string;
}