Improve URL handling

Shared/OpenURLHandler.swift

@@ -1,8 +1,15 @@
+import CoreMedia
 import Foundation
+import Siesta
 
 struct OpenURLHandler {
+    static let yatteeProtocol = "yattee://"
+
     var accounts: AccountsModel
+    var navigation: NavigationModel
+    var recents: RecentsModel
     var player: PlayerModel
+    var search: SearchModel
 
     func handle(_ url: URL) {
         if accounts.current.isNil {
@@ -19,11 +26,69 @@ struct OpenURLHandler {
             }
         #endif
 
-        let parser = VideoURLParser(url: url)
+        let parser = URLParser(url: urlByRemovingYatteeProtocol(url))
 
-        guard let id = parser.id,
-              id != player.currentVideo?.id
-        else {
+        switch parser.destination {
+        case .video:
+            handleVideoUrlOpen(parser)
+        case .playlist:
+            handlePlaylistUrlOpen(parser)
+        case .channel:
+            handleChannelUrlOpen(parser)
+        case .search:
+            handleSearchUrlOpen(parser)
+        case .favorites:
+            hideViewsAboveBrowser()
+            navigation.tabSelection = .favorites
+            #if os(macOS)
+                focusMainWindow()
+            #endif
+        case .subscriptions:
+            guard accounts.app.supportsSubscriptions, accounts.signedIn else { return }
+            hideViewsAboveBrowser()
+            navigation.tabSelection = .subscriptions
+            #if os(macOS)
+                focusMainWindow()
+            #endif
+        case .popular:
+            guard accounts.app.supportsPopular else { return }
+            hideViewsAboveBrowser()
+            navigation.tabSelection = .popular
+            #if os(macOS)
+                focusMainWindow()
+            #endif
+        case .trending:
+            hideViewsAboveBrowser()
+            navigation.tabSelection = .trending
+            #if os(macOS)
+                focusMainWindow()
+            #endif
+        default:
+            navigation.presentAlert(title: "Error", message: "This URL could not be opened")
+        }
+    }
+
+    private func hideViewsAboveBrowser() {
+        player.hide()
+        navigation.presentingChannel = false
+        navigation.presentingPlaylist = false
+    }
+
+    private func urlByRemovingYatteeProtocol(_ url: URL) -> URL! {
+        var urlAbsoluteString = url.absoluteString
+
+        guard urlAbsoluteString.hasPrefix(Self.yatteeProtocol) else {
+            return url
+        }
+
+        urlAbsoluteString = String(urlAbsoluteString.dropFirst(Self.yatteeProtocol.count))
+
+        return URL(string: urlAbsoluteString)
+    }
+
+    private func handleVideoUrlOpen(_ parser: URLParser) {
+        guard let id = parser.videoID, id != player.currentVideo?.id else {
+            navigation.presentAlert(title: "Could not open video", message: "Could not extract video ID")
             return
         }
 
@@ -31,11 +96,145 @@ struct OpenURLHandler {
             Windows.main.open()
         #endif
 
-        accounts.api.video(id).load().onSuccess { response in
-            if let video: Video = response.typedContent() {
-                self.player.playNow(video, at: parser.time)
-                self.player.show()
+        accounts.api.video(id)
+            .load()
+            .onSuccess { response in
+                if let video: Video = response.typedContent() {
+                    let time = parser.time.isNil ? nil : CMTime.secondsInDefaultTimescale(TimeInterval(parser.time!))
+                    self.player.playNow(video, at: time)
+                    self.player.show()
+                } else {
+                    navigation.presentAlert(title: "Error", message: "This video could not be opened")
+                }
+            }
+            .onFailure { responseError in
+                navigation.presentAlert(title: "Could not open video", message: responseError.userMessage)
+            }
+    }
+
+    private func handlePlaylistUrlOpen(_ parser: URLParser) {
+        #if os(macOS)
+            if alertIfNoMainWindowOpen() { return }
+        #endif
+
+        guard let playlistID = parser.playlistID else {
+            navigation.presentAlert(title: "Could not open playlist", message: "Could not extract playlist ID")
+            return
+        }
+
+        accounts.api.channelPlaylist(playlistID)?
+            .load()
+            .onSuccess { response in
+                if var playlist: ChannelPlaylist = response.typedContent() {
+                    playlist.id = playlistID
+                    DispatchQueue.main.async {
+                        NavigationModel.openChannelPlaylist(
+                            playlist,
+                            player: player,
+                            recents: recents,
+                            navigation: navigation
+                        )
+                    }
+                } else {
+                    navigation.presentAlert(title: "Could not open playlist", message: "Playlist could not be found")
+                }
+            }
+            .onFailure { responseError in
+                navigation.presentAlert(title: "Could not open playlist", message: responseError.userMessage)
+            }
+    }
+
+    private func handleChannelUrlOpen(_ parser: URLParser) {
+        #if os(macOS)
+            if alertIfNoMainWindowOpen() { return }
+        #endif
+
+        guard let resource = resourceForChannelUrl(parser) else {
+            navigation.presentAlert(title: "Could not open channel", message: "Could not extract channel information")
+            return
+        }
+
+        resource
+            .load()
+            .onSuccess { response in
+                if let channel: Channel = response.typedContent() {
+                    DispatchQueue.main.async {
+                        NavigationModel.openChannel(
+                            channel,
+                            player: player,
+                            recents: recents,
+                            navigation: navigation
+                        )
+                    }
+                } else {
+                    navigation.presentAlert(title: "Could not open channel", message: "Channel could not be found")
+                }
+            }
+            .onFailure { responseError in
+                navigation.presentAlert(title: "Could not open channel", message: responseError.userMessage)
+            }
+    }
+
+    private func resourceForChannelUrl(_ parser: URLParser) -> Resource? {
+        if let id = parser.channelID {
+            return accounts.api.channel(id)
+        }
+
+        guard let name = parser.channelName else {
+            return nil
+        }
+
+        if accounts.app.supportsOpeningChannelsByName {
+            return accounts.api.channelByName(name)
+        }
+
+        if let instance = InstancesModel.all.first(where: { $0.app.supportsOpeningChannelsByName }) {
+            return instance.anonymous.channelByName(name)
+        }
+
+        return nil
+    }
+
+    private func handleSearchUrlOpen(_ parser: URLParser) {
+        #if os(macOS)
+            if alertIfNoMainWindowOpen() { return }
+        #endif
+
+        NavigationModel.openSearchQuery(
+            parser.searchQuery,
+            player: player,
+            recents: recents,
+            navigation: navigation,
+            search: search
+        )
+
+        #if os(macOS)
+            focusMainWindow()
+        #endif
+    }
+
+    #if os(macOS)
+        private func focusMainWindow() {
+            DispatchQueue.main.asyncAfter(deadline: .now() + 1) {
+                Windows.main.focus()
             }
         }
-    }
+
+        private func alertIfNoMainWindowOpen() -> Bool {
+            guard !Windows.main.isOpen else {
+                return false
+            }
+
+            navigation.presentAlert(
+                title: "Restart the app to open this link",
+                message:
+                "To open this link in the app you need to close and open it manually to have browser window, " +
+                    "then you can try opening links again.\n\nThis is a limitation of SwiftUI on macOS versions earlier than Ventura."
+            )
+
+            navigation.presentingAlertInVideoPlayer = true
+
+            return true
+        }
+    #endif
 }