Mercurial > jhg
diff src/org/tmatesoft/hg/internal/WorkingDirFileWriter.java @ 525:0be5be8d57e9
Repository checkout support, first iteration
author | Artem Tikhomirov <tikhomirov.artem@gmail.com> |
---|---|
date | Fri, 11 Jan 2013 18:12:39 +0100 |
parents | |
children | 2f9ed6bcefa2 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/org/tmatesoft/hg/internal/WorkingDirFileWriter.java Fri Jan 11 18:12:39 2013 +0100 @@ -0,0 +1,83 @@ +/* + * Copyright (c) 2012-2013 TMate Software Ltd + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; version 2 of the License. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * For information on how to redistribute this software under + * the terms of a license other than GNU General Public License + * contact TMate Software at support@hg4j.com + */ +package org.tmatesoft.hg.internal; + +import java.io.File; +import java.io.FileOutputStream; +import java.io.IOException; +import java.nio.ByteBuffer; +import java.nio.channels.FileChannel; + +import org.tmatesoft.hg.repo.HgDataFile; +import org.tmatesoft.hg.repo.HgRepository; +import org.tmatesoft.hg.util.ByteChannel; +import org.tmatesoft.hg.util.CancelledException; +import org.tmatesoft.hg.util.Path; +import org.tmatesoft.hg.util.LogFacility.Severity; + +/** + * + * @author Artem Tikhomirov + * @author TMate Software Ltd. + */ +public class WorkingDirFileWriter implements ByteChannel { + + + private final HgRepository repo; + private File dest; + private FileChannel destChannel; + private int totalBytesWritten; + + public WorkingDirFileWriter(HgRepository hgRepo) { + repo = hgRepo; + } + + public void processFile(HgDataFile df, int fileRevIndex) throws IOException { + try { + prepare(df.getPath()); + df.contentWithFilters(fileRevIndex, this); + } catch (CancelledException ex) { + repo.getSessionContext().getLog().dump(getClass(), Severity.Error, ex, "Our impl doesn't throw cancellation"); + } + finish(); + } + + public void prepare(Path fname) throws IOException { + String fpath = fname.toString(); + dest = new File(repo.getWorkingDir(), fpath); + if (fpath.indexOf('/') != -1) { + dest.getParentFile().mkdirs(); + } + destChannel = new FileOutputStream(dest).getChannel(); + totalBytesWritten = 0; + } + + public int write(ByteBuffer buffer) throws IOException, CancelledException { + int written = destChannel.write(buffer); + totalBytesWritten += written; + return written; + } + + public void finish() throws IOException { + destChannel.close(); + dest = null; + } + + public int bytesWritten() { + return totalBytesWritten; + } +}