ChecksumTrackingOutputStream.java

/* ******************************************************************************
 * Copyright (c) 2006-2016 XMind Ltd. and others.
 * 
 * This file is a part of XMind 3. XMind releases 3 and
 * above are dual-licensed under the Eclipse Public License (EPL),
 * which is available at http://www.eclipse.org/legal/epl-v10.html
 * and the GNU Lesser General Public License (LGPL), 
 * which is available at http://www.gnu.org/licenses/lgpl.html
 * See http://www.xmind.net/license.html for details.
 * 
 * Contributors:
 *     XMind Ltd. - initial API and implementation
 *******************************************************************************/
/**
 * 
 */
package org.xmind.core.io;

import java.io.FilterOutputStream;
import java.io.IOException;
import java.io.OutputStream;

import org.xmind.core.IChecksumStream;
import org.xmind.core.IEncryptionData;

/**
 * @author Frank Shaka
 *
 */
public class ChecksumTrackingOutputStream extends FilterOutputStream {

    private final IEncryptionData encData;

    /**
     * @param out
     */
    public ChecksumTrackingOutputStream(IEncryptionData encData,
            OutputStream out) {
        super(out);
        this.encData = encData;
    }

    /*
     * (non-Javadoc)
     * 
     * @see java.io.FilterOutputStream#write(byte[])
     */
    @Override
    public void write(byte[] b) throws IOException {
        out.write(b);
    }

    /*
     * (non-Javadoc)
     * 
     * @see java.io.FilterOutputStream#write(byte[], int, int)
     */
    @Override
    public void write(byte[] b, int off, int len) throws IOException {
        out.write(b, off, len);
    }

    /*
     * (non-Javadoc)
     * 
     * @see java.io.FilterOutputStream#write(int)
     */
    @Override
    public void write(int b) throws IOException {
        out.write(b);
    }

    /*
     * (non-Javadoc)
     * 
     * @see java.io.FilterOutputStream#close()
     */
    @Override
    public void close() throws IOException {
        super.close();

        if (out instanceof IChecksumStream) {
            String checksum = ((IChecksumStream) out).getChecksum();
            encData.setChecksum(checksum);
        }
    }

}