tikhomirov@673: /*
tikhomirov@673:  * Copyright (c) 2013 TMate Software Ltd
tikhomirov@673:  *  
tikhomirov@673:  * This program is free software; you can redistribute it and/or modify
tikhomirov@673:  * it under the terms of the GNU General Public License as published by
tikhomirov@673:  * the Free Software Foundation; version 2 of the License.
tikhomirov@673:  *
tikhomirov@673:  * This program is distributed in the hope that it will be useful,
tikhomirov@673:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@673:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@673:  * GNU General Public License for more details.
tikhomirov@673:  *
tikhomirov@673:  * For information on how to redistribute this software under
tikhomirov@673:  * the terms of a license other than GNU General Public License
tikhomirov@673:  * contact TMate Software at support@hg4j.com
tikhomirov@673:  */
tikhomirov@673: package org.tmatesoft.hg.internal;
tikhomirov@673: 
tikhomirov@673: import java.io.File;
tikhomirov@673: import java.io.FileInputStream;
tikhomirov@673: import java.io.IOException;
tikhomirov@673: 
tikhomirov@673: import org.tmatesoft.hg.core.HgIOException;
tikhomirov@673: import org.tmatesoft.hg.core.SessionContext;
tikhomirov@673: import org.tmatesoft.hg.repo.HgBundle;
tikhomirov@673: import org.tmatesoft.hg.repo.HgInternals;
tikhomirov@673: import org.tmatesoft.hg.repo.HgRuntimeException;
tikhomirov@673: import org.tmatesoft.hg.util.LogFacility;
tikhomirov@673: 
tikhomirov@673: /**
tikhomirov@673:  * 
tikhomirov@673:  * @author Artem Tikhomirov
tikhomirov@673:  * @author TMate Software Ltd.
tikhomirov@673:  */
tikhomirov@673: public class BundleSerializer implements DataSerializer.DataSource {
tikhomirov@673: 	private final LogFacility log;
tikhomirov@673: 	private final File bundleFile;
tikhomirov@673: 	
tikhomirov@673: 	public static BundleSerializer newInstance(SessionContext ctx, HgBundle bundle) {
tikhomirov@673: 		return new BundleSerializer(ctx.getLog(), HgInternals.getBundleFile(bundle));
tikhomirov@673: 	}
tikhomirov@673: 
tikhomirov@673: 	public BundleSerializer(LogFacility logFacility, File bundleFile) {
tikhomirov@673: 		log = logFacility;
tikhomirov@673: 		this.bundleFile = bundleFile;
tikhomirov@673: 	}
tikhomirov@673: 
tikhomirov@673: 	public void serialize(DataSerializer out) throws HgIOException, HgRuntimeException {
tikhomirov@673: 		FileInputStream fis = null;
tikhomirov@673: 		try {
tikhomirov@673: 			fis = new FileInputStream(bundleFile);
tikhomirov@673: 			byte[] buffer = new byte[8*1024];
tikhomirov@673: 			int r;
tikhomirov@673: 			while ((r = fis.read(buffer, 0, buffer.length)) > 0) {
tikhomirov@673: 				out.write(buffer, 0, r);
tikhomirov@673: 			}
tikhomirov@673: 			
tikhomirov@673: 		} catch (IOException ex) {
tikhomirov@673: 			throw new HgIOException("Failed to serialize bundle", bundleFile);
tikhomirov@673: 		} finally {
tikhomirov@673: 			new FileUtils(log, this).closeQuietly(fis, bundleFile);
tikhomirov@673: 		}
tikhomirov@673: 	}
tikhomirov@673: 
tikhomirov@673: 	public int serializeLength() throws HgRuntimeException {
tikhomirov@673: 		return Internals.ltoi(bundleFile.length());
tikhomirov@673: 	}
tikhomirov@673: }