Update 4anime server recommendations

SuperMarcus · Dec 11, 2019 · e7d47f6 · e7d47f6
1 parent e376758
commit e7d47f6
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/NineAnimator/Models/Anime Source/4anime.to/FourAnime.swift b/NineAnimator/Models/Anime Source/4anime.to/FourAnime.swift
@@ -58,6 +58,15 @@ class NASourceFourAnime: BaseSource, Source, PromiseSource {
             ? DummyParser.registeredInstance : nil
     }
 
+    override func recommendServer(for anime: Anime) -> Anime.ServerIdentifier? {
+        return recommendServers(for: anime, ofPurpose: .playback).first
+    }
+
+    override func recommendServers(for anime: Anime, ofPurpose purpose: VideoProviderParserParsingPurpose) -> [Anime.ServerIdentifier] {
+        return anime.servers.keys.contains(NASourceFourAnime.FourAnimeStream)
+            ? [NASourceFourAnime.FourAnimeStream] : []
+    }
+
     func link(from url: URL) -> NineAnimatorPromise<AnyLink> {
         return .fail()
     }