improve isStoreURL check logic

brave · Sep 5, 2023 · c13f3d4 · c13f3d4
1 parent ee0ca25
commit c13f3d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/...e/Frontend/Browser/BrowserViewController/BrowserViewController+WKNavigationDelegate.swift b/...e/Frontend/Browser/BrowserViewController/BrowserViewController+WKNavigationDelegate.swift
@@ -111,11 +111,11 @@ extension BrowserViewController: WKNavigationDelegate {
  // leave a 'Back to Safari' button in the status bar, which we do not want.
  fileprivate func isStoreURL(_ url: URL) -> Bool {
  if url.scheme == "http" || url.scheme == "https" {
- if url.host == "itunes.apple.com" {
+ if url.host == "itunes.apple.com" || url.host == "apps.apple.com" {
  return true
  }
  }
- if url.scheme == "itms-appss" || url.scheme == "itmss" {
+ if url.scheme == "itms-appss" || url.scheme == "itms-apps" || url.scheme == "itmss" {
  return true
  }
  return false

diff --git a/Sources/Brave/Frontend/Browser/Playlist/Managers & Cache/PlaylistCacheLoader.swift b/Sources/Brave/Frontend/Browser/Playlist/Managers & Cache/PlaylistCacheLoader.swift
@@ -512,11 +512,11 @@ extension PlaylistWebLoader: WKNavigationDelegate {
  // leave a 'Back to Safari' button in the status bar, which we do not want.
  fileprivate func isStoreURL(_ url: URL) -> Bool {
  if url.scheme == "http" || url.scheme == "https" {
- if url.host == "itunes.apple.com" {
+ if url.host == "itunes.apple.com" || url.host == "apps.apple.com" {
  return true
  }
  }
- if url.scheme == "itms-appss" || url.scheme == "itmss" {
+ if url.scheme == "itms-appss" || url.scheme == "itms-apps" || url.scheme == "itmss" {
  return true
  }
  return false