Add support for Authentication Session (#305)

* Add support for authorising with ASWebAuthenticationSession
* Add fallback support for SFAuthenticationSession on iOS 11

Author: blork

Sources/Base/OAuth2AuthConfig.swift

@@ -43,6 +43,9 @@ public struct OAuth2AuthConfig {
 		/// Starting with iOS 9, `SFSafariViewController` will be used for embedded authorization instead of our custom class. You can turn this off here.
 		public var useSafariView = true
 
+		/// Starting with iOS 12, `ASWebAuthenticationSession` can be used for embedded authorization instead of our custom class. You can turn this on here.
+		public var useAuthenticationSession = false
+		
 		#if os(iOS)
 		/// By assigning your own style you can configure how the embedded authorization is presented.
 		public var modalPresentationStyle = UIModalPresentationStyle.fullScreen
@@ -60,7 +63,7 @@ public struct OAuth2AuthConfig {
 
 	/// Whether to automatically dismiss the auto-presented authorization screen.
 	public var authorizeEmbeddedAutoDismiss = true
-
+	
 	/// Context information for the authorization flow:
 	/// - iOS:   The parent view controller to present from
 	/// - macOS: An NSWindow from which to present a modal sheet _or_ `nil` to present in a new window

Sources/iOS/OAuth2Authorizer+iOS.swift

@@ -21,6 +21,7 @@
 
 import UIKit
 import SafariServices
+import AuthenticationServices
 #if !NO_MODULE_IMPORT
 import Base
 #endif
@@ -39,6 +40,8 @@ open class OAuth2Authorizer: OAuth2AuthorizerUI {
 	/// Used to store the `SFSafariViewControllerDelegate`.
 	private var safariViewDelegate: AnyObject?
 
+	/// Used to store the authentication session.
+	private var authenticationSession: AnyObject?
 
 	public init(oauth2: OAuth2) {
 		self.oauth2 = oauth2
@@ -72,23 +75,31 @@ open class OAuth2Authorizer: OAuth2AuthorizerUI {
 	- parameter at:   The authorize URL to open
 	*/
 	public func authorizeEmbedded(with config: OAuth2AuthConfig, at url: URL) throws {
-		guard let controller = config.authorizeContext as? UIViewController else {
-			throw (nil == config.authorizeContext) ? OAuth2Error.noAuthorizationContext : OAuth2Error.invalidAuthorizationContext
-		}
-		
-		if #available(iOS 9, *), config.ui.useSafariView {
-			let web = try authorizeSafariEmbedded(from: controller, at: url)
-			if config.authorizeEmbeddedAutoDismiss {
-				oauth2.internalAfterAuthorizeOrFail = { wasFailure, error in
-					web.dismiss(animated: true)
+		if #available(iOS 11, *), config.ui.useAuthenticationSession {
+			guard let redirect = oauth2.redirect else {
+				throw OAuth2Error.noRedirectURL
+			}
+			
+			authenticationSessionEmbedded(at: url, withRedirect: redirect)
+		} else {
+			guard let controller = config.authorizeContext as? UIViewController else {
+				throw (nil == config.authorizeContext) ? OAuth2Error.noAuthorizationContext : OAuth2Error.invalidAuthorizationContext
+			}
+			
+			if #available(iOS 9, *), config.ui.useSafariView {
+				let web = try authorizeSafariEmbedded(from: controller, at: url)
+				if config.authorizeEmbeddedAutoDismiss {
+					oauth2.internalAfterAuthorizeOrFail = { wasFailure, error in
+						web.dismiss(animated: true)
+					}
 				}
 			}
-		}
-		else {
-			let web = try authorizeEmbedded(from: controller, at: url)
-			if config.authorizeEmbeddedAutoDismiss {
-				oauth2.internalAfterAuthorizeOrFail = { wasFailure, error in
-					web.dismiss(animated: true)
+			else {
+				let web = try authorizeEmbedded(from: controller, at: url)
+				if config.authorizeEmbeddedAutoDismiss {
+					oauth2.internalAfterAuthorizeOrFail = { wasFailure, error in
+						web.dismiss(animated: true)
+					}
 				}
 			}
 		}
@@ -104,6 +115,47 @@ open class OAuth2Authorizer: OAuth2AuthorizerUI {
 	open func willPresent(viewController: UIViewController, in naviController: UINavigationController?) {
 	}
 
+	// MARK: - SFAuthenticationSession / ASWebAuthenticationSession
+	
+	/**
+	Use SFAuthenticationSession or ASWebAuthenticationSession to manage authorisation.
+	
+	On iOS 11, use SFAuthenticationSession. On iOS 12+, use ASWebAuthenticationSession.
+	
+	The mechanism works just like when you're using Safari itself to log the user in, hence you **need to implement**
+	`application(application:openURL:sourceApplication:annotation:)` in your application delegate.
+	
+	This method dismisses the view controller automatically - this cannot be disabled.
+	
+	- parameter at:   The authorize URL to open
+	- returns:        A Boolean value indicating whether the web authentication session starts successfully.
+	*/
+	@available(iOS 11.0, *)
+	@discardableResult
+	public func authenticationSessionEmbedded(at url: URL, withRedirect redirect: String) -> Bool {
+		let completionHandler: (URL?, Error?) -> Void = { url, error in
+			if let url = url {
+				do {
+					try self.oauth2.handleRedirectURL(url as URL)
+				}
+				catch let err {
+					self.oauth2.logger?.warn("OAuth2", msg: "Cannot intercept redirect URL: \(err)")
+				}
+			} else {
+				self.oauth2.didFail(with: nil)
+			}
+			self.authenticationSession = nil
+		}
+		
+		if #available(iOS 12, *) {
+			authenticationSession = ASWebAuthenticationSession(url: url, callbackURLScheme: redirect, completionHandler: completionHandler)
+			return (authenticationSession as! ASWebAuthenticationSession).start()
+		} else {
+			authenticationSession = SFAuthenticationSession(url: url, callbackURLScheme: redirect, completionHandler: completionHandler)
+			return (authenticationSession as! SFAuthenticationSession).start()
+		}
+	}
+	
 
 	// MARK: - Safari Web View Controller
 
@@ -135,13 +187,14 @@ open class OAuth2Authorizer: OAuth2AuthorizerUI {
 			web.preferredControlTintColor = tint
 		}
 		web.modalPresentationStyle = oauth2.authConfig.ui.modalPresentationStyle
-        
+		
 		willPresent(viewController: web, in: nil)
 		controller.present(web, animated: true, completion: nil)
 
 		return web
 	}
 
+	
 	/**
 	Called from our delegate, which reacts to users pressing "Done". We can assume this is always a cancel as nomally the Safari view
 	controller is dismissed automatically.