tikhomirov@484: /*
tikhomirov@484:  * Copyright (c) 2012 TMate Software Ltd
tikhomirov@484:  *  
tikhomirov@484:  * This program is free software; you can redistribute it and/or modify
tikhomirov@484:  * it under the terms of the GNU General Public License as published by
tikhomirov@484:  * the Free Software Foundation; version 2 of the License.
tikhomirov@484:  *
tikhomirov@484:  * This program is distributed in the hope that it will be useful,
tikhomirov@484:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@484:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@484:  * GNU General Public License for more details.
tikhomirov@484:  *
tikhomirov@484:  * For information on how to redistribute this software under
tikhomirov@484:  * the terms of a license other than GNU General Public License
tikhomirov@484:  * contact TMate Software at support@hg4j.com
tikhomirov@484:  */
tikhomirov@484: package org.tmatesoft.hg.repo;
tikhomirov@484: 
tikhomirov@484: import java.io.File;
tikhomirov@484: import java.util.ArrayList;
tikhomirov@484: import java.util.Collection;
tikhomirov@484: import java.util.Collections;
tikhomirov@484: import java.util.LinkedHashMap;
tikhomirov@484: import java.util.Map;
tikhomirov@484: 
tikhomirov@484: import org.tmatesoft.hg.core.Nodeid;
tikhomirov@490: import org.tmatesoft.hg.internal.Internals;
tikhomirov@484: import org.tmatesoft.hg.internal.LineReader;
tikhomirov@484: import org.tmatesoft.hg.util.LogFacility;
tikhomirov@484: 
tikhomirov@484: /**
tikhomirov@484:  * 
tikhomirov@484:  * @author Artem Tikhomirov
tikhomirov@484:  * @author TMate Software Ltd.
tikhomirov@484:  */
tikhomirov@484: public final class HgBookmarks {
tikhomirov@490: 	private final Internals internalRepo;
tikhomirov@484: 	private Map<String, Nodeid> bookmarks = Collections.emptyMap();
tikhomirov@484: 	private String activeBookmark; 
tikhomirov@484: 
tikhomirov@490: 	HgBookmarks(Internals internals) {
tikhomirov@490: 		internalRepo = internals;
tikhomirov@484: 	}
tikhomirov@484: 	
tikhomirov@484: 	/*package-local*/ void read() throws HgInvalidControlFileException {
tikhomirov@501: 		final LogFacility log = internalRepo.getSessionContext().getLog();
tikhomirov@490: 		final HgRepository repo = internalRepo.getRepo();
tikhomirov@490: 		File all = internalRepo.getFileFromRepoDir(HgRepositoryFiles.Bookmarks.getName());
tikhomirov@484: 		LinkedHashMap<String, Nodeid> bm = new LinkedHashMap<String, Nodeid>();
tikhomirov@484: 		if (all.canRead()) {
tikhomirov@484: 			LineReader lr1 = new LineReader(all, log);
tikhomirov@484: 			ArrayList<String> c = new ArrayList<String>();
tikhomirov@484: 			lr1.read(new LineReader.SimpleLineCollector(), c);
tikhomirov@484: 			for (String s : c) {
tikhomirov@484: 				int x = s.indexOf(' ');
tikhomirov@484: 				try {
tikhomirov@484: 					if (x > 0) {
tikhomirov@484: 						Nodeid nid = Nodeid.fromAscii(s.substring(0, x));
tikhomirov@484: 						String name = new String(s.substring(x+1));
tikhomirov@484: 						if (repo.getChangelog().isKnown(nid)) {
tikhomirov@484: 							// copy name part not to drag complete line
tikhomirov@484: 							bm.put(name, nid);
tikhomirov@484: 						} else {
tikhomirov@484: 							log.dump(getClass(), LogFacility.Severity.Info, "Bookmark %s points to non-existent revision %s, ignored.", name, nid);
tikhomirov@484: 						}
tikhomirov@484: 					} else {
tikhomirov@484: 						log.dump(getClass(), LogFacility.Severity.Warn, "Can't parse bookmark entry: %s", s);
tikhomirov@484: 					}
tikhomirov@484: 				} catch (IllegalArgumentException ex) {
tikhomirov@484: 					log.dump(getClass(), LogFacility.Severity.Warn, ex, String.format("Can't parse bookmark entry: %s", s));
tikhomirov@484: 				}
tikhomirov@484: 			}
tikhomirov@484: 			bookmarks = bm;
tikhomirov@484: 		} else {
tikhomirov@484: 			bookmarks = Collections.emptyMap();
tikhomirov@484: 		}
tikhomirov@484: 		
tikhomirov@484: 		activeBookmark = null;
tikhomirov@490: 		File active = internalRepo.getFileFromRepoDir(HgRepositoryFiles.BookmarksCurrent.getName());
tikhomirov@484: 		if (active.canRead()) {
tikhomirov@490: 			LineReader lr2 = new LineReader(active, log);
tikhomirov@484: 			ArrayList<String> c = new ArrayList<String>(2);
tikhomirov@484: 			lr2.read(new LineReader.SimpleLineCollector(), c);
tikhomirov@484: 			if (c.size() > 0) {
tikhomirov@484: 				activeBookmark = c.get(0);
tikhomirov@484: 			}
tikhomirov@484: 		}
tikhomirov@484: 	}
tikhomirov@484: 
tikhomirov@484: 	/**
tikhomirov@484: 	 * Tell name of the active bookmark 
tikhomirov@484: 	 * @return <code>null</code> if none active
tikhomirov@484: 	 */
tikhomirov@484: 	public String getActiveBookmarkName() {
tikhomirov@484: 		return activeBookmark;
tikhomirov@484: 	}
tikhomirov@484: 
tikhomirov@484: 	/**
tikhomirov@484: 	 * Retrieve revision associated with the named bookmark.
tikhomirov@484: 	 * 
tikhomirov@484: 	 * @param bookmarkName name of the bookmark
tikhomirov@484: 	 * @return revision or <code>null</code> if bookmark is not known
tikhomirov@484: 	 */
tikhomirov@484: 	public Nodeid getRevision(String bookmarkName) {
tikhomirov@484: 		return bookmarks.get(bookmarkName);
tikhomirov@484: 	}
tikhomirov@484: 
tikhomirov@484: 	/**
tikhomirov@484: 	 * Retrieve all bookmarks known in the repository
tikhomirov@484: 	 * @return collection with names, never <code>null</code>
tikhomirov@484: 	 */
tikhomirov@484: 	public Collection<String> getAllBookmarks() {
tikhomirov@484: 		// bookmarks are initialized with atomic assignment,
tikhomirov@484: 		// hence can use view (not a synchronized copy) here
tikhomirov@484: 		return Collections.unmodifiableSet(bookmarks.keySet());
tikhomirov@484: 	}
tikhomirov@484: }