tikhomirov@13: /*
tikhomirov@74:  * Copyright (c) 2010-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@2:  */
tikhomirov@74: package org.tmatesoft.hg.repo;
tikhomirov@74: 
tikhomirov@157: import java.io.IOException;
tikhomirov@157: 
tikhomirov@157: import org.tmatesoft.hg.core.HgBadStateException;
tikhomirov@74: import org.tmatesoft.hg.core.Nodeid;
tikhomirov@157: import org.tmatesoft.hg.internal.DataAccess;
tikhomirov@196: import org.tmatesoft.hg.internal.Pool;
tikhomirov@77: import org.tmatesoft.hg.internal.RevlogStream;
tikhomirov@74: 
tikhomirov@2: 
tikhomirov@2: /**
tikhomirov@2:  *
tikhomirov@74:  * @author Artem Tikhomirov
tikhomirov@74:  * @author TMate Software Ltd.
tikhomirov@2:  */
tikhomirov@2: public class HgManifest extends Revlog {
tikhomirov@2: 
tikhomirov@13: 	/*package-local*/ HgManifest(HgRepository hgRepo, RevlogStream content) {
tikhomirov@21: 		super(hgRepo, content);
tikhomirov@13: 	}
tikhomirov@13: 
tikhomirov@19: 	public void walk(int start, int end, final Inspector inspector) {
tikhomirov@196: 		if (inspector == null) {
tikhomirov@196: 			throw new IllegalArgumentException();
tikhomirov@196: 		}
tikhomirov@196: 		content.iterate(start, end, true, new ManifestParser(inspector));
tikhomirov@19: 	}
tikhomirov@19: 
tikhomirov@19: 	public interface Inspector {
tikhomirov@19: 		boolean begin(int revision, Nodeid nid);
tikhomirov@19: 		boolean next(Nodeid nid, String fname, String flags);
tikhomirov@19: 		boolean end(int revision);
tikhomirov@2: 	}
tikhomirov@196: 
tikhomirov@196: 	private static class ManifestParser implements RevlogStream.Inspector {
tikhomirov@196: 		private boolean gtg = true; // good to go
tikhomirov@196: 		private final Inspector inspector;
tikhomirov@196: 		private final Pool<Nodeid> nodeidPool;
tikhomirov@196: 		private final Pool<String> fnamePool;
tikhomirov@196: 		private final Pool<String> flagsPool;
tikhomirov@196: 
tikhomirov@196: 		public ManifestParser(Inspector delegate) {
tikhomirov@196: 			assert delegate != null;
tikhomirov@196: 			inspector = delegate;
tikhomirov@196: 			nodeidPool = new Pool<Nodeid>();
tikhomirov@196: 			fnamePool = new Pool<String>();
tikhomirov@196: 			flagsPool = new Pool<String>();
tikhomirov@196: 		}
tikhomirov@196: 
tikhomirov@196: 		public void next(int revisionNumber, int actualLen, int baseRevision, int linkRevision, int parent1Revision, int parent2Revision, byte[] nodeid, DataAccess da) {
tikhomirov@196: 			if (!gtg) {
tikhomirov@196: 				return;
tikhomirov@196: 			}
tikhomirov@196: 			try {
tikhomirov@196: 				gtg = gtg && inspector.begin(revisionNumber, new Nodeid(nodeid, true));
tikhomirov@196: 				int i;
tikhomirov@196: 				String fname = null;
tikhomirov@196: 				String flags = null;
tikhomirov@196: 				Nodeid nid = null;
tikhomirov@196: 				byte[] data = da.byteArray();
tikhomirov@196: 				for (i = 0; gtg && i < actualLen; i++) {
tikhomirov@196: 					int x = i;
tikhomirov@196: 					for( ; data[i] != '\n' && i < actualLen; i++) {
tikhomirov@196: 						if (fname == null && data[i] == 0) {
tikhomirov@196: 							fname = fnamePool.unify(new String(data, x, i - x));
tikhomirov@196: 							x = i+1;
tikhomirov@196: 						}
tikhomirov@196: 					}
tikhomirov@196: 					if (i < actualLen) {
tikhomirov@196: 						assert data[i] == '\n'; 
tikhomirov@196: 						int nodeidLen = i - x < 40 ? i-x : 40;
tikhomirov@196: 						nid = nodeidPool.unify(Nodeid.fromAscii(data, x, nodeidLen));
tikhomirov@196: 						if (nodeidLen + x < i) {
tikhomirov@196: 							// 'x' and 'l' for executable bits and symlinks?
tikhomirov@196: 							// hg --debug manifest shows 644 for each regular file in my repo
tikhomirov@196: 							flags = flagsPool.unify(new String(data, x + nodeidLen, i-x-nodeidLen));
tikhomirov@196: 						}
tikhomirov@196: 						gtg = gtg && inspector.next(nid, fname, flags);
tikhomirov@196: 					}
tikhomirov@196: 					nid = null;
tikhomirov@196: 					fname = flags = null;
tikhomirov@196: 				}
tikhomirov@196: 				gtg = gtg && inspector.end(revisionNumber);
tikhomirov@196: 			} catch (IOException ex) {
tikhomirov@196: 				throw new HgBadStateException(ex);
tikhomirov@196: 			}
tikhomirov@196: 		}
tikhomirov@196: 	}
tikhomirov@2: }