firebase-ios-sdk/FirebaseAuth/Sources/MultiFactor/TOTP/FIRTOTPMultiFactorGenerator.m

/*
 * 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/LICENSE2.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 <TargetConditionals.h>
#if TARGET_OS_IOS

#import "FirebaseAuth/Sources/Auth/FIRAuth_Internal.h"
#import "FirebaseAuth/Sources/Backend/FIRAuthBackend+MultiFactor.h"
#import "FirebaseAuth/Sources/Backend/FIRAuthBackend.h"
#import "FirebaseAuth/Sources/Backend/RPC/MultiFactor/Enroll/FIRStartMFAEnrollmentRequest.h"
#import "FirebaseAuth/Sources/Backend/RPC/MultiFactor/Enroll/FIRStartMFAEnrollmentResponse.h"
#import "FirebaseAuth/Sources/Backend/RPC/Proto/TOTP/FIRAuthProtoStartMFATOTPEnrollmentResponseInfo.h"
#import "FirebaseAuth/Sources/MultiFactor/FIRMultiFactorSession+Internal.h"
#import "FirebaseAuth/Sources/MultiFactor/TOTP/FIRTOTPMultiFactorAssertion+Internal.h"
#import "FirebaseAuth/Sources/MultiFactor/TOTP/FIRTOTPSecret+Internal.h"
#import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRTOTPMultiFactorAssertion.h"
#import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRTOTPMultiFactorGenerator.h"
#import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRTOTPSecret.h"

@implementation FIRTOTPMultiFactorGenerator

+ (void)generateSecretWithMultiFactorSession:(FIRMultiFactorSession *)session
                                  completion:(void (^)(FIRTOTPSecret *_Nullable secret,
                                                       NSError *_Nullable error))completion {
  if (session.IDToken) {
    FIRStartMFAEnrollmentRequest *request = [[FIRStartMFAEnrollmentRequest alloc]
             initWithIDToken:session.IDToken
          TOTPEnrollmentInfo:[[FIRAuthProtoStartMFATOTPEnrollmentRequestInfo alloc] init]
        requestConfiguration:session.currentUser.auth.requestConfiguration];
    [FIRAuthBackend
        startMultiFactorEnrollment:request
                          callback:^(FIRStartMFAEnrollmentResponse *_Nullable response,
                                     NSError *_Nullable error) {
                            if (error) {
                              if (completion) {
                                completion(nil, error);
                              }
                            } else if (response.TOTPSessionInfo) {
                              FIRTOTPSecret *secret = [[FIRTOTPSecret alloc]
                                             initWithSecretKey:response.TOTPSessionInfo
                                                                   .sharedSecretKey
                                              hashingAlgorithm:response.TOTPSessionInfo
                                                                   .hashingAlgorithm
                                                    codeLength:response.TOTPSessionInfo
                                                                   .verificationCodeLength
                                           codeIntervalSeconds:response.TOTPSessionInfo.periodSec
                                  enrollmentCompletionDeadline:response.TOTPSessionInfo
                                                                   .finalizeEnrollmentTime
                                                   sessionInfo:response.TOTPSessionInfo
                                                                   .sessionInfo];
                              if (completion) {
                                completion(secret, nil);
                              }
                            } else {
                              NSError *error =
                                  [NSError errorWithDomain:FIRAuthErrorDomain
                                                      code:FIRAuthErrorCodeInternalError
                                                  userInfo:@{
                                                    NSLocalizedDescriptionKey :
                                                        @"Error generating TOTP secret."
                                                  }];
                              if (completion) {
                                completion(nil, error);
                              }
                            }
                          }];
  } else {
    NSError *error = [NSError errorWithDomain:FIRAuthErrorDomain
                                         code:FIRAuthErrorCodeInternalError
                                     userInfo:@{NSLocalizedDescriptionKey : @"Invalid ID token."}];
    if (completion) {
      completion(nil, error);
    }
  }
}

+ (FIRTOTPMultiFactorAssertion *)assertionForEnrollmentWithSecret:(FIRTOTPSecret *)secret
                                                  oneTimePassword:(NSString *)oneTimePassword {
  return [[FIRTOTPMultiFactorAssertion alloc] initWithSecret:secret
                                             oneTimePassword:oneTimePassword];
}

+ (FIRTOTPMultiFactorAssertion *)assertionForSignInWithEnrollmentID:(NSString *)enrollmentID
                                                    oneTimePassword:(NSString *)oneTimePassword {
  return [[FIRTOTPMultiFactorAssertion alloc] initWithEnrollmentID:enrollmentID
                                                   oneTimePassword:oneTimePassword];
}

@end

#endif