Examples with LocalStorageUtil.getSubdirOrFile se.light.assembly64.util.LocalStorageUtil.getSubdirOrFile used on opensource projects

Example 6 with LocalStorageUtil.getSubdirOrFile

use of se.light.assembly64.util.LocalStorageUtil.getSubdirOrFile in project assembly64fx by freabemania.

the class InstallationService method installEntry.

private File installEntry(LocationAndInstallationStatus installLocation, Artifact entry, String remotePath, String remoteName, String version, Boolean skipUnpack, Boolean updateProgressBar) throws Exception {
    CancelableTask cancelTask = CancelableTask.of();
    int ctr = 0;
    // to keep track of the progressupdates if something fails
    int progressUpdated = 0;
    if (entry.getType() == ArtifactType.REMOTE) {
        while (ctr < 3 && cancelTask.isRunning()) {
            LOGGER.info("Installing " + entry.getName());
            ProgressMessageBox.setMessage("Fetching " + entry.getName());
            ProgressControlWrapper.getInstance().setProgressLabel("Fetching file " + remoteName);
            File target = new File(pathService.getTmpFolderAsString() + remoteName);
            if (target.exists() && ftpService.registerDownload(remoteName, DOWNLOAD_STATUS.CHECK)) {
                ProgressControlWrapper.getInstance().setProgressLabel("Waiting for other worker to finish downloading " + remoteName);
                ProgressControlWrapper.getInstance().increaseCurrentProgressStep(1);
                while (true) {
                    if (!ftpService.registerDownload(remoteName, DOWNLOAD_STATUS.CHECK)) {
                        LOGGER.info(remoteName + " was downloaded by another thread");
                        break;
                    } else if (cancelTask.isCancelled()) {
                        return null;
                    }
                    Thread.sleep(500);
                }
            } else if (!target.exists()) {
                if (!ftpService.registerDownload(remoteName, DOWNLOAD_STATUS.ADD)) {
                    ftpService.getFile(remotePath, remoteName, target);
                    if (cancelTask.isCancelled()) {
                        return null;
                    }
                } else {
                    ProgressControlWrapper.getInstance().setProgressLabel("Waiting for other worker to finish downloading " + remoteName);
                    ProgressControlWrapper.getInstance().increaseCurrentProgressStep(1);
                    while (true) {
                        if (!ftpService.registerDownload(remoteName, DOWNLOAD_STATUS.CHECK)) {
                            LOGGER.info(remoteName + " was downloaded by another thread");
                            break;
                        } else if (cancelTask.isCancelled()) {
                            return null;
                        }
                        Thread.sleep(1000);
                    }
                }
            } else if (target.exists()) {
                LOGGER.info("Target " + target.getAbsolutePath() + " was already downloaded");
            }
            if (progressUpdated < 2 && updateProgressBar) {
                ProgressControlWrapper.getInstance().increaseProgress();
            }
            progressUpdated++;
            if (cancelTask.isCancelled()) {
                return null;
            }
            if (!skipUnpack) {
                try {
                    ProgressMessageBox.setMessage("Installing file " + entry.getName() + "at " + LocalStorageUtil.getSubdirOrFile(installLocation.getLocation(), entry.getPrefix()));
                    unzipService.extractZip(target, LocalStorageUtil.getSubdirOrFile(installLocation.getLocation(), entry.getPrefix()), entry);
                    // unzipService.extractZip(target, installLocation.getLocation(), entry);
                    ProgressControlWrapper.getInstance().increaseProgress();
                    if (progressUpdated < 2 && updateProgressBar) {
                        ProgressControlWrapper.getInstance().setProgressLabel("Fetched file " + remoteName);
                    }
                    progressUpdated++;
                    if (!cancelTask.isCancelled()) {
                        updateDb(installLocation, entry, version);
                    }
                    ProgressControlWrapper.getInstance().setProgressLabel("Done");
                    return target;
                } catch (Exception e) {
                    LOGGER.error("Something bad", e);
                    Analytics.sendEvent("invalid_zip", entry.getName());
                    FileUtils.forceDelete(target);
                    ctr++;
                }
            } else {
                ProgressControlWrapper.getInstance().setProgressLabel("Done");
                return target;
            }
        }
    }
    throw new IllegalStateException("U1nable to install entry " + remoteName);
}

Example 7 with LocalStorageUtil.getSubdirOrFile

use of se.light.assembly64.util.LocalStorageUtil.getSubdirOrFile in project assembly64fx by freabemania.

the class UnzipService method extractZip.

public void extractZip(File source, File target, Artifact entry) throws Exception {
    byte[] buffer = new byte[1024];
    ZipInputStream zis = null;
    CancelableTask cancelTask = CancelableTask.of();
    int logCtr = 0;
    try {
        FileUtils.forceMkdir(target);
        zis = new ZipInputStream(new FileInputStream(source));
        ZipEntry ze = zis.getNextEntry();
        if (ze == null) {
            Analytics.sendEvent("invalid_zip", entry.getName());
            throw new ZipException("Invalid zipentry");
        }
        int ctr = 0;
        while (ze != null && cancelTask.isRunning()) {
            String fileName = ze.getName();
            File newFile = new File(target + File.separator + fileName);
            ctr++;
            try {
                if (ze.isDirectory()) {
                    FileUtils.forceMkdir(newFile);
                } else {
                    try {
                        FileUtils.forceMkdir(new File(newFile.getParent()));
                        FileOutputStream fos = new FileOutputStream(newFile);
                        int len;
                        while ((len = zis.read(buffer)) > 0 && cancelTask.isRunning()) {
                            fos.write(buffer, 0, len);
                        }
                        fos.close();
                        if (logCtr % 10 == 0) {
                            ProgressControlWrapper.getInstance().setProgressLabel("Installing " + newFile.getName());
                            double finished = (double) ((double) (((double) ctr / (double) entry.getEntries())));
                            ProgressControlWrapper.getInstance().increaseCurrentProgressStep(finished);
                            // lt 100k then give up
                            if (target.getFreeSpace() < 100000) {
                                GenericMessageDialogController.withErrorProps("Oopss...", "No space left on device!", true).showAndWait();
                                CancelTaskRegistry.getInstance().broadcastCancel();
                            }
                        }
                    } catch (Exception e) {
                        LOGGER.info("Unable to extract entry " + fileName);
                    }
                }
            } catch (Exception e) {
                LOGGER.error("Unable to extract " + ze.getName());
            }
            ze = zis.getNextEntry();
        }
        if (cancelTask.isCancelled()) {
            ProgressControlWrapper.getInstance().setProgressLabel("Cleaning out (may take a while), please wait...");
            FileUtils.forceDelete(LocalStorageUtil.getSubdirOrFile(target, entry.getFolder()));
        }
    } finally {
        try {
            zis.closeEntry();
        } catch (Exception e) {
            LOGGER.error("Unable to close zip");
        }
        try {
            zis.close();
        } catch (Exception e) {
            LOGGER.error("Unable to close zipstream");
        }
    }
}