firebase-ios-sdk/FirebaseAppCheck/Sources/Core/FIRAppCheckSettings.m

/*
 * 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 "FirebaseAppCheck/Sources/Core/FIRAppCheckSettings.h"

#import "FirebaseCore/Extension/FirebaseCoreInternal.h"

NS_ASSUME_NONNULL_BEGIN

NSString *const kFIRAppCheckTokenAutoRefreshEnabledUserDefaultsPrefix =
    @"FIRAppCheckTokenAutoRefreshEnabled_";
NSString *const kFIRAppCheckTokenAutoRefreshEnabledInfoPlistKey =
    @"FirebaseAppCheckTokenAutoRefreshEnabled";

@interface FIRAppCheckSettings ()

@property(nonatomic, weak, readonly) FIRApp *firebaseApp;
@property(nonatomic, readonly) NSUserDefaults *userDefaults;
@property(nonatomic, readonly) NSBundle *mainBundle;
@property(nonatomic, readonly) NSString *userDefaultKey;
@property(nonatomic, assign) BOOL isTokenAutoRefreshConfigured;

@end

@implementation FIRAppCheckSettings

- (instancetype)initWithApp:(FIRApp *)firebaseApp
                userDefault:(NSUserDefaults *)userDefaults
                 mainBundle:(NSBundle *)mainBundle {
  self = [super init];
  if (self) {
    _firebaseApp = firebaseApp;
    _userDefaults = userDefaults;
    _mainBundle = mainBundle;
    _userDefaultKey = [kFIRAppCheckTokenAutoRefreshEnabledUserDefaultsPrefix
        stringByAppendingString:firebaseApp.name];
    [super setIsTokenAutoRefreshEnabled:NO];
    _isTokenAutoRefreshConfigured = NO;
  }
  return self;
}

- (BOOL)isTokenAutoRefreshEnabled {
  @synchronized(self) {
    if (self.isTokenAutoRefreshConfigured) {
      // Return value form the in-memory cache to avoid accessing the user default or bundle when
      // not required.
      return [super isTokenAutoRefreshEnabled];
    }

    // Check user defaults for a value set during the previous launch.
    NSNumber *isTokenAutoRefreshEnabledNumber =
        [self.userDefaults objectForKey:self.userDefaultKey];

    // Check Info.plist if no user defaults value found.
    if (isTokenAutoRefreshEnabledNumber == nil) {
      isTokenAutoRefreshEnabledNumber = [self.mainBundle
          objectForInfoDictionaryKey:kFIRAppCheckTokenAutoRefreshEnabledInfoPlistKey];
    }

    if (isTokenAutoRefreshEnabledNumber != nil) {
      // Update in-memory cache.
      self.isTokenAutoRefreshConfigured = YES;
      self.isTokenAutoRefreshEnabled = isTokenAutoRefreshEnabledNumber.boolValue;
      // Return the value.
      return [super isTokenAutoRefreshEnabled];
    }

    // Fallback to the global data collection flag.
    if (self.firebaseApp) {
      return self.firebaseApp.isDataCollectionDefaultEnabled;
    } else {
      // If `self.firebaseApp == nil`, then the app has been de-initialized. No auto-refresh in this
      // case.
      return NO;
    }
  }
}

- (void)setIsTokenAutoRefreshEnabled:(BOOL)isTokenAutoRefreshEnabled {
  @synchronized(self) {
    self.isTokenAutoRefreshConfigured = YES;
    [super setIsTokenAutoRefreshEnabled:isTokenAutoRefreshEnabled];
    [self.userDefaults setBool:isTokenAutoRefreshEnabled forKey:self.userDefaultKey];
  }
}

@end

NS_ASSUME_NONNULL_END