TransferTableDataSource.java

package ch.cyberduck;

/*
 *  Copyright (c) 2005 David Kocher. All rights reserved.
 *  http://cyberduck.ch/
 *
 *  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 2 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.
 *
 *  Bug fixes, suggestions and comments should be sent to:
 *  dkocher@cyberduck.ch
 */

import ch.cyberduck.core.*;
//import ch.cyberduck.ui.cocoa.application.NSDraggingInfo;
//import ch.cyberduck.ui.cocoa.application.NSPasteboard;
//import ch.cyberduck.ui.cocoa.application.NSTableColumn;
//import ch.cyberduck.ui.cocoa.application.NSTableView;
//import ch.cyberduck.ui.cocoa.foundation.*;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
//import org.rococoa.cocoa.foundation.NSInteger;
//import org.rococoa.cocoa.foundation.NSUInteger;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

/**
 * @version $Id$
 */
public class TransferTableDataSource extends ListDataSource {
    private static Logger log = Logger.getLogger(TransferTableDataSource.class);

    public static final String PROGRESS_COLUMN = "PROGRESS";
    // virtual column to implement keyboard selection
    protected static final String TYPEAHEAD_COLUMN = "TYPEAHEAD";

    /**
	 * @uml.property  name="controllers"
	 * @uml.associationEnd  qualifier="t:ch.cyberduck.core.Transfer ch.cyberduck.ProgressController"
	 */
    private final Map<Transfer, ProgressController> controllers = new HashMap<Transfer, ProgressController>();

    public TransferTableDataSource() {
        TransferCollection.instance().addListener(new AbstractCollectionListener<Transfer>() {
            @Override
            public void collectionItemRemoved(Transfer item) {
                final ProgressController controller = controllers.remove(item);
                if(controller != null) {
                    controller.invalidate();
                }
            }
        });
    }

    @Override
    protected void invalidate() {
        cache.clear();
        super.invalidate();
    }

    /**
	 * @uml.property  name="filter"
	 * @uml.associationEnd  
	 */
    private TransferFilter filter = new NullTransferFilter();

    /**
     * @param searchString
     */
    public void setFilter(final String searchString) {
        if(StringUtils.isBlank(searchString)) {
            // Revert to the default filter
            this.filter = new NullTransferFilter();
        }
        else {
            // Setting up a custom filter
            this.filter = new TransferFilter() {
                public boolean accept(Transfer transfer) {
                    // Match for pathnames and hostname
                    return transfer.getName().toLowerCase().contains(searchString.toLowerCase())
                            || transfer.getSession().getHost().getHostname().toLowerCase().contains(searchString.toLowerCase());
                }
            };
        }
    }

    /**
     * @return The filtered collection currently to be displayed within the constraints
     */
    protected Collection<Transfer> getSource() {
        if(null == filter) {
            return TransferCollection.instance();
        }
        Collection<Transfer> filtered = new Collection<Transfer>(TransferCollection.instance());
        for(Iterator<Transfer> i = filtered.iterator(); i.hasNext();) {
            if(!filter.accept(i.next())) {
                //temporarly remove the t from the collection
                i.remove();
            }
        }
        return filtered;
    }

    /**
     * @param view
     */
//    public NSInteger numberOfRowsInTableView(NSTableView view) {
//        return new NSInteger(this.getSource().size());
//    }

    /**
	 * Second cache because it is expensive to create proxy instances
	 * @uml.property  name="cache"
	 * @uml.associationEnd  
	 */
    private AttributeCache<Transfer> cache = new AttributeCache<Transfer>(
            Preferences.instance().getInteger("queue.model.cache.size")
    );

    /**
     * @param view
     * @param tableColumn
     * @param row
     */
//    public NSObject tableView_objectValueForTableColumn_row(NSTableView view, NSTableColumn tableColumn, NSInteger row) {
//        if(row.intValue() >= this.numberOfRowsInTableView(view).intValue()) {
//            return null;
//        }
//        final String identifier = tableColumn.identifier();
//        final Transfer item = this.getSource().get(row.intValue());
//        final NSObject cached = cache.get(item, identifier);
//        if(null == cached) {
//            if(identifier.equals(PROGRESS_COLUMN)) {
//                return cache.put(item, identifier, null);
//            }
//            if(identifier.equals(TYPEAHEAD_COLUMN)) {
//                return cache.put(item, identifier, NSString.stringWithString(item.getName()));
//            }
//            throw new IllegalArgumentException("Unknown identifier: " + identifier);
//        }
//        return cached;
//    }

    // ----------------------------------------------------------
    // Drop methods
    // ----------------------------------------------------------

//    @Override
//    public NSUInteger tableView_validateDrop_proposedRow_proposedDropOperation(NSTableView view, NSDraggingInfo draggingInfo, NSInteger row, NSUInteger operation) {
//        log.debug("tableViewValidateDrop:row:" + row + ",operation:" + operation);
//        if(draggingInfo.draggingPasteboard().availableTypeFromArray(NSArray.arrayWithObject(NSPasteboard.StringPboardType)) != null) {
//            view.setDropRow(row, NSTableView.NSTableViewDropAbove);
//            return NSDraggingInfo.NSDragOperationCopy;
//        }
//        if(!PathPasteboard.allPasteboards().isEmpty()) {
//            view.setDropRow(row, NSTableView.NSTableViewDropAbove);
//            return NSDraggingInfo.NSDragOperationCopy;
//        }
//        log.debug("tableViewValidateDrop:DragOperationNone");
//        return NSDraggingInfo.NSDragOperationNone;
//    }

    /**
     * Invoked by tableView when the mouse button is released over a table view that previously decided to allow a drop.
     *
     * @param draggingInfo contains details on this dragging operation.
     * @param row          The proposed location is row and action is operation.
     */
//    @Override
//    public boolean tableView_acceptDrop_row_dropOperation(NSTableView view, NSDraggingInfo draggingInfo, NSInteger row, NSUInteger operation) {
//        if(draggingInfo.draggingPasteboard().availableTypeFromArray(NSArray.arrayWithObject(NSPasteboard.StringPboardType)) != null) {
//            String droppedText = draggingInfo.draggingPasteboard().stringForType(NSPasteboard.StringPboardType);// get the data from paste board
//            if(StringUtils.isNotBlank(droppedText)) {
//                log.info("NSPasteboard.StringPboardType:" + droppedText);
//                DownloadController c = new DownloadController(TransferController.instance(), droppedText);
//                c.beginSheet();
//                return true;
//            }
//            return false;
//        }
//        final Map<Host, PathPasteboard<NSDictionary>> boards = PathPasteboard.allPasteboards();
//        if(!boards.isEmpty()) {
//            for(PathPasteboard<NSDictionary> pasteboard : boards.values()) {
//                TransferCollection.instance().add(row.intValue(), new DownloadTransfer(pasteboard.getFiles()));
//                view.reloadData();
//                view.selectRowIndexes(NSIndexSet.indexSetWithIndex(row), false);
//                view.scrollRowToVisible(row);
//            }
//            boards.clear();
//            return true;
//        }
//        return false;
//    }

    /**
     * @param row
     * @return
     */
    public ProgressController getController(int row) {
        return this.getController(this.getSource().get(row));
    }

    /**
     * @param t
     * @return
     */
    public ProgressController getController(Transfer t) {
        if(!controllers.containsKey(t)) {
            controllers.put(t, new ProgressController(t));
        }
        return controllers.get(t);
    }

    /**
     * @param row
     * @param highlighted
     */
//    public void setHighlighted(int row, boolean highlighted) {
//        this.getController(row).setHighlighted(highlighted);
//    }
}