torrentfilter.java

这是一个基于java编写的torrent的P2P源码

/*
 * Created on 26/09/2005
 * Created by Paul Duran
 * Copyright (C) 2004 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, SARL au capital de 30,000 euros
 * 8 Allee Lenotre, La Grille Royale, 78600 Le Mesnil le Roi, France.
 *
 */

package org.gudy.azureus2.ui.console.commands;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.gudy.azureus2.core3.download.DownloadManager;

import com.aelitis.azureus.core.AzureusCoreException;

public class TorrentFilter {

	private static final Pattern rangePattern = Pattern.compile("([0-9]+)\\s*((-)|(-\\s*([0-9]+)))?");
	
	public TorrentFilter() {
		super();
	}
	
	/**
	 * matches a range of torrents. eg: 3-5 or a single torrent. eg: 3. or from 3 onwards: 3-
	 * @param torrents torrents to match
	 * @param filter range expression
	 * @return list of matched DownloadManager objects
	 */
	private List matchRange( List torrents, String filter )
	{
		Matcher matcher = rangePattern.matcher(filter);
		List list = new ArrayList();
		if( matcher.matches() )
		{
			int minId = Integer.parseInt(matcher.group(1));
			if( minId == 0 )
				throw new AzureusCoreException("lower range must be greater than 0");
			if( minId > torrents.size() )
				throw new AzureusCoreException("lower range specified (" + minId + ") is outside number of torrents (" + torrents.size() + ")");
			if( matcher.group(2) == null )
			{				
				// received a single number. eg: 3
				list.add(torrents.get(minId-1));
				return list;
			}
			int maxId;
			if( matcher.group(3) == null )
				// received bound range. eg: 3-5
				maxId = Integer.parseInt(matcher.group(5));
			else
				// received open ended range. eg: 3-
				maxId = torrents.size();
			
			if( minId > maxId )
				throw new AzureusCoreException("when specifying a range, the max value must be greater than or equal to the min value");
			
			for( int i = (minId-1) ; i < maxId && i < torrents.size() ; i++ )
			{
				list.add(torrents.get(i));
			}
		}
		return list;
	}
	
	/**
	 * attempst to match a wildcard against the list of torrents by
	 * checking their display name
	 * @param torrents list of available torrents to match
	 * @param filter wildcard (glob) filter
	 * @return list of matched DownloadManager objects
	 */
	private List matchWildcard( List torrents, String filter )
	{
		Pattern pattern = Pattern.compile(wildcardToPattern(filter), Pattern.CASE_INSENSITIVE);
		List list = new ArrayList();
		for (Iterator iter = torrents.iterator(); iter.hasNext();) {
			DownloadManager dm = (DownloadManager) iter.next();
			if( pattern.matcher(dm.getDisplayName()).matches() )
				list.add(dm);
		}
		return list;
	}
	
	/**
	 * converts the wildcard (eg: tran*) into a regular expression - (tran.*)
	 * @param wild wildcard (glob) expression
	 * @return regular expression string
	 */
	private String wildcardToPattern(String wild)
	{
		if (wild == null)
			return null;

		StringBuffer buffer = new StringBuffer();

		char[] chars = wild.toCharArray();

		for (int i = 0; i < chars.length; ++i) {
			if (chars[i] == '*')
				buffer.append(".*");
			else if (chars[i] == '?')
				buffer.append(".");
			else if ("+()^$.{}[]|\\".indexOf(chars[i]) != -1)
				buffer.append('\\').append(chars[i]); // prefix all metacharacters
													// with backslash
			else
				buffer.append(chars[i]);
		}
		return buffer.toString().toLowerCase();
	}
	
	/**
	 * tries our two different matching algorithms using the 
	 * supplied filter against the list of torrents in the ConsoleInput object
	 * @param torrentsToMatch list of DownloadManager objects to attempt to match against
	 * @param filter filter - eg: range or glob filter
	 * @return list of matched DownloadManager objects
	 */
	public List getTorrents(List torrentsToMatch, String filter) 
	{
		List torrents = new ArrayList();
		torrents.addAll(matchRange(torrentsToMatch, filter) );
		torrents.addAll(matchWildcard(torrentsToMatch, filter) );
		return torrents;
	}

	/**
	 * first tries to match torrents by concatenating all of the arguments.
	 * if that doesn't work, attempts to match each argument individually.
	 * @param torrentsToMatch list of DownloadManager objects to attempt to match against
	 * @param args arguments to try to match
	 * @return list of matched DownloadManager objects
	 */
	public List getTorrents(List torrentsToMatch, List args)
	{
		// first, try to match the whole list concatenated as a string
		StringBuffer allArgs = new StringBuffer();
		boolean first = true;
		for (Iterator iter = args.iterator(); iter.hasNext();) {
			if( ! first ) 
				allArgs.append(",");
			else
				first = false;
			allArgs.append(iter.next());
		}
		List torrents;
		torrents = matchWildcard(torrentsToMatch, allArgs.toString());
		if( torrents.size() > 0 )
			return torrents;
		torrents = matchRange(torrentsToMatch, allArgs.toString());
		if( torrents.size() > 0 )
			return torrents;
		// if no torrents then handle each argument individually
		for (Iterator iter = args.iterator(); iter.hasNext();) {
			torrents.addAll(getTorrents(torrentsToMatch, (String)iter.next()) );
		}
		return torrents;
	}
}