tikhomirov@536: /*
tikhomirov@536:  * Copyright (c) 2013 TMate Software Ltd
tikhomirov@536:  *  
tikhomirov@536:  * This program is free software; you can redistribute it and/or modify
tikhomirov@536:  * it under the terms of the GNU General Public License as published by
tikhomirov@536:  * the Free Software Foundation; version 2 of the License.
tikhomirov@536:  *
tikhomirov@536:  * This program is distributed in the hope that it will be useful,
tikhomirov@536:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@536:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@536:  * GNU General Public License for more details.
tikhomirov@536:  *
tikhomirov@536:  * For information on how to redistribute this software under
tikhomirov@536:  * the terms of a license other than GNU General Public License
tikhomirov@536:  * contact TMate Software at support@hg4j.com
tikhomirov@536:  */
tikhomirov@536: package org.tmatesoft.hg.test;
tikhomirov@536: 
tikhomirov@536: import static org.junit.Assert.assertEquals;
tikhomirov@536: import static org.junit.Assert.assertTrue;
tikhomirov@623: import static org.tmatesoft.hg.util.LogFacility.Severity.Debug;
tikhomirov@536: 
tikhomirov@536: import java.io.File;
tikhomirov@536: import java.io.FileOutputStream;
tikhomirov@559: import java.io.FileWriter;
tikhomirov@536: import java.io.IOException;
tikhomirov@536: import java.util.ArrayList;
tikhomirov@623: import java.util.Arrays;
tikhomirov@623: import java.util.Iterator;
tikhomirov@623: import java.util.LinkedList;
tikhomirov@637: import java.util.List;
tikhomirov@536: 
tikhomirov@635: import junit.framework.Assert;
tikhomirov@635: 
tikhomirov@623: import org.tmatesoft.hg.core.HgException;
tikhomirov@637: import org.tmatesoft.hg.core.HgInitCommand;
tikhomirov@653: import org.tmatesoft.hg.core.Nodeid;
tikhomirov@623: import org.tmatesoft.hg.internal.FileUtils;
tikhomirov@623: import org.tmatesoft.hg.internal.StreamLogFacility;
tikhomirov@536: import org.tmatesoft.hg.repo.HgRepository;
tikhomirov@637: import org.tmatesoft.hg.util.CancelledException;
tikhomirov@536: 
tikhomirov@536: /**
tikhomirov@536:  * 
tikhomirov@536:  * @author Artem Tikhomirov
tikhomirov@536:  * @author TMate Software Ltd.
tikhomirov@536:  */
tikhomirov@536: public class RepoUtils {
tikhomirov@536: 
tikhomirov@637: 	static File initEmptyTempRepo(String dirName) throws IOException, HgException {
tikhomirov@536: 		File dest = createEmptyDir(dirName);
tikhomirov@637: 		try {
tikhomirov@637: 			new HgInitCommand().location(dest).revlogV1().execute();
tikhomirov@637: 		} catch (CancelledException ex) {
tikhomirov@637: 			Assert.fail(ex.toString());
tikhomirov@637: 		}
tikhomirov@536: 		return dest;
tikhomirov@536: 	}
tikhomirov@536: 
tikhomirov@536: 	static File createEmptyDir(String dirName) throws IOException {
tikhomirov@536: 		File dest = new File(Configuration.get().getTempDir(), dirName);
tikhomirov@536: 		if (dest.exists()) {
tikhomirov@637: 			rmdir(dest);
tikhomirov@536: 		}
tikhomirov@536: 		dest.mkdirs();
tikhomirov@536: 		return dest;
tikhomirov@536: 	}
tikhomirov@536: 
tikhomirov@623: 	static File cloneRepoToTempLocation(String configRepoName, String name, boolean noupdate) throws HgException, IOException, InterruptedException {
tikhomirov@536: 		return cloneRepoToTempLocation(Configuration.get().find(configRepoName), name, noupdate);
tikhomirov@536: 	}
tikhomirov@536: 
tikhomirov@536: 	static File cloneRepoToTempLocation(HgRepository repo, String name, boolean noupdate) throws IOException, InterruptedException {
tikhomirov@631: 		return cloneRepoToTempLocation(repo.getWorkingDir(), name, noupdate, false);
tikhomirov@631: 	}
tikhomirov@631: 
tikhomirov@631: 	static File cloneRepoToTempLocation(File repoLoc, String name, boolean noupdate, boolean usePull) throws IOException, InterruptedException {
tikhomirov@536: 		File testRepoLoc = createEmptyDir(name);
tikhomirov@536: 		ExecHelper eh = new ExecHelper(new OutputParser.Stub(), testRepoLoc.getParentFile());
tikhomirov@536: 		ArrayList<String> cmd = new ArrayList<String>();
tikhomirov@559: 		cmd.add("hg");
tikhomirov@559: 		cmd.add("clone");
tikhomirov@536: 		if (noupdate) {
tikhomirov@536: 			cmd.add("--noupdate");
tikhomirov@536: 		}
tikhomirov@631: 		if (usePull) {
tikhomirov@631: 			cmd.add("--pull");
tikhomirov@631: 		}
tikhomirov@631: 		cmd.add(repoLoc.toString());
tikhomirov@536: 		cmd.add(testRepoLoc.getName());
tikhomirov@536: 		eh.run(cmd.toArray(new String[cmd.size()]));
tikhomirov@536: 		assertEquals("[sanity]", 0, eh.getExitValue());
tikhomirov@536: 		return testRepoLoc;
tikhomirov@536: 	}
tikhomirov@623: 	
tikhomirov@623: 	static File copyRepoToTempLocation(String configRepoName, String newRepoName) throws HgException, IOException {
tikhomirov@623: 		File testRepoLoc = createEmptyDir(newRepoName);
tikhomirov@623: 		final File srcDir = Configuration.get().find(configRepoName).getWorkingDir();
tikhomirov@623: 		Iterator<File> it = new Iterator<File>() {
tikhomirov@623: 			private final LinkedList<File> queue = new LinkedList<File>();
tikhomirov@623: 			{
tikhomirov@623: 				queue.addAll(Arrays.asList(srcDir.listFiles()));
tikhomirov@623: 			}
tikhomirov@623: 			public boolean hasNext() {
tikhomirov@623: 				return !queue.isEmpty();
tikhomirov@623: 			}
tikhomirov@623: 			public File next() {
tikhomirov@623: 				File n = queue.removeFirst();
tikhomirov@623: 				if (n.isDirectory()) {
tikhomirov@623: 					queue.addAll(Arrays.asList(n.listFiles()));
tikhomirov@623: 				}
tikhomirov@623: 				return n;
tikhomirov@623: 			}
tikhomirov@623: 			public void remove() {
tikhomirov@623: 				throw new UnsupportedOperationException();
tikhomirov@623: 			}
tikhomirov@623: 		};
tikhomirov@654: 		FileUtils fu = new FileUtils(new StreamLogFacility(Debug, true, System.err), RepoUtils.class);
tikhomirov@623: 		String srcPrefix = srcDir.getAbsolutePath();
tikhomirov@623: 		while (it.hasNext()) {
tikhomirov@623: 			File next = it.next();
tikhomirov@623: 			assert next.getAbsolutePath().startsWith(srcPrefix);
tikhomirov@623: 			String relPath = next.getAbsolutePath().substring(srcPrefix.length());
tikhomirov@623: 			File dest = new File(testRepoLoc, relPath);
tikhomirov@623: 			if (next.isDirectory()) {
tikhomirov@623: 				dest.mkdir();
tikhomirov@623: 			} else {
tikhomirov@623: 				fu.copy(next, dest);
tikhomirov@623: 				dest.setLastModified(next.lastModified());
tikhomirov@623: 			}
tikhomirov@623: 		}
tikhomirov@623: 		return testRepoLoc;
tikhomirov@623: 	}
tikhomirov@536: 
tikhomirov@559: 	static void modifyFileAppend(File f, Object content) throws IOException {
tikhomirov@536: 		assertTrue(f.isFile());
tikhomirov@536: 		FileOutputStream fos = new FileOutputStream(f, true);
tikhomirov@559: 		if (content == null) {
tikhomirov@559: 			content = "XXX".getBytes();
tikhomirov@559: 		}
tikhomirov@559: 		if (content instanceof byte[]) {
tikhomirov@559: 			fos.write((byte[]) content);
tikhomirov@559: 		} else {
tikhomirov@559: 			fos.write(String.valueOf(content).getBytes());
tikhomirov@559: 		}
tikhomirov@536: 		fos.close();
tikhomirov@536: 	}
tikhomirov@536: 
tikhomirov@559: 	static void createFile(File f, Object content) throws IOException {
tikhomirov@559: 		if (content == null) {
tikhomirov@559: 			f.createNewFile();
tikhomirov@559: 			return;
tikhomirov@559: 		}
tikhomirov@559: 		if (content instanceof byte[]) {
tikhomirov@559: 			FileOutputStream fos = new FileOutputStream(f);
tikhomirov@559: 			fos.write((byte[]) content);
tikhomirov@559: 			fos.close();
tikhomirov@559: 		} else {
tikhomirov@559: 			FileWriter fw = new FileWriter(f);
tikhomirov@559: 			fw.write(String.valueOf(content));
tikhomirov@559: 			fw.close();
tikhomirov@559: 		}
tikhomirov@559: 	}
tikhomirov@635: 
tikhomirov@635: 	static void exec(File wd, int expectedRetVal, String... args) throws Exception {
tikhomirov@635: 		OutputParser.Stub s = new OutputParser.Stub();
tikhomirov@635: 		try {
tikhomirov@635: 			ExecHelper eh = new ExecHelper(s, wd);
tikhomirov@635: 			eh.run(args);
tikhomirov@635: 			Assert.assertEquals(expectedRetVal, eh.getExitValue());
tikhomirov@635: 		} catch (Exception ex) {
tikhomirov@635: 			System.err.println(s.result());
tikhomirov@635: 			throw ex;
tikhomirov@635: 		}
tikhomirov@635: 	}
tikhomirov@637: 
tikhomirov@637: 	static void rmdir(File dest) throws IOException {
tikhomirov@637: 		LinkedList<File> queue = new LinkedList<File>();
tikhomirov@637: 		queue.addAll(Arrays.asList(dest.listFiles()));
tikhomirov@637: 		while (!queue.isEmpty()) {
tikhomirov@637: 			File next = queue.removeFirst();
tikhomirov@637: 			if (next.isDirectory()) {
tikhomirov@637: 				List<File> files = Arrays.asList(next.listFiles());
tikhomirov@637: 				if (!files.isEmpty()) {
tikhomirov@637: 					queue.addAll(files);
tikhomirov@637: 					queue.add(next);
tikhomirov@637: 				}
tikhomirov@637: 				// fall through
tikhomirov@637: 			} 
tikhomirov@637: 			next.delete();
tikhomirov@637: 		}
tikhomirov@637: 		dest.delete();
tikhomirov@637: 	}
tikhomirov@653: 
tikhomirov@653: 	static Nodeid[] allRevisions(HgRepository repo) {
tikhomirov@653: 		Nodeid[] allRevs = new Nodeid[repo.getChangelog().getRevisionCount()];
tikhomirov@653: 		for (int i = 0; i < allRevs.length; i++) {
tikhomirov@653: 			allRevs[i] = repo.getChangelog().getRevision(i);
tikhomirov@653: 		}
tikhomirov@653: 		return allRevs;
tikhomirov@653: 	}
tikhomirov@660: 
tikhomirov@660: 	static void assertHgVerifyOk(ErrorCollectorExt errorCollector, File repoLoc) throws InterruptedException, IOException {
tikhomirov@660: 		ExecHelper verifyRun = new ExecHelper(new OutputParser.Stub(), repoLoc);
tikhomirov@660: 		verifyRun.run("hg", "verify");
tikhomirov@660: 		errorCollector.assertEquals("hg verify", 0, verifyRun.getExitValue());
tikhomirov@660: 	}
tikhomirov@536: }