added custom user agent handling

Author: xtools-at

ios-pwa-wrapper/Constants.swift

@@ -9,11 +9,18 @@
 //import Foundation
 import UIKit
 
-// Basic configuration
+// Basic App-/WebView-configuration
+let appTitle = "iOS PWA Wrapper"
 let webAppUrl = URL(string: "https://www.duckduckgo.com")
 let allowedOrigin = "duckduckgo.com"
-let appTitle = "iOS PWA Wrapper"
-let menuButtonJavascript = "document.querySelector('body').remove();"
+let menuButtonJavascript = """
+    document.querySelector('body').remove();
+"""
+let useCustomUserAgent = false
+let useUserAgentPostfix = true
+let customUserAgent = "Mozilla/5.0 (iPhone; CPU iPhone OS 11_0 like Mac OS X) AppleWebKit/604.1.38 (KHTML, like Gecko) Version/11.0 Mobile/15A356"
+let userAgentPostfix = "iOSApp"
+
 
 // Settings
 let changeAppTitleToPageTitle = false

ios-pwa-wrapper/ViewController.swift

@@ -59,19 +59,34 @@ class ViewController: UIViewController {
         webView = WKWebView(frame: CGRect(x: 0, y: 0, width: webViewContainer.frame.width, height: webViewContainer.frame.height))
         webView.navigationDelegate = self
         webView.uiDelegate = self
-        webViewContainer.addSubview(webView)
         webView.autoresizingMask = [.flexibleWidth, .flexibleHeight]
+        webViewContainer.addSubview(webView)
 
         // settings
         webView.allowsBackForwardNavigationGestures = true
+        // user agent
+        if (useCustomUserAgent) {
+            webView.customUserAgent = customUserAgent
+        }
+        if (useUserAgentPostfix) {
+            if (useCustomUserAgent) {
+                webView.customUserAgent = customUserAgent + " " + userAgentPostfix
+            } else {
+                webView.evaluateJavaScript("navigator.userAgent", completionHandler: { (result, error) in
+                    if let resultObject = result {
+                        self.webView.customUserAgent = (String(describing: resultObject) + " " + userAgentPostfix)
+                    }
+                })
+            }
+        }
 
         // init observers
         webView.addObserver(self, forKeyPath: #keyPath(WKWebView.isLoading), options: NSKeyValueObservingOptions.new, context: nil)
         webView.addObserver(self, forKeyPath: #keyPath(WKWebView.estimatedProgress), options: NSKeyValueObservingOptions.new, context: nil)
     }
 
     // Initialize UI
-    // Call after WebView has been initialized
+    // call after WebView has been initialized
     func setupUI() {
         // UI elements
         leftButton.isEnabled = false
@@ -87,7 +102,6 @@ class ViewController: UIViewController {
             navigationItem.largeTitleDisplayMode = UINavigationItem.LargeTitleDisplayMode.always
         }
         if (useLightStatusBarStyle) {
-            // UIApplication.shared.setStatusBarStyle(.lightContent, animated: true) // iOS < 9
             self.navigationController?.navigationBar.barStyle = UIBarStyle.black
         }
     }