org.gudy.azureus2.ui.console.commands.Priority.java Source code

Java tutorial

Introduction

Here is the source code for org.gudy.azureus2.ui.console.commands.Priority.java

Source

/*
 * 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]);
    }

}