source: java/installer2/src/net/oni2/aeinstaller/backend/oni/management/tools/ToolsManager.java@ 853

package net.oni2.aeinstaller.backend.oni.management.tools;

import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashSet;
import java.util.TreeSet;

import net.oni2.aeinstaller.backend.CaseInsensitiveFile;
import net.oni2.aeinstaller.backend.FileChecksum;
import net.oni2.aeinstaller.backend.Paths;
import net.oni2.aeinstaller.backend.packages.Package;
import net.oni2.aeinstaller.backend.packages.PackageManager;

import org.apache.commons.io.FileUtils;

/**
 * @author Christian Illy
 */
public class ToolsManager {

        /**
         * Verify integrity of installed tools
         */
        public static void verifyToolsIntegrity() {
                final ToolInstallationList til = ToolInstallationList.getInstance();
                for (final Package m : PackageManager.getInstance().getInstalledTools()) {
                        ToolFileIterator.iteratePlatformToolFiles(m,
                                        new ToolFileIteratorEntry() {
                                                @Override
                                                public void toolFile(File source, File target) {
                                                        byte[] chkSrc = FileChecksum
                                                                        .calculateFileMD5(source);
                                                        if (!target.exists()) {
                                                                til.markModified(m.getPackageNumber(), true);
                                                        } else {
                                                                byte[] chkTrg = FileChecksum
                                                                                .calculateFileMD5(target);
                                                                if (!Arrays.equals(chkSrc, chkTrg))
                                                                        til.markModified(m.getPackageNumber(), true);
                                                        }
                                                }
                                        });
                }
        }

        /**
         * @param tools
         *            Tools to (un)install
         * @param uninstall
         *            Uninstall tools or install?
         */
        public static void installTools(TreeSet<Package> tools, boolean uninstall) {
                ToolInstallationList til = ToolInstallationList.getInstance();
                for (Package m : tools) {
                        if (!uninstall) { // Install:
                                final HashSet<String> files = new HashSet<String>();
                                ToolFileIterator.iteratePlatformToolFiles(m,
                                                new ToolFileIteratorEntry() {
                                                        @Override
                                                        public void toolFile(File source, File target) {
                                                                copyToolsFiles(source, target, files);
                                                        }
                                                });
                                til.addTool(m.getPackageNumber(), files);
                        } else { // Uninstall:
                                if (til.isInstalled(m.getPackageNumber())) {
                                        removeTool(til.getFiles(m.getPackageNumber()));
                                        til.removeTool(m.getPackageNumber());
                                }
                        }
                }
                til.saveList();
        }

        private static void copyToolsFiles(File src, File target,
                        HashSet<String> files) {
                try {
                        File targetFile = CaseInsensitiveFile.getCaseInsensitiveFile(
                                        target.getParentFile(), target.getName());

                        // Case mismatch?
                        if (!targetFile.getName().equals(src.getName()))
                                targetFile.delete();

                        files.add(target.getPath().replace(
                                        Paths.getEditionBasePath().getPath(), ""));

                        FileUtils.copyFile(src, target);
                        if (src.canExecute())
                                target.setExecutable(true);
                } catch (IOException e) {
                        e.printStackTrace();
                }
        }

        private static void removeTool(HashSet<String> files) {
                for (String p : files) {
                        File targetFile = new File(Paths.getEditionBasePath().getPath() + p);
                        if (targetFile.getPath().contains(
                                        Paths.getEditionBasePath().getPath())) {
                                File targetFolder = targetFile.getParentFile();

                                if (targetFile.exists())
                                        targetFile.delete();
                                if (targetFolder.list().length == 0)
                                        targetFolder.delete();
                        }
                }
        }
}