diff --git a/app/src/main/java/eu/kanade/tachiyomi/data/database/DatabaseHelper.java b/app/src/main/java/eu/kanade/tachiyomi/data/database/DatabaseHelper.java index 5c2c0c1ed7..36b41a7544 100644 --- a/app/src/main/java/eu/kanade/tachiyomi/data/database/DatabaseHelper.java +++ b/app/src/main/java/eu/kanade/tachiyomi/data/database/DatabaseHelper.java @@ -40,6 +40,7 @@ import eu.kanade.tachiyomi.data.database.tables.MangaCategoryTable; import eu.kanade.tachiyomi.data.database.tables.MangaSyncTable; import eu.kanade.tachiyomi.data.database.tables.MangaTable; import eu.kanade.tachiyomi.data.mangasync.base.MangaSyncService; +import eu.kanade.tachiyomi.data.source.base.Source; import eu.kanade.tachiyomi.util.ChapterRecognition; import rx.Observable; @@ -240,13 +241,14 @@ public class DatabaseHelper { } // Add new chapters or delete if the source deletes them - public Observable> insertOrRemoveChapters(Manga manga, List sourceChapters) { + public Observable> insertOrRemoveChapters(Manga manga, List sourceChapters, Source source) { List dbChapters = getChapters(manga).executeAsBlocking(); Observable> newChapters = Observable.from(sourceChapters) .filter(c -> !dbChapters.contains(c)) .doOnNext(c -> { c.manga_id = manga.id; + source.parseChapterNumber(c); ChapterRecognition.parseChapterNumber(c, manga); }) .toList(); diff --git a/app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt b/app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt index c5a4a22d0e..19d683d40e 100644 --- a/app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt +++ b/app/src/main/java/eu/kanade/tachiyomi/data/library/LibraryUpdateService.kt @@ -201,9 +201,10 @@ class LibraryUpdateService : Service() { * @return a pair of the inserted and removed chapters. */ fun updateManga(manga: Manga): Observable> { - return sourceManager.get(manga.source)!! + val source = sourceManager.get(manga.source) + return source!! .pullChaptersFromNetwork(manga.url) - .flatMap { db.insertOrRemoveChapters(manga, it) } + .flatMap { db.insertOrRemoveChapters(manga, it, source) } } /** diff --git a/app/src/main/java/eu/kanade/tachiyomi/data/source/base/Source.java b/app/src/main/java/eu/kanade/tachiyomi/data/source/base/Source.java index 89e2738aa6..0b4680a082 100644 --- a/app/src/main/java/eu/kanade/tachiyomi/data/source/base/Source.java +++ b/app/src/main/java/eu/kanade/tachiyomi/data/source/base/Source.java @@ -235,6 +235,11 @@ public abstract class Source extends BaseSource { return getId() + chapterUrl; } + // Overridable method to allow custom parsing. + public void parseChapterNumber(Chapter chapter) { + + } + protected LazyHeaders.Builder glideHeadersBuilder() { LazyHeaders.Builder builder = new LazyHeaders.Builder(); for (Map.Entry> entry : requestHeaders.toMultimap().entrySet()) { diff --git a/app/src/main/java/eu/kanade/tachiyomi/ui/manga/chapter/ChaptersPresenter.java b/app/src/main/java/eu/kanade/tachiyomi/ui/manga/chapter/ChaptersPresenter.java index b2e9404488..8a86e277b2 100644 --- a/app/src/main/java/eu/kanade/tachiyomi/ui/manga/chapter/ChaptersPresenter.java +++ b/app/src/main/java/eu/kanade/tachiyomi/ui/manga/chapter/ChaptersPresenter.java @@ -119,7 +119,7 @@ public class ChaptersPresenter extends BasePresenter { private Observable> getOnlineChaptersObs() { return source.pullChaptersFromNetwork(manga.url) .subscribeOn(Schedulers.io()) - .flatMap(chapters -> db.insertOrRemoveChapters(manga, chapters)) + .flatMap(chapters -> db.insertOrRemoveChapters(manga, chapters, source)) .observeOn(AndroidSchedulers.mainThread()); }