lnreader-plugins/plugins/chinese/69shu.ts

import { load as parseHTML } from 'cheerio';
import { fetchText } from '@/lib/fetch';
import { FilterTypes, Filters } from '@libs/filterInputs';
import { Plugin } from '@/types/plugin';
import { NovelStatus } from '@libs/novelStatus';

class Shu69 implements Plugin.PluginBase {
  private fetchOptions = {
    headers: {
      'User-Agent':
        'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:137.0) Gecko/20100101 Firefox/137.0',
      'Accept':
        'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8',
      'Accept-Language': 'en-US,us;q=0.5',
      'Referer': 'https://www.69shu.xyz/', // Referer
      'DNT': '1', // Do Not Track
      'Upgrade-Insecure-Requests': '1', // Upgrade-Insecure-Requests
    },
  };

  id = '69shu';
  name = '69书吧';
  icon = 'src/cn/69shu/icon.png';
  site = 'https://www.69shu.xyz';
  version = '0.2.2';

  async popularNovels(
    pageNo: number,
    {
      showLatestNovels,
      filters,
    }: Plugin.PopularNovelsOptions<typeof this.filters>,
  ): Promise<Plugin.NovelItem[]> {
    let url: string;
    if (showLatestNovels) {
      url = `${this.site}/rank/lastupdate/${pageNo}.html`;
    } else if (filters.sort.value === 'none') {
      url = `${this.site}/rank/${filters.rank.value}/${pageNo}.html`;
    } else {
      url = `${this.site}/sort/${filters.sort.value}/${pageNo}.html`;
    }

    const body = await fetchText(url, this.fetchOptions);
    if (body === '') throw Error('无法获取小说列表，请检查网络');

    const loadedCheerio = parseHTML(body);

    const novels: Plugin.NovelItem[] = [];

    loadedCheerio('div.book-coverlist').each((i, el) => {
      const url = loadedCheerio(el).find('a.cover').attr('href');

      const novelName = loadedCheerio(el).find('h4.name').text().trim();
      const novelCover = loadedCheerio(el).find('a.cover > img').attr('src');
      if (!url) return;

      const novel = {
        name: novelName,
        cover: novelCover,
        path: url.replace(this.site, ''),
      };

      novels.push(novel);
    });

    return novels;
  }

  async parseNovel(novelPath: string): Promise<Plugin.SourceNovel> {
    const url = this.site + novelPath;

    const body = await fetchText(url, this.fetchOptions);
    if (body === '') throw Error('无法获取小说内容，请检查网络');

    const loadedCheerio = parseHTML(body);

    const novel: Plugin.SourceNovel = {
      path: novelPath,
      chapters: [],
      name: loadedCheerio('h1').text().trim(),
    };

    novel.cover = loadedCheerio('div.cover > img').attr('src');

    novel.summary = loadedCheerio('#bookIntro').text().trim();

    const bookInfo = loadedCheerio('div.caption-bookinfo > p');

    novel.author = bookInfo.find('a').attr('title');

    novel.artist = undefined;

    novel.status = bookInfo.text().includes('连载')
      ? NovelStatus.Ongoing
      : NovelStatus.Completed;

    novel.genres = '';

    // Table of Content is on a different page than the summary page
    const chapters: Plugin.ChapterItem[] = [];

    const allUrl = loadedCheerio('dd.all > a').attr('href');
    if (allUrl) {
      // --- Start: Fetch chapters with pagination (Sequential) ---
      let currentChaptersUrl = new URL(allUrl, this.site).toString();
      let hasMorePages = true;

      while (hasMorePages) {
        const chaptersBody = await fetchText(
          currentChaptersUrl,
          this.fetchOptions,
        );
        const chaptersLoadedCheerio = parseHTML(chaptersBody);

        // Extract chapters from the current page
        chaptersLoadedCheerio('dl.panel-chapterlist dd').each((i, el) => {
          const chapterUrl = chaptersLoadedCheerio(el).find('a').attr('href');
          const chapterName = chaptersLoadedCheerio(el).find('a').text().trim();
          if (chapterUrl) {
            // Ensure relative path, handle both absolute/relative cases
            const relativeChapterUrl = chapterUrl.startsWith('http')
              ? chapterUrl.replace(this.site, '')
              : chapterUrl;
            // Avoid duplicates if the same chapter appears on multiple pages (unlikely but safe)
            if (!chapters.some(chap => chap.path === relativeChapterUrl)) {
              chapters.push({
                name: chapterName,
                path: relativeChapterUrl,
              });
            }
          }
        });

        // Find the link to the next page using the text "下一页"
        const nextPageLinkElement = chaptersLoadedCheerio(
          'div.listpage a.onclick',
        ).filter((i, el) =>
          chaptersLoadedCheerio(el).text().includes('下一页'),
        );
        const nextPageLink = nextPageLinkElement.attr('href');

        if (nextPageLink && nextPageLink !== 'javascript:void(0);') {
          // Check if it's a valid relative or absolute URL before creating the URL object
          try {
            const absoluteNextPageUrl = new URL(
              nextPageLink,
              this.site,
            ).toString();
            if (absoluteNextPageUrl === currentChaptersUrl) {
              // Break if the next page URL is the same as the current one (prevents infinite loops)
              hasMorePages = false;
            } else {
              currentChaptersUrl = absoluteNextPageUrl;
            }
          } catch (e) {
            // Handle cases where the link might be invalid or unexpected
            console.warn(`Invalid next page link found: ${nextPageLink}`);
            hasMorePages = false;
          }
        } else {
          hasMorePages = false;
        }
      }
      // --- End: Fetch chapters with pagination (Sequential) ---
    } else {
      // Fallback if no "all chapters" link is found
      loadedCheerio(
        'div.panel.hidden-xs > dl.panel-chapterlist:nth-child(2) > dd',
      ).each((i, el) => {
        const chapterUrl = loadedCheerio(el).find('a').attr('href');
        const chapterName = loadedCheerio(el).find('a').text().trim();
        if (chapterUrl) {
          const relativeChapterUrl = chapterUrl.startsWith('http')
            ? chapterUrl.replace(this.site, '')
            : chapterUrl;
          chapters.push({
            name: chapterName,
            path: relativeChapterUrl,
          });
        }
      });
    }

    // Remove duplicates just in case (though less likely with sequential fetching)
    const uniqueChapters = chapters.filter(
      (chapter, index, self) =>
        index === self.findIndex(c => c.path === chapter.path),
    );

    novel.chapters = uniqueChapters;

    return novel;
  }

  async parseChapter(chapterPath: string): Promise<string> {
    const chapterUrl = new URL(chapterPath, this.site).toString();
    const body = await fetchText(chapterUrl, this.fetchOptions); // Header hinzugefügt

    const loadedCheerio = parseHTML(body);

    const chapterText = loadedCheerio('#chaptercontent p')
      .map((i, el) => loadedCheerio(el).text())
      .get()
      // remove empty lines and 69shu ads
      .map((line: string) => line.trim())
      .filter((line: string) => line !== '' && !line.includes('69书吧'))
      .map((line: string) => `<p>${line}</p>`)
      .join('\n');

    return chapterText;
  }

  async searchNovels(
    searchTerm: string,
    pageNo: number,
  ): Promise<Plugin.NovelItem[]> {
    if (pageNo > 1) return [];

    const searchUrl = `${this.site}/search`;
    const formData = new FormData();
    formData.append('searchkey', searchTerm);

    const searchOptions = {
      ...this.fetchOptions,
      method: 'post',
      body: formData,
      headers: {
        ...this.fetchOptions.headers,
      },
    };

    const body = await fetchText(searchUrl, searchOptions);
    if (body === '') throw Error('无法获取搜索结果，请检查网络');

    const loadedCheerio = parseHTML(body);

    const novels: Plugin.NovelItem[] = [];

    loadedCheerio('div.book-coverlist').each((i, el) => {
      const url = loadedCheerio(el).find('a.cover').attr('href');

      const novelName = loadedCheerio(el).find('h4.name').text().trim();
      const novelCover = loadedCheerio(el).find('a.cover > img').attr('src');

      if (!url) return;

      const novel = {
        name: novelName,
        cover: novelCover,
        path: url.replace(this.site, ''),
      };

      novels.push(novel);
    });

    return novels;
  }

  filters = {
    rank: {
      label: '排行榜',
      value: 'allvisit',
      options: [
        { label: '总排行榜', value: 'allvisit' },
        { label: '月排行榜', value: 'monthvisit' },
        { label: '周排行榜', value: 'weekvisit' },
        { label: '日排行榜', value: 'dayvisit' },
        { label: '收藏榜', value: 'goodnum' },
        { label: '字数榜', value: 'words' },
        { label: '推荐榜', value: 'allvote' },
        { label: '新书榜', value: 'postdate' },
        { label: '更新榜', value: 'lastupdate' },
      ],
      type: FilterTypes.Picker,
    },
    sort: {
      label: '分类',
      value: 'none',
      options: [
        { label: '无', value: 'none' },
        { label: '全部', value: 'all' },
        { label: '玄幻', value: 'xuanhuan' },
        { label: '仙侠', value: 'xianxia' },
        { label: '都市', value: 'dushi' },
        { label: '历史', value: 'lishi' },
        { label: '游戏', value: 'youxi' },
        { label: '科幻', value: 'kehuan' },
        { label: '灵异', value: 'kongbu' },
        { label: '言情', value: 'nvsheng' },
        { label: '其它', value: 'qita' },
      ],
      type: FilterTypes.Picker,
    },
  } satisfies Filters;
}

export default new Shu69();