tikhomirov@74: /*
tikhomirov@74:  * Copyright (c) 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@74:  */
tikhomirov@74: package org.tmatesoft.hg.internal;
tikhomirov@74: 
tikhomirov@114: import static org.tmatesoft.hg.internal.RequiresFile.*;
tikhomirov@74: 
tikhomirov@202: import java.io.File;
tikhomirov@202: import java.io.FileOutputStream;
tikhomirov@202: import java.io.IOException;
tikhomirov@114: import java.util.ArrayList;
tikhomirov@114: import java.util.List;
tikhomirov@114: 
tikhomirov@295: import org.tmatesoft.hg.repo.HgInternals;
tikhomirov@331: import org.tmatesoft.hg.repo.HgRepoConfig.ExtensionsSection;
tikhomirov@114: import org.tmatesoft.hg.repo.HgRepository;
tikhomirov@74: import org.tmatesoft.hg.util.PathRewrite;
tikhomirov@74: 
tikhomirov@74: /**
tikhomirov@74:  * Fields/members that shall not be visible  
tikhomirov@74:  * 
tikhomirov@74:  * @author Artem Tikhomirov
tikhomirov@74:  * @author TMate Software Ltd.
tikhomirov@74:  */
tikhomirov@74: public class Internals {
tikhomirov@74: 	
tikhomirov@74: 	private int requiresFlags = 0;
tikhomirov@114: 	private List<Filter.Factory> filterFactories;
tikhomirov@114: 	
tikhomirov@114: 
tikhomirov@114: 	public Internals() {
tikhomirov@114: 	}
tikhomirov@295: 	
tikhomirov@295: 	public void parseRequires(HgRepository hgRepo, File requiresFile) {
tikhomirov@295: 		try {
tikhomirov@295: 			new RequiresFile().parse(this, requiresFile);
tikhomirov@295: 		} catch (IOException ex) {
tikhomirov@295: 			// FIXME not quite sure error reading requires file shall be silently logged only.
tikhomirov@295: 			HgInternals.getContext(hgRepo).getLog().error(getClass(), ex, null);
tikhomirov@295: 		}
tikhomirov@295: 	}
tikhomirov@74: 
tikhomirov@83: 	public/*for tests, otherwise pkg*/ void setStorageConfig(int version, int flags) {
tikhomirov@74: 		requiresFlags = flags;
tikhomirov@74: 	}
tikhomirov@74: 
tikhomirov@74: 	// XXX perhaps, should keep both fields right here, not in the HgRepository
tikhomirov@74: 	public PathRewrite buildDataFilesHelper() {
tikhomirov@74: 		return new StoragePathHelper((requiresFlags & STORE) != 0, (requiresFlags & FNCACHE) != 0, (requiresFlags & DOTENCODE) != 0);
tikhomirov@74: 	}
tikhomirov@74: 
tikhomirov@74: 	public PathRewrite buildRepositoryFilesHelper() {
tikhomirov@74: 		if ((requiresFlags & STORE) != 0) {
tikhomirov@74: 			return new PathRewrite() {
tikhomirov@292: 				public CharSequence rewrite(CharSequence path) {
tikhomirov@74: 					return "store/" + path;
tikhomirov@74: 				}
tikhomirov@74: 			};
tikhomirov@74: 		} else {
tikhomirov@292: 			return new PathRewrite.Empty();
tikhomirov@74: 		}
tikhomirov@74: 	}
tikhomirov@114: 
tikhomirov@331: 	public List<Filter.Factory> getFilters(HgRepository hgRepo) {
tikhomirov@114: 		if (filterFactories == null) {
tikhomirov@114: 			filterFactories = new ArrayList<Filter.Factory>();
tikhomirov@331: 			ExtensionsSection cfg = hgRepo.getConfiguration().getExtensions();
tikhomirov@331: 			if (cfg.isEnabled("eol")) {
tikhomirov@114: 				NewlineFilter.Factory ff = new NewlineFilter.Factory();
tikhomirov@331: 				ff.initialize(hgRepo);
tikhomirov@114: 				filterFactories.add(ff);
tikhomirov@114: 			}
tikhomirov@331: 			if (cfg.isEnabled("keyword")) {
tikhomirov@114: 				KeywordFilter.Factory ff = new KeywordFilter.Factory();
tikhomirov@331: 				ff.initialize(hgRepo);
tikhomirov@114: 				filterFactories.add(ff);
tikhomirov@114: 			}
tikhomirov@114: 		}
tikhomirov@114: 		return filterFactories;
tikhomirov@114: 	}
tikhomirov@202: 	
tikhomirov@202: 	public void initEmptyRepository(File hgDir) throws IOException {
tikhomirov@202: 		hgDir.mkdir();
tikhomirov@202: 		FileOutputStream requiresFile = new FileOutputStream(new File(hgDir, "requires"));
tikhomirov@202: 		StringBuilder sb = new StringBuilder(40);
tikhomirov@202: 		sb.append("revlogv1\n");
tikhomirov@202: 		if ((requiresFlags & STORE) != 0) {
tikhomirov@202: 			sb.append("store\n");
tikhomirov@202: 		}
tikhomirov@202: 		if ((requiresFlags & FNCACHE) != 0) {
tikhomirov@202: 			sb.append("fncache\n");
tikhomirov@202: 		}
tikhomirov@202: 		if ((requiresFlags & DOTENCODE) != 0) {
tikhomirov@202: 			sb.append("dotencode\n");
tikhomirov@202: 		}
tikhomirov@202: 		requiresFile.write(sb.toString().getBytes());
tikhomirov@202: 		requiresFile.close();
tikhomirov@202: 		new File(hgDir, "store").mkdir(); // with that, hg verify says ok.
tikhomirov@202: 	}
tikhomirov@202: 
tikhomirov@292: 	public static boolean runningOnWindows() {
tikhomirov@292: 		return System.getProperty("os.name").indexOf("Windows") != -1;
tikhomirov@292: 	}
tikhomirov@292: 
tikhomirov@331: 	public ConfigFile readConfiguration(HgRepository hgRepo, File repoRoot) throws IOException {
tikhomirov@331: 		ConfigFile configFile = new ConfigFile();
tikhomirov@351: 		if (runningOnWindows()) {
tikhomirov@351: 			// FIXME read install-dir
tikhomirov@351: 			//
tikhomirov@351: 			// XXX perhaps, makes sense to compare getenv(USERPROFILE) and getenv(HOME) and use 
tikhomirov@351: 			// them if set (and use both if their values do not match). Only if both not set, rely to user.home?
tikhomirov@351: 			configFile.addLocation(new File(System.getProperty("user.home"), "Mercurial.ini"));
tikhomirov@351: 		} else {
tikhomirov@351: 			// FIXME read from install-root
tikhomirov@351: 			//
tikhomirov@351: 			File d = new File("/etc/mercurial/hgrc.d/");
tikhomirov@351: 			if (d.isDirectory()) {
tikhomirov@351: 				for (File f : d.listFiles()) {
tikhomirov@351: 					// XXX in fact, need to sort in alphabetical order 
tikhomirov@351: 					if (f.getName().endsWith(".rc")) {
tikhomirov@351: 						configFile.addLocation(f);
tikhomirov@351: 					}
tikhomirov@351: 				}
tikhomirov@351: 			}
tikhomirov@351: 			configFile.addLocation(new File("/etc/mercurial/hgrc"));
tikhomirov@351: 		}
tikhomirov@331: 		configFile.addLocation(new File(System.getProperty("user.home"), ".hgrc"));
tikhomirov@331: 		// last one, overrides anything else
tikhomirov@331: 		// <repo>/.hg/hgrc
tikhomirov@331: 		configFile.addLocation(new File(repoRoot, "hgrc"));
tikhomirov@331: 		return configFile;
tikhomirov@331: 	}
tikhomirov@74: }