/**
 * Copyright (c) 2016-present Invertase Limited & Contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this library 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 "RNFBFirestoreTransactionRegistry.h"
#import "RNFBFirestoreTransactionAttempt.h"

#if __has_include("RNFBHandleMap.h")
#import "RNFBHandleMap.h"
#else
#import "RNFBApp/RNFBHandleMap.h"
#endif

#if __has_include(<RNFBFirestore/RNFBFirestore-Swift.h>)
#import <RNFBFirestore/RNFBFirestore-Swift.h>
#elif __has_include("RNFBFirestore-Swift.h")
#import "RNFBFirestore-Swift.h"
#elif __has_include("RNFBHandleMapStorage-Swift.inc")
#import "RNFBHandleMapStorage-Swift.inc"
#else
#error "RNFBFirestoreTransactionRegistryStorage Swift interface not found"
#endif

@interface RNFBFirestoreTransactionRegistry ()
@property(nonatomic, strong) RNFBFirestoreTransactionRegistryStorage *storage;
@end

@implementation RNFBFirestoreTransactionRegistry

- (instancetype)init {
  self = [super init];
  if (self) {
    _storage = [[RNFBFirestoreTransactionRegistryStorage alloc] init];
  }
  return self;
}

- (void)rnfb_abortState:(id)state {
  if (![state isKindOfClass:[RNFBFirestoreTransactionAttempt class]]) {
    return;
  }
  [(RNFBFirestoreTransactionAttempt *)state abort];
}

- (BOOL)put:(id)key value:(id)value error:(NSError **)error {
  if (![self.storage putIfAbsent:key value:value]) {
    if (error != nil) {
      NSString *message = [NSString stringWithFormat:@"Handle id already registered: %@", key];
      *error = [NSError errorWithDomain:RNFBHandleMapErrorDomain
                                   code:RNFBHandleMapErrorCollision
                               userInfo:@{NSLocalizedDescriptionKey : message}];
    }
    return NO;
  }
  return YES;
}

- (BOOL)putOrSkip:(id)key value:(id)value {
  // Single HandleMap lock: store if absent, or YES when existing == value.
  return [self.storage putIfAbsentOrSame:key value:value];
}

- (id)get:(id)key {
  return [self.storage get:key];
}

- (id)take:(id)key {
  return [self.storage take:key];
}

- (void)abortAll {
  NSArray *remaining = [self.storage takeAll];
  for (id state in remaining) {
    [self rnfb_abortState:state];
  }
}

@end