/**
 * Copyright (c) 2015-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the BSD-style license found in the
 * LICENSE file in the root directory of this source tree. An additional grant
 * of patent rights can be found in the PATENTS file in the same directory.
 */

#import "RCTSparseArray.h"

@implementation RCTSparseArray
{
  NSMutableDictionary *_storage;
}

- (instancetype)init
{
  return [self initWithCapacity:0];
}

- (instancetype)initWithCapacity:(NSUInteger)capacity
{
  if ((self = [super init])) {
    _storage = [NSMutableDictionary dictionaryWithCapacity:capacity];
  }
  return self;
}

- (instancetype)initWithSparseArray:(RCTSparseArray *)sparseArray
{
  if ((self = [super init])) {
    _storage = [sparseArray->_storage copy];
  }
  return self;
}

+ (instancetype)sparseArray
{
  return [self new];
}

+ (instancetype)sparseArrayWithCapacity:(NSUInteger)capacity
{
  return [[self alloc] initWithCapacity:capacity];
}

+ (instancetype)sparseArrayWithSparseArray:(RCTSparseArray *)sparseArray
{
  return [[self alloc] initWithSparseArray:sparseArray];
}

- (id)objectAtIndexedSubscript:(NSUInteger)idx
{
  return _storage[@(idx)];
}

- (void)setObject:(id)obj atIndexedSubscript:(NSUInteger)idx
{
  self[@(idx)] = obj;
}

- (id)objectForKeyedSubscript:(NSNumber *)key
{
  return _storage[key];
}

- (void)setObject:(id)obj forKeyedSubscript:(NSNumber *)key
{
  if (obj) {
    _storage[key] = obj;
  } else {
    [_storage removeObjectForKey:key];
  }
}

- (NSUInteger)count
{
  return _storage.count;
}

- (NSArray *)allIndexes
{
  return _storage.allKeys;
}

- (NSArray *)allObjects
{
  return _storage.allValues;
}

- (void)enumerateObjectsUsingBlock:(void (^)(id obj, NSNumber *idx, BOOL *stop))block
{
  NSParameterAssert(block != nil);
  [_storage enumerateKeysAndObjectsUsingBlock:^(NSNumber *key, id obj, BOOL *stop) {
    block(obj, key, stop);
  }];
}

- (void)enumerateObjectsWithOptions:(NSEnumerationOptions)opts usingBlock:(void (^)(id obj, NSNumber *idx, BOOL *stop))block
{
  NSParameterAssert(block != nil);
  [_storage enumerateKeysAndObjectsWithOptions:opts usingBlock:^(NSNumber *key, id obj, BOOL *stop) {
    block(obj, key, stop);
  }];
}

- (void)removeAllObjects
{
  [_storage removeAllObjects];
}

- (id)copyWithZone:(NSZone *)zone
{
  return [[[self class] allocWithZone:zone] initWithSparseArray:self];
}

- (NSString *)description
{
  return [super.description stringByAppendingString:_storage.description];
}

@end