MacPass/MacPass/MPOutlineDataSource.m

//
//  MPOutlineDataSource.m
//  MacPass
//
//  Created by Michael Starke on 19.07.12.
//  Copyright (c) 2012 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 "MPOutlineDataSource.h"
#import "MPDocument.h"
#import "MPConstants.h"

#import "KeePassKit/KeePassKit.h"

@interface MPOutlineDataSource ()

@property (weak) KPKGroup *localDraggedGroup;
@property (weak) KPKEntry *localDraggedEntry;

@property (strong) NSArray<KPKGroup *> *draggedGroups;
@property (strong) NSArray<KPKEntry *> *draggedEntries;

@end

@implementation MPOutlineDataSource

- (id<NSPasteboardWriting>)outlineView:(NSOutlineView *)outlineView pasteboardWriterForItem:(id)item {
  id representedObject = [item representedObject];
  if([representedObject isKindOfClass:KPKGroup.class]) {
    KPKGroup *group = representedObject;
    return group;
  }
  return nil;
}

- (NSDragOperation)outlineView:(NSOutlineView *)outlineView validateDrop:(id<NSDraggingInfo>)info proposedItem:(id)item proposedChildIndex:(NSInteger)index {
  
  /* Clean up our local search */
  self.localDraggedEntry = nil;
  self.localDraggedGroup = nil;
  
  info.animatesToDestination = YES;
  NSDragOperation operationMask = NSDragOperationMove;
  /*
   If we can support copy on drag, this can be used
   to obtain the dragging modifier mask the user presses
   */
  BOOL localCopy = NO;
  if([info draggingSourceOperationMask] == NSDragOperationCopy) {
    operationMask = NSDragOperationCopy;
    localCopy = YES;
  }
  
  
  /* Check if the Target is the root group */
  id targetItem = [item representedObject];
  if( ![targetItem isKindOfClass:KPKGroup.class] ) {
    return NSDragOperationNone; // Block all unknown types
  }
  
  NSPasteboard *pasteBoard = [info draggingPasteboard];
  KPKGroup *draggedGroup = nil;
  KPKEntry *draggedEntry = nil;
  BOOL couldReadPasteboard = [self _readDataFromPasteboard:pasteBoard group:&draggedGroup entry:&draggedEntry];
  if(!couldReadPasteboard) {
    return NSDragOperationNone;
  }
  
  KPKGroup *targetGroup = targetItem;
  BOOL validTarget = YES;
  MPDocument *document = outlineView.window.windowController.document;
  /* Dragging Groups */
  if(draggedGroup) {
    self.localDraggedGroup = [document findGroup:draggedGroup.uuid];
    if( [draggedGroup.uuid isEqual:targetGroup.uuid] ) {
      return NSDragOperationNone; // Groups cannot be moved inside themselves
    }
    if(self.localDraggedGroup) {
      if( self.localDraggedGroup.parent == targetGroup ) {
        validTarget &= index != NSOutlineViewDropOnItemIndex;
        validTarget &= index != [self.localDraggedGroup.parent.groups indexOfObject:self.localDraggedGroup];
      }
      BOOL isAnchesor = [self.localDraggedGroup isAnchestorOf:targetGroup];
      validTarget &= !isAnchesor;
    }
    else {
      /* Copy can always work in this case */
      operationMask = NSDragOperationCopy;
    }
  }
  else if(draggedEntry) {
    self.localDraggedEntry = [document findEntry:draggedEntry.uuid];
    if(self.localDraggedEntry) {
      /* local Copy is always valid regardless of parent */
      validTarget = localCopy ? YES : self.localDraggedEntry.parent != targetGroup;
      [outlineView setDropItem:item dropChildIndex:NSOutlineViewDropOnItemIndex];
    }
    else {
      /* Entry copy is always valid */
      operationMask = NSDragOperationCopy;
    }
  }
  return validTarget ? operationMask : NSDragOperationNone;
}

- (BOOL)outlineView:(NSOutlineView *)outlineView acceptDrop:(id<NSDraggingInfo>)info item:(id)item childIndex:(NSInteger)index {
  info.animatesToDestination = YES;
  
  id targetItem = [item representedObject];
  if(![targetItem isKindOfClass:KPKGroup.class]) {
    return NO; // Wrong
  }
  
  NSPasteboard *pasteBoard = [info draggingPasteboard];
  KPKGroup *draggedGroup = nil;
  KPKEntry *draggedEntry = nil;
  BOOL validPateboard = [self _readDataFromPasteboard:pasteBoard group:&draggedGroup entry:&draggedEntry];
  if(!validPateboard) {
    return NO;
  }
  
  BOOL copyItem = ([info draggingSourceOperationMask] == NSDragOperationCopy);
  
  KPKGroup *targetGroup = (KPKGroup *)targetItem;
  if(draggedGroup) {
    if(copyItem || (nil == self.localDraggedGroup) ) {
      draggedGroup = [draggedGroup copyWithTitle:nil options:kKPKCopyOptionNone];
      [draggedGroup addToGroup:targetGroup atIndex:index];
      [draggedGroup.undoManager setActionName:NSLocalizedString(@"COPY_GROUP", "Action title for copying a group via drag and drop")];
      return YES;
    }
    else if(self.localDraggedGroup) {
      /* Simple move */
      [self.localDraggedGroup moveToGroup:targetGroup atIndex:index];
      [self.localDraggedGroup.undoManager setActionName:NSLocalizedString(@"MOVE_GROUP", "Action title for moving a group via drag and drop")];
      return YES;
    }
    /* Nothing valid */
    return NO;
  }
  else if(draggedEntry) {
    if(copyItem || (nil == self.localDraggedEntry)) {
      draggedEntry = [draggedEntry copyWithTitle:nil options:kKPKCopyOptionNone];
      [draggedEntry addToGroup:targetGroup];
      [draggedEntry.undoManager setActionName:NSLocalizedString(@"COPY_ENTRY", "Action title for copying an entry via drag and drop")];
      return YES;
    }
    else if(self.localDraggedEntry) {
      [self.localDraggedEntry moveToGroup:targetGroup];
      [self.localDraggedEntry.undoManager setActionName:NSLocalizedString(@"MOVE_ENTRY", "Action title for moving an entry via drag and drop")];
      return YES;
    }
  }
  return NO;
}

- (BOOL)_readDataFromPasteboard:(NSPasteboard *)pasteboard group:(KPKGroup **)group entry:(KPKEntry **)entry;{
  
  if(entry == NULL || group == NULL) {
    return NO; // Need valid pointers
  }
  /* Cleanup old stuff */
  
  NSArray *types = pasteboard.types;
  if(types.count > 1 || types.count == 0) {
    return NO;
  }
  
  NSString *draggedType = types.lastObject;
  if([draggedType isEqualToString:KPKGroupUTI]) {
    // dragging group
    NSArray *groups = [pasteboard readObjectsForClasses:@[KPKGroup.class] options:nil];
    if(groups.count != 1) {
      return NO;
    }
    *group = groups.lastObject;
    return YES;
  }
  else if([draggedType isEqualToString:KPKEntryUTI]) {
    NSArray *entries = [pasteboard readObjectsForClasses:@[KPKEntry.class] options:nil];
    if([entries count] != 1) {
      return NO; // NO entry readable
    }
    *entry = entries.lastObject;
    return YES;
  }
  return NO;
}

@end