tikhomirov@539: /*
tikhomirov@539:  * Copyright (c) 2013 TMate Software Ltd
tikhomirov@539:  *  
tikhomirov@539:  * This program is free software; you can redistribute it and/or modify
tikhomirov@539:  * it under the terms of the GNU General Public License as published by
tikhomirov@539:  * the Free Software Foundation; version 2 of the License.
tikhomirov@539:  *
tikhomirov@539:  * This program is distributed in the hope that it will be useful,
tikhomirov@539:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@539:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@539:  * GNU General Public License for more details.
tikhomirov@539:  *
tikhomirov@539:  * For information on how to redistribute this software under
tikhomirov@539:  * the terms of a license other than GNU General Public License
tikhomirov@539:  * contact TMate Software at support@hg4j.com
tikhomirov@539:  */
tikhomirov@539: package org.tmatesoft.hg.internal;
tikhomirov@539: 
tikhomirov@539: import java.io.File;
tikhomirov@539: import java.io.FileOutputStream;
tikhomirov@539: import java.io.IOException;
tikhomirov@539: import java.nio.charset.Charset;
tikhomirov@539: import java.util.ArrayList;
tikhomirov@539: 
tikhomirov@539: import org.tmatesoft.hg.util.Path;
tikhomirov@539: 
tikhomirov@539: /**
tikhomirov@539:  * <blockquote>
tikhomirov@539:  * The fncache file contains the paths of all filelog files in the store as encoded by mercurial.filelog.encodedir. The paths are separated by '\n' (LF).
tikhomirov@539:  * </blockquote>
tikhomirov@539:  * @see http://mercurial.selenic.com/wiki/fncacheRepoFormat
tikhomirov@539:  * @author Artem Tikhomirov
tikhomirov@539:  * @author TMate Software Ltd.
tikhomirov@539:  */
tikhomirov@539: public class FNCacheFile {
tikhomirov@539: 	
tikhomirov@539: 	private final Internals repo;
tikhomirov@539: 	private final ArrayList<Path> files;
tikhomirov@539: 
tikhomirov@539: 	public FNCacheFile(Internals internalRepo) {
tikhomirov@539: 		repo = internalRepo;
tikhomirov@539: 		files = new ArrayList<Path>();
tikhomirov@539: 	}
tikhomirov@539: 
tikhomirov@539: 	public void read(Path.Source pathFactory) throws IOException {
tikhomirov@539: 		File f = fncacheFile();
tikhomirov@539: 		files.clear();
tikhomirov@539: 		if (!f.exists()) {
tikhomirov@539: 			return;
tikhomirov@539: 		}
tikhomirov@539: 		ArrayList<String> entries = new ArrayList<String>();
tikhomirov@539: 		// names in fncache are in local encoding, shall translate to unicode
tikhomirov@539: 		new LineReader(f, repo.getSessionContext().getLog(), repo.getFilenameEncoding()).read(new LineReader.SimpleLineCollector(), entries);
tikhomirov@539: 		for (String e : entries) {
tikhomirov@539: 			files.add(pathFactory.path(e));
tikhomirov@539: 		}
tikhomirov@539: 	}
tikhomirov@539: 	
tikhomirov@539: 	public void write() throws IOException {
tikhomirov@539: 		if (files.isEmpty()) {
tikhomirov@539: 			return;
tikhomirov@539: 		}
tikhomirov@539: 		File f = fncacheFile();
tikhomirov@539: 		f.getParentFile().mkdirs();
tikhomirov@539: 		final Charset filenameEncoding = repo.getFilenameEncoding();
tikhomirov@539: 		FileOutputStream fncacheFile = new FileOutputStream(f);
tikhomirov@539: 		for (Path p : files) {
tikhomirov@539: 			String s = "data/" + p.toString() + ".i"; // TODO post-1.0 this is plain wrong. (a) likely need .d files, too; (b) what about dh/ location? 
tikhomirov@539: 			fncacheFile.write(s.getBytes(filenameEncoding));
tikhomirov@539: 			fncacheFile.write(0x0A); // http://mercurial.selenic.com/wiki/fncacheRepoFormat
tikhomirov@539: 		}
tikhomirov@539: 		fncacheFile.close();
tikhomirov@539: 	}
tikhomirov@539: 
tikhomirov@539: 	public void add(Path p) {
tikhomirov@539: 		files.add(p);
tikhomirov@539: 	}
tikhomirov@539: 
tikhomirov@539: 	private File fncacheFile() {
tikhomirov@539: 		return repo.getFileFromStoreDir("fncache");
tikhomirov@539: 	}
tikhomirov@539: }