/*
 *  RapidMiner
 *
 *  Copyright (C) 2001-2011 by Rapid-I and the contributors
 *
 *  Complete list of developers available at our web site:
 *
 *       http://rapid-i.com
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU Affero General Public License as published by
 *  the Free Software Foundation, either version 3 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 Affero General Public License for more details.
 *
 *  You should have received a copy of the GNU Affero General Public License
 *  along with this program.  If not, see http://www.gnu.org/licenses/.
 */
package com.rapidminer.operator.meta;

import java.util.LinkedList;
import java.util.List;
import java.util.regex.Pattern;

import com.rapidminer.operator.Annotations;
import com.rapidminer.operator.IOObject;
import com.rapidminer.operator.OperatorDescription;
import com.rapidminer.operator.OperatorException;
import com.rapidminer.operator.nio.file.RepositoryBlobObject;
import com.rapidminer.parameter.ParameterType;
import com.rapidminer.parameter.ParameterTypeRepositoryLocation;
import com.rapidminer.repository.BlobEntry;
import com.rapidminer.repository.Entry;
import com.rapidminer.repository.Folder;
import com.rapidminer.repository.IOObjectEntry;
import com.rapidminer.repository.RepositoryException;
import com.rapidminer.repository.RepositoryLocation;

public class RepositoryIterator extends AbstractRepositoryIterator {

	public static final String PARAMETER_DIRECTORY = "repository_folder";

	private RepositoryLocation repositoryLocation;

	public RepositoryIterator(OperatorDescription description) {
		super(description);
	}

	@Override
	protected void iterate(Object currentParent, Pattern filter,
			boolean recursive, int type)
			throws OperatorException {

		if (currentParent == null) {
			currentParent = repositoryLocation;
		}
		try {
			Entry entry;
			if(currentParent instanceof RepositoryLocation){
			entry = ((RepositoryLocation) currentParent).locateEntry();
			} else {
				entry = (Entry) currentParent;
			}
			String entryType = entry.getType();
			if (entryType.equals(Folder.TYPE_NAME)) {
				for (Entry child : ((Folder) entry).getDataEntries()) {
					iterate(child, filter, recursive, type);
				}
				if (recursive) {
					for (Entry child : ((Folder) entry).getSubfolders()) {
						iterate(child, filter, recursive, type);
					}
				}

			}
			if (entryType.equals(IOObjectEntry.TYPE_NAME)
					&& type == IO_OBJECT) {
				String fileName = entry.getName();
				String fullPath = entry.getLocation().getAbsoluteLocation();
				String parentPath = entry.getContainingFolder().getName();
				if (matchesFilter(filter, fileName, fullPath, parentPath)) {
					
					IOObject data = ((IOObjectEntry)entry).retrieveData(null);					
					data.setSource(getName());
					data.getAnnotations().setAnnotation(Annotations.KEY_SOURCE, entry.getLocation().toString());
					doWorkForSingleIterationStep(fileName, fullPath,
							parentPath, data);
				}
			}
			if (entryType.equals(BlobEntry.TYPE_NAME)
					&& type == BLOB) {
				String fileName = entry.getName();
				String fullPath = entry.getLocation().getAbsoluteLocation();
				String parentPath = entry.getContainingFolder().getName();
				if (matchesFilter(filter, fileName, fullPath, parentPath)) {
					RepositoryLocation location = entry.getLocation();
					String source = location.getAbsoluteLocation();
					RepositoryBlobObject result2 = new RepositoryBlobObject(location);
					result2.getAnnotations().setAnnotation(Annotations.KEY_SOURCE,
							source);
					result2.setSource(getName());
					result2.getAnnotations().setAnnotation(Annotations.KEY_SOURCE, location.toString());
					doWorkForSingleIterationStep(fileName, fullPath,
							parentPath, result2);
				}
			}
		} catch (RepositoryException e) {
			throw new OperatorException("Error ocured during working with repository: "+e.toString(), e);
		}
	}

	@Override
	public void doWork() throws OperatorException {
		repositoryLocation = getParameterAsRepositoryLocation(PARAMETER_DIRECTORY);

		super.doWork();
	}

	@Override
	public List<ParameterType> getParameterTypes() {
		List<ParameterType> types = new LinkedList<ParameterType>();

		ParameterTypeRepositoryLocation folder = new ParameterTypeRepositoryLocation(
				PARAMETER_DIRECTORY, "Folder in the repository to iterate over", false, true, false);
		folder.setExpert(false);
		types.add(folder);

		types.addAll(super.getParameterTypes());

		return types;
	}

}