//
//  MPContextToolbarButton.m
//  MacPass
//
//  Created by Michael Starke on 26.07.13.
//  Copyright (c) 2013 HicknHack Software GmbH. All rights reserved.
//
//  This program is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  This program is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with this program.  If not, see <http://www.gnu.org/licenses/>.
//

#import "MPContextButton.h"
#import "MPSegmentedContextCell.h"

@interface MPContextButton ()

@end

@implementation MPContextButton

- (instancetype)initWithCoder:(NSCoder *)coder {
  self = [super initWithCoder:coder];
  if(self) {
    [self _setup];
  }
  return self;
}

- (instancetype)initWithFrame:(NSRect)frame {
  self = [super initWithFrame:frame];
  if (self) {
    [self _setup];
  }
  return self;
}

- (void)_setup {
  NSData *data = [NSKeyedArchiver archivedDataWithRootObject:[self cell]];
  NSKeyedUnarchiver *unarchiver = [[NSKeyedUnarchiver alloc] initForReadingWithData:data];
  MPSegmentedContextCell *cell = [[MPSegmentedContextCell alloc] initWithCoder:unarchiver];
  self.cell = cell;
  
  self.focusRingType = NSFocusRingTypeNone;
  self.segmentStyle = NSSegmentStyleTexturedSquare;
  self.segmentCount = MPContextButtonSegmentCount;
  cell.trackingMode = NSSegmentSwitchTrackingMomentary;
  [cell setWidth:31 forSegment:MPContextButtonSegmentButton];
  [cell setWidth:17 forSegment:MPContextButtonSegmentContextButton];
  cell.trackingMode = NSSegmentSwitchTrackingMomentary;
  
  NSImage *contextTriangle = [NSBundle.mainBundle imageForResource:@"contextTriangleTemplate"];
  [self setImage:contextTriangle forSegment:MPContextButtonSegmentContextButton];
  
  cell.contextMenuAction = @selector(showContextMenu:);
  cell.contextMenuTarget = self;
}

- (void)setContextMenu:(NSMenu *)menu {
  if(_contextMenu != menu) {
    _contextMenu = menu;
  }
}

/*
 Block the segment setter to prevent accidental settings
 */
- (void)setImage:(NSImage *)image forSegment:(NSInteger)segment {
  if(segment < MPContextButtonSegmentCount) {
    [super setImage:image forSegment:segment];
  }
}

- (void)setSegmentCount:(NSInteger)count {
  if(count == MPContextButtonSegmentCount) {
    super.segmentCount = count;
  }
}

- (void)setImage:(NSImage *)image {
  [self setImage:image forSegment:MPContextButtonSegmentButton];
}

- (void)showContextMenu:(id)sender {
  NSPoint point = self.frame.origin;
  point.x = [self.cell widthForSegment:MPContextButtonSegmentButton];
  point.y = NSHeight(self.frame) + 3;
  [_contextMenu popUpMenuPositioningItem:nil atLocation:point inView:self];
}

- (void)setControlSize:(NSControlSize)controlSize {
  NSImageRep *rep = [[self imageForSegment:MPContextButtonSegmentButton] bestRepresentationForRect:NSMakeRect(0, 0, 100, 100) context:nil hints:nil];
  CGFloat scale = rep.size.width / rep.size.height;
  switch (controlSize) {
    case NSRegularControlSize:
      [self imageForSegment:MPContextButtonSegmentButton].size = NSMakeSize(16 * scale, 16);
      break;
      
    case NSSmallControlSize:
      [self imageForSegment:MPContextButtonSegmentButton].size = NSMakeSize(14 * scale, 14);
      break;
      
    case NSMiniControlSize:
      [self imageForSegment:MPContextButtonSegmentButton].size = NSMakeSize(8 * scale, 8);
      
    default:
      break;
  }
  super.controlSize = controlSize;
}

- (NSControlSize)controlSize {
  return super.controlSize;
}

- (void)_updateContextButtonState {
  BOOL hasContextMenu = (self.contextMenu != nil);
  [self setEnabled:hasContextMenu forSegment:MPContextButtonSegmentContextButton];
}

@end