GoogleSignIn-iOS/Samples/Swift/DaysUntilBirthday/Shared/ViewModels/AuthenticationViewModel.swift

/*
 * Copyright 2021 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 SwiftUI
import GoogleSignIn

/// A class conforming to `ObservableObject` used to represent a user's authentication status.
final class AuthenticationViewModel: ObservableObject {
  /// The user's log in status.
  /// - note: This will publish updates when its value changes.
  @Published var state: State
  private var authenticator: GoogleSignInAuthenticator {
    return GoogleSignInAuthenticator(authViewModel: self)
  }
  /// The user-authorized scopes.
  /// - note: If the user is logged out, then this will default to empty.
  var authorizedScopes: [String] {
    switch state {
    case .signedIn(let user):
      return user.grantedScopes ?? []
    case .signedOut:
      return []
    }
  }

  /// Creates an instance of this view model.
  init() {
    if let user = GIDSignIn.sharedInstance.currentUser {
      self.state = .signedIn(user)
    } else {
      self.state = .signedOut
    }
  }

  /// Signs the user in.
  func signIn() {
#if os(iOS)
    guard let rootViewController = UIApplication.shared.windows.first?.rootViewController else {
      print("There is no root view controller!")
      return
    }
#elseif os(macOS)
    guard let presentingWindow = NSApplication.shared.windows.first else {
      print("There is no presenting window!")
      return
    }
#endif

    Task { @MainActor in
      do {
#if os(iOS)
        let signInResult = try await authenticator.signIn(with: rootViewController)
#elseif os(macOS)
        let signInResult = try await authenticator.signIn(with: presentingWindow)
#endif
        self.state = .signedIn(signInResult.user)
      } catch {
        print("Error signing in: \(error)")
      }
    }
  }

  /// Signs the user out.
  func signOut() {
    authenticator.signOut()
  }

  /// Disconnects the previously granted scope and signs the user out.
  func disconnect() {
    Task { @MainActor in
      do {
        try await authenticator.disconnect()
      } catch {
        print("Error disconnecting: \(error)")
      }
    }
  }

  var hasBirthdayReadScope: Bool {
    return authorizedScopes.contains(BirthdayLoader.birthdayReadScope)
  }

#if os(iOS)
  /// Adds the requested birthday read scope.
  /// - parameter viewController: A `UIViewController` to use while presenting the flow.
  /// - returns: The `GIDSignInResult`.
  /// - throws: Any error that may arise while adding the read birthday scope.
  func addBirthdayReadScope(viewController: UIViewController) async throws -> GIDSignInResult {
    return try await authenticator.addBirthdayReadScope(viewController: viewController)
  }
#endif


#if os(macOS)
  /// adds the requested birthday read scope.
  /// - parameter window: An `NSWindow` to use while presenting the flow.
  /// - returns: The `GIDSignInResult`.
  /// - throws: Any error that may arise while adding the read birthday scope.
  func addBirthdayReadScope(window: NSWindow) async throws -> GIDSignInResult {
    return try await authenticator.addBirthdayReadScope(window: window)
  }
#endif
}

extension AuthenticationViewModel {
  /// An enumeration representing logged in status.
  enum State {
    /// The user is logged in and is the associated value of this case.
    case signedIn(GIDGoogleUser)
    /// The user is logged out.
    case signedOut
  }
}