Package org.gudy.azureus2.ui.console.commands

Source Code of org.gudy.azureus2.ui.console.commands.Priority

/*
* Created on 07/04/2006
* Created by David Mohr, based on Alias.java by Paul Duran
* Copyright (C) 2006 Aelitis, 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, 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, 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.console.commands;

import java.io.PrintStream;
import java.util.LinkedList;
import java.util.List;

import org.apache.commons.cli.CommandLine;
import org.gudy.azureus2.core3.disk.DiskManagerFileInfo;
import org.gudy.azureus2.core3.download.DownloadManager;
import org.gudy.azureus2.ui.console.ConsoleInput;

/**
* the priority command changes the priority of files within a torrent
* @author dmohr
*/
public class Priority extends OptionsConsoleCommand {

  public Priority() {
    super("prio");
  }

  public String getCommandDescriptions() {
    return "prio [#torrent] [#file|range(i.e. 1-2,5)|all] [normal|high|dnd|del]";
  }

  public void printHelpExtra(PrintStream out, List args) {
    out.println("> -----");
    out.println("Usage: prio [torrent] [file(s)] [priority]");
    out.println("Options:");
    out.println("\t[torrent]\tThe torrent number from 'show torrents'");
    out.println("\t[file(s)] is one of:");
    out.println("\t\t\t#file:\tthe file number from 'show [#torrent]',");
    out.println("\t\t\trange:\ta range of file numbers, i.e. 1-3 or 1-10,12-15 or 1,3,5-8 ,");
    out.println("\t\t\tall:\t 'all' applies priority to all files of the torrent");
    out.println("\t[priority] is one of:");
    out.println("\t\t\tnormal\tNormal priority");
    out.println("\t\t\thigh  \tHigh priority");
    out.println("\t\t\tdnd   \tDo not download (skip)");
    out.println("\t\t\tdel   \tDo not download & delete file");
    out.println("> -----");
  }
 
  private static final int NORMAL=1;
  private static final int HIGH=2;
  private static final int DONOTDOWNLOAD=3;
  private static final int DELETE=4;
  private static final String[] priostr = { "Normal", "High", "DoNotDownload", "Delete" };
 
  private int newprio;
 
  public void execute(String commandName, ConsoleInput console, CommandLine commandLine) {

    String tnumstr, fnumstr, newpriostr;
    int tnumber;
    DiskManagerFileInfo[] files;
    String[] sections;
    List args = commandLine.getArgList();
    LinkedList fs,fe;
    DownloadManager dm;
   
    if( args.isEmpty() )
    {
      console.out.println("Torrent # required!");
      return;
    } else {
      tnumstr = (String) args.remove(0);
    }
    if( args.isEmpty() )
    {
      console.out.println("File # required!");
      return;
    } else {
      fnumstr = (String) args.remove(0);
    }

    if ((console.torrents == null) || console.torrents.isEmpty()) {
      console.out.println("> Command 'prio': No torrents in list (try 'show torrents' first).");
      return;
    }
   
    try {
      tnumber = Integer.parseInt(tnumstr);
      if ((tnumber == 0) || (tnumber > console.torrents.size())) {
        console.out.println("> Command 'prio': Torrent #" + tnumber + " unknown.");
        return;
      }

      dm = (DownloadManager) console.torrents.get(tnumber - 1);
      files = dm.getDiskManagerFileInfo();
    }
    catch (Exception e) {
      e.printStackTrace();
      console.out.println("> Command 'prio': Torrent # '" + tnumstr + "' unknown.");
      return;
    }

    if( args.isEmpty() )
    {
      console.out.println("> Command 'prio': missing parameter for new priority");
      return;
    } else {
      newpriostr = (String) args.remove(0);
    }

    if (newpriostr.equalsIgnoreCase("normal")) {
      newprio = NORMAL;
    } else if (newpriostr.equalsIgnoreCase("high")) {
      newprio = HIGH;
    } else if (newpriostr.equalsIgnoreCase("dnd")) {
      newprio = DONOTDOWNLOAD;
    } else if (newpriostr.equalsIgnoreCase("del")) {
      newprio = DELETE;
    } else {
      console.out.println("> Command 'prio': unknown priority "
          + newpriostr);
      return;
    }

    if (fnumstr.equalsIgnoreCase("all")) {
      sections = new String[1];
      sections[0] = "1-"+files.length;
    } else
      sections = fnumstr.split(",");
   
    fs = new LinkedList();
    fe = new LinkedList();
   
    int dash,start,end;
    for (int i=0; i<sections.length; i++) {
      try {
        if ((dash = sections[i].indexOf('-')) != -1) {
          start = Integer.parseInt(sections[i].substring(0,dash));
          end = Integer.parseInt(sections[i].substring(dash+1));
        } else
          start = end = Integer.parseInt(sections[i]);
        if ((start == 0) || (end > files.length)) {
          console.out.println("> Command 'prio': Invalid file range " + sections[i]);
          return;
        }
        if (start > end) {
          console.out.println("> Command 'prio': Invalid file range '"+sections[i]+"'");
        }
       
        // -1 compensates for 0-based offsets
        fs.add(new Integer(start - 1));
        fe.add(new Integer(end - 1));
      } catch (Exception e) {
        console.out.println("> Command 'prio': File # '" + sections[i]
            + "' unknown.");
        return;
      }
    }
   
//    console.out.println("DM was " + dm.getState());
    if ((newprio == DELETE) && (dm.getState() != DownloadManager.STATE_STOPPED)) {
      try {
        dm.stopIt( DownloadManager.STATE_STOPPED, false, false );
      } catch (Exception e) {
        console.out.println("Failed to stop torrent " + tnumber);
        return;
      }
    }
   
//    console.out.println("DM is " + dm.getState());
    int nummod = 0;
    while (fs.size() > 0) {
      start = ((Integer) fs.removeFirst()).intValue();
      end = ((Integer) fe.removeFirst()).intValue();
      for (int i = start; i <= end; i++) {
        nummod++;
        // DEBUG
//        console.out.println("Setting priority for file " + i + " to " + newprio);
        if (newprio == NORMAL) {
          files[i].setPriority(0);
          files[i].setSkipped(false);
        } else if (newprio == HIGH) {
          files[i].setPriority(1);
          files[i].setSkipped(false);
        } else if (newprio == DONOTDOWNLOAD) {
          files[i].setPriority(0);
          files[i].setSkipped(true);
        } else if (newprio == DELETE) {
          int st = files[i].getStorageType();
          int target_st = -1;
          if ( st == DiskManagerFileInfo.ST_LINEAR ){
            target_st = DiskManagerFileInfo.ST_COMPACT;
          }else if ( st == DiskManagerFileInfo.ST_REORDER ){
            target_st = DiskManagerFileInfo.ST_REORDER_COMPACT;
          }
          if (target_st != -1 &&
            files[i].setStorageType(target_st)) {
            files[i].setPriority(0);
            files[i].setSkipped(true);
          } else {
            console.out.println("> Command 'prio': Failed to delete file " + (i+1));
            nummod--;
          }
        }
      }
    }
    if ((newprio == DELETE) && (dm.getState() == DownloadManager.STATE_STOPPED)) {
      try {
        dm.stopIt( DownloadManager.STATE_QUEUED, false, false );
      } catch (Exception e) {
        console.out.println("Failed to restart torrent " + tnumber);
        return;
      }
    }

//    console.out.println("DM is again " + dm.getState());
   
    console.out.println(nummod + " file(s) priority set to " + priostr[newprio-1]);
  }

}
TOP

Related Classes of org.gudy.azureus2.ui.console.commands.Priority

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.