firebase-ios-sdk/FirebaseAuth/Sources/Swift/Backend/RPC/SetAccountInfoResponse.swift

// Copyright 2023 Google LLC
//
// 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

/** @class FIRSetAccountInfoResponseProviderUserInfo
    @brief Represents the provider user info part of the response from the setAccountInfo endpoint.
    @see https://developers.google.com/identity/toolkit/web/reference/relyingparty/setAccountInfo
 */
class SetAccountInfoResponseProviderUserInfo: NSObject {
  /** @property providerID
      @brief The ID of the identity provider.
   */
  var providerID: String?

  /** @property displayName
      @brief The user's display name at the identity provider.
   */
  var displayName: String?

  /** @property photoURL
      @brief The user's photo URL at the identity provider.
   */
  var photoURL: URL?

  /** @fn initWithAPIKey:
      @brief Designated initializer.
      @param dictionary The provider user info data from endpoint.
   */
  init(dictionary: [String: Any]) {
    providerID = dictionary["providerId"] as? String
    displayName = dictionary["displayName"] as? String
    if let photoURL = dictionary["photoUrl"] as? String {
      self.photoURL = URL(string: photoURL)
    }
  }
}

/** @class FIRSetAccountInfoResponse
    @brief Represents the response from the setAccountInfo endpoint.
    @see https://developers.google.com/identity/toolkit/web/reference/relyingparty/setAccountInfo
 */
class SetAccountInfoResponse: AuthRPCResponse {
  required init() {}

  /** @property email
      @brief The email or the user.
   */
  var email: String?

  /** @property displayName
      @brief The display name of the user.
   */
  var displayName: String?

  /** @property providerUserInfo
      @brief The user's profiles at the associated identity providers.
   */
  var providerUserInfo: [SetAccountInfoResponseProviderUserInfo]?

  /** @property idToken
      @brief Either an authorization code suitable for performing an STS token exchange, or the
          access token from Secure Token Service, depending on whether @c returnSecureToken is set
          on the request.
   */
  var idToken: String?

  /** @property approximateExpirationDate
      @brief The approximate expiration date of the access token.
   */
  var approximateExpirationDate: Date?

  /** @property refreshToken
      @brief The refresh token from Secure Token Service.
   */
  var refreshToken: String?

  func setFields(dictionary: [String: AnyHashable]) throws {
    email = dictionary["email"] as? String
    displayName = dictionary["displayName"] as? String
    idToken = dictionary["idToken"] as? String
    if let expiresIn = dictionary["expiresIn"] as? String {
      approximateExpirationDate = Date(timeIntervalSinceNow: (expiresIn as NSString)
        .doubleValue)
    }
    refreshToken = dictionary["refreshToken"] as? String
    if let providerUserInfoData = dictionary["providerUserInfo"] as? [[String: Any]] {
      providerUserInfo = providerUserInfoData.map { .init(dictionary: $0) }
    }
  }
}