tikhomirov@58: /*
tikhomirov@74:  * Copyright (c) 2011 TMate Software Ltd
tikhomirov@74:  *  
tikhomirov@74:  * This program is free software; you can redistribute it and/or modify
tikhomirov@74:  * it under the terms of the GNU General Public License as published by
tikhomirov@74:  * the Free Software Foundation; version 2 of the License.
tikhomirov@74:  *
tikhomirov@74:  * This program is distributed in the hope that it will be useful,
tikhomirov@74:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@74:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@74:  * GNU General Public License for more details.
tikhomirov@74:  *
tikhomirov@74:  * For information on how to redistribute this software under
tikhomirov@74:  * the terms of a license other than GNU General Public License
tikhomirov@102:  * contact TMate Software at support@hg4j.com
tikhomirov@58:  */
tikhomirov@74: package org.tmatesoft.hg.util;
tikhomirov@58: 
tikhomirov@58: import java.io.File;
tikhomirov@58: import java.util.LinkedList;
tikhomirov@58: import java.util.NoSuchElementException;
tikhomirov@58: 
tikhomirov@58: /**
tikhomirov@74:  * 
tikhomirov@74:  * @author Artem Tikhomirov
tikhomirov@74:  * @author TMate Software Ltd.
tikhomirov@58:  */
tikhomirov@58: public class FileWalker {
tikhomirov@58: 
tikhomirov@58: 	private final File startDir;
tikhomirov@58: 	private final LinkedList<File> dirQueue;
tikhomirov@58: 	private final LinkedList<File> fileQueue;
tikhomirov@58: 	private File nextFile;
tikhomirov@58: 	private String nextPath;
tikhomirov@58: 
tikhomirov@58: 	public FileWalker(File startDir) {
tikhomirov@58: 		this.startDir = startDir;
tikhomirov@58: 		dirQueue = new LinkedList<File>();
tikhomirov@58: 		fileQueue = new LinkedList<File>();
tikhomirov@58: 		reset();
tikhomirov@58: 	}
tikhomirov@58: 
tikhomirov@58: 	public void reset() {
tikhomirov@58: 		fileQueue.clear();
tikhomirov@58: 		dirQueue.clear();
tikhomirov@58: 		dirQueue.add(startDir);
tikhomirov@58: 		nextFile = null;
tikhomirov@58: 		nextPath = null;
tikhomirov@58: 	}
tikhomirov@58: 	
tikhomirov@58: 	public boolean hasNext() {
tikhomirov@58: 		return fill();
tikhomirov@58: 	}
tikhomirov@58: 
tikhomirov@58: 	public void next() {
tikhomirov@58: 		if (!fill()) {
tikhomirov@58: 			throw new NoSuchElementException();
tikhomirov@58: 		}
tikhomirov@58: 		nextFile = fileQueue.removeFirst();
tikhomirov@58: 		nextPath = path(nextFile);
tikhomirov@58: 	}
tikhomirov@58: 
tikhomirov@58: 	public String name() {
tikhomirov@58: 		return nextPath;
tikhomirov@58: 	}
tikhomirov@58: 	
tikhomirov@58: 	public File file() {
tikhomirov@58: 		return nextFile;
tikhomirov@58: 	}
tikhomirov@58: 	
tikhomirov@58: 	private String path(File f) {
tikhomirov@58: 		// XXX LocalHgRepo#normalize
tikhomirov@58: 		String p = f.getPath().substring(startDir.getPath().length() + 1);
tikhomirov@58: 		return p.replace('\\', '/').replace("//", "/");
tikhomirov@58: 	}
tikhomirov@58: 
tikhomirov@58: 	private File[] listFiles(File f) {
tikhomirov@58: 		// in case we need to solve os-related file issues (mac with some encodings?)
tikhomirov@58: 		return f.listFiles();
tikhomirov@58: 	}
tikhomirov@58: 
tikhomirov@58: 	// return true when fill added any elements to fileQueue. 
tikhomirov@58: 	private boolean fill() {
tikhomirov@58: 		while (fileQueue.isEmpty()) {
tikhomirov@58: 			if (dirQueue.isEmpty()) {
tikhomirov@58: 				return false;
tikhomirov@58: 			}
tikhomirov@58: 			while (!dirQueue.isEmpty()) {
tikhomirov@58: 				File dir = dirQueue.removeFirst();
tikhomirov@58: 				for (File f : listFiles(dir)) {
tikhomirov@58: 					if (f.isDirectory()) {
tikhomirov@58: 						if (!".hg".equals(f.getName())) {
tikhomirov@58: 							dirQueue.addLast(f);
tikhomirov@58: 						}
tikhomirov@58: 					} else {
tikhomirov@58: 						fileQueue.addLast(f);
tikhomirov@58: 					}
tikhomirov@58: 				}
tikhomirov@58: 				break;
tikhomirov@58: 			}
tikhomirov@58: 		}
tikhomirov@58: 		return !fileQueue.isEmpty();
tikhomirov@58: 	}
tikhomirov@58: }