2020-12-18 17:23:56 +00:00
|
|
|
//
|
|
|
|
// Copyright 2020 New Vector Ltd
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
// You may obtain a copy of the License at
|
|
|
|
//
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
//
|
|
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
// limitations under the License.
|
|
|
|
//
|
|
|
|
|
|
|
|
import Foundation
|
|
|
|
|
|
|
|
enum SSOAuthenticationServiceError: Error {
|
|
|
|
case tokenNotFound
|
|
|
|
case userCanceled
|
|
|
|
case unknown
|
|
|
|
}
|
|
|
|
|
|
|
|
@objcMembers
|
|
|
|
final class SSOAuthenticationService: NSObject {
|
|
|
|
|
|
|
|
// MARK: - Constants
|
|
|
|
|
|
|
|
// MARK: - Properties
|
|
|
|
|
|
|
|
private let homeserverStringURL: String
|
|
|
|
|
2021-01-07 13:27:18 +00:00
|
|
|
let callBackURLScheme: String?
|
2020-12-18 17:23:56 +00:00
|
|
|
|
|
|
|
// MARK: - Setup
|
|
|
|
|
|
|
|
init(homeserverStringURL: String) {
|
2021-01-07 13:27:18 +00:00
|
|
|
self.homeserverStringURL = homeserverStringURL
|
|
|
|
self.callBackURLScheme = BuildSettings.applicationURLScheme
|
2020-12-18 17:23:56 +00:00
|
|
|
super.init()
|
|
|
|
}
|
|
|
|
|
|
|
|
// MARK: - Public
|
|
|
|
|
2021-01-07 13:27:18 +00:00
|
|
|
func authenticationURL(for identityProvider: String?, transactionId: String) -> URL? {
|
2020-12-18 17:23:56 +00:00
|
|
|
guard var authenticationComponent = URLComponents(string: self.homeserverStringURL) else {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
let ssoRedirectPath: String
|
|
|
|
|
|
|
|
if let identityProvider = identityProvider {
|
2021-01-08 15:21:12 +00:00
|
|
|
ssoRedirectPath = SSOURLConstants.Paths.unstableRedirect + identityProvider
|
2020-12-18 17:23:56 +00:00
|
|
|
} else {
|
2021-01-08 15:21:12 +00:00
|
|
|
ssoRedirectPath = SSOURLConstants.Paths.redirect
|
2020-12-18 17:23:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
authenticationComponent.path = ssoRedirectPath
|
|
|
|
|
|
|
|
var queryItems: [URLQueryItem] = []
|
|
|
|
|
2021-01-07 13:27:18 +00:00
|
|
|
if let callBackURLScheme = self.buildCallBackURL(with: transactionId) {
|
|
|
|
queryItems.append(URLQueryItem(name: SSOURLConstants.Parameters.redirectURL, value: callBackURLScheme))
|
2020-12-18 17:23:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
authenticationComponent.queryItems = queryItems
|
|
|
|
|
|
|
|
return authenticationComponent.url
|
|
|
|
}
|
|
|
|
|
|
|
|
func loginToken(from url: URL) -> String? {
|
|
|
|
guard let components = URLComponents(string: url.absoluteString) else {
|
|
|
|
return nil
|
|
|
|
}
|
2021-01-07 13:27:18 +00:00
|
|
|
return components.vc_getQueryItemValue(for: SSOURLConstants.Parameters.callbackLoginToken)
|
2020-12-18 17:23:56 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// MARK: - Private
|
|
|
|
|
2021-01-07 13:27:18 +00:00
|
|
|
private func buildCallBackURL(with transactionId: String) -> String? {
|
|
|
|
guard let callBackURLScheme = self.callBackURLScheme else {
|
2020-12-18 17:23:56 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
var urlComponents = URLComponents()
|
2021-01-07 13:27:18 +00:00
|
|
|
urlComponents.scheme = callBackURLScheme
|
|
|
|
urlComponents.host = CustomSchemeURLConstants.Hosts.connect
|
|
|
|
|
|
|
|
// Transaction id is used to indentify the request
|
|
|
|
urlComponents.queryItems = [URLQueryItem(name: CustomSchemeURLConstants.Parameters.transactionId, value: transactionId)]
|
2020-12-18 17:23:56 +00:00
|
|
|
return urlComponents.string
|
|
|
|
}
|
|
|
|
}
|