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 java.util.ArrayList;
tikhomirov@64: import java.util.Collections;
tikhomirov@64: import java.util.List;
tikhomirov@64: 
tikhomirov@64: import org.tmatesoft.hg.core.LogCommand.FileRevision;
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:  * TODO rename to Changeset along with original Changeset moved to .repo and renamed to HgChangeset?
tikhomirov@64:  * Not thread-safe, don't try to read from different threads
tikhomirov@64:  * 
tikhomirov@64:  * @author Artem Tikhomirov
tikhomirov@64:  * @author TMate Software Ltd.
tikhomirov@64:  */
tikhomirov@64: public class Cset implements Cloneable {
tikhomirov@64: 	private final StatusCollector statusHelper;
tikhomirov@64: 	private final PathPool pathHelper;
tikhomirov@64: 
tikhomirov@64: 	//
tikhomirov@64: 	private Changeset changeset;
tikhomirov@64: 	private Nodeid nodeid;
tikhomirov@64: 
tikhomirov@64: 	//
tikhomirov@64: 	private List<FileRevision> modifiedFiles, addedFiles;
tikhomirov@64: 	private List<Path> deletedFiles;
tikhomirov@64: 	private int revNumber;
tikhomirov@64: 
tikhomirov@64: 	// XXX consider CommandContext with StatusCollector, PathPool etc. Commands optionally get CC through a cons or create new
tikhomirov@64: 	// and pass it around
tikhomirov@64: 	/*package-local*/Cset(StatusCollector statusCollector, PathPool pathPool) {
tikhomirov@64: 		statusHelper = statusCollector;
tikhomirov@64: 		pathHelper = pathPool;
tikhomirov@64: 	}
tikhomirov@64: 	
tikhomirov@64: 	/*package-local*/
tikhomirov@64: 	void init(int localRevNumber, Nodeid nid, Changeset rawChangeset) {
tikhomirov@64: 		revNumber = localRevNumber;
tikhomirov@64: 		nodeid = nid;
tikhomirov@64: 		changeset = rawChangeset;
tikhomirov@64: 	}
tikhomirov@64: 	
tikhomirov@64: 	public String getUser() {
tikhomirov@64: 		return changeset.user();
tikhomirov@64: 	}
tikhomirov@64: 	public String getComment() {
tikhomirov@64: 		return changeset.comment();
tikhomirov@64: 	}
tikhomirov@64: 	public String getBranch() {
tikhomirov@64: 		return changeset.branch();
tikhomirov@64: 	}
tikhomirov@64: 	public String getDate() {
tikhomirov@64: 		return changeset.dateString();
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	public List<Path> getAffectedFiles() {
tikhomirov@64: 		ArrayList<Path> rv = new ArrayList<Path>(changeset.files().size());
tikhomirov@64: 		for (String name : changeset.files()) {
tikhomirov@64: 			rv.add(pathHelper.path(name));
tikhomirov@64: 		}
tikhomirov@64: 		return rv;
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	public List<FileRevision> getModifiedFiles() {
tikhomirov@64: 		if (modifiedFiles == null) {
tikhomirov@64: 			initFileChanges();
tikhomirov@64: 		}
tikhomirov@64: 		return modifiedFiles;
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	public List<FileRevision> getAddedFiles() {
tikhomirov@64: 		if (addedFiles == null) {
tikhomirov@64: 			initFileChanges();
tikhomirov@64: 		}
tikhomirov@64: 		return addedFiles;
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	public List<Path> getRemovedFiles() {
tikhomirov@64: 		if (deletedFiles == null) {
tikhomirov@64: 			initFileChanges();
tikhomirov@64: 		}
tikhomirov@64: 		return deletedFiles;
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	@Override
tikhomirov@64: 	public Cset clone() {
tikhomirov@64: 		try {
tikhomirov@64: 			Cset copy = (Cset) super.clone();
tikhomirov@64: 			copy.changeset = changeset.clone();
tikhomirov@64: 			return copy;
tikhomirov@64: 		} catch (CloneNotSupportedException ex) {
tikhomirov@64: 			throw new InternalError(ex.toString());
tikhomirov@64: 		}
tikhomirov@64: 	}
tikhomirov@64: 
tikhomirov@64: 	private /*synchronized*/ void initFileChanges() {
tikhomirov@64: 		ArrayList<Path> deleted = new ArrayList<Path>();
tikhomirov@64: 		ArrayList<FileRevision> modified = new ArrayList<FileRevision>();
tikhomirov@64: 		ArrayList<FileRevision> added = new ArrayList<FileRevision>();
tikhomirov@64: 		StatusCollector.Record r = new StatusCollector.Record();
tikhomirov@64: 		statusHelper.change(revNumber, r);
tikhomirov@64: 		final HgRepository repo = statusHelper.getRepo();
tikhomirov@64: 		for (String s : r.getModified()) {
tikhomirov@64: 			Path p = pathHelper.path(s);
tikhomirov@64: 			Nodeid nid = r.nodeidAfterChange(s);
tikhomirov@64: 			if (nid == null) {
tikhomirov@64: 				throw new IllegalArgumentException();
tikhomirov@64: 			}
tikhomirov@64: 			modified.add(new FileRevision(repo, nid, p));
tikhomirov@64: 		}
tikhomirov@64: 		for (String s : r.getAdded()) {
tikhomirov@64: 			Path p = pathHelper.path(s);
tikhomirov@64: 			Nodeid nid = r.nodeidAfterChange(s);
tikhomirov@64: 			if (nid == null) {
tikhomirov@64: 				throw new IllegalArgumentException();
tikhomirov@64: 			}
tikhomirov@64: 			added.add(new FileRevision(repo, nid, p));
tikhomirov@64: 		}
tikhomirov@64: 		for (String s : r.getRemoved()) {
tikhomirov@64: 			deleted.add(pathHelper.path(s));
tikhomirov@64: 		}
tikhomirov@64: 		modified.trimToSize();
tikhomirov@64: 		added.trimToSize();
tikhomirov@64: 		deleted.trimToSize();
tikhomirov@64: 		modifiedFiles = Collections.unmodifiableList(modified);
tikhomirov@64: 		addedFiles = Collections.unmodifiableList(added);
tikhomirov@64: 		deletedFiles = Collections.unmodifiableList(deleted);
tikhomirov@64: 	}
tikhomirov@64: }