com.mindsnacks.zinc.classes.jobs.ZincUnarchiveBundleJob Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of JavaZinc Show documentation
Zinc client for Java
There is a newer version: 1.6.2
package com.mindsnacks.zinc.classes.jobs;

import com.mindsnacks.zinc.classes.ZincJobFactory;
import com.mindsnacks.zinc.classes.data.*;
import com.mindsnacks.zinc.classes.fileutils.FileHelper;
import com.mindsnacks.zinc.exceptions.ZincRuntimeException;

import java.io.File;
import java.io.IOException;
import java.util.Map;

/**
 * @author NachoSoto.
 *
 * This job creates the result bundle by extracting the contents of the archive,
 * with the information in the ZincManifest.
 */
public class ZincUnarchiveBundleJob extends ZincJob {
    private final ZincBundle mDownloadedBundle;
    private final ZincCloneBundleRequest mRequest;
    private final ZincJobFactory mJobFactory;
    private final FileHelper mFileHelper;

    public ZincUnarchiveBundleJob(final ZincBundle downloadedBundle,
                                  final ZincCloneBundleRequest request,
                                  final ZincJobFactory jobFactory,
                                  final FileHelper fileHelper) {
        mDownloadedBundle = downloadedBundle;
        mRequest = request;
        mJobFactory = jobFactory;
        mFileHelper = fileHelper;
    }

    @Override
    public ZincBundle run() throws Exception {
        final BundleID bundleID = mRequest.getBundleID();
        final int version = mDownloadedBundle.getVersion();

        final File temporaryFolder = getTemporaryBundleFolder(bundleID),
                   resultFolder = getBundleFolder(bundleID);

        unarchiveBundle(mDownloadedBundle, temporaryFolder, getManifest(version, bundleID));

        cleanUpDownloadedFolder();
        moveToBundlesFolder(temporaryFolder, resultFolder);

        return new ZincBundle(resultFolder, bundleID, version);
    }

    private File getTemporaryBundleFolder(final BundleID bundleID) {
        return new File(
            mRequest.getRepoFolder(),
            PathHelper.getLocalTemporaryBundleFolder(bundleID, mDownloadedBundle.getVersion(), mRequest.getFlavorName())
        );
    }

    private File getBundleFolder(final BundleID bundleID) {
        return new File(
            mRequest.getRepoFolder(),
            PathHelper.getLocalBundleFolder(bundleID, mDownloadedBundle.getVersion(), mRequest.getFlavorName())
        );
    }

    private ZincManifest getManifest(final int version,
                                     final BundleID bundleID) throws Exception {
        return mJobFactory.downloadManifest(
                mRequest.getSourceURL(),
                bundleID.getBundleName(),
                version
        ).call();
    }

    private void unarchiveBundle(final File downloadedBundle,
                                 final File temporaryFolder,
                                 final ZincManifest manifest) throws IOException {
        logMessage("unarchiving");

        final Map files = manifest.getFilesWithFlavor(mRequest.getFlavorName());

        for (final Map.Entry entry : files.entrySet()) {
            final ZincManifest.FileInfo fileInfo = entry.getValue();

            final String originFilename = fileInfo.getHashWithExtension();
            final String destinationFilename = entry.getKey();

            if (fileInfo.isGzipped()) {
                mFileHelper.unzipFile(downloadedBundle, originFilename, temporaryFolder, destinationFilename);
            } else {
                mFileHelper.copyFile(downloadedBundle, originFilename, temporaryFolder, destinationFilename);
            }
        }
    }

    private void cleanUpDownloadedFolder() {
        logMessage("cleaning up archive");

        mFileHelper.removeDirectory(mDownloadedBundle);
    }

    private void moveToBundlesFolder(final File temporaryFolder, final File bundleFolder) {
        logMessage("moving bundle");

        if (!bundleFolder.mkdirs() || !mFileHelper.moveFile(temporaryFolder, bundleFolder)) {
            throw new ZincRuntimeException(String.format("Error moving bundle from '%s' to '%s'", temporaryFolder, bundleFolder));
        }
    }

    @Override
    protected String getJobName() {
        return super.getJobName() + " (" + mRequest.getBundleID() + ")";
    }
}