tikhomirov@239: /*
tikhomirov@571:  * Copyright (c) 2011-2013 TMate Software Ltd
tikhomirov@239:  *  
tikhomirov@239:  * This program is free software; you can redistribute it and/or modify
tikhomirov@239:  * it under the terms of the GNU General Public License as published by
tikhomirov@239:  * the Free Software Foundation; version 2 of the License.
tikhomirov@239:  *
tikhomirov@239:  * This program is distributed in the hope that it will be useful,
tikhomirov@239:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@239:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@239:  * GNU General Public License for more details.
tikhomirov@239:  *
tikhomirov@239:  * For information on how to redistribute this software under
tikhomirov@239:  * the terms of a license other than GNU General Public License
tikhomirov@239:  * contact TMate Software at support@hg4j.com
tikhomirov@239:  */
tikhomirov@239: package org.tmatesoft.hg.internal;
tikhomirov@239: 
tikhomirov@239: import java.io.BufferedReader;
tikhomirov@239: import java.io.File;
tikhomirov@239: import java.io.FileReader;
tikhomirov@239: import java.io.IOException;
tikhomirov@239: import java.util.Arrays;
tikhomirov@239: import java.util.Collections;
tikhomirov@239: import java.util.HashMap;
tikhomirov@239: import java.util.LinkedList;
tikhomirov@239: import java.util.List;
tikhomirov@239: import java.util.Map;
tikhomirov@239: 
tikhomirov@442: import org.tmatesoft.hg.core.Nodeid;
tikhomirov@442: import org.tmatesoft.hg.repo.HgInternals;
tikhomirov@423: import org.tmatesoft.hg.repo.HgInvalidControlFileException;
tikhomirov@239: import org.tmatesoft.hg.repo.HgRepository;
tikhomirov@239: import org.tmatesoft.hg.repo.HgSubrepoLocation;
tikhomirov@442: import org.tmatesoft.hg.util.Path;
tikhomirov@239: 
tikhomirov@239: /**
tikhomirov@239:  * 
tikhomirov@442:  * @see http://mercurial.selenic.com/wiki/SubrepoWork
tikhomirov@442:  * @see http://mercurial.selenic.com/wiki/Subrepository
tikhomirov@239:  * @author Artem Tikhomirov
tikhomirov@239:  * @author TMate Software Ltd.
tikhomirov@239:  */
tikhomirov@239: public class SubrepoManager /* XXX RepoChangeNotifier, RepoChangeListener */{
tikhomirov@239: 
tikhomirov@239: 	private final HgRepository repo;
tikhomirov@239: 	private List<HgSubrepoLocation> subRepos;
tikhomirov@239: 
tikhomirov@239: 	public SubrepoManager(HgRepository hgRepo) {
tikhomirov@239: 		assert hgRepo != null;
tikhomirov@239: 		repo = hgRepo;
tikhomirov@239: 	}
tikhomirov@239: 
tikhomirov@348: 	private List<HgSubrepoLocation> readActualState() throws HgInvalidControlFileException {
tikhomirov@239: 		File hgsubFile = new File(repo.getWorkingDir(), ".hgsub");
tikhomirov@239: 		if (!hgsubFile.canRead()) {
tikhomirov@239: 			return Collections.emptyList();
tikhomirov@239: 		}
tikhomirov@348: 		Map<String, String> state; // path -> revision
tikhomirov@348: 		File hgstateFile = null;
tikhomirov@239: 		try {
tikhomirov@348: 			hgstateFile = new File(repo.getWorkingDir(), ".hgsubstate");
tikhomirov@239: 			if (hgstateFile.canRead()) {
tikhomirov@239: 				state = readState(new BufferedReader(new FileReader(hgstateFile)));
tikhomirov@239: 			} else {
tikhomirov@239: 				state = Collections.emptyMap();
tikhomirov@239: 			}
tikhomirov@348: 		} catch (IOException ex) {
tikhomirov@348: 			throw new HgInvalidControlFileException("Subrepo state read failed", ex, hgstateFile);
tikhomirov@348: 		}
tikhomirov@348: 		try {
tikhomirov@239: 			BufferedReader br = new BufferedReader(new FileReader(hgsubFile));
tikhomirov@239: 			return readConfig(br, state);
tikhomirov@239: 		} catch (IOException ex) {
tikhomirov@348: 			throw new HgInvalidControlFileException("Subrepo state read failed", ex, hgsubFile);
tikhomirov@239: 		}
tikhomirov@239: 	}
tikhomirov@239: 
tikhomirov@239: 	private List<HgSubrepoLocation> readConfig(BufferedReader br, Map<String, String> substate) throws IOException {
tikhomirov@239: 		try {
tikhomirov@239: 			String line;
tikhomirov@239: 			LinkedList<HgSubrepoLocation> res = new LinkedList<HgSubrepoLocation>();
tikhomirov@442: 			HgInternals hgRepoInternal = new HgInternals(repo);
tikhomirov@571: 			final Path.Source pathFactory = repo.getSessionContext().getPathFactory();
tikhomirov@239: 			while ((line = br.readLine()) != null) {
tikhomirov@239: 				int sep = line.indexOf('=');
tikhomirov@239: 				if (sep == -1) {
tikhomirov@239: 					continue;
tikhomirov@239: 				}
tikhomirov@239: 				// since both key and value are referenced from HgSubrepoLocation, doesn't make sense
tikhomirov@239: 				// to have separate String instances (new String(line.substring()))
tikhomirov@239: 				String key = line.substring(0, sep).trim();
tikhomirov@239: 				String value = line.substring(sep + 1).trim();
tikhomirov@442: 				if (key.length() == 0 || value.length() == 0) {
tikhomirov@239: 					// XXX log bad line?
tikhomirov@239: 					continue;
tikhomirov@239: 				}
tikhomirov@239: 				HgSubrepoLocation.Kind kind = HgSubrepoLocation.Kind.Hg;
tikhomirov@239: 				int kindEnd = value.indexOf(']', 1);
tikhomirov@239: 				if (value.charAt(0) == '[' && kindEnd != -1) {
tikhomirov@239: 					String kindStr = value.substring(1, kindEnd);
tikhomirov@239: 					value = value.substring(kindEnd + 1);
tikhomirov@239: 					if ("svn".equals(kindStr)) {
tikhomirov@239: 						kind = HgSubrepoLocation.Kind.SVN;
tikhomirov@239: 					} else if ("git".equals(kindStr)) {
tikhomirov@239: 						kind = HgSubrepoLocation.Kind.Git;
tikhomirov@239: 					}
tikhomirov@239: 				}
tikhomirov@239: 				// TODO respect paths mappings in config file
tikhomirov@442: 				//
tikhomirov@442: 				// apparently, key value can't end with '/', `hg commit` fails if it does:
tikhomirov@442: 				// abort: path ends in directory separator: fourth/
tikhomirov@571: 				Path p = pathFactory.path(key.charAt(key.length()-1) == '/' ? key : key + '/');
tikhomirov@442: 				String revValue = substate.get(key);
tikhomirov@442: 				HgSubrepoLocation loc = hgRepoInternal.newSubrepo(p, value, kind, revValue == null ? null : Nodeid.fromAscii(revValue));
tikhomirov@239: 				res.add(loc);
tikhomirov@239: 			}
tikhomirov@239: 			return Arrays.asList(res.toArray(new HgSubrepoLocation[res.size()]));
tikhomirov@239: 		} finally {
tikhomirov@239: 			br.close();
tikhomirov@239: 		}
tikhomirov@239: 	}
tikhomirov@239: 
tikhomirov@239: 	private Map<String, String> readState(BufferedReader br) throws IOException {
tikhomirov@442: 		// TODO reuse for other files with <revision><space><value> format, like .hgtags
tikhomirov@239: 		HashMap<String, String> rv = new HashMap<String, String>();
tikhomirov@239: 		try {
tikhomirov@239: 			String line;
tikhomirov@239: 			while ((line = br.readLine()) != null) {
tikhomirov@239: 				int sep = line.trim().indexOf(' ');
tikhomirov@239: 				if (sep != -1) {
tikhomirov@239: 					rv.put(line.substring(sep+1).trim(), line.substring(0, sep).trim());
tikhomirov@239: 				}
tikhomirov@239: 			}
tikhomirov@239: 		} finally {
tikhomirov@239: 			br.close();
tikhomirov@239: 		}
tikhomirov@239: 		return rv;
tikhomirov@239: 	}
tikhomirov@239: 
tikhomirov@348: 	/*public to allow access from HgRepository, otherwise package-local*/
tikhomirov@348: 	public void read() throws HgInvalidControlFileException {
tikhomirov@348: 		subRepos = readActualState();
tikhomirov@348: 	}
tikhomirov@348: 	
tikhomirov@239: 	public List<HgSubrepoLocation> all(/*int revision, or TIP|WC*/) {
tikhomirov@348: 		assert subRepos != null;
tikhomirov@239: 		return subRepos;
tikhomirov@239: 	}
tikhomirov@239: }