Package org.jnode.apps.vmware.disk.tools

Source Code of org.jnode.apps.vmware.disk.tools.DiskCopier

/*
* $Id$
*
* Copyright (C) 2003-2014 JNode.org
*
* This library is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as published
* by the Free Software Foundation; either version 2.1 of the License, or
* (at your option) any later version.
*
* This library 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 Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this library; If not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
package org.jnode.apps.vmware.disk.tools;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FilenameFilter;
import java.io.IOException;
import java.nio.channels.FileChannel;

import org.apache.log4j.Logger;

/**
* Utility methods for copying VMware disks.
*
* @author Fabien DUMINY (fduminy@jnode.org)
*
*/
public class DiskCopier {
    private static final Logger LOG = Logger.getLogger(DiskCopier.class);

    /**
     * Copy a VMware disk to a given directory.
     * @param mainFile main file for the VMware disk.
     * @param toDirectory destination directory for the copied VMware disk.
     * @return the main file for the copied VMware disk
     * @throws IOException
     */
    public static File copyDisk(final File mainFile, final File toDirectory) throws IOException {
        final String name = mainFile.getName();
        final int idx = name.lastIndexOf('.');
        final String beginName = name.substring(0, idx);
        final String endName = name.substring(idx);

        final File parentDir = mainFile.getParentFile();
        File[] files = parentDir.listFiles(new FilenameFilter() {
            public boolean accept(File dir, String name) {
                boolean ok = name.startsWith(beginName) && name.endsWith(endName);
                return ok;
            }
        });

        File mainFileCopy = null;
        for (File file : files) {
            File f = copyFile(file, toDirectory);
            if (file.getName().equals(mainFile.getName())) {
                mainFileCopy = f;
            }
        }

        return mainFileCopy;
    }

    /**
     * Copy a file to a directory.
     * @param file to copy
     * @param toDirectory destination directory
     * @return the copied file
     * @throws IOException
     */
    public static File copyFile(File file, File toDirectory) throws IOException {
        LOG.debug("copying file " + file.getName() + " to " + toDirectory.getName());
        FileInputStream fis = null;
        FileOutputStream fos = null;
        File outFile = null;

        try {
            fis = new FileInputStream(file);
            FileChannel inCh = fis.getChannel();

            outFile = new File(toDirectory, file.getName());
            fos = new FileOutputStream(outFile);
            FileChannel outCh = fos.getChannel();

            outCh.transferFrom(inCh, 0, inCh.size());

            return outFile;
        } catch (IOException ioe) {
            LOG.error(ioe);
            throw ioe;
        } finally {
            try {
                if (fos != null) {
                    fos.close();
                }
            } finally {
                if (fis != null) {
                    fis.close();
                }
            }
        }
    }
}
TOP

Related Classes of org.jnode.apps.vmware.disk.tools.DiskCopier

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.