diff --git a/Sora/Views/MediaInfoView/MediaInfoView.swift b/Sora/Views/MediaInfoView/MediaInfoView.swift index b32d41f..58925c1 100644 --- a/Sora/Views/MediaInfoView/MediaInfoView.swift +++ b/Sora/Views/MediaInfoView/MediaInfoView.swift @@ -20,6 +20,7 @@ struct MediaItem: Identifiable { struct MediaInfoView: View { let title: String + let originalImageUrl: String @State var imageUrl: String let href: String let module: ScrapingModule @@ -532,10 +533,19 @@ struct MediaInfoView: View { } } - Button(action: { - fetchTMDBPosterImageAndSet() - }) { - Label("Use TMDB Poster Image", systemImage: "photo") + if let tmdbPoster = UserDefaults.standard.string(forKey: "tmdbPosterURL_\(href)"), imageUrl == tmdbPoster { + Button(action: { + UserDefaults.standard.removeObject(forKey: "tmdbPosterURL_\(href)") + self.imageUrl = originalImageUrl + }) { + Label("Use Module Image", systemImage: "arrow.uturn.backward") + } + } else { + Button(action: { + fetchTMDBPosterImageAndSet() + }) { + Label("Use TMDB Poster Image", systemImage: "photo") + } } Divider() @@ -859,9 +869,7 @@ struct MediaInfoView: View { let apiType = tmdbType.rawValue let urlString = "https://api.themoviedb.org/3/\(apiType)/\(tmdbID)?api_key=738b4edd0a156cc126dc4a4b8aea4aca" guard let url = URL(string: urlString) else { return } - let tmdbImageWidth = UserDefaults.standard.string(forKey: "tmdbImageWidth") ?? "original" - URLSession.custom.dataTask(with: url) { data, _, error in guard let data = data, error == nil else { return } do {