#import "MLRNSource.h"
#import <React/RCTLog.h>
#import <React/UIView+React.h>
#import "MLRNMapView.h"

@implementation MLRNSource

- (instancetype)initWithFrame:(CGRect)frame {
  if (self = [super initWithFrame:frame]) {
    _layers = [[NSMutableArray alloc] init];
    _reactSubviews = [[NSMutableArray alloc] init];
    _hitbox = @{@"top" : @22.0f, @"right" : @22.0f, @"bottom" : @22.0f, @"left" : @22.0f};
  }
  return self;
}

#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wobjc-missing-super-calls"
- (void)insertReactSubview:(id<RCTComponent>)subview atIndex:(NSInteger)atIndex {
  if ([subview isKindOfClass:[MLRNLayer class]]) {
    MLRNLayer *layer = (MLRNLayer *)subview;

    if (layer.sourceID == nil) {
      layer.sourceID = _id;
    }

    if (_map.style != nil) {
      [layer addToMap:_map style:_map.style];
    }

    [_layers addObject:layer];
    [_reactSubviews insertObject:layer atIndex:atIndex];
  }
}
#pragma clang diagnostic pop

#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wobjc-missing-super-calls"
- (void)removeReactSubview:(id<RCTComponent>)subview {
  if ([subview isKindOfClass:[MLRNLayer class]]) {
    MLRNLayer *layer = (MLRNLayer *)subview;
    [layer removeFromMap:_map.style];
    [_layers removeObject:layer];
    [_reactSubviews removeObject:layer];
  }
}
#pragma clang diagnostic pop

#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wobjc-missing-super-calls"
- (NSArray<id<RCTComponent>> *)reactSubviews {
  return _reactSubviews;
}
#pragma clang diagnostic pop

- (void)setMap:(MLRNMapView *)map {
  if (map == nil) {
    [self removeFromMap];
    _map = nil;
  } else {
    _map = map;
    [self addToMap];
  }
}

- (void)addToMap {
  if (_map.style == nil) {
    return;
  }

  MLNSource *existingSource = [_map.style sourceWithIdentifier:_id];
  if (existingSource != nil) {
    _source = existingSource;
  } else {
    _source = [self makeSource];
    if (_source != nil) {
      [_map.style addSource:_source];
    } else {
      RCTLogError(@"Could not create source with id:%@. Make sure url/shape is provided", _id);
    }
  }

  if (_layers.count > 0) {
    for (int i = 0; i < _layers.count; i++) {
      MLRNLayer *layer = [_layers objectAtIndex:i];
      if (layer.sourceID == nil) {
        layer.sourceID = _id;
      }
      [layer addToMap:_map style:_map.style];
    }
  }
}

- (void)removeFromMap {
  if (_map.style == nil) {
    return;
  }

  for (int i = 0; i < _layers.count; i++) {
    MLRNLayer *layer = [_layers objectAtIndex:i];
    [layer removeFromMap:_map.style];
  }

  if (_source != nil) {
    [_map.style removeSource:_source];
  }
}

- (nullable MLNSource *)makeSource {
  @throw [NSException
      exceptionWithName:NSInternalInconsistencyException
                 reason:[NSString stringWithFormat:@"You must override %@ in a subclass",
                                                   NSStringFromSelector(_cmd)]
               userInfo:nil];
}

- (NSArray<NSString *> *)getLayerIDs {
  NSMutableArray *layerIDs = [[NSMutableArray alloc] init];

  for (MLRNLayer *layer in _layers) {
    [layerIDs addObject:layer.id];
  }

  return layerIDs;
}

@end