org.jwat.tools.tasks.UnpackTask Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jwat-tools Show documentation

JWAT-Tools uses the available JWAT libraries to make high level tasks available either from command-line or programmatically. Common tasks include: Test, Compress, Decompress, CDX, Arc2Warc. More specialised tasks include: Changed, ContainerMD, Delete, Extract, Interval, PathIndex, Unpack, Headers2CDX.

There is a newer version: 0.7.1

Show newest version

package org.jwat.tools.tasks;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.util.List;

import org.jwat.common.ByteCountingPushBackInputStream;
import org.jwat.common.RandomAccessFileInputStream;
import org.jwat.gzip.GzipEntry;
import org.jwat.gzip.GzipReader;
import org.jwat.tools.JWATTools;
import org.jwat.tools.core.CommandLine;

public class UnpackTask extends ProcessTask {

	public static final String commandName = "unpack";

	public static final String commandDescription = "unpack multifile GZip";

	public UnpackTask() {
	}

	@Override
	public void show_help() {
		System.out.println("Work in progress...");
	}

	@Override
	public void command(CommandLine.Arguments arguments) {
		CommandLine.Argument argument = arguments.idMap.get( JWATTools.A_FILES );
		List filesList = argument.values;
		filelist_feeder( filesList, this );
	}

	@Override
	public void process(File srcFile) {
		String srcFname = srcFile.getName();
		RandomAccessFile raf = null;
		RandomAccessFileInputStream rafin;
		ByteCountingPushBackInputStream pbin = null;
		int count = 0;
		try {
			raf = new RandomAccessFile( srcFile, "r" );
			rafin = new RandomAccessFileInputStream( raf );
			pbin = new ByteCountingPushBackInputStream( new BufferedInputStream( rafin, 8192 ), 32 );
			if (GzipReader.isGzipped(pbin)) {
				String dstFname;
				if (srcFname.endsWith(".gz")) {
					dstFname = srcFname.substring( 0, srcFname.length() - 3 );
				}
				else {
					dstFname = srcFname + ".org";
				}
				GzipReader gzipReader = new GzipReader( pbin );
				GzipEntry gzipEntry;
				InputStream in;
				byte[] buffer = new byte[ 8192 ];
				int read;
				while ( (gzipEntry = gzipReader.getNextEntry()) != null ) {
					++count;
					File dstFile = new File( srcFile.getParentFile(), dstFname + "." + count );
					System.out.println( srcFname + " -> " + dstFname + "." + count );
					raf = new RandomAccessFile( dstFile, "rw" );
					in = gzipEntry.getInputStream();
					while ( (read = in.read(buffer)) != -1 ) {
						raf.write( buffer, 0, read );
					}
					in.close();
					gzipEntry.close();
					raf.close();
				}
				gzipReader.close();
				/*
				if ( !dstFile.exists() ) {
				}
				else {
					System.out.println( dstFile.getName() + " already exists, skipping." );
				}
				*/
			}
			else if ( srcFname.toLowerCase().endsWith( ".gz" ) ) {
				System.out.println( "Invalid extension: " + srcFname );
			}
		}
		catch (FileNotFoundException e) {
			e.printStackTrace();
		}
		catch (IOException e) {
			e.printStackTrace();
		}
		finally {
			if (pbin != null) {
				try {
					pbin.close();
				}
				catch (IOException e) {
				}
			}
			if (raf != null) {
				try {
					raf.close();
				}
				catch (IOException e) {
				}
			}
		}
	}

}