/*
* File : MyTrackerView.java
* Created : 30-Oct-2003
* By : parg
*
* Copyright (C) 2004, 2005, 2006 Aelitis SAS, 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 2 of the License.
*
* 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 ( see the LICENSE file ).
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
* AELITIS, SAS au capital de 46,603.30 euros,
* 8 Allee Lenotre, La Grille Royale, 78600 Le Mesnil le Roi, France.
*/
package org.gudy.azureus2.ui.swt.views;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetSocketAddress;
import java.net.URL;
import java.util.Arrays;
import java.util.Map;
import org.eclipse.swt.SWT;
import org.eclipse.swt.graphics.Color;
import org.eclipse.swt.widgets.*;
import org.gudy.azureus2.core3.category.Category;
import org.gudy.azureus2.core3.category.CategoryManager;
import org.gudy.azureus2.core3.category.CategoryManagerListener;
import org.gudy.azureus2.core3.download.DownloadManager;
import org.gudy.azureus2.core3.logging.LogAlert;
import org.gudy.azureus2.core3.logging.Logger;
import org.gudy.azureus2.core3.torrent.TOTorrent;
import org.gudy.azureus2.core3.tracker.host.TRHostListener;
import org.gudy.azureus2.core3.tracker.host.TRHostTorrent;
import org.gudy.azureus2.core3.tracker.host.TRHostTorrentRemovalVetoException;
import org.gudy.azureus2.core3.util.*;
import org.gudy.azureus2.plugins.torrent.TorrentAttribute;
import org.gudy.azureus2.plugins.tracker.TrackerTorrent;
import org.gudy.azureus2.plugins.ui.tables.TableManager;
import org.gudy.azureus2.pluginsimpl.local.torrent.TorrentManagerImpl;
import org.gudy.azureus2.ui.swt.CategoryAdderWindow;
import org.gudy.azureus2.ui.swt.Messages;
import org.gudy.azureus2.ui.swt.Utils;
import org.gudy.azureus2.ui.swt.mainwindow.Colors;
import org.gudy.azureus2.ui.swt.mainwindow.SWTThread;
import org.gudy.azureus2.ui.swt.shells.CoreWaiterSWT;
import org.gudy.azureus2.ui.swt.views.table.TableRowSWT;
import org.gudy.azureus2.ui.swt.views.table.TableViewSWT;
import org.gudy.azureus2.ui.swt.views.table.TableViewSWTMenuFillListener;
import org.gudy.azureus2.ui.swt.views.table.impl.TableViewSWTImpl;
import org.gudy.azureus2.ui.swt.views.table.impl.TableViewTab;
import org.gudy.azureus2.ui.swt.views.tableitems.mytracker.*;
import com.aelitis.azureus.core.AzureusCore;
import com.aelitis.azureus.core.AzureusCoreFactory;
import com.aelitis.azureus.core.AzureusCoreRunningListener;
import com.aelitis.azureus.ui.UIFunctions;
import com.aelitis.azureus.ui.UIFunctionsManager;
import com.aelitis.azureus.ui.common.table.*;
/**
* @author parg
* @author TuxPaper
* 2004/Apr/20: Remove need for tableItemToObject
* 2004/Apr/21: extends TableView instead of IAbstractView
*/
public class MyTrackerView
extends TableViewTab
implements TRHostListener, CategoryManagerListener, TableLifeCycleListener,
TableSelectionListener, TableViewSWTMenuFillListener, TableRefreshListener
{
private static TableColumnCore[] basicItems = null;
protected static final TorrentAttribute category_attribute =
TorrentManagerImpl.getSingleton().getAttribute( TorrentAttribute.TA_CATEGORY );
private Menu menuCategory;
private TableViewSWT<TRHostTorrent> tv;
public MyTrackerView() {
super("MyTrackerView");
if (basicItems == null) {
basicItems = new TableColumnCore[] {
new NameItem(),
new TrackerItem(),
new StatusItem(),
new CategoryItem(),
new PassiveItem(),
new SeedCountItem(),
new PeerCountItem(),
new BadNATCountItem(),
new AnnounceCountItem(),
new ScrapeCountItem(),
new CompletedCountItem(),
new UploadedItem(),
new DownloadedItem(),
new LeftItem(),
new TotalBytesInItem(),
new AverageBytesInItem(),
new TotalBytesOutItem(),
new AverageBytesOutItem(),
new DateAddedItem(),
};
}
tv = new TableViewSWTImpl<TRHostTorrent>(TrackerTorrent.class,
TableManager.TABLE_MYTRACKER, getPropertiesPrefix(), basicItems, "name",
SWT.MULTI | SWT.FULL_SELECTION | SWT.BORDER | SWT.VIRTUAL);
tv.addLifeCycleListener(this);
tv.addSelectionListener(this, false);
tv.addMenuFillListener(this);
tv.addRefreshListener(this, false);
}
public TableViewSWT initYourTableView() {
return tv;
}
// @see com.aelitis.azureus.ui.common.table.TableLifeCycleListener#tableViewInitialized()
public void tableViewInitialized() {
AzureusCoreFactory.addCoreRunningListener(new AzureusCoreRunningListener() {
public void azureusCoreRunning(AzureusCore core) {
core.getTrackerHost().addListener(MyTrackerView.this);
}
});
}
// @see com.aelitis.azureus.ui.common.table.TableLifeCycleListener#tableViewDestroyed()
public void tableViewDestroyed() {
try {
AzureusCoreFactory.getSingleton().getTrackerHost().removeListener( this );
} catch (Exception ignore) {
}
}
// @see com.aelitis.azureus.ui.common.table.TableSelectionListener#defaultSelected(com.aelitis.azureus.ui.common.table.TableRowCore[])
public void defaultSelected(TableRowCore[] rows, int stateMask) {
final TRHostTorrent torrent = (TRHostTorrent) tv.getFirstSelectedDataSource();
if (torrent == null)
return;
CoreWaiterSWT.waitForCoreRunning(new AzureusCoreRunningListener() {
public void azureusCoreRunning(AzureusCore core) {
DownloadManager dm = core.getGlobalManager().getDownloadManager(
torrent.getTorrent());
if (dm != null) {
UIFunctions uiFunctions = UIFunctionsManager.getUIFunctions();
if (uiFunctions != null) {
uiFunctions.openView(UIFunctions.VIEW_DM_DETAILS, dm);
}
}
}
});
}
public void fillMenu(String sColumnName, final Menu menu) {
menuCategory = new Menu(getComposite().getShell(), SWT.DROP_DOWN);
final MenuItem itemCategory = new MenuItem(menu, SWT.CASCADE);
Messages.setLanguageText(itemCategory, "MyTorrentsView.menu.setCategory"); //$NON-NLS-1$
//itemCategory.setImage(ImageRepository.getImage("speed"));
itemCategory.setMenu(menuCategory);
addCategorySubMenu();
new MenuItem(menu, SWT.SEPARATOR);
final MenuItem itemStart = new MenuItem(menu, SWT.PUSH);
Messages.setLanguageText(itemStart, "MyTorrentsView.menu.start"); //$NON-NLS-1$
Utils.setMenuItemImage(itemStart, "start");
final MenuItem itemStop = new MenuItem(menu, SWT.PUSH);
Messages.setLanguageText(itemStop, "MyTorrentsView.menu.stop"); //$NON-NLS-1$
Utils.setMenuItemImage(itemStop, "stop");
final MenuItem itemRemove = new MenuItem(menu, SWT.PUSH);
Messages.setLanguageText(itemRemove, "MyTorrentsView.menu.remove"); //$NON-NLS-1$
Utils.setMenuItemImage(itemRemove, "delete");
Object[] hostTorrents = tv.getSelectedDataSources().toArray();
itemStart.setEnabled(false);
itemStop.setEnabled(false);
itemRemove.setEnabled(false);
if (hostTorrents.length > 0) {
boolean start_ok = true;
boolean stop_ok = true;
boolean remove_ok = true;
for (int i = 0; i < hostTorrents.length; i++) {
TRHostTorrent host_torrent = (TRHostTorrent)hostTorrents[i];
int status = host_torrent.getStatus();
if ( status != TRHostTorrent.TS_STOPPED ){
start_ok = false;
}
if ( status != TRHostTorrent.TS_STARTED ){
stop_ok = false;
}
/*
try{
host_torrent.canBeRemoved();
}catch( TRHostTorrentRemovalVetoException f ){
remove_ok = false;
}
*/
}
itemStart.setEnabled(start_ok);
itemStop.setEnabled(stop_ok);
itemRemove.setEnabled(remove_ok);
}
itemStart.addListener(SWT.Selection, new Listener() {
public void handleEvent(Event e) {
startSelectedTorrents();
}
});
itemStop.addListener(SWT.Selection, new Listener() {
public void handleEvent(Event e) {
stopSelectedTorrents();
}
});
itemRemove.addListener(SWT.Selection, new Listener() {
public void handleEvent(Event e) {
removeSelectedTorrents();
}
});
new MenuItem(menu, SWT.SEPARATOR);
}
// @see org.gudy.azureus2.ui.swt.views.TableViewSWTMenuFillListener#addThisColumnSubMenu(java.lang.String, org.eclipse.swt.widgets.Menu)
public void addThisColumnSubMenu(String columnName, Menu menuThisColumn) {
}
public void
torrentAdded(
TRHostTorrent host_torrent )
{
tv.addDataSource(host_torrent);
}
public void torrentChanged(TRHostTorrent t) { }
public void
torrentRemoved(
TRHostTorrent host_torrent )
{
tv.removeDataSource(host_torrent);
}
public boolean
handleExternalRequest(
InetSocketAddress client,
String user,
String url,
URL absolute_url,
String header,
InputStream is,
OutputStream os,
AsyncController async )
throws IOException
{
return( false );
}
// @see com.aelitis.azureus.ui.common.table.TableRefreshListener#tableRefresh()
public void tableRefresh() {
if (getComposite() == null || getComposite().isDisposed()){
return;
}
UIFunctions uiFunctions = UIFunctionsManager.getUIFunctions();
if (uiFunctions != null) {
uiFunctions.refreshIconBar();
}
// Store values for columns that are calculate from peer information, so
// that we only have to do one loop. (As opposed to each cell doing a loop)
// Calculate code copied from TrackerTableItem
TableRowCore[] rows = tv.getRows();
for (int x = 0; x < rows.length; x++) {
TableRowSWT row = (TableRowSWT)rows[x];
if (row == null){
continue;
}
TRHostTorrent host_torrent = (TRHostTorrent)rows[x].getDataSource(true);
if (host_torrent == null){
continue;
}
long uploaded = host_torrent.getTotalUploaded();
long downloaded = host_torrent.getTotalDownloaded();
long left = host_torrent.getTotalLeft();
int seed_count = host_torrent.getSeedCount();
host_torrent.setData("GUI_PeerCount", new Long(host_torrent.getLeecherCount()));
host_torrent.setData("GUI_SeedCount", new Long(seed_count));
host_torrent.setData("GUI_BadNATCount", new Long(host_torrent.getBadNATCount()));
host_torrent.setData("GUI_Uploaded", new Long(uploaded));
host_torrent.setData("GUI_Downloaded", new Long(downloaded));
host_torrent.setData("GUI_Left", new Long(left));
if ( seed_count != 0 ){
Color fg = row.getForeground();
if (fg != null && fg.equals(Colors.blues[Colors.BLUES_MIDDARK])) {
row.setForeground(Colors.blues[Colors.BLUES_MIDDARK]);
}
}
}
}
@Override
public void refreshToolBar(Map list) {
boolean start = false, stop = false, remove = false;
Object[] hostTorrents = tv.getSelectedDataSources().toArray();
if (hostTorrents.length > 0) {
remove = true;
for (int i = 0; i < hostTorrents.length; i++) {
TRHostTorrent host_torrent = (TRHostTorrent)hostTorrents[i];
int status = host_torrent.getStatus();
if ( status == TRHostTorrent.TS_STOPPED ){
start = true;
}
if ( status == TRHostTorrent.TS_STARTED ){
stop = true;
}
/*
try{
host_torrent.canBeRemoved();
}catch( TRHostTorrentRemovalVetoException f ){
remove = false;
}
*/
}
}
list.put("start", start);
list.put("stop", stop);
list.put("remove", remove);
super.refreshToolBar(list);
}
public boolean toolBarItemActivated(String itemKey) {
if(itemKey.equals("start")) {
startSelectedTorrents();
return true;
}
if(itemKey.equals("stop")){
stopSelectedTorrents();
return true;
}
if(itemKey.equals("remove")){
removeSelectedTorrents();
return true;
}
return super.toolBarItemActivated(itemKey);
}
private void stopSelectedTorrents() {
tv.runForSelectedRows(new TableGroupRowRunner() {
public void run(TableRowCore row) {
TRHostTorrent torrent = (TRHostTorrent)row.getDataSource(true);
if (torrent.getStatus() == TRHostTorrent.TS_STARTED)
torrent.stop();
}
});
}
private void startSelectedTorrents() {
tv.runForSelectedRows(new TableGroupRowRunner() {
public void run(TableRowCore row) {
TRHostTorrent torrent = (TRHostTorrent)row.getDataSource(true);
if (torrent.getStatus() == TRHostTorrent.TS_STOPPED)
torrent.start();
}
});
}
private void removeSelectedTorrents() {
tv.runForSelectedRows(new TableGroupRowRunner() {
public void run(TableRowCore row) {
TRHostTorrent torrent = (TRHostTorrent)row.getDataSource(true);
try{
torrent.remove();
}catch( TRHostTorrentRemovalVetoException f ){
Logger.log(new LogAlert(torrent, false,
"{globalmanager.download.remove.veto}", f));
}
}
});
}
private void addCategorySubMenu() {
MenuItem[] items = menuCategory.getItems();
int i;
for (i = 0; i < items.length; i++) {
items[i].dispose();
}
Category[] categories = CategoryManager.getCategories();
Arrays.sort(categories);
if (categories.length > 0) {
Category catUncat = CategoryManager.getCategory(Category.TYPE_UNCATEGORIZED);
if (catUncat != null) {
final MenuItem itemCategory = new MenuItem(menuCategory, SWT.PUSH);
Messages.setLanguageText(itemCategory, catUncat.getName());
itemCategory.setData("Category", catUncat);
itemCategory.addListener(SWT.Selection, new Listener() {
public void handleEvent(Event event) {
MenuItem item = (MenuItem)event.widget;
assignSelectedToCategory((Category)item.getData("Category"));
}
});
new MenuItem(menuCategory, SWT.SEPARATOR);
}
for (i = 0; i < categories.length; i++) {
if (categories[i].getType() == Category.TYPE_USER) {
final MenuItem itemCategory = new MenuItem(menuCategory, SWT.PUSH);
itemCategory.setText(categories[i].getName());
itemCategory.setData("Category", categories[i]);
itemCategory.addListener(SWT.Selection, new Listener() {
public void handleEvent(Event event) {
MenuItem item = (MenuItem)event.widget;
assignSelectedToCategory((Category)item.getData("Category"));
}
});
}
}
new MenuItem(menuCategory, SWT.SEPARATOR);
}
final MenuItem itemAddCategory = new MenuItem(menuCategory, SWT.PUSH);
Messages.setLanguageText(itemAddCategory,
"MyTorrentsView.menu.setCategory.add");
itemAddCategory.addListener(SWT.Selection, new Listener() {
public void handleEvent(Event event) {
addCategory();
}
});
}
public void
categoryAdded(Category category)
{
Utils.execSWTThread(
new AERunnable()
{
public void
runSupport()
{
addCategorySubMenu();
}
});
}
public void
categoryRemoved(
Category category)
{
Utils.execSWTThread(
new AERunnable()
{
public void
runSupport()
{
addCategorySubMenu();
}
});
}
public void categoryChanged(Category category) {
}
private void addCategory() {
CategoryAdderWindow adderWindow = new CategoryAdderWindow(SWTThread.getInstance().getDisplay());
Category newCategory = adderWindow.getNewCategory();
if (newCategory != null)
assignSelectedToCategory(newCategory);
}
private void assignSelectedToCategory(final Category category) {
CoreWaiterSWT.waitForCoreRunning(new AzureusCoreRunningListener() {
public void azureusCoreRunning(final AzureusCore core) {
assignSelectedToCategory(core, category);
}
});
}
private void assignSelectedToCategory(final AzureusCore core,
final Category category) {
tv.runForSelectedRows(new TableGroupRowRunner() {
public void run(TableRowCore row) {
TRHostTorrent tr_torrent = (TRHostTorrent)row.getDataSource(true);
final TOTorrent torrent = tr_torrent.getTorrent();
DownloadManager dm = core.getGlobalManager().getDownloadManager( torrent );
if ( dm != null ){
dm.getDownloadState().setCategory( category );
}else{
String cat_str;
if ( category == null ){
cat_str = null;
}else if ( category == CategoryManager.getCategory(Category.TYPE_UNCATEGORIZED)){
cat_str = null;
}else{
cat_str = category.getName();
}
// bit of a hack-alert here
TorrentUtils.setPluginStringProperty( torrent, "azcoreplugins.category", cat_str );
try{
TorrentUtils.writeToFile( torrent );
}catch( Throwable e ){
Debug.printStackTrace( e );
}
}
}
});
}
// @see com.aelitis.azureus.ui.common.table.TableSelectionListener#deselected(com.aelitis.azureus.ui.common.table.TableRowCore[])
public void deselected(TableRowCore[] rows) {
UIFunctions uiFunctions = UIFunctionsManager.getUIFunctions();
if (uiFunctions != null) {
uiFunctions.refreshIconBar();
}
}
// @see com.aelitis.azureus.ui.common.table.TableSelectionListener#focusChanged(com.aelitis.azureus.ui.common.table.TableRowCore)
public void focusChanged(TableRowCore focus) {
UIFunctions uiFunctions = UIFunctionsManager.getUIFunctions();
if (uiFunctions != null) {
uiFunctions.refreshIconBar();
}
}
// @see com.aelitis.azureus.ui.common.table.TableSelectionListener#selected(com.aelitis.azureus.ui.common.table.TableRowCore[])
public void selected(TableRowCore[] rows) {
UIFunctions uiFunctions = UIFunctionsManager.getUIFunctions();
if (uiFunctions != null) {
uiFunctions.refreshIconBar();
}
}
// @see com.aelitis.azureus.ui.common.table.TableSelectionListener#mouseEnter(com.aelitis.azureus.ui.common.table.TableRowCore)
public void mouseEnter(TableRowCore row) {
}
// @see com.aelitis.azureus.ui.common.table.TableSelectionListener#mouseExit(com.aelitis.azureus.ui.common.table.TableRowCore)
public void mouseExit(TableRowCore row) {
}
}