tikhomirov@463: /*
tikhomirov@463:  * Copyright (c) 2012 TMate Software Ltd
tikhomirov@463:  *  
tikhomirov@463:  * This program is free software; you can redistribute it and/or modify
tikhomirov@463:  * it under the terms of the GNU General Public License as published by
tikhomirov@463:  * the Free Software Foundation; version 2 of the License.
tikhomirov@463:  *
tikhomirov@463:  * This program is distributed in the hope that it will be useful,
tikhomirov@463:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@463:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@463:  * GNU General Public License for more details.
tikhomirov@463:  *
tikhomirov@463:  * For information on how to redistribute this software under
tikhomirov@463:  * the terms of a license other than GNU General Public License
tikhomirov@463:  * contact TMate Software at support@hg4j.com
tikhomirov@463:  */
tikhomirov@463: package org.tmatesoft.hg.repo.ext;
tikhomirov@463: 
tikhomirov@463: import java.io.BufferedReader;
tikhomirov@463: import java.io.File;
tikhomirov@463: import java.io.FileReader;
tikhomirov@463: import java.io.IOException;
tikhomirov@464: import java.util.ArrayList;
tikhomirov@464: import java.util.Collection;
tikhomirov@463: import java.util.Collections;
tikhomirov@464: import java.util.HashMap;
tikhomirov@463: import java.util.LinkedList;
tikhomirov@463: import java.util.List;
tikhomirov@464: import java.util.Map;
tikhomirov@463: 
tikhomirov@463: import org.tmatesoft.hg.core.HgInvalidControlFileException;
tikhomirov@463: import org.tmatesoft.hg.core.HgInvalidFileException;
tikhomirov@463: import org.tmatesoft.hg.core.Nodeid;
tikhomirov@463: import org.tmatesoft.hg.repo.HgInternals;
tikhomirov@463: import org.tmatesoft.hg.repo.HgRepository;
tikhomirov@463: import org.tmatesoft.hg.util.LogFacility;
tikhomirov@463: import org.tmatesoft.hg.util.Path;
tikhomirov@463: 
tikhomirov@463: /**
tikhomirov@463:  * Mercurial Queues Support. 
tikhomirov@463:  * Access to MqExtension functionality.
tikhomirov@463:  * 
tikhomirov@463:  * @author Artem Tikhomirov
tikhomirov@463:  * @author TMate Software Ltd.
tikhomirov@463:  */
tikhomirov@463: public class MqManager {
tikhomirov@463: 	
tikhomirov@463: 	private final HgRepository repo;
tikhomirov@463: 	private List<PatchRecord> applied = Collections.emptyList();
tikhomirov@463: 	private List<PatchRecord> allKnown = Collections.emptyList();
tikhomirov@464: 	private List<String> queueNames = Collections.emptyList();
tikhomirov@464: 	private String activeQueue = "patches";
tikhomirov@463: 
tikhomirov@463: 	public MqManager(HgRepository hgRepo) {
tikhomirov@463: 		repo = hgRepo;
tikhomirov@463: 	}
tikhomirov@463: 	
tikhomirov@463: 	/**
tikhomirov@463: 	 * Updates manager with up-to-date state of the mercurial queues.
tikhomirov@463: 	 */
tikhomirov@463: 	public void refresh() throws HgInvalidControlFileException {
tikhomirov@464: 		applied = allKnown = Collections.emptyList();
tikhomirov@464: 		queueNames = Collections.emptyList();
tikhomirov@463: 		File repoDir = HgInternals.getRepositoryDir(repo);
tikhomirov@463: 		final LogFacility log = HgInternals.getContext(repo).getLog();
tikhomirov@463: 		final File fileStatus = new File(repoDir, "patches/status");
tikhomirov@463: 		final File fileSeries = new File(repoDir, "patches/series");
tikhomirov@463: 		try {
tikhomirov@464: 			File queues = new File(repoDir, "patches.queues");
tikhomirov@464: 			if (queues.isFile()) {
tikhomirov@464: 				LineReader lr = new LineReader(queues, log).trimLines(true).skipEmpty(true);
tikhomirov@464: 				lr.read(new SimpleLineCollector(), queueNames = new LinkedList<String>());
tikhomirov@464: 			}
tikhomirov@464: 			File activeQueueFile = new File(repoDir, "patches.queue");
tikhomirov@464: 			ArrayList<String> contents = new ArrayList<String>();
tikhomirov@464: 			if (activeQueueFile.isFile()) {
tikhomirov@464: 				new LineReader(activeQueueFile, log).read(new SimpleLineCollector(), contents);
tikhomirov@464: 				if (contents.isEmpty()) {
tikhomirov@464: 					log.warn(getClass(), "File %s with active queue name is empty", activeQueueFile.getName());
tikhomirov@464: 					activeQueue = "patches";
tikhomirov@464: 				} else {
tikhomirov@464: 					activeQueue = contents.get(0);
tikhomirov@464: 				}
tikhomirov@464: 			} else {
tikhomirov@464: 				activeQueue = "patches";
tikhomirov@464: 			}
tikhomirov@463: 			if (fileStatus.isFile()) {
tikhomirov@463: 				new LineReader(fileStatus, log).read(new LineConsumer<List<PatchRecord>>() {
tikhomirov@463: 	
tikhomirov@463: 					public boolean consume(String line, List<PatchRecord> result) throws IOException {
tikhomirov@463: 						int sep = line.indexOf(':');
tikhomirov@463: 						if (sep == -1) {
tikhomirov@463: 							log.warn(MqManager.class, "Bad line in %s:%s", fileStatus.getPath(), line);
tikhomirov@463: 							return true;
tikhomirov@463: 						}
tikhomirov@463: 						Nodeid nid = Nodeid.fromAscii(line.substring(0, sep));
tikhomirov@463: 						String name = new String(line.substring(sep+1));
tikhomirov@463: 						result.add(new PatchRecord(nid, name, Path.create(".hg/patches/" + name)));
tikhomirov@463: 						return true;
tikhomirov@463: 					}
tikhomirov@463: 				}, applied = new LinkedList<PatchRecord>());
tikhomirov@463: 			}
tikhomirov@463: 			if (fileSeries.isFile()) {
tikhomirov@464: 				final Map<String,PatchRecord> name2patch = new HashMap<String, PatchRecord>();
tikhomirov@464: 				for (PatchRecord pr : applied) {
tikhomirov@464: 					name2patch.put(pr.getName(), pr);
tikhomirov@464: 				}
tikhomirov@464: 				LinkedList<String> knownPatchNames = new LinkedList<String>();
tikhomirov@464: 				new LineReader(fileSeries, log).read(new SimpleLineCollector(), knownPatchNames);
tikhomirov@464: 				// XXX read other queues?
tikhomirov@464: 				allKnown = new ArrayList<PatchRecord>(knownPatchNames.size());
tikhomirov@464: 				for (String name : knownPatchNames) {
tikhomirov@464: 					PatchRecord pr = name2patch.get(name);
tikhomirov@464: 					if (pr == null) {
tikhomirov@464: 						pr = new PatchRecord(null, name, Path.create(".hg/patches/" + name));
tikhomirov@463: 					}
tikhomirov@464: 					allKnown.add(pr);
tikhomirov@464: 				}
tikhomirov@463: 			}
tikhomirov@463: 		} catch (HgInvalidFileException ex) {
tikhomirov@463: 			HgInvalidControlFileException th = new HgInvalidControlFileException(ex.getMessage(), ex.getCause(), ex.getFile());
tikhomirov@463: 			th.setStackTrace(ex.getStackTrace());
tikhomirov@463: 			throw th;
tikhomirov@463: 		}
tikhomirov@463: 	}
tikhomirov@464: 	
tikhomirov@464: 	static class SimpleLineCollector implements LineConsumer<Collection<String>> {
tikhomirov@464: 
tikhomirov@464: 		public boolean consume(String line, Collection<String> result) throws IOException {
tikhomirov@464: 			result.add(line);
tikhomirov@464: 			return true;
tikhomirov@464: 		}
tikhomirov@464: 	}
tikhomirov@464: 	
tikhomirov@464: 	/**
tikhomirov@464: 	 * Number of patches not yet applied
tikhomirov@464: 	 * @return positive value when there are 
tikhomirov@464: 	 */
tikhomirov@464: 	public int getQueueSize() {
tikhomirov@464: 		return getAllKnownPatches().size() - getAppliedPatches().size();
tikhomirov@464: 	}
tikhomirov@463: 
tikhomirov@463: 	/**
tikhomirov@463: 	 * Subset of the patches from the queue that were already applied to the repository
tikhomirov@464: 	 * <p>Analog of 'hg qapplied'
tikhomirov@463: 	 * 
tikhomirov@463: 	 * <p>Clients shall call {@link #refresh()} prior to first use
tikhomirov@463: 	 * @return collection of records in no particular order, may be empty if none applied
tikhomirov@463: 	 */
tikhomirov@463: 	public List<PatchRecord> getAppliedPatches() {
tikhomirov@463: 		return Collections.unmodifiableList(applied);
tikhomirov@463: 	}
tikhomirov@463: 	
tikhomirov@463: 	/**
tikhomirov@463: 	 * All of the patches that MQ knows about for this repository
tikhomirov@463: 	 * 
tikhomirov@463: 	 * <p>Clients shall call {@link #refresh()} prior to first use
tikhomirov@463: 	 * @return collection of records in no particular order, may be empty if there are no patches in the queue
tikhomirov@463: 	 */
tikhomirov@463: 	public List<PatchRecord> getAllKnownPatches() {
tikhomirov@463: 		return Collections.unmodifiableList(allKnown);
tikhomirov@463: 	}
tikhomirov@463: 	
tikhomirov@464: 	/**
tikhomirov@464: 	 * Name of the patch queue <code>hg qqueue --active</code> which is active now.
tikhomirov@464: 	 * @return patch queue name
tikhomirov@464: 	 */
tikhomirov@464: 	public String getActiveQueueName() {
tikhomirov@464: 		return activeQueue;
tikhomirov@464: 	}
tikhomirov@464: 
tikhomirov@464: 	/**
tikhomirov@464: 	 * Patch queues known in the repository, <code>hg qqueue -l</code> analog.
tikhomirov@464: 	 * There's at least one patch queue (default one names 'patches'). Only one patch queue at a time is active.
tikhomirov@464: 	 * 
tikhomirov@464: 	 * @return names of patch queues
tikhomirov@464: 	 */
tikhomirov@464: 	public List<String> getQueueNames() {
tikhomirov@464: 		return Collections.unmodifiableList(queueNames);
tikhomirov@464: 	}
tikhomirov@464: 	
tikhomirov@463: 	public class PatchRecord {
tikhomirov@463: 		private final Nodeid nodeid;
tikhomirov@463: 		private final String name;
tikhomirov@463: 		private final Path location;
tikhomirov@464: 		
tikhomirov@464: 		// hashCode/equals might be useful if cons becomes public
tikhomirov@463: 
tikhomirov@463: 		PatchRecord(Nodeid revision, String name, Path diffLocation) {
tikhomirov@463: 			nodeid = revision;
tikhomirov@463: 			this.name = name;
tikhomirov@463: 			this.location = diffLocation;
tikhomirov@463: 		}
tikhomirov@463: 
tikhomirov@463: 		/**
tikhomirov@463: 		 * Identifies changeset of the patch that has been applied to the repository
tikhomirov@463: 		 * 
tikhomirov@463: 		 * @return changeset revision or <code>null</code> if this patch is not yet applied
tikhomirov@463: 		 */
tikhomirov@463: 		public Nodeid getRevision() {
tikhomirov@463: 			return nodeid;
tikhomirov@463: 		}
tikhomirov@463: 
tikhomirov@463: 		/**
tikhomirov@463: 		 * Identifies patch, either based on a user-supplied name (<code>hg qnew <i>patch-name</i></code>) or 
tikhomirov@463: 		 * an automatically generated name (like <code><i>revisionIndex</i>.diff</code> for imported changesets).
tikhomirov@463: 		 * Clients shall not rely on this naming scheme, though.
tikhomirov@463: 		 * 
tikhomirov@463: 		 * @return never <code>null</code>
tikhomirov@463: 		 */
tikhomirov@463: 		public String getName() {
tikhomirov@463: 			return name;
tikhomirov@463: 		}
tikhomirov@463: 		
tikhomirov@463: 		/**
tikhomirov@463: 		 * Location of diff file with the patch, relative to repository root
tikhomirov@463: 		 * @return path to the patch, never <code>null</code>
tikhomirov@463: 		 */
tikhomirov@463: 		public Path getPatchLocation() {
tikhomirov@463: 			return location;
tikhomirov@463: 		}
tikhomirov@463: 	}
tikhomirov@463: 
tikhomirov@463: 	// TODO refine API and extract into separate classes
tikhomirov@463: 
tikhomirov@463: 	interface LineConsumer<T> {
tikhomirov@463: //		boolean begin(File f, T paramObj) throws IOException;
tikhomirov@463: 		boolean consume(String line, T paramObj) throws IOException;
tikhomirov@463: //		boolean end(File f, T paramObj) throws IOException;
tikhomirov@463: 	}
tikhomirov@463: 	
tikhomirov@463: 	class LineReader {
tikhomirov@463: 		
tikhomirov@463: 		private final File file;
tikhomirov@463: 		private final LogFacility log;
tikhomirov@464: 		private boolean trimLines = true;
tikhomirov@464: 		private boolean skipEmpty = true;
tikhomirov@464: 		private String ignoreThatStars = null;
tikhomirov@463: 
tikhomirov@463: 		LineReader(File f, LogFacility logFacility) {
tikhomirov@463: 			file = f;
tikhomirov@463: 			log = logFacility;
tikhomirov@463: 		}
tikhomirov@464: 		
tikhomirov@464: 		/**
tikhomirov@464: 		 * default: <code>true</code>
tikhomirov@464: 		 * <code>false</code> to return line as is
tikhomirov@464: 		 */
tikhomirov@464: 		LineReader trimLines(boolean trim) {
tikhomirov@464: 			trimLines = trim;
tikhomirov@464: 			return this;
tikhomirov@464: 		}
tikhomirov@464: 		
tikhomirov@464: 		/**
tikhomirov@464: 		 * default: <code>true</code>
tikhomirov@464: 		 * <code>false</code> to pass empty lines to consumer
tikhomirov@464: 		 */
tikhomirov@464: 		LineReader skipEmpty(boolean skip) {
tikhomirov@464: 			skipEmpty = skip;
tikhomirov@464: 			return this;
tikhomirov@464: 		}
tikhomirov@464: 		
tikhomirov@464: 		/**
tikhomirov@464: 		 * default: doesn't skip any line.
tikhomirov@464: 		 * set e.g. to "#" or "//" to skip lines that start with such prefix
tikhomirov@464: 		 */
tikhomirov@464: 		LineReader ignoreLineComments(String lineStart) {
tikhomirov@464: 			ignoreThatStars = lineStart;
tikhomirov@464: 			return this;
tikhomirov@464: 		}
tikhomirov@463: 
tikhomirov@463: 		<T> void read(LineConsumer<T> consumer, T paramObj) throws HgInvalidFileException {
tikhomirov@463: 			BufferedReader statusFileReader = null;
tikhomirov@463: 			try {
tikhomirov@463: //				consumer.begin(file, paramObj);
tikhomirov@463: 				statusFileReader = new BufferedReader(new FileReader(file));
tikhomirov@463: 				String line;
tikhomirov@463: 				boolean ok = true;
tikhomirov@463: 				while (ok && (line = statusFileReader.readLine()) != null) {
tikhomirov@464: 					if (trimLines) {
tikhomirov@464: 						line = line.trim();
tikhomirov@464: 					}
tikhomirov@464: 					if (ignoreThatStars != null && line.startsWith(ignoreThatStars)) {
tikhomirov@464: 						continue;
tikhomirov@464: 					}
tikhomirov@464: 					if (!skipEmpty || line.length() > 0) {
tikhomirov@463: 						ok = consumer.consume(line, paramObj);
tikhomirov@463: 					}
tikhomirov@463: 				}
tikhomirov@463: 			} catch (IOException ex) {
tikhomirov@463: 				throw new HgInvalidFileException(ex.getMessage(), ex, file);
tikhomirov@463: 			} finally {
tikhomirov@463: 				try {
tikhomirov@463: 					statusFileReader.close();
tikhomirov@463: 				} catch (IOException ex) {
tikhomirov@463: 					log.warn(MqManager.class, ex, null);
tikhomirov@463: 				}
tikhomirov@463: //				try {
tikhomirov@463: //					consumer.end(file, paramObj);
tikhomirov@463: //				} catch (IOException ex) {
tikhomirov@463: //					log.warn(MqManager.class, ex, null);
tikhomirov@463: //				}
tikhomirov@463: 			}
tikhomirov@463: 		}
tikhomirov@463: 	}
tikhomirov@463: }