mirror of
https://github.com/MacPass/MacPass.git
synced 2025-12-14 09:22:33 +00:00
350 lines
13 KiB
Objective-C
350 lines
13 KiB
Objective-C
//
|
|
// MPOutlineViewController.m
|
|
// MacPass
|
|
//
|
|
// Created by michael starke on 19.02.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 "MPOutlineViewController.h"
|
|
#import "MPActionHelper.h"
|
|
#import "MPConstants.h"
|
|
#import "MPContextMenuHelper.h"
|
|
#import "MPDocument.h"
|
|
#import "MPDocumentWindowController.h"
|
|
#import "MPIconHelper.h"
|
|
#import "MPNotifications.h"
|
|
#import "MPOutlineContextMenuDelegate.h"
|
|
#import "MPOutlineDataSource.h"
|
|
|
|
#import "KeePassKit/KeePassKit.h"
|
|
#import "KPKNode+IconImage.h"
|
|
|
|
#import "HNHUi/HNHUi.h"
|
|
|
|
#define EXPIRED_GROUP_REFRESH_SECONDS 60
|
|
|
|
NSString *const MPOutlineViewDidChangeGroupSelection = @"com.macpass.MPOutlineViewDidChangeGroupSelection";
|
|
|
|
NSString *const _MPOutlineViewDataViewIdentifier = @"DataCell";
|
|
NSString *const _MPOutlinveViewHeaderViewIdentifier = @"HeaderCell";
|
|
|
|
@interface MPOutlineViewController () {
|
|
BOOL _bindingEstablished;
|
|
MPOutlineContextMenuDelegate *_menuDelegate;
|
|
|
|
}
|
|
@property (weak) IBOutlet NSOutlineView *outlineView;
|
|
|
|
@property (strong) NSTreeController *treeController;
|
|
@property (strong) MPOutlineDataSource *datasource;
|
|
|
|
@property (copy, nonatomic) NSString *databaseNameWrapper;
|
|
|
|
@end
|
|
|
|
@implementation MPOutlineViewController
|
|
|
|
- (NSString *)nibName {
|
|
return @"OutlineView";
|
|
}
|
|
|
|
- (id)initWithNibName:(NSString *)nibNameOrNil bundle:(NSBundle *)nibBundleOrNil {
|
|
self = [super initWithNibName:nibNameOrNil bundle:nibBundleOrNil];
|
|
if (self) {
|
|
_treeController = [[NSTreeController alloc] init];
|
|
_bindingEstablished = NO;
|
|
_datasource = [[MPOutlineDataSource alloc] init];
|
|
_databaseNameWrapper = NSLocalizedString(@"NEW_DATABASE", "Name for a newly created Database");
|
|
_menuDelegate = [[MPOutlineContextMenuDelegate alloc] init];
|
|
_menuDelegate.viewController = self;
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
- (void)dealloc {
|
|
[self.outlineView unbind:NSContentBinding];
|
|
[self.treeController unbind:NSContentBinding];
|
|
[NSNotificationCenter.defaultCenter removeObserver:self];
|
|
[self.outlineView setDelegate:nil];
|
|
}
|
|
|
|
- (void)viewDidLoad {
|
|
self.outlineView.menu = [self _contextMenu];
|
|
self.outlineView.allowsEmptySelection = YES;
|
|
self.outlineView.floatsGroupRows = NO;
|
|
self.outlineView.doubleAction = @selector(_doubleClickedGroup:);
|
|
self.outlineView.allowsMultipleSelection = YES;
|
|
self.outlineView.delegate = self;
|
|
[self.outlineView registerForDraggedTypes:@[ KPKGroupUTI, KPKEntryUTI ]];
|
|
[self.outlineView setDraggingSourceOperationMask:NSDragOperationEvery forLocal:YES];
|
|
|
|
[NSNotificationCenter.defaultCenter addObserver:self
|
|
selector:@selector(_didBecomeFirstResponder:)
|
|
name:MPDidActivateViewNotification
|
|
object:self.outlineView];
|
|
|
|
|
|
NSView *clipView = self.outlineView.enclosingScrollView.contentView;
|
|
[NSNotificationCenter.defaultCenter addObserver:self
|
|
selector:@selector(_outlineDidScroll:)
|
|
name:NSViewBoundsDidChangeNotification
|
|
object:clipView];
|
|
|
|
}
|
|
|
|
- (NSResponder *)reconmendedFirstResponder {
|
|
return self.outlineView;
|
|
}
|
|
|
|
#pragma mark Outline handling
|
|
|
|
- (void)showOutline {
|
|
if(!_bindingEstablished) {
|
|
MPDocument *document = self.windowController.document;
|
|
self.treeController.childrenKeyPath = NSStringFromSelector(@selector(groups));
|
|
[self.treeController bind:NSContentBinding toObject:document withKeyPath:NSStringFromSelector(@selector(tree)) options:nil];
|
|
[self.outlineView bind:NSContentBinding toObject:self.treeController withKeyPath:NSStringFromSelector(@selector(arrangedObjects)) options:nil];
|
|
[self.outlineView bind:NSSelectionIndexPathsBinding toObject:self.treeController withKeyPath:NSStringFromSelector(@selector(selectionIndexPaths)) options:nil];
|
|
[self bind:NSStringFromSelector(@selector(databaseNameWrapper)) toObject:document.tree.metaData withKeyPath:NSStringFromSelector(@selector(databaseName)) options:nil];
|
|
self.outlineView.dataSource = self.datasource;
|
|
_bindingEstablished = YES;
|
|
}
|
|
NSTreeNode *node = [_outlineView itemAtRow:0];
|
|
NSInteger topRow = 0;
|
|
[self _expandItems:node topRow:&topRow];
|
|
if(topRow > 0) {
|
|
NSRect rowRect = [self.outlineView rectOfRow:topRow];
|
|
[self.outlineView scrollPoint:rowRect.origin];
|
|
}
|
|
}
|
|
|
|
- (void)_expandItems:(NSTreeNode *)node topRow:(NSInteger *)topRow {
|
|
NSAssert(NULL != topRow, @"Invalid paramter!");
|
|
id nodeItem = node.representedObject;
|
|
if([nodeItem isKindOfClass:[KPKTree class]]) {
|
|
[self.outlineView expandItem:node expandChildren:NO];
|
|
*topRow = -1;
|
|
}
|
|
else if([nodeItem respondsToSelector:@selector(isExpanded)]) {
|
|
if([nodeItem isExpanded]) {
|
|
[self.outlineView expandItem:node];
|
|
}
|
|
else {
|
|
[self.outlineView collapseItem:node];
|
|
}
|
|
}
|
|
for(NSTreeNode *child in node.childNodes) {
|
|
[self _expandItems:child topRow:topRow];
|
|
}
|
|
if([nodeItem respondsToSelector:@selector(uuid)]) {
|
|
MPDocument *document = self.windowController.document;
|
|
NSUUID *uuid = [nodeItem uuid];
|
|
if(*topRow != 1 && [document.tree.metaData.lastTopVisibleGroup isEqual:uuid]) {
|
|
*topRow = [self.outlineView rowForItem:node];
|
|
}
|
|
if([uuid isEqual:document.tree.metaData.lastSelectedGroup]) {
|
|
NSInteger selectedRow = [self.outlineView rowForItem:node];
|
|
if(selectedRow >= 0) {
|
|
[self.outlineView selectRowIndexes:[NSIndexSet indexSetWithIndex:selectedRow] byExtendingSelection:NO];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#pragma mark Custom Setter/Getter
|
|
- (void)setDatabaseNameWrapper:(NSString *)databaseNameWrapper {
|
|
if(![_databaseNameWrapper isEqualToString:databaseNameWrapper]) {
|
|
_databaseNameWrapper = (databaseNameWrapper.length == 0) ? NSLocalizedString(@"DATABASE", "Default display name for KDB databases") : [databaseNameWrapper copy];
|
|
}
|
|
}
|
|
|
|
#pragma mark MPTargetNodeResolving
|
|
- (NSArray<KPKGroup *> *)currentTargetGroups {
|
|
/* there are instances where we get the KPKTree tree as selection, so filter this out! */
|
|
if([self.treeController.selectedObjects.firstObject isKindOfClass:[KPKTree class]] ) {
|
|
return @[];
|
|
}
|
|
return self.treeController.selectedObjects;
|
|
}
|
|
|
|
- (NSArray<KPKNode *> *)currentTargetNodes {
|
|
NSArray *groups = [self currentTargetGroups];
|
|
if(groups.count > 0) {
|
|
return groups;
|
|
}
|
|
MPDocument *document = self.windowController.document;
|
|
return document.selectedNodes;
|
|
}
|
|
|
|
#pragma mark Notifications
|
|
- (void)registerNotificationsForDocument:(MPDocument *)document {
|
|
[NSNotificationCenter.defaultCenter addObserver:self selector:@selector(_didAddGroup:) name:MPDocumentDidAddGroupNotification object:document];
|
|
}
|
|
|
|
- (void)clearSelection {
|
|
[self.outlineView deselectAll:nil];
|
|
NSNotification *notification = [NSNotification notificationWithName:NSOutlineViewSelectionDidChangeNotification object:self.outlineView];
|
|
[self outlineViewSelectionDidChange:notification];
|
|
}
|
|
|
|
- (void)_didBecomeFirstResponder:(NSNotification *)notification {
|
|
if( notification.object != self.outlineView ) {
|
|
return; // Nothing we need to worry about
|
|
}
|
|
MPDocument *document = self.windowController.document;
|
|
document.selectedGroups = [self currentTargetGroups];
|
|
}
|
|
|
|
- (void)_outlineDidScroll:(NSNotification *)notification {
|
|
NSView *clipView = notification.object;
|
|
if(nil == clipView || self.outlineView.enclosingScrollView.contentView != clipView) {
|
|
return; // Wrong view
|
|
}
|
|
/* padding to compensate for clipped items */
|
|
CGPoint point = CGPointMake(clipView.bounds.origin.x, clipView.bounds.origin.y + 11);
|
|
NSInteger topRow = [self.outlineView rowAtPoint:point];
|
|
id item = [[self.outlineView itemAtRow:topRow] representedObject];
|
|
if([item isKindOfClass:[KPKGroup class]]) {
|
|
KPKGroup *group = item;
|
|
MPDocument *document = self.windowController.document;
|
|
document.tree.metaData.lastTopVisibleGroup = group.uuid;
|
|
}
|
|
}
|
|
|
|
# pragma mark MPDocument Notifications
|
|
- (void)_didAddGroup:(NSNotification *)notification {
|
|
NSDictionary *userInfo = [notification userInfo];
|
|
KPKGroup *group = userInfo[MPDocumentGroupKey];
|
|
NSIndexPath *groupIndexPath = [group indexPath];
|
|
NSTreeNode *groupNode = [[self.treeController arrangedObjects] descendantNodeAtIndexPath:groupIndexPath];
|
|
[self.outlineView expandItem:groupNode.parentNode];
|
|
NSInteger groupRow = [self.outlineView rowForItem:groupNode];
|
|
[self.outlineView selectRowIndexes:[NSIndexSet indexSetWithIndex:groupRow] byExtendingSelection:NO];
|
|
[self.outlineView scrollRowToVisible:groupRow];
|
|
}
|
|
|
|
- (id)itemUnderMouse {
|
|
NSPoint mouseLocation = [self.outlineView.window mouseLocationOutsideOfEventStream];
|
|
NSPoint localPoint = [self.outlineView convertPoint:mouseLocation fromView:self.outlineView.window.contentView];
|
|
NSInteger row = [self.outlineView rowAtPoint:localPoint];
|
|
if(row == -1) {
|
|
return nil; // No row was hit
|
|
}
|
|
return [[self.outlineView itemAtRow:row] representedObject];
|
|
}
|
|
|
|
#pragma mark -
|
|
#pragma mark Actions
|
|
|
|
- (void)_doubleClickedGroup:(id)sender {
|
|
[(MPDocumentWindowController *)self.windowController showInspector:sender];
|
|
}
|
|
|
|
#pragma mark NSOutlineViewDelegate
|
|
- (NSView *)outlineView:(NSOutlineView *)outlineView viewForTableColumn:(NSTableColumn *)tableColumn item:(id)item {
|
|
NSTableCellView *view;
|
|
if( [self _itemIsRootNode:item] ) {
|
|
view = [outlineView makeViewWithIdentifier:_MPOutlinveViewHeaderViewIdentifier owner:self];
|
|
[view.textField bind:NSValueBinding toObject:self withKeyPath:NSStringFromSelector(@selector(databaseNameWrapper)) options:nil];
|
|
}
|
|
else {
|
|
view = [outlineView makeViewWithIdentifier:_MPOutlineViewDataViewIdentifier owner:self];
|
|
|
|
NSString *iconImageKeyPath = [NSString stringWithFormat:@"%@.%@", NSStringFromSelector(@selector(representedObject)), NSStringFromSelector(@selector(iconImage))];
|
|
NSString *titleKeyPath = [NSString stringWithFormat:@"%@.%@", NSStringFromSelector(@selector(representedObject)), NSStringFromSelector(@selector(title))];
|
|
[view.imageView bind:NSValueBinding toObject:item withKeyPath:iconImageKeyPath options:nil];
|
|
[view.textField bind:NSValueBinding toObject:item withKeyPath:titleKeyPath options:nil];
|
|
|
|
|
|
NSString *entriesCountKeyPath = [[NSString alloc] initWithFormat:@"%@.%@.%@", NSStringFromSelector(@selector(representedObject)), KPKEntriesArrayBinding, @"@count"];
|
|
[view.textField bind:NSStringFromSelector(@selector(count)) toObject:item withKeyPath:entriesCountKeyPath options:nil];
|
|
}
|
|
|
|
return view;
|
|
}
|
|
|
|
- (BOOL)outlineView:(NSOutlineView *)outlineView isGroupItem:(id)item {
|
|
return [self _itemIsRootNode:item];
|
|
}
|
|
|
|
- (BOOL)outlineView:(NSOutlineView *)outlineView shouldSelectItem:(id)item {
|
|
return ![self _itemIsRootNode:item];
|
|
}
|
|
|
|
- (void)outlineViewSelectionDidChange:(NSNotification *)notification {
|
|
MPDocument *document = self.windowController.document;
|
|
NSArray<KPKGroup *> *groups = [self currentTargetGroups];
|
|
document.tree.metaData.lastSelectedGroup = (groups.count == 1 ? groups.firstObject.uuid : [NSUUID kpk_nullUUID]);
|
|
document.selectedGroups = groups;
|
|
}
|
|
|
|
- (BOOL)outlineView:(NSOutlineView *)outlineView shouldShowOutlineCellForItem:(id)item {
|
|
return ![self _itemIsRootNode:item];
|
|
}
|
|
|
|
- (void)outlineViewItemDidExpand:(NSNotification *)notification {
|
|
NSDictionary *userInfo = [notification userInfo];
|
|
id item = userInfo[NSStringFromClass([NSObject class])];
|
|
id representedObject = [item representedObject];
|
|
if([representedObject isKindOfClass:[KPKGroup class]]) {
|
|
KPKGroup *group = (KPKGroup *)representedObject;
|
|
group.isExpanded = YES;
|
|
}
|
|
}
|
|
- (void)outlineViewItemDidCollapse:(NSNotification *)notification {
|
|
NSDictionary *userInfo = notification.userInfo;
|
|
id item = userInfo[NSStringFromClass([NSObject class])];
|
|
id representedObject = [item representedObject];
|
|
if([representedObject isKindOfClass:[KPKGroup class]]) {
|
|
KPKGroup *group = (KPKGroup *)representedObject;
|
|
group.isExpanded = NO;
|
|
}
|
|
}
|
|
|
|
- (void)outlineView:(NSOutlineView *)outlineView didRemoveRowView:(NSTableRowView *)rowView forRow:(NSInteger)row {
|
|
/* Deletion of an item */
|
|
if(row == -1) {
|
|
NSNotification *notification = [NSNotification notificationWithName:NSOutlineViewSelectionDidChangeNotification object:outlineView];
|
|
[self outlineViewSelectionDidChange:notification];
|
|
}
|
|
}
|
|
|
|
#pragma mark Validation
|
|
- (BOOL)validateMenuItem:(NSMenuItem *)menuItem {
|
|
MPDocument *document = self.windowController.document;
|
|
if(![document validateUserInterfaceItem:menuItem]) {
|
|
return NO;
|
|
}
|
|
KPKGroup *group = [self currentTargetNodes].firstObject.asGroup;
|
|
return group.isTrash && group.isTrashed;
|
|
}
|
|
|
|
- (NSMenu *)_contextMenu {
|
|
NSMenu *menu = [[NSMenu alloc] init];
|
|
menu.delegate = _menuDelegate;
|
|
return menu;
|
|
}
|
|
|
|
- (BOOL)_itemIsRootNode:(id)item {
|
|
id node = [item representedObject];
|
|
return [node isKindOfClass:[KPKTree class]];
|
|
}
|
|
|
|
@end
|