mangayomi-mirror/lib/services/get_chapter_pages.dart

import 'dart:convert';
import 'dart:io';
import 'dart:typed_data';
import 'package:path/path.dart' as p;
import 'package:mangayomi/eval/lib.dart';
import 'package:mangayomi/eval/javascript/http.dart';
import 'package:mangayomi/main.dart';
import 'package:mangayomi/models/chapter.dart';
import 'package:mangayomi/models/page.dart';
import 'package:mangayomi/models/settings.dart';
import 'package:mangayomi/modules/manga/archive_reader/providers/archive_reader_providers.dart';
import 'package:mangayomi/modules/manga/reader/reader_view.dart';
import 'package:mangayomi/providers/storage_provider.dart';
import 'package:mangayomi/utils/utils.dart';
import 'package:mangayomi/utils/reg_exp_matcher.dart';
import 'package:mangayomi/modules/more/providers/incognito_mode_state_provider.dart';
import 'package:riverpod_annotation/riverpod_annotation.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';
part 'get_chapter_pages.g.dart';

class GetChapterPagesModel {
  Directory? path;
  List<PageUrl> pageUrls = [];
  List<bool> isLocaleList = [];
  List<Uint8List?> archiveImages = [];
  List<UChapDataPreload> uChapDataPreload;
  GetChapterPagesModel({
    required this.path,
    required this.pageUrls,
    required this.isLocaleList,
    required this.archiveImages,
    required this.uChapDataPreload,
  });
}

@riverpod
Future<GetChapterPagesModel> getChapterPages(
  Ref ref, {
  required Chapter chapter,
}) async {
  List<UChapDataPreload> uChapDataPreloadp = [];
  Directory? path;
  List<PageUrl> pageUrls = [];
  List<bool> isLocaleList = [];
  final settings = isar.settings.getSync(227);
  List<ChapterPageurls>? chapterPageUrlsList =
      settings!.chapterPageUrlsList ?? [];
  final isarPageUrls = chapterPageUrlsList
      .where((element) => element.chapterId == chapter.id)
      .firstOrNull;
  final incognitoMode = ref.read(incognitoModeStateProvider);
  final storageProvider = StorageProvider();
  final mangaDirectory = await storageProvider.getMangaMainDirectory(chapter);
  path = await storageProvider.getMangaChapterDirectory(
    chapter,
    mangaMainDirectory: mangaDirectory,
  );

  List<Uint8List?> archiveImages = [];
  final isLocalArchive = (chapter.archivePath ?? '').isNotEmpty;
  if (!chapter.manga.value!.isLocalArchive!) {
    final source = getSource(
      chapter.manga.value!.lang!,
      chapter.manga.value!.source!,
    )!;
    if ((isarPageUrls?.urls?.isNotEmpty ?? false) &&
        (isarPageUrls?.chapterUrl ?? chapter.url) == chapter.url) {
      for (var i = 0; i < isarPageUrls!.urls!.length; i++) {
        Map<String, String>? headers;
        if (isarPageUrls.headers?.isNotEmpty ?? false) {
          headers =
              (jsonDecode(isarPageUrls.headers![i]) as Map?)?.toMapStringString;
        }
        pageUrls.add(PageUrl(isarPageUrls.urls![i], headers: headers));
      }
    } else {
      pageUrls = await getExtensionService(source).getPageList(chapter.url!);
    }
  }

  if (pageUrls.isNotEmpty || isLocalArchive) {
    if (await File(
          p.join(mangaDirectory!.path, "${chapter.name}.cbz"),
        ).exists() ||
        isLocalArchive) {
      final path = isLocalArchive
          ? chapter.archivePath
          : p.join(mangaDirectory.path, "${chapter.name}.cbz");
      final local = await ref.read(
        getArchiveDataFromFileProvider(path!).future,
      );
      for (var image in local.images!) {
        archiveImages.add(image.image!);
        isLocaleList.add(true);
      }
    } else {
      for (var i = 0; i < pageUrls.length; i++) {
        archiveImages.add(null);
        if (await File(p.join(path!.path, '${padIndex(i)}.jpg')).exists()) {
          isLocaleList.add(true);
        } else {
          isLocaleList.add(false);
        }
      }
    }
    if (isLocalArchive) {
      for (var i = 0; i < archiveImages.length; i++) {
        pageUrls.add(PageUrl(""));
      }
    }
    if (!incognitoMode) {
      List<ChapterPageurls>? chapterPageUrls = [];
      for (var chapterPageUrl in settings.chapterPageUrlsList ?? []) {
        if (chapterPageUrl.chapterId != chapter.id) {
          chapterPageUrls.add(chapterPageUrl);
        }
      }
      final chapterPageHeaders = pageUrls
          .map((e) => e.headers == null ? null : jsonEncode(e.headers))
          .toList();
      chapterPageUrls.add(
        ChapterPageurls()
          ..chapterId = chapter.id
          ..urls = pageUrls.map((e) => e.url).toList()
          ..chapterUrl = chapter.url
          ..headers = chapterPageHeaders.first != null
              ? chapterPageHeaders.map((e) => e.toString()).toList()
              : null,
      );
      isar.writeTxnSync(() {
        isar.settings.putSync(
          settings
            ..chapterPageUrlsList = chapterPageUrls
            ..updatedAt = DateTime.now().millisecondsSinceEpoch,
        );
      });
    }
    for (var i = 0; i < pageUrls.length; i++) {
      uChapDataPreloadp.add(
        UChapDataPreload(
          chapter,
          path,
          pageUrls[i],
          isLocaleList[i],
          archiveImages[i],
          i,
          GetChapterPagesModel(
            path: path,
            pageUrls: pageUrls,
            isLocaleList: isLocaleList,
            archiveImages: archiveImages,
            uChapDataPreload: uChapDataPreloadp,
          ),
          i,
        ),
      );
    }
  }

  return GetChapterPagesModel(
    path: path,
    pageUrls: pageUrls,
    isLocaleList: isLocaleList,
    archiveImages: archiveImages,
    uChapDataPreload: uChapDataPreloadp,
  );
}