tikhomirov@64: /*
tikhomirov@64:  * Copyright (c) 2011 TMate Software Ltd
tikhomirov@64:  *  
tikhomirov@64:  * This program is free software; you can redistribute it and/or modify
tikhomirov@64:  * it under the terms of the GNU General Public License as published by
tikhomirov@64:  * the Free Software Foundation; version 2 of the License.
tikhomirov@64:  *
tikhomirov@64:  * This program is distributed in the hope that it will be useful,
tikhomirov@64:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@64:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@64:  * GNU General Public License for more details.
tikhomirov@64:  *
tikhomirov@64:  * For information on how to redistribute this software under
tikhomirov@64:  * the terms of a license other than GNU General Public License
tikhomirov@64:  * contact TMate Software at support@svnkit.com
tikhomirov@64:  */
tikhomirov@64: package org.tmatesoft.hg.core;
tikhomirov@64: 
tikhomirov@64: import static com.tmate.hgkit.ll.HgRepository.TIP;
tikhomirov@64: 
tikhomirov@64: import java.util.Calendar;
tikhomirov@64: import java.util.Collections;
tikhomirov@64: import java.util.ConcurrentModificationException;
tikhomirov@64: import java.util.LinkedList;
tikhomirov@64: import java.util.List;
tikhomirov@64: import java.util.Set;
tikhomirov@64: import java.util.TreeSet;
tikhomirov@64: 
tikhomirov@64: import org.tmatesoft.hg.util.PathPool;
tikhomirov@64: 
tikhomirov@64: import com.tmate.hgkit.ll.Changeset;
tikhomirov@64: import com.tmate.hgkit.ll.HgRepository;
tikhomirov@64: import com.tmate.hgkit.ll.Nodeid;
tikhomirov@64: import com.tmate.hgkit.ll.StatusCollector;
tikhomirov@64: 
tikhomirov@64: /**
tikhomirov@64:  * <pre>
tikhomirov@70:  *   new LogCommand().limit(20).branch("maintenance-2.1").user("me").execute(new MyHandler());
tikhomirov@64:  * </pre>
tikhomirov@64:  * Not thread-safe (each thread has to use own {@link LogCommand} instance).
tikhomirov@64:  * 
tikhomirov@64:  * @author Artem Tikhomirov
tikhomirov@64:  * @author TMate Software Ltd.
tikhomirov@64:  */
tikhomirov@64: public class LogCommand implements Changeset.Inspector {
tikhomirov@64: 
tikhomirov@64: 	private final HgRepository repo;
tikhomirov@64: 	private Set<String> users;
tikhomirov@64: 	private Set<String> branches;
tikhomirov@64: 	private int limit = 0, count = 0;
tikhomirov@64: 	private int startRev = 0, endRev = TIP;
tikhomirov@64: 	private Handler delegate;
tikhomirov@64: 	private Calendar date;
tikhomirov@64: 	private Cset changeset;
tikhomirov@64: 
tikhomirov@64: 	public LogCommand(HgRepository hgRepo) {
tikhomirov@64: 		this.repo = hgRepo;
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	/**
tikhomirov@64: 	 * Limit search to specified user. Multiple user names may be specified.
tikhomirov@64: 	 * @param user - full or partial name of the user, case-insensitive, non-null.
tikhomirov@64: 	 * @return <code>this</code> instance for convenience
tikhomirov@64: 	 */
tikhomirov@64: 	public LogCommand user(String user) {
tikhomirov@64: 		if (user == null) {
tikhomirov@64: 			throw new IllegalArgumentException();
tikhomirov@64: 		}
tikhomirov@64: 		if (users == null) {
tikhomirov@64: 			users = new TreeSet<String>();
tikhomirov@64: 		}
tikhomirov@64: 		users.add(user.toLowerCase());
tikhomirov@64: 		return this;
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	/**
tikhomirov@64: 	 * Limit search to specified branch. Multiple branch specification possible (changeset from any of these 
tikhomirov@64: 	 * would be included in result). If unspecified, all branches are considered.
tikhomirov@64: 	 * @param branch - branch name, case-sensitive, non-null.
tikhomirov@64: 	 * @return <code>this</code> instance for convenience
tikhomirov@64: 	 */
tikhomirov@64: 	public LogCommand branch(String branch) {
tikhomirov@64: 		if (branch == null) {
tikhomirov@64: 			throw new IllegalArgumentException();
tikhomirov@64: 		}
tikhomirov@64: 		if (branches == null) {
tikhomirov@64: 			branches = new TreeSet<String>();
tikhomirov@64: 		}
tikhomirov@64: 		branches.add(branch);
tikhomirov@64: 		return this;
tikhomirov@64: 	}
tikhomirov@64: 	
tikhomirov@64: 	// limit search to specific date
tikhomirov@64: 	// multiple?
tikhomirov@64: 	public LogCommand date(Calendar date) {
tikhomirov@64: 		this.date = date;
tikhomirov@64: 		// FIXME implement
tikhomirov@64: 		// isSet(field) - false => don't use in detection of 'same date'
tikhomirov@64: 		throw HgRepository.notImplemented();
tikhomirov@64: 	}
tikhomirov@64: 	
tikhomirov@64: 	/**
tikhomirov@64: 	 * 
tikhomirov@64: 	 * @param num - number of changeset to produce. Pass 0 to clear the limit. 
tikhomirov@64: 	 * @return <code>this</code> instance for convenience
tikhomirov@64: 	 */
tikhomirov@64: 	public LogCommand limit(int num) {
tikhomirov@64: 		limit = num;
tikhomirov@64: 		return this;
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	/**
tikhomirov@64: 	 * Limit to specified subset of Changelog, [min(rev1,rev2), max(rev1,rev2)], inclusive.
tikhomirov@64: 	 * Revision may be specified with {@link HgRepository#TIP}  
tikhomirov@64: 	 * @param rev1
tikhomirov@64: 	 * @param rev2
tikhomirov@64: 	 * @return <code>this</code> instance for convenience
tikhomirov@64: 	 */
tikhomirov@64: 	public LogCommand range(int rev1, int rev2) {
tikhomirov@64: 		if (rev1 != TIP && rev2 != TIP) {
tikhomirov@64: 			startRev = rev2 < rev1 ? rev2 : rev1;
tikhomirov@64: 			endRev = startRev == rev2 ? rev1 : rev2;
tikhomirov@64: 		} else if (rev1 == TIP && rev2 != TIP) {
tikhomirov@64: 			startRev = rev2;
tikhomirov@64: 			endRev = rev1;
tikhomirov@64: 		} else {
tikhomirov@64: 			startRev = rev1;
tikhomirov@64: 			endRev = rev2;
tikhomirov@64: 		}
tikhomirov@64: 		return this;
tikhomirov@64: 	}
tikhomirov@64: 	
tikhomirov@64: 	// multiple? Bad idea, would need to include extra method into Handler to tell start of next file
tikhomirov@64: 	public LogCommand file(Path file) {
tikhomirov@64: 		// implicit --follow in this case
tikhomirov@64: 		throw HgRepository.notImplemented();
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	/**
tikhomirov@64: 	 * Similar to {@link #execute(com.tmate.hgkit.ll.Changeset.Inspector)}, collects and return result as a list.
tikhomirov@64: 	 */
tikhomirov@64: 	public List<Cset> execute() {
tikhomirov@64: 		CollectHandler collector = new CollectHandler();
tikhomirov@64: 		execute(collector);
tikhomirov@64: 		return collector.getChanges();
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	/**
tikhomirov@64: 	 * 
tikhomirov@64: 	 * @param inspector
tikhomirov@64: 	 * @throws IllegalArgumentException when inspector argument is null
tikhomirov@64: 	 * @throws ConcurrentModificationException if this log command instance is already running
tikhomirov@64: 	 */
tikhomirov@64: 	public void execute(Handler handler) {
tikhomirov@64: 		if (handler == null) {
tikhomirov@64: 			throw new IllegalArgumentException();
tikhomirov@64: 		}
tikhomirov@64: 		if (delegate != null) {
tikhomirov@64: 			throw new ConcurrentModificationException();
tikhomirov@64: 		}
tikhomirov@64: 		try {
tikhomirov@64: 			delegate = handler;
tikhomirov@64: 			count = 0;
tikhomirov@64: 			changeset = new Cset(new StatusCollector(repo), new PathPool(repo.getPathHelper()));
tikhomirov@64: 			repo.getChangelog().range(startRev, endRev, this);
tikhomirov@64: 		} finally {
tikhomirov@64: 			delegate = null;
tikhomirov@64: 			changeset = null;
tikhomirov@64: 		}
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	//
tikhomirov@64: 	
tikhomirov@64: 	public void next(int revisionNumber, Nodeid nodeid, Changeset cset) {
tikhomirov@64: 		if (limit > 0 && count >= limit) {
tikhomirov@64: 			return;
tikhomirov@64: 		}
tikhomirov@64: 		if (branches != null && !branches.contains(cset.branch())) {
tikhomirov@64: 			return;
tikhomirov@64: 		}
tikhomirov@64: 		if (users != null) {
tikhomirov@64: 			String csetUser = cset.user().toLowerCase();
tikhomirov@64: 			boolean found = false;
tikhomirov@64: 			for (String u : users) {
tikhomirov@64: 				if (csetUser.indexOf(u) != -1) {
tikhomirov@64: 					found = true;
tikhomirov@64: 					break;
tikhomirov@64: 				}
tikhomirov@64: 			}
tikhomirov@64: 			if (!found) {
tikhomirov@64: 				return;
tikhomirov@64: 			}
tikhomirov@64: 		}
tikhomirov@64: 		if (date != null) {
tikhomirov@64: 			// FIXME
tikhomirov@64: 		}
tikhomirov@64: 		count++;
tikhomirov@64: 		changeset.init(revisionNumber, nodeid, cset);
tikhomirov@64: 		delegate.next(changeset);
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	public interface Handler {
tikhomirov@64: 		/**
tikhomirov@64: 		 * @param changeset not necessarily a distinct instance each time, {@link Cset#clone() clone()} if need a copy.
tikhomirov@64: 		 */
tikhomirov@64: 		void next(Cset changeset);
tikhomirov@64: 	}
tikhomirov@64: 	
tikhomirov@64: 	public static class CollectHandler implements Handler {
tikhomirov@64: 		private final List<Cset> result = new LinkedList<Cset>();
tikhomirov@64: 
tikhomirov@64: 		public List<Cset> getChanges() {
tikhomirov@64: 			return Collections.unmodifiableList(result);
tikhomirov@64: 		}
tikhomirov@64: 
tikhomirov@64: 		public void next(Cset changeset) {
tikhomirov@64: 			result.add(changeset.clone());
tikhomirov@64: 		}
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	public static final class FileRevision {
tikhomirov@64: 		private final HgRepository repo;
tikhomirov@64: 		private final Nodeid revision;
tikhomirov@64: 		private final Path path;
tikhomirov@64: 		
tikhomirov@64: 		public FileRevision(HgRepository hgRepo, Nodeid rev, Path p) {
tikhomirov@64: 			if (hgRepo == null || rev == null || p == null) {
tikhomirov@64: 				throw new IllegalArgumentException();
tikhomirov@64: 			}
tikhomirov@64: 			repo = hgRepo;
tikhomirov@64: 			revision = rev;
tikhomirov@64: 			path = p;
tikhomirov@64: 		}
tikhomirov@64: 		
tikhomirov@64: 		public Path getPath() {
tikhomirov@64: 			return path;
tikhomirov@64: 		}
tikhomirov@64: 		public Nodeid getRevision() {
tikhomirov@64: 			return revision;
tikhomirov@64: 		}
tikhomirov@64: 		public byte[] getContent() {
tikhomirov@64: 			// XXX Content wrapper, to allow formats other than byte[], e.g. Stream, DataAccess, etc?
tikhomirov@64: 			return repo.getFileNode(path).content();
tikhomirov@64: 		}
tikhomirov@64: 	}
tikhomirov@64: }