comaps/iphone/Maps/Classes/CustomViews/MapViewControls/TrafficButton/MWMTrafficButtonViewController.mm

#import "MWMTrafficButtonViewController.h"

#import <CoreApi/MWMMapOverlayManager.h>

#import "MWMAlertViewController.h"
#import "MWMButton.h"
#import "MWMMapViewControlsCommon.h"
#import "MWMMapViewControlsManager.h"
#import "MapViewController.h"
#import "SwiftBridge.h"
#import "base/assert.hpp"

namespace {
CGFloat const kTopOffset = 6;

NSArray<UIImage *> *imagesWithName(NSString *name) {
  NSUInteger const imagesCount = 3;
  NSMutableArray<UIImage *> *images = [NSMutableArray arrayWithCapacity:imagesCount];
  NSString *mode = [UIColor isNightMode] ? @"dark" : @"light";
  for (NSUInteger i = 1; i <= imagesCount; i += 1) {
    NSString *imageName = [NSString stringWithFormat:@"%@_%@_%@", name, mode, @(i).stringValue];
    [images addObject:static_cast<UIImage *_Nonnull>([UIImage imageNamed:imageName])];
  }
  return [images copy];
}
}  // namespace

@interface MWMMapViewControlsManager ()

@property(nonatomic) MWMTrafficButtonViewController *trafficButton;

@end

@interface MWMTrafficButtonViewController () <MWMMapOverlayManagerObserver, ThemeListener>

@property(nonatomic) NSLayoutConstraint *topOffset;
@property(nonatomic) NSLayoutConstraint *leftOffset;
@property(nonatomic) CGRect availableArea;

@end

@implementation MWMTrafficButtonViewController

+ (MWMTrafficButtonViewController *)controller {
  return [MWMMapViewControlsManager manager].trafficButton;
}

- (instancetype)init {
  self = [super init];
  if (self) {
    MapViewController *ovc = [MapViewController sharedController];
    [ovc addChildViewController:self];
    [ovc.controlsView addSubview:self.view];
    [self configLayout];
    [self applyTheme];
    [StyleManager.shared addListener:self];
    [MWMMapOverlayManager addObserver:self];
  }
  return self;
}

- (void)dealloc {
  [StyleManager.shared removeListener:self];
}

- (void)viewWillDisappear:(BOOL)animated {
  [super viewWillDisappear:animated];
  [Toast hideAll];
}

- (void)configLayout {
  UIView *sv = self.view;
  UIView *ov = sv.superview;

  self.topOffset = [sv.topAnchor constraintEqualToAnchor:ov.topAnchor constant:kTopOffset];
  self.topOffset.active = YES;
  self.leftOffset = [sv.leadingAnchor constraintEqualToAnchor:ov.leadingAnchor constant:kViewControlsOffsetToBounds];
  self.leftOffset.active = YES;
}

- (void)setHidden:(BOOL)hidden {
  _hidden = hidden;
  [self refreshLayout];
}

- (void)refreshLayout {
  dispatch_async(dispatch_get_main_queue(), ^{
    auto const availableArea = self.availableArea;
    auto const leftOffset = self.hidden ? -self.view.width : availableArea.origin.x + kViewControlsOffsetToBounds;
    [self.view.superview animateConstraintsWithAnimations:^{
      self.topOffset.constant = availableArea.origin.y + kTopOffset;
      self.leftOffset.constant = leftOffset;
      self.view.alpha = self.hidden ? 0 : 1;
    }];
  });
}

- (void)handleTrafficState:(MWMMapOverlayTrafficState)state {
  MWMButton *btn = (MWMButton *)self.view;
  UIImageView *iv = btn.imageView;
  switch (state) {
    case MWMMapOverlayTrafficStateDisabled:
      CHECK(false, ("Incorrect traffic manager state."));
      break;
    case MWMMapOverlayTrafficStateEnabled:
      btn.imageName = @"btn_traffic_on";
      break;
    case MWMMapOverlayTrafficStateWaitingData:
      iv.animationImages = imagesWithName(@"btn_traffic_update");
      iv.animationDuration = 0.8;
      [iv startAnimating];
      break;
    case MWMMapOverlayTrafficStateOutdated:
      btn.imageName = @"btn_traffic_outdated";
      break;
    case MWMMapOverlayTrafficStateNoData:
      btn.imageName = @"btn_traffic_on";
      [Toast showWithText:L(@"traffic_data_unavailable")];
      break;
    case MWMMapOverlayTrafficStateNetworkError:
      [MWMMapOverlayManager setTrafficEnabled:NO];
      [[MWMAlertViewController activeAlertController] presentNoConnectionAlert];
      break;
    case MWMMapOverlayTrafficStateExpiredData:
      btn.imageName = @"btn_traffic_outdated";
      [Toast showWithText:L(@"traffic_update_maps_text")];
      break;
    case MWMMapOverlayTrafficStateExpiredApp:
      btn.imageName = @"btn_traffic_outdated";
      [Toast showWithText:L(@"traffic_update_app_message")];
      break;
  }
}

- (void)handleIsolinesState:(MWMMapOverlayIsolinesState)state {
  switch (state) {
    case MWMMapOverlayIsolinesStateDisabled:
      break;
    case MWMMapOverlayIsolinesStateEnabled:
      if (![MWMMapOverlayManager isolinesVisible])
        [Toast showWithText:L(@"isolines_toast_zooms_1_10")];
      break;
    case MWMMapOverlayIsolinesStateExpiredData:
      [MWMAlertViewController.activeAlertController presentInfoAlert:L(@"isolines_activation_error_dialog")];
      [MWMMapOverlayManager setIsoLinesEnabled:NO];
      break;
    case MWMMapOverlayIsolinesStateNoData:
      [MWMAlertViewController.activeAlertController presentInfoAlert:L(@"isolines_location_error_dialog")];
      [MWMMapOverlayManager setIsoLinesEnabled:NO];
      break;
  }
}

- (void)applyTheme {
  MWMButton *btn = static_cast<MWMButton *>(self.view);
  UIImageView *iv = btn.imageView;

  // Traffic state machine: https://confluence.mail.ru/pages/viewpage.action?pageId=103680959
  [iv stopAnimating];
  if ([MWMMapOverlayManager trafficEnabled]) {
    [self handleTrafficState:[MWMMapOverlayManager trafficState]];
  } else if ([MWMMapOverlayManager transitEnabled]) {
    btn.imageName = @"btn_subway_on";
    if ([MWMMapOverlayManager transitState] == MWMMapOverlayTransitStateNoData)
      [Toast showWithText:L(@"subway_data_unavailable")];
  } else if ([MWMMapOverlayManager isoLinesEnabled]) {
    btn.imageName = @"btn_isoMap_on";
    [self handleIsolinesState:[MWMMapOverlayManager isolinesState]];
  } else if ([MWMMapOverlayManager outdoorEnabled]) {
    btn.imageName = @"btn_isoMap_on";
  } else {
    btn.imageName = @"btn_layers";
  }
}

- (IBAction)buttonTouchUpInside {
  BOOL needsToDisableMapLayer =
  [MWMMapOverlayManager trafficEnabled] ||
  [MWMMapOverlayManager transitEnabled] ||
  [MWMMapOverlayManager isoLinesEnabled] ||
  [MWMMapOverlayManager outdoorEnabled];

  if (needsToDisableMapLayer) {
    [MWMMapOverlayManager setTrafficEnabled:NO];
    [MWMMapOverlayManager setTransitEnabled:NO];
    [MWMMapOverlayManager setIsoLinesEnabled:NO];
    [MWMMapOverlayManager setOutdoorEnabled:NO];
  } else {
    MWMMapViewControlsManager.manager.menuState = MWMBottomMenuStateLayers;
  }
}

+ (void)updateAvailableArea:(CGRect)frame {
  auto controller = [self controller];
  if (CGRectEqualToRect(controller.availableArea, frame))
    return;
  controller.availableArea = frame;
  BOOL isHidden = frame.origin.y + frame.size.height < controller.view.origin.y + controller.view.height + kTopOffset;
  [MapViewController.sharedController.controlsManager setTrafficButtonHidden:isHidden];
  [controller refreshLayout];
}

#pragma mark - MWMMapOverlayManagerObserver

- (void)onTrafficStateUpdated {
  [self applyTheme];
}
- (void)onTransitStateUpdated {
  [self applyTheme];
}
- (void)onIsoLinesStateUpdated {
  [self applyTheme];
}
- (void)onOutdoorStateUpdated {
  [self applyTheme];
}

@end