CollectionViewController.m revision 960ee46ac6eaff8d3a1638319e95a4f1047ff5f3
//
// FreeOTP
//
// Authors: Nathaniel McCallum <npmccallum@redhat.com>
//
// Copyright (C) 2014 Nathaniel McCallum, Red Hat
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
#import "CollectionViewController.h"
#import "AddTokenViewController.h"
#import "QRCodeScanViewController.h"
#import "RenameTokenViewController.h"
#import "TokenImagePickerController.h"
#import "BlockActionSheet.h"
#import "TokenCell.h"
#import "TokenStore.h"
@interface CollectionViewController () <UICollectionViewDataSource, UICollectionViewDelegate, UIPopoverControllerDelegate>
@property (nonatomic, strong) UIPopoverController* popover;
@end
@implementation CollectionViewController
{
TokenStore* store;
NSIndexPath* lastPath;
UILongPressGestureRecognizer* longPressGesture;
}
- (NSInteger)numberOfSectionsInCollectionView:(UICollectionView *)collectionView {
return 1;
}
- (NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSection:(NSInteger)section {
return store.count;
}
- (CGSize)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout sizeForItemAtIndexPath:(NSIndexPath *)indexPath
{
switch ((int) collectionView.frame.size.width) {
case 1024: // iPad
case 768: // iPad
return CGSizeMake(328, 96);
case 568: // iPhone5 landscape
case 320: // iPhone* portrait
return CGSizeMake(269, 80);
case 480: // iPhone4 landscape
default:
return CGSizeMake(225, 64);
}
}
- (UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cellForItemAtIndexPath:(NSIndexPath *)indexPath {
NSString* name = nil;
switch ((int) collectionView.frame.size.width) {
case 1024: // iPad
case 768: // iPad
name = @"iPad";
break;
case 568: // iPhone5 landscape
case 320: // iPhone* portrait
name = @"iPhone5";
break;
case 480: // iPhone4 landscape
default:
name = @"iPhone4";
break;
}
TokenCell* cell = [collectionView dequeueReusableCellWithReuseIdentifier:name forIndexPath:indexPath];
return [cell bind:[store get:indexPath.row]] ? cell : nil;
}
- (void)willRotateToInterfaceOrientation:(UIInterfaceOrientation)toInterfaceOrientation duration:(NSTimeInterval)duration
{
// If the device is smaller than an iPhone5,
// then reload the data to pick up the new cell size.
// This is unfortunate because it resets token UI state.
// However, this works until we get completely dynamic resizing.
if ([[UIScreen mainScreen] bounds].size.height < 568)
[self.collectionView reloadData];
}
- (void)collectionView:(UICollectionView *)collectionView didSelectItemAtIndexPath:(NSIndexPath *)indexPath
{
// Perform animation.
[collectionView deselectItemAtIndexPath:indexPath animated:YES];
// If we are not in edit mode, generate the token.
if (self.navigationItem.leftBarButtonItem.style == UIBarButtonItemStylePlain) {
// Get the current cell.
TokenCell* cell = (TokenCell*)[collectionView cellForItemAtIndexPath:indexPath];
if (cell == nil)
return;
// Get the selected token.
Token* token = [store get:indexPath.row];
if (token == nil)
return;
// Get the token code and save the token state.
TokenCode* tc = token.code;
[store save:token];
// Show the token code.
cell.state = tc;
// Copy the token code to the clipboard.
NSString* code = tc.currentCode;
if (code != nil)
[[UIPasteboard generalPasteboard] setString:code];
return;
}
// Get the cell.
TokenCell* cell = (TokenCell*)[self.collectionView cellForItemAtIndexPath:indexPath];
if (cell == nil)
return;
// Create the action sheet.
BlockActionSheet* as = [[BlockActionSheet alloc] init];
// On iPads, the sheet points to the token.
// Otherwise, add a title to make the context clear.
if (UI_USER_INTERFACE_IDIOM() != UIUserInterfaceIdiomPad)
// Add the remaining buttons.
[as addButtonWithTitle:NSLocalizedString(@"Change Icon", nil)];
[as addButtonWithTitle:NSLocalizedString(@"Rename", nil)];
as.destructiveButtonIndex = [as addButtonWithTitle:NSLocalizedString(@"Delete", nil)];
as.cancelButtonIndex = [as addButtonWithTitle:NSLocalizedString(@"Cancel", nil)];
[as showFromRect:cell.frame inView:self.collectionView animated:YES];
as.callback = ^(NSInteger offset) {
switch (offset) {
case 1: { // Delete
BlockActionSheet* as = [[BlockActionSheet alloc] init];
as.title = NSLocalizedString(@"Are you sure?", nil);
if (UI_USER_INTERFACE_IDIOM() != UIUserInterfaceIdiomPad)
as.title = [NSString stringWithFormat:@"%@\n\n%@\n%@", as.title, cell.issuer.text, cell.label.text];
as.destructiveButtonIndex = [as addButtonWithTitle:NSLocalizedString(@"Delete", nil)];
as.cancelButtonIndex = [as addButtonWithTitle:NSLocalizedString(@"Cancel", nil)];
[as showFromRect:cell.frame inView:self.collectionView animated:YES];
as.callback = ^(NSInteger offset) {
if (offset != 1)
return;
[[[TokenStore alloc] init] del:indexPath.row];
[self.collectionView deleteItemsAtIndexPaths:@[indexPath]];
};
break;
}
case 2: { // Rename
RenameTokenViewController* c = [self.storyboard instantiateViewControllerWithIdentifier:@"renameToken"];
UINavigationController* nc = [[UINavigationController alloc] initWithRootViewController:c];
if (UI_USER_INTERFACE_IDIOM() != UIUserInterfaceIdiomPad) {
[self presentViewController:nc animated:YES completion:nil];
} else {
c.popover = self.popover = [[UIPopoverController alloc] initWithContentViewController:nc];
self.popover.delegate = self;
self.popover.popoverContentSize = CGSizeMake(320, 375);
[self.popover presentPopoverFromRect:cell.frame inView:self.collectionView permittedArrowDirections:UIPopoverArrowDirectionAny animated:YES];
}
break;
}
case 3: { // Change Icon
TokenImagePickerController* ipc = [[TokenImagePickerController alloc] initWithTokenID:indexPath.row];
if (UI_USER_INTERFACE_IDIOM() != UIUserInterfaceIdiomPad) {
[self presentViewController:ipc animated:YES completion:nil];
} else {
ipc.popover = self.popover = [[UIPopoverController alloc] initWithContentViewController:ipc];
self.popover.delegate = self;
[self.popover presentPopoverFromRect:cell.frame inView:self.collectionView permittedArrowDirections:UIPopoverArrowDirectionAny animated:YES];
}
}
}
};
}
- (void)handleLongPress:(UILongPressGestureRecognizer *)gestureRecognizer
{
UICollectionViewCell* cell = nil;
// Get the current index path.
CGPoint p = [gestureRecognizer locationInView:self.collectionView];
NSIndexPath *currPath = [self.collectionView indexPathForItemAtPoint:p];
switch (gestureRecognizer.state) {
case UIGestureRecognizerStateBegan:
if (currPath != nil)
cell = [self.collectionView cellForItemAtIndexPath:currPath];
if (cell == nil)
return; // Invalid state
lastPath = currPath;
// Animate to the "lifted" state.
cell = [self.collectionView cellForItemAtIndexPath:currPath];
{[UIView animateWithDuration:0.3f animations:^{
cell.transform = CGAffineTransformMakeScale(1.1f, 1.1f);
[self.collectionView bringSubviewToFront:cell];
}];}
return;
case UIGestureRecognizerStateChanged:
if (lastPath != nil)
cell = [self.collectionView cellForItemAtIndexPath:lastPath];
if (cell == nil)
return; // Invalid state
if (currPath != nil && lastPath.row != currPath.row) {
// Move the display.
[self.collectionView moveItemAtIndexPath:lastPath toIndexPath:currPath];
// Scroll the display to handle moving tokens up or down.
if (lastPath.row < currPath.row)
[self.collectionView scrollToItemAtIndexPath:currPath atScrollPosition:UICollectionViewScrollPositionTop animated:YES];
else
[self.collectionView scrollToItemAtIndexPath:currPath atScrollPosition:UICollectionViewScrollPositionBottom animated:YES];
// Write changes.
TokenStore* ts = [[TokenStore alloc] init];
[ts moveFrom:lastPath.row to:currPath.row];
// Reset state.
cell.transform = CGAffineTransformMakeScale(1.1f, 1.1f); // Moving the token resets the size...
[self.collectionView bringSubviewToFront:cell]; // ... and Z index.
lastPath = currPath;
}
cell.center = [gestureRecognizer locationInView:self.collectionView];
return;
case UIGestureRecognizerStateEnded:
// Animate back to the original state, but in the new location.
if (lastPath != nil) {
cell = [self.collectionView cellForItemAtIndexPath:lastPath];
{[UIView animateWithDuration:0.3f animations:^{
UICollectionViewLayout* l = self.collectionView.collectionViewLayout;
cell.center = [l layoutAttributesForItemAtIndexPath:lastPath].center;
cell.transform = CGAffineTransformMakeScale(1.0f, 1.0f);
} completion:^(BOOL c){
lastPath = nil;
}];}
}
break;
default:
break;
}
[self.collectionView reloadData];
}
- (IBAction)addClicked:(id)sender
{
if (UI_USER_INTERFACE_IDIOM() != UIUserInterfaceIdiomPad) {
[self performSegueWithIdentifier:@"addToken" sender:self];
return;
}
AddTokenViewController* c = [self.storyboard instantiateViewControllerWithIdentifier:@"addToken"];
UINavigationController* nc = [[UINavigationController alloc] initWithRootViewController:c];
c.popover = self.popover = [[UIPopoverController alloc] initWithContentViewController:nc];
self.popover.delegate = self;
self.popover.popoverContentSize = CGSizeMake(320, 715);
[self.popover presentPopoverFromBarButtonItem:sender permittedArrowDirections:UIPopoverArrowDirectionAny animated:YES];
}
- (IBAction)scanClicked:(id)sender
{
[self performSegueWithIdentifier:@"scanToken" sender:self];
}
- (IBAction)editClicked:(id)sender
{
UIBarButtonItem* edit = sender;
[self.collectionView reloadData];
// Enable/disable the add/scan buttons.
for (UIBarButtonItem* i in self.navigationItem.rightBarButtonItems)
[i setEnabled:edit.style != UIBarButtonItemStylePlain];
switch (edit.style) {
case UIBarButtonItemStylePlain:
edit.title = NSLocalizedString(@"Done", nil);
edit.style = UIBarButtonItemStyleDone;
// Setup gesture.
longPressGesture = [[UILongPressGestureRecognizer alloc] initWithTarget:self action:@selector(handleLongPress:)];
[self.collectionView addGestureRecognizer:longPressGesture];
break;
default:
edit.title = NSLocalizedString(@"Edit", nil);
edit.style = UIBarButtonItemStylePlain;
// Remove gesture.
[self.collectionView removeGestureRecognizer:longPressGesture];
longPressGesture = nil;
break;
}
}
- (void)popoverControllerDidDismissPopover:(UIPopoverController *)popoverController
{
self.popover = nil;
[self.collectionView reloadData];
}
- (void)viewDidLoad
{
[super viewDidLoad];
// Setup store.
store = [[TokenStore alloc] init];
// Setup collection view.
self.collectionView.delegate = self;
// Setup buttons.
UIBarButtonItem* add = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemAdd target:self action:@selector(addClicked:)];
if ([AVCaptureDevice defaultDeviceWithMediaType:AVMediaTypeVideo] == nil) {
self.navigationItem.rightBarButtonItems = @[add];
} else {
id icon = [UIImage imageNamed:@"qrcode.png"];
id scan = [[UIBarButtonItem alloc] initWithImage:icon style:UIBarButtonItemStylePlain target:self action:@selector(scanClicked:)];
self.navigationItem.rightBarButtonItems = @[add, scan];
}
}
- (void)viewWillAppear:(BOOL)animated
{
[super viewWillAppear:animated];
[self.collectionView reloadData];
}
@end