/**
 *
 * Copyright (c) 2014, the Railo Company Ltd. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either 
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library 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
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public 
 * License along with this library.  If not, see <http://www.gnu.org/licenses/>.
 * 
 **/
package lucee.commons.io.res.filter;


import lucee.commons.io.res.Resource;
import lucee.commons.lang.StringUtil;

/**
 * Filter fuer die <code>listFiles</code> Methode des FIle Objekt, 
 * zum filtern von FIles mit einer bestimmten Extension.
 */
public final class ExtensionResourceFilter implements ResourceFilter {
	
	private String[] extensions;
	private final boolean allowDir;
	private final boolean ignoreCase;
    //private int extLen;

	public static final ExtensionResourceFilter EXTENSION_JAR_NO_DIR = new ExtensionResourceFilter(".jar", false);
	public static final ExtensionResourceFilter EXTENSION_CLASS_DIR  = new ExtensionResourceFilter(".class", true);


	/**
	 * Konstruktor des Filters
	 * @param extension Endung die geprueft werden soll.
	 */
	public ExtensionResourceFilter(String extension) {
		this(new String[]{extension},false,true);
	}

	/**
	 * Konstruktor des Filters
	 * @param extension Endung die geprueft werden soll.
	 */
	public ExtensionResourceFilter(String extension, boolean allowDir) {
		this(new String[]{extension},allowDir,true);
	}
	
	public ExtensionResourceFilter(String[] extensions) {
		this(extensions,false,true);
	}
	
	public ExtensionResourceFilter(String[] extensions, boolean allowDir) {
		this(extensions,allowDir,true);
	}

	
	public ExtensionResourceFilter(String[] extensions, boolean allowDir, boolean ignoreCase) {
		String[] tmp=new String[extensions.length];
		for(int i=0;i<extensions.length;i++) {
			if(!StringUtil.startsWith(extensions[i],'.'))
	            tmp[i]="."+extensions[i];
			else 
				tmp[i]=extensions[i];
		}
		this.extensions=tmp;
    	this.allowDir=allowDir;
    	this.ignoreCase=ignoreCase;
	}
	
	public void addExtension(String extension) {
		String[] tmp=new String[extensions.length+1];
		// add existing
		for(int i=0;i<extensions.length;i++) {
			tmp[i]=extensions[i];
		}
		// add the new one
		if(!StringUtil.startsWith(extension,'.'))
            tmp[extensions.length]="."+extension;
		else 
			tmp[extensions.length]=extension;
		
		this.extensions=tmp;
	}

	@Override
	public boolean accept(Resource res) {
		if(res.isDirectory()) return allowDir;
		if(res.exists()) {
			String name=res.getName();
			for(int i=0;i<extensions.length;i++) {
				if(ignoreCase){
					if(StringUtil.endsWithIgnoreCase(name,extensions[i]))
						return true;
				}
				else {
					if(name.endsWith(extensions[i]))
						return true;
				}
			}
		}
		return false;
	}
	
	public boolean accept(String name) {
			for(int i=0;i<extensions.length;i++) {
				if(ignoreCase){
					if(StringUtil.endsWithIgnoreCase(name,extensions[i]))
						return true;
				}
				else {
					if(name.endsWith(extensions[i]))
						return true;
				}
			}
		
		return false;
	}
	
    /**
     * @return Returns the extension.
     */
    public String[] getExtensions() {
        return extensions;
    }
}