tikhomirov@10: /* 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@0: */ tikhomirov@74: package org.tmatesoft.hg.repo; tikhomirov@74: tikhomirov@74: import java.io.File; tikhomirov@74: import java.io.IOException; tikhomirov@234: import java.io.StringReader; tikhomirov@74: import java.lang.ref.SoftReference; tikhomirov@114: import java.util.ArrayList; tikhomirov@114: import java.util.Collections; tikhomirov@74: import java.util.HashMap; tikhomirov@114: import java.util.List; tikhomirov@0: tikhomirov@234: import org.tmatesoft.hg.core.HgDataStreamException; tikhomirov@235: import org.tmatesoft.hg.core.Nodeid; tikhomirov@234: import org.tmatesoft.hg.internal.ByteArrayChannel; tikhomirov@114: import org.tmatesoft.hg.internal.ConfigFile; tikhomirov@74: import org.tmatesoft.hg.internal.DataAccessProvider; tikhomirov@231: import org.tmatesoft.hg.internal.Experimental; tikhomirov@114: import org.tmatesoft.hg.internal.Filter; tikhomirov@74: import org.tmatesoft.hg.internal.RequiresFile; tikhomirov@77: import org.tmatesoft.hg.internal.RevlogStream; tikhomirov@239: import org.tmatesoft.hg.internal.SubrepoManager; tikhomirov@234: import org.tmatesoft.hg.util.CancelledException; tikhomirov@235: import org.tmatesoft.hg.util.Pair; tikhomirov@133: import org.tmatesoft.hg.util.Path; tikhomirov@284: import org.tmatesoft.hg.util.PathPool; tikhomirov@64: import org.tmatesoft.hg.util.PathRewrite; tikhomirov@220: import org.tmatesoft.hg.util.ProgressSupport; tikhomirov@64: tikhomirov@1: tikhomirov@74: tikhomirov@0: /** tikhomirov@64: * Shall be as state-less as possible, all the caching happens outside the repo, in commands/walkers tikhomirov@74: * tikhomirov@74: * @author Artem Tikhomirov tikhomirov@74: * @author TMate Software Ltd. tikhomirov@0: */ tikhomirov@74: public final class HgRepository { tikhomirov@0: tikhomirov@197: // if new constants added, consider fixing HgInternals#wrongLocalRevision tikhomirov@197: public static final int TIP = -3; tikhomirov@68: public static final int BAD_REVISION = Integer.MIN_VALUE; tikhomirov@68: public static final int WORKING_COPY = -2; tikhomirov@252: tikhomirov@252: public static final String DEFAULT_BRANCH_NAME = "default"; tikhomirov@5: tikhomirov@2: // temp aux marker method tikhomirov@2: public static IllegalStateException notImplemented() { tikhomirov@2: return new IllegalStateException("Not implemented"); tikhomirov@2: } tikhomirov@148: tikhomirov@74: private final File repoDir; // .hg folder tikhomirov@237: private final File workingDir; // .hg/../ tikhomirov@74: private final String repoLocation; tikhomirov@74: private final DataAccessProvider dataAccess; tikhomirov@142: private final PathRewrite normalizePath; tikhomirov@74: private final PathRewrite dataPathHelper; tikhomirov@74: private final PathRewrite repoPathHelper; tikhomirov@74: tikhomirov@97: private HgChangelog changelog; tikhomirov@2: private HgManifest manifest; tikhomirov@50: private HgTags tags; tikhomirov@220: private HgBranches branches; tikhomirov@231: private HgMergeState mergeState; tikhomirov@239: private SubrepoManager subRepos; tikhomirov@220: tikhomirov@74: // XXX perhaps, shall enable caching explicitly tikhomirov@74: private final HashMap> streamsCache = new HashMap>(); tikhomirov@74: tikhomirov@74: private final org.tmatesoft.hg.internal.Internals impl = new org.tmatesoft.hg.internal.Internals(); tikhomirov@91: private HgIgnore ignore; tikhomirov@114: private ConfigFile configFile; tikhomirov@220: tikhomirov@74: HgRepository(String repositoryPath) { tikhomirov@74: repoDir = null; tikhomirov@237: workingDir = null; tikhomirov@74: repoLocation = repositoryPath; tikhomirov@74: dataAccess = null; tikhomirov@74: dataPathHelper = repoPathHelper = null; tikhomirov@142: normalizePath = null; tikhomirov@1: } tikhomirov@1: tikhomirov@148: HgRepository(String repositoryPath, File repositoryRoot) { tikhomirov@74: assert ".hg".equals(repositoryRoot.getName()) && repositoryRoot.isDirectory(); tikhomirov@148: assert repositoryPath != null; tikhomirov@148: assert repositoryRoot != null; tikhomirov@74: repoDir = repositoryRoot; tikhomirov@237: workingDir = repoDir.getParentFile(); tikhomirov@237: if (workingDir == null) { tikhomirov@237: throw new IllegalArgumentException(repoDir.toString()); tikhomirov@237: } tikhomirov@148: repoLocation = repositoryPath; tikhomirov@74: dataAccess = new DataAccessProvider(); tikhomirov@142: final boolean runningOnWindows = System.getProperty("os.name").indexOf("Windows") != -1; tikhomirov@142: if (runningOnWindows) { tikhomirov@142: normalizePath = new PathRewrite() { tikhomirov@142: tikhomirov@142: public String rewrite(String path) { tikhomirov@142: // TODO handle . and .. (although unlikely to face them from GUI client) tikhomirov@142: path = path.replace('\\', '/').replace("//", "/"); tikhomirov@142: if (path.startsWith("/")) { tikhomirov@142: path = path.substring(1); tikhomirov@142: } tikhomirov@142: return path; tikhomirov@142: } tikhomirov@142: }; tikhomirov@142: } else { tikhomirov@142: normalizePath = new PathRewrite.Empty(); // or strip leading slash, perhaps? tikhomirov@142: } tikhomirov@74: parseRequires(); tikhomirov@74: dataPathHelper = impl.buildDataFilesHelper(); tikhomirov@74: repoPathHelper = impl.buildRepositoryFilesHelper(); tikhomirov@1: } tikhomirov@0: tikhomirov@145: @Override tikhomirov@145: public String toString() { tikhomirov@145: return getClass().getSimpleName() + "[" + getLocation() + (isInvalid() ? "(BAD)" : "") + "]"; tikhomirov@145: } tikhomirov@74: tikhomirov@74: public String getLocation() { tikhomirov@74: return repoLocation; tikhomirov@74: } tikhomirov@74: tikhomirov@74: public boolean isInvalid() { tikhomirov@74: return repoDir == null || !repoDir.exists() || !repoDir.isDirectory(); tikhomirov@74: } tikhomirov@74: tikhomirov@97: public HgChangelog getChangelog() { tikhomirov@0: if (this.changelog == null) { tikhomirov@74: String storagePath = repoPathHelper.rewrite("00changelog.i"); tikhomirov@202: RevlogStream content = resolve(Path.create(storagePath), true); tikhomirov@97: this.changelog = new HgChangelog(this, content); tikhomirov@0: } tikhomirov@0: return this.changelog; tikhomirov@0: } tikhomirov@2: tikhomirov@74: public HgManifest getManifest() { tikhomirov@2: if (this.manifest == null) { tikhomirov@202: RevlogStream content = resolve(Path.create(repoPathHelper.rewrite("00manifest.i")), true); tikhomirov@13: this.manifest = new HgManifest(this, content); tikhomirov@2: } tikhomirov@2: return this.manifest; tikhomirov@2: } tikhomirov@50: tikhomirov@231: public HgTags getTags() { tikhomirov@50: if (tags == null) { tikhomirov@234: tags = new HgTags(this); tikhomirov@104: try { tikhomirov@234: HgDataFile hgTags = getFileNode(".hgtags"); tikhomirov@234: if (hgTags.exists()) { tikhomirov@234: for (int i = 0; i <= hgTags.getLastRevision(); i++) { // FIXME in fact, would be handy to have walk(start,end) tikhomirov@234: // method for data files as well, though it looks odd. tikhomirov@234: try { tikhomirov@234: ByteArrayChannel sink = new ByteArrayChannel(); tikhomirov@234: hgTags.content(i, sink); tikhomirov@234: final String content = new String(sink.toArray(), "UTF8"); tikhomirov@234: tags.readGlobal(new StringReader(content)); tikhomirov@234: } catch (CancelledException ex) { tikhomirov@234: ex.printStackTrace(); // IGNORE, can't happen, we did not configure cancellation tikhomirov@234: } catch (HgDataStreamException ex) { tikhomirov@234: ex.printStackTrace(); // FIXME need to react tikhomirov@234: } catch (IOException ex) { tikhomirov@234: // UnsupportedEncodingException can't happen (UTF8) tikhomirov@234: // only from readGlobal. Need to reconsider exceptions thrown from there tikhomirov@234: ex.printStackTrace(); // XXX need to decide what to do this. failure to read single revision shall not break complete cycle tikhomirov@234: } tikhomirov@234: } tikhomirov@234: } tikhomirov@252: tags.readGlobal(new File(getWorkingDir(), ".hgtags")); // XXX replace with HgDataFile.workingCopy tikhomirov@104: tags.readLocal(new File(repoDir, "localtags")); tikhomirov@104: } catch (IOException ex) { tikhomirov@104: ex.printStackTrace(); // FIXME log or othewise report tikhomirov@104: } tikhomirov@50: } tikhomirov@50: return tags; tikhomirov@50: } tikhomirov@50: tikhomirov@231: public HgBranches getBranches() { tikhomirov@220: if (branches == null) { tikhomirov@220: branches = new HgBranches(this); tikhomirov@220: branches.collect(ProgressSupport.Factory.get(null)); tikhomirov@220: } tikhomirov@220: return branches; tikhomirov@220: } tikhomirov@231: tikhomirov@231: @Experimental(reason="Perhaps, shall not cache instance, and provide loadMergeState as it may change often") tikhomirov@231: public HgMergeState getMergeState() { tikhomirov@231: if (mergeState == null) { tikhomirov@231: mergeState = new HgMergeState(this); tikhomirov@231: } tikhomirov@231: return mergeState; tikhomirov@231: } tikhomirov@220: tikhomirov@74: public HgDataFile getFileNode(String path) { tikhomirov@74: String nPath = normalizePath.rewrite(path); tikhomirov@74: String storagePath = dataPathHelper.rewrite(nPath); tikhomirov@202: RevlogStream content = resolve(Path.create(storagePath), false); tikhomirov@115: Path p = Path.create(nPath); tikhomirov@115: if (content == null) { tikhomirov@115: return new HgDataFile(this, p); tikhomirov@115: } tikhomirov@115: return new HgDataFile(this, p, content); tikhomirov@74: } tikhomirov@1: tikhomirov@74: public HgDataFile getFileNode(Path path) { tikhomirov@77: String storagePath = dataPathHelper.rewrite(path.toString()); tikhomirov@202: RevlogStream content = resolve(Path.create(storagePath), false); tikhomirov@115: // XXX no content when no file? or HgDataFile.exists() to detect that? tikhomirov@115: if (content == null) { tikhomirov@115: return new HgDataFile(this, path); tikhomirov@115: } tikhomirov@74: return new HgDataFile(this, path, content); tikhomirov@74: } tikhomirov@2: tikhomirov@142: /* clients need to rewrite path from their FS to a repository-friendly paths, and, perhaps, vice versa*/ tikhomirov@142: public PathRewrite getToRepoPathHelper() { tikhomirov@74: return normalizePath; tikhomirov@74: } tikhomirov@284: tikhomirov@284: /** tikhomirov@284: * @return pair of values, {@link Pair#first()} and {@link Pair#second()} are respective parents, never null. tikhomirov@284: */ tikhomirov@235: public Pair getWorkingCopyParents() { tikhomirov@284: return HgDirstate.readParents(this, new File(repoDir, "dirstate")); tikhomirov@235: } tikhomirov@252: tikhomirov@252: /** tikhomirov@252: * @return name of the branch associated with working directory, never null. tikhomirov@252: */ tikhomirov@252: public String getWorkingCopyBranchName() { tikhomirov@284: return HgDirstate.readBranch(this); tikhomirov@252: } tikhomirov@2: tikhomirov@237: /** tikhomirov@237: * @return location where user files (shall) reside tikhomirov@237: */ tikhomirov@237: public File getWorkingDir() { tikhomirov@237: return workingDir; tikhomirov@237: } tikhomirov@239: tikhomirov@239: /** tikhomirov@239: * Provides access to sub-repositories defined in this repository. Enumerated sub-repositories are those directly tikhomirov@239: * known, not recursive collection of all nested sub-repositories. tikhomirov@239: * @return list of all known sub-repositories in this repository, or empty list if none found. tikhomirov@239: */ tikhomirov@239: public List getSubrepositories() { tikhomirov@239: if (subRepos == null) { tikhomirov@239: subRepos = new SubrepoManager(this); tikhomirov@239: } tikhomirov@239: return subRepos.all(); tikhomirov@239: } tikhomirov@237: tikhomirov@237: // shall be of use only for internal classes tikhomirov@74: /*package-local*/ File getRepositoryRoot() { tikhomirov@74: return repoDir; tikhomirov@74: } tikhomirov@74: tikhomirov@74: // XXX package-local, unless there are cases when required from outside (guess, working dir/revision walkers may hide dirstate access and no public visibility needed) tikhomirov@280: // XXX consider passing Path pool or factory to produce (shared) Path instead of Strings tikhomirov@284: /*package-local*/ final HgDirstate loadDirstate(PathPool pathPool) { tikhomirov@284: return new HgDirstate(this, new File(repoDir, "dirstate"), pathPool); tikhomirov@74: } tikhomirov@74: tikhomirov@74: // package-local, see comment for loadDirstate tikhomirov@91: /*package-local*/ final HgIgnore getIgnore() { tikhomirov@91: // TODO read config for additional locations tikhomirov@91: if (ignore == null) { tikhomirov@91: ignore = new HgIgnore(); tikhomirov@91: try { tikhomirov@252: File ignoreFile = new File(getWorkingDir(), ".hgignore"); tikhomirov@91: ignore.read(ignoreFile); tikhomirov@91: } catch (IOException ex) { tikhomirov@91: ex.printStackTrace(); // log warn tikhomirov@91: } tikhomirov@91: } tikhomirov@91: return ignore; tikhomirov@74: } tikhomirov@74: tikhomirov@74: /*package-local*/ DataAccessProvider getDataAccess() { tikhomirov@74: return dataAccess; tikhomirov@74: } tikhomirov@74: tikhomirov@2: /** tikhomirov@2: * Perhaps, should be separate interface, like ContentLookup tikhomirov@74: * path - repository storage path (i.e. one usually with .i or .d) tikhomirov@2: */ tikhomirov@202: /*package-local*/ RevlogStream resolve(Path path, boolean shallFakeNonExistent) { tikhomirov@74: final SoftReference ref = streamsCache.get(path); tikhomirov@74: RevlogStream cached = ref == null ? null : ref.get(); tikhomirov@74: if (cached != null) { tikhomirov@74: return cached; tikhomirov@74: } tikhomirov@74: File f = new File(repoDir, path.toString()); tikhomirov@74: if (f.exists()) { tikhomirov@74: RevlogStream s = new RevlogStream(dataAccess, f); tikhomirov@74: streamsCache.put(path, new SoftReference(s)); tikhomirov@74: return s; tikhomirov@202: } else { tikhomirov@202: if (shallFakeNonExistent) { tikhomirov@202: try { tikhomirov@202: File fake = File.createTempFile(f.getName(), null); tikhomirov@202: fake.deleteOnExit(); tikhomirov@202: return new RevlogStream(dataAccess, fake); tikhomirov@202: } catch (IOException ex) { tikhomirov@202: ex.printStackTrace(); // FIXME report in debug tikhomirov@202: } tikhomirov@202: } tikhomirov@74: } tikhomirov@74: return null; // XXX empty stream instead? tikhomirov@74: } tikhomirov@114: tikhomirov@114: // can't expose internal class, otherwise seems reasonable to have it in API tikhomirov@114: /*package-local*/ ConfigFile getConfigFile() { tikhomirov@114: if (configFile == null) { tikhomirov@114: configFile = impl.newConfigFile(); tikhomirov@114: configFile.addLocation(new File(System.getProperty("user.home"), ".hgrc")); tikhomirov@114: // last one, overrides anything else tikhomirov@114: // /.hg/hgrc tikhomirov@114: configFile.addLocation(new File(getRepositoryRoot(), "hgrc")); tikhomirov@114: } tikhomirov@114: return configFile; tikhomirov@114: } tikhomirov@114: tikhomirov@114: /*package-local*/ List getFiltersFromRepoToWorkingDir(Path p) { tikhomirov@114: return instantiateFilters(p, new Filter.Options(Filter.Direction.FromRepo)); tikhomirov@114: } tikhomirov@114: tikhomirov@114: /*package-local*/ List getFiltersFromWorkingDirToRepo(Path p) { tikhomirov@114: return instantiateFilters(p, new Filter.Options(Filter.Direction.ToRepo)); tikhomirov@114: } tikhomirov@237: tikhomirov@237: /*package-local*/ File getFile(HgDataFile dataFile) { tikhomirov@237: return new File(getWorkingDir(), dataFile.getPath().toString()); tikhomirov@237: } tikhomirov@114: tikhomirov@114: private List instantiateFilters(Path p, Filter.Options opts) { tikhomirov@114: List factories = impl.getFilters(this, getConfigFile()); tikhomirov@114: if (factories.isEmpty()) { tikhomirov@114: return Collections.emptyList(); tikhomirov@114: } tikhomirov@114: ArrayList rv = new ArrayList(factories.size()); tikhomirov@114: for (Filter.Factory ff : factories) { tikhomirov@114: Filter f = ff.create(p, opts); tikhomirov@114: if (f != null) { tikhomirov@114: rv.add(f); tikhomirov@114: } tikhomirov@114: } tikhomirov@114: return rv; tikhomirov@114: } tikhomirov@74: tikhomirov@74: private void parseRequires() { tikhomirov@74: new RequiresFile().parse(impl, new File(repoDir, "requires")); tikhomirov@74: } tikhomirov@114: tikhomirov@0: }