transshift/TransmissionRPCClient/FileListController.m

//
//  FileListController.m
//  TransmissionRPCClient
//
//  Created by Alexey Chechetkin on 30.06.15.
//  Copyright (c) 2015 Alexey Chechetkin. All rights reserved.
//

#import "FileListController.h"
//#import "FileListCell.h"
#import "FSDirectory.h"
#import "FileListFSCell.h"
#import "TRFileInfo.h"
#import "NSObject+DataObject.h"

#define ICON_FILE             @"iconFile"
#define ICON_FOLDER_OPENED    @"iconFolderOpened"
#define ICON_FOLDER_CLOSED    @"iconFolderClosed"

@interface FileListController ()
@end

@implementation FileListController

{
    UIImage *_iconImgFile;
    UIImage *_iconImgFolderOpened;
    UIImage *_iconImgFolderClosed;
    
    //FSDirectory *_fsDir;
    BOOL     _isSelectOnly;
}

- (void)viewDidLoad
{
    [super viewDidLoad];
    
    // preload images
    _iconImgFile = [[UIImage imageNamed:ICON_FILE] imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate];
    _iconImgFolderOpened = [[UIImage imageNamed:ICON_FOLDER_OPENED] imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate];
    _iconImgFolderClosed = [[UIImage imageNamed:ICON_FOLDER_CLOSED] imageWithRenderingMode:UIImageRenderingModeAlwaysTemplate];
    
    
    UIRefreshControl *refreshControl = [[UIRefreshControl alloc] init];
    self.refreshControl = refreshControl;
    
    [self.refreshControl addTarget:self action:@selector(askDelegateForDataUpdate) forControlEvents:UIControlEventValueChanged];
 }

- (void)viewWillAppear:(BOOL)animated
{
    [super viewWillAppear:animated];
    
    self.navigationItem.leftBarButtonItem.title =  NSLocalizedString(@"Info", @"FileListController nav left button title");
}

- (void)askDelegateForDataUpdate
{
    [self.refreshControl endRefreshing];
    if( _delegate && [_delegate respondsToSelector:@selector(fileListControllerNeedUpdateFilesForTorrentWithId:)])
        [_delegate fileListControllerNeedUpdateFilesForTorrentWithId:_torrentId];
}

- (void)setFsDir:(FSDirectory *)fsDir
{
    _fsDir = fsDir;
    _isSelectOnly = YES;
    
    [self.tableView reloadData];
}

// update file infos
- (void)setFileInfos:(NSArray *)fileInfos
{
     BOOL needToSort = NO;
    if( !_fsDir )
    {
        _fsDir = [FSDirectory directory];
        needToSort = YES;
    }
    
    for( int i = 0; i < fileInfos.count; i++ )
    {
        TRFileInfo *fileInfo = fileInfos[i];
        FSItem *item = [_fsDir addFilePath:fileInfo.name withIndex:i];
        item.info = fileInfo;
    }
    
    if( needToSort )
        [_fsDir sort];
    
    [_fsDir setNeedToRecalcStats];
    
    [self.tableView reloadData];
}

- (void)toggleFolderDownloading:(UIGestureRecognizer*)sender
{
    sender.view.userInteractionEnabled = NO;
        
    FSItem *item = sender.dataObject;
    NSArray *fileIndexes = item.fileIndexes;
    
    BOOL wanted = !item.isAllFilesWanted;
    
    if( _isSelectOnly )
    {
        item.isAllFilesWanted = wanted;
        [self.tableView reloadData];
        return;
    }
    
    if( _delegate && wanted &&
       [_delegate respondsToSelector:@selector(fileListControllerResumeDownloadingFilesWithIndexes:forTorrentWithId:)])
    {
        [_delegate fileListControllerResumeDownloadingFilesWithIndexes:fileIndexes
                                                      forTorrentWithId:_torrentId];
    }
    else if( _delegate && !wanted &&
            [_delegate respondsToSelector:@selector(fileListControllerStopDownloadingFilesWithIndexes:forTorrentWithId:)])
    {
        [_delegate fileListControllerStopDownloadingFilesWithIndexes:fileIndexes
                                                    forTorrentWithId:_torrentId];
    }
    
    [self askDelegateForDataUpdate];
}


- (void)toggleFileDownloading:(UIGestureRecognizer*)sender
{
    FSItem* item = sender.dataObject;
    BOOL wanted = !item.info.wanted;
    
    if( _isSelectOnly )
    {
        item.info.wanted = wanted;
        [self.tableView reloadData];
        return;
    }
    
    //sender.enabled = NO;
    sender.view.userInteractionEnabled = NO;
    
    if( _delegate && wanted && [_delegate respondsToSelector:@selector(fileListControllerResumeDownloadingFilesWithIndexes:forTorrentWithId:)])
    {
        [_delegate fileListControllerResumeDownloadingFilesWithIndexes:@[@(item.index)]
                                                      forTorrentWithId:_torrentId];
    }
    else if( _delegate && !wanted &&
            [_delegate respondsToSelector:@selector(fileListControllerStopDownloadingFilesWithIndexes:forTorrentWithId:)])
    {
        [_delegate fileListControllerStopDownloadingFilesWithIndexes:@[@(item.index)]
                                                    forTorrentWithId:_torrentId];
    }
    
    [self askDelegateForDataUpdate];
}

- (void)prioritySegmentToggled:(UISegmentedControl*)sender
{
    sender.enabled = NO;
    
    int priority = (int)sender.selectedSegmentIndex - 1;
    
    if( _delegate && [_delegate respondsToSelector:@selector(fileListControllerSetPriority:forFilesWithIndexes:forTorrentWithId:)])
    {
        [_delegate fileListControllerSetPriority:priority forFilesWithIndexes:@[sender.dataObject] forTorrentWithId:_torrentId];
    }
    
    [self askDelegateForDataUpdate];
}

// toggle collapse flag
- (void)folderTapped:(UITapGestureRecognizer*)sender
{
    FSItem *item = sender.dataObject;
    
    NSArray *indexes = [_fsDir childIndexesForItem:item];
    item.collapsed = !item.collapsed;
    
    if( indexes.count > 0 )
    {
        NSMutableArray *indexPaths = [NSMutableArray array];
        
        for( NSNumber *idx in indexes )
        {
            NSIndexPath *path = [NSIndexPath indexPathForRow:[idx intValue] inSection:0];
            [indexPaths addObject:path];
        }
        
        [self.tableView beginUpdates];
        
        if( item.collapsed )
        {
            [self.tableView deleteRowsAtIndexPaths:indexPaths withRowAnimation:UITableViewRowAnimationTop];
        }
        else
        {
            [self.tableView insertRowsAtIndexPaths:indexPaths withRowAnimation:UITableViewRowAnimationBottom];
        }
        
        [self.tableView endUpdates];
        
        // update folder icon
        NSUInteger itemIdx = [[indexes firstObject] intValue] - 1;
        FileListFSCell *cell = (FileListFSCell *)[self.tableView cellForRowAtIndexPath:[NSIndexPath indexPathForRow:itemIdx inSection:0]];
        if( cell )
        {
            cell.iconImg.image = item.isCollapsed ? _iconImgFolderClosed : _iconImgFolderOpened;
        }
    }
    //[self.tableView reloadData];
}

#pragma mark - Table view data source

- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView
{
    return _fsDir ? 1 : 0;
}

- (NSString *)tableView:(UITableView *)tableView titleForHeaderInSection:(NSInteger)section
{
    return NSLocalizedString( @"Files & Folders", @"");
}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section
{
    return _fsDir.count;
}

- (UITableViewCell*)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath
{
    FileListFSCell *cell = [tableView dequeueReusableCellWithIdentifier:CELL_ID_FILELISTFSCELL forIndexPath:indexPath];
    
    FSItem *item = [_fsDir itemAtIndex:(int)indexPath.row];
   
    cell.nameLabel.text = item.name;
    cell.iconImg.image = item.isFile ? _iconImgFile : (  item.isCollapsed ? _iconImgFolderClosed : _iconImgFolderOpened );
    
    // make indentation
    float leftIdent = FILELISTFSCELL_LEFTLABEL_WIDTH + ( (item.level - 1) * FILELISTFSCELL_LEFTLABEL_LEVEL_INDENTATION );
    
    if( _fsDir.rootItem.folderDownloadProgress >= 1.0 )
        leftIdent = ((item.level - 1) * FILELISTFSCELL_LEFTLABEL_LEVEL_INDENTATION);
    
    cell.leftIndentConstraint.constant = leftIdent;
    
    cell.iconImg.tintColor = cell.tintColor;            // default (blue) tintColor
    cell.prioritySegment.hidden = YES;                  // by default folders don't have priority segment
    cell.nameLabel.textColor = [UIColor blackColor];    // by default file/folder names are black
    
    cell.nameLabelTrailConstraint.priority = 751;
    cell.nameLabelTrailToSegmentConstraint.priority = 750;
    
    cell.touchView.userInteractionEnabled = NO;
    cell.leftLabel.userInteractionEnabled = NO;
    
    if( item.isFile )
    {
        if (_isSelectOnly)
        {
            cell.detailLabel.text = item.info.lengthString;
        }
        else
        {
            cell.detailLabel.text = [NSString stringWithFormat: NSLocalizedString(@"%@ of %@, %@ downloaded", @"FileList cell file info"),
                                     item.info.bytesComplitedString,
                                     item.info.lengthString,
                                     item.info.downloadProgressString];
        }
        
        if( item.info.downloadProgress < 1.0f )
        {
            if( !_isSelectOnly )
            {
                // configure priority segment control
                [cell.prioritySegment addTarget:self action:@selector(prioritySegmentToggled:) forControlEvents:UIControlEventValueChanged];
                cell.prioritySegment.dataObject = @(item.index);
                cell.prioritySegment.selectedSegmentIndex = item.info.priority + 1;
                cell.prioritySegment.enabled = YES;
                cell.prioritySegment.hidden = NO;
            }

            // configure left checkBox control
            cell.leftLabel.userInteractionEnabled = YES;
            UITapGestureRecognizer *tapRecognizer = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(toggleFileDownloading:)];
            tapRecognizer.dataObject = item;
            [cell.leftLabel addGestureRecognizer:tapRecognizer];
            
            if( !item.info.wanted || _isSelectOnly )
            {
                cell.prioritySegment.hidden = YES;
                
                if( !item.info.wanted )
                {
                    cell.iconImg.tintColor = [UIColor grayColor];
                    cell.nameLabel.textColor = [UIColor grayColor];
                }
            }
            else
            {
                cell.nameLabelTrailConstraint.priority = 750;
                cell.nameLabelTrailToSegmentConstraint.priority = 751;
            }
            
            cell.leftLabel.text = item.info.wanted ? @"☑︎" : @"◻︎";
        }
        else
        {
            cell.leftLabel.text = @"";
        }
    }
    else // it is folder
    {
        
        if (_isSelectOnly)
        {
            cell.detailLabel.text = [NSString stringWithFormat:NSLocalizedString(  @"%i files, %@", @"" ), item.filesCount, item.folderSizeString];
        }
        else
        {
            cell.detailLabel.text = [NSString stringWithFormat: NSLocalizedString(@"%i files, %@ of %@, %@ downloaded", @"FileList cell folder info"),
                                     item.filesCount,
                                     item.folderDownloadedString,
                                     item.folderSizeString,
                                     item.folderDownloadProgressString];
        }
        
        if( item.folderDownloadProgress < 1.0 )
        {
            // get info for files within folder
            cell.leftLabel.text = item.isAllFilesWanted ? @"☑︎" : @"◻︎";
            cell.nameLabel.textColor = item.isAllFilesWanted ? [UIColor blackColor] : [UIColor grayColor];
            cell.iconImg.tintColor = item.isAllFilesWanted ? cell.tintColor : [UIColor grayColor];
            
            // add handler for checking wanted/unwanted files
            cell.leftLabel.userInteractionEnabled = YES;
            
             // add recognizer for unwanted files
             UITapGestureRecognizer *recognizer = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(toggleFolderDownloading:)];
             recognizer.dataObject = item;
             [cell.leftLabel addGestureRecognizer:recognizer];
        }
        else
        {
            cell.leftLabel.text = @" ";
        }
        
        // Add tap handeler for folder - open/close
        [cell.touchView layoutIfNeeded];
        UITapGestureRecognizer *tapFolderRec = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(folderTapped:)];
        tapFolderRec.dataObject = item;
        
        cell.touchView.userInteractionEnabled = YES;
        [cell.touchView addGestureRecognizer:tapFolderRec];
    }
    
    return cell;
}

@end