oni/management/Installer.java

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

import java.io.File;
import java.io.FileFilter;
import java.io.FileNotFoundException;
import java.io.FilenameFilter;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.TreeMap;
import java.util.TreeSet;
import java.util.Vector;
import java.util.regex.Pattern;

import net.oni2.SettingsManager;
import net.oni2.aeinstaller.AEInstaller2;
import net.oni2.aeinstaller.backend.CaseInsensitiveFile;
import net.oni2.aeinstaller.backend.Paths;
import net.oni2.aeinstaller.backend.oni.OniSplit;
import net.oni2.aeinstaller.backend.oni.PersistDat;
import net.oni2.aeinstaller.backend.oni.XMLTools;
import net.oni2.aeinstaller.backend.packages.EBSLInstallType;
import net.oni2.aeinstaller.backend.packages.Package;
import net.oni2.aeinstaller.backend.packages.PackageManager;
import net.oni2.platformtools.PlatformInformation;
import net.oni2.platformtools.PlatformInformation.Platform;
import net.oni2.platformtools.applicationinvoker.ApplicationInvocationResult;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.filefilter.RegexFileFilter;
import org.apache.commons.io.filefilter.TrueFileFilter;
import org.javabuilders.swing.SwingJavaBuilder;

import com.paour.NaturalOrderComparator;

/**
 * @author Christian Illy
 */
public class Installer {
        private static FileFilter dirFileFilter = new FileFilter() {
                @Override
                public boolean accept(File pathname) {
                        return pathname.isDirectory();
                }
        };

        /**
         * Verify that the Edition is within a subfolder to vanilla Oni
         * (..../Oni/Edition/AEInstaller)
         * 
         * @return true if GDF can be found in the parent's parent-path
         */
        public static boolean verifyRunningDirectory() {
                return Paths.getVanillaGDF().exists()
                                && Paths.getVanillaGDF().isDirectory();
        }

        /**
         * @return Is Edition Core initialized
         */
        public static boolean isEditionInitialized() {
                return Paths.getVanillaOnisPath().exists();
        }

        /**
         * @return list of currently installed mods
         */
        public static Vector<Integer> getInstalledMods() {
                File installCfg = new File(Paths.getEditionGDF(), "installed_mods.xml");
                return PackageManager.getInstance().loadModSelection(installCfg);
        }

        /**
         * Install the given set of mods
         * 
         * @param mods
         *            Mods to install
         * @param listener
         *            Listener for install progress updates
         */
        public static void install(TreeSet<Package> mods,
                        InstallProgressListener listener) {
                File logFile = new File(Paths.getInstallerPath(), "Installation.log");
                Logger log = null;
                try {
                        log = new Logger(logFile);
                } catch (FileNotFoundException e) {
                        e.printStackTrace();
                }
                SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
                Date start = new Date();
                log.println("Installation of mods started at " + sdf.format(start));

                log.println();
                log.println("AEI2 version: "
                                + SwingJavaBuilder.getConfig().getResource("appversion"));
                log.println("Installed tools:");
                for (Package t : PackageManager.getInstance().getInstalledTools()) {
                        log.println(String.format(" - %s (%s)", t.getName(), t.getVersion()));
                }
                log.println("Installing mods:");
                for (Package m : mods) {
                        log.println(String.format(" - %s (%s)", m.getName(), m.getVersion()));
                }
                log.println();

                Paths.getEditionGDF().mkdirs();
                for (File f : Paths.getEditionGDF().listFiles(new FilenameFilter() {
                        public boolean accept(File arg0, String arg1) {
                                String s = arg1.toLowerCase();
                                return s.endsWith(".dat")
                                                || s.endsWith(".raw")
                                                || s.endsWith(".sep")
                                                || (s.equals("intro.bik") && !SettingsManager
                                                                .getInstance().get("copyintro", false))
                                                || (s.equals("outro.bik") && !SettingsManager
                                                                .getInstance().get("copyoutro", false));
                        }
                })) {
                        f.delete();
                }
                File IGMD = new File(Paths.getEditionGDF(), "IGMD");
                if (IGMD.exists()) {
                        for (File f : IGMD.listFiles(new FileFilter() {
                                @Override
                                public boolean accept(File pathname) {
                                        return pathname.isDirectory();
                                }
                        })) {
                                File ignore = CaseInsensitiveFile.getCaseInsensitiveFile(f,
                                                "ignore.txt");
                                if (!ignore.exists()) {
                                        try {
                                                FileUtils.deleteDirectory(f);
                                        } catch (IOException e) {
                                                e.printStackTrace();
                                        }
                                }
                        }
                }

                File installCfg = new File(Paths.getEditionGDF(), "installed_mods.xml");
                PackageManager.getInstance().saveModSelection(installCfg, mods);

                TreeSet<Integer> unlockLevels = new TreeSet<Integer>();

                Vector<File> foldersOni = new Vector<File>();
                foldersOni.add(Paths.getVanillaOnisPath());

                Vector<File> foldersPatches = new Vector<File>();

                for (Package m : mods) {
                        for (int lev : m.getUnlockLevels())
                                unlockLevels.add(lev);

                        File oni = CaseInsensitiveFile.getCaseInsensitiveFile(
                                        m.getLocalPath(), "oni");
                        if (oni.exists()) {
                                if (m.hasSeparatePlatformDirs()) {
                                        File oniCommon = CaseInsensitiveFile
                                                        .getCaseInsensitiveFile(oni, "common");
                                        File oniMac = CaseInsensitiveFile.getCaseInsensitiveFile(
                                                        oni, "mac_only");
                                        File oniWin = CaseInsensitiveFile.getCaseInsensitiveFile(
                                                        oni, "win_only");
                                        if (oniCommon.exists())
                                                foldersOni.add(oniCommon);
                                        if (PlatformInformation.getPlatform() == Platform.MACOS
                                                        && oniMac.exists())
                                                foldersOni.add(oniMac);
                                        else if (oniWin.exists())
                                                foldersOni.add(oniWin);
                                } else {
                                        foldersOni.add(oni);
                                }
                        }

                        File patches = CaseInsensitiveFile.getCaseInsensitiveFile(
                                        m.getLocalPath(), "patches");
                        if (patches.exists()) {
                                if (m.hasSeparatePlatformDirs()) {
                                        File patchesCommon = CaseInsensitiveFile
                                                        .getCaseInsensitiveFile(patches, "common");
                                        File patchesMac = CaseInsensitiveFile
                                                        .getCaseInsensitiveFile(patches, "mac_only");
                                        File patchesWin = CaseInsensitiveFile
                                                        .getCaseInsensitiveFile(patches, "win_only");
                                        if (patchesCommon.exists())
                                                foldersPatches.add(patchesCommon);
                                        if (PlatformInformation.getPlatform() == Platform.MACOS
                                                        && patchesMac.exists())
                                                foldersPatches.add(patchesMac);
                                        else if (patchesWin.exists())
                                                foldersPatches.add(patchesWin);
                                } else {
                                        foldersPatches.add(patches);
                                }
                        }
                }

                TreeMap<String, Vector<File>> levels = new TreeMap<String, Vector<File>>(
                                new NaturalOrderComparator());
                for (File path : foldersOni) {
                        for (File levelF : path.listFiles()) {
                                String fn = levelF.getName().toLowerCase();
                                String levelN = null;
                                if (levelF.isDirectory()) {
                                        levelN = fn;
                                } else if (fn.endsWith(".dat")) {
                                        levelN = fn.substring(0, fn.lastIndexOf('.')).toLowerCase();
                                }
                                if (levelN != null) {
                                        if (!levels.containsKey(levelN))
                                                levels.put(levelN, new Vector<File>());
                                        levels.get(levelN).add(levelF);
                                }
                        }
                }

                applyPatches(levels, foldersPatches, listener, log);

                combineBinaryFiles(levels, listener, log);
                combineBSLFolders(mods, listener, log);

                copyVideos(log);

                if (unlockLevels.size() > 0) {
                        unlockLevels(unlockLevels, log);
                }

                log.println();
                Date end = new Date();
                log.println("Initialization ended at " + sdf.format(end));
                log.println("Process took "
                                + ((end.getTime() - start.getTime()) / 1000) + " seconds");
                log.close();
        }

        private static void combineBSLFolders(TreeSet<Package> mods,
                        InstallProgressListener listener, Logger log) {
                listener.installProgressUpdate(95, 100, "Installing BSL files");
                log.println();
                log.println("Installing BSL files");

                HashMap<EBSLInstallType, Vector<Package>> modsToInclude = new HashMap<EBSLInstallType, Vector<Package>>();
                modsToInclude.put(EBSLInstallType.NORMAL, new Vector<Package>());
                modsToInclude.put(EBSLInstallType.ADDON, new Vector<Package>());

                for (Package m : mods.descendingSet()) {
                        File bsl = CaseInsensitiveFile.getCaseInsensitiveFile(
                                        m.getLocalPath(), "bsl");
                        if (bsl.exists()) {
                                if (m.hasSeparatePlatformDirs()) {
                                        File bslCommon = CaseInsensitiveFile
                                                        .getCaseInsensitiveFile(bsl, "common");
                                        File bslMac = CaseInsensitiveFile.getCaseInsensitiveFile(
                                                        bsl, "mac_only");
                                        File bslWin = CaseInsensitiveFile.getCaseInsensitiveFile(
                                                        bsl, "win_only");
                                        if ((PlatformInformation.getPlatform() == Platform.MACOS && bslMac
                                                        .exists())
                                                        || ((PlatformInformation.getPlatform() == Platform.WIN || PlatformInformation
                                                                        .getPlatform() == Platform.LINUX) && bslWin
                                                                        .exists()) || bslCommon.exists()) {
                                                modsToInclude.get(m.getBSLInstallType()).add(m);
                                        }
                                } else {
                                        modsToInclude.get(m.getBSLInstallType()).add(m);
                                }
                        }
                }

                for (Package m : modsToInclude.get(EBSLInstallType.NORMAL)) {
                        copyBSL(m, false, log);
                }
                Vector<Package> addons = modsToInclude.get(EBSLInstallType.ADDON);
                for (int i = addons.size() - 1; i >= 0; i--) {
                        copyBSL(addons.get(i), true, log);
                }
        }

        private static void copyBSL(Package sourceMod, boolean addon, Logger log) {
                File targetBaseFolder = new File(Paths.getEditionGDF(), "IGMD");
                if (!targetBaseFolder.exists())
                        targetBaseFolder.mkdir();

                Vector<File> sources = new Vector<File>();
                File bsl = CaseInsensitiveFile.getCaseInsensitiveFile(
                                sourceMod.getLocalPath(), "bsl");
                if (sourceMod.hasSeparatePlatformDirs()) {
                        File bslCommon = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
                                        "common");
                        File bslMac = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
                                        "mac_only");
                        File bslWin = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
                                        "win_only");
                        if (PlatformInformation.getPlatform() == Platform.MACOS
                                        && bslMac.exists()) {
                                for (File f : bslMac.listFiles(dirFileFilter)) {
                                        File targetBSL = new File(targetBaseFolder, f.getName());
                                        if (addon || !targetBSL.exists())
                                                sources.add(f);
                                }
                        }
                        if ((PlatformInformation.getPlatform() == Platform.WIN || PlatformInformation
                                        .getPlatform() == Platform.LINUX) && bslWin.exists()) {
                                for (File f : bslWin.listFiles(dirFileFilter)) {
                                        File targetBSL = new File(targetBaseFolder, f.getName());
                                        if (addon || !targetBSL.exists())
                                                sources.add(f);
                                }
                        }
                        if (bslCommon.exists()) {
                                for (File f : bslCommon.listFiles(dirFileFilter)) {
                                        File targetBSL = new File(targetBaseFolder, f.getName());
                                        if (addon || !targetBSL.exists())
                                                sources.add(f);
                                }
                        }
                } else {
                        for (File f : bsl.listFiles(dirFileFilter)) {
                                File targetBSL = new File(targetBaseFolder, f.getName());
                                if (addon || !targetBSL.exists())
                                        sources.add(f);
                        }
                }

                log.println("\tMod \"" + sourceMod.getName() + "\"");
                for (File f : sources) {
                        log.println("\t\t" + f.getName());
                        File targetPath = new File(targetBaseFolder, f.getName());
                        if (!targetPath.exists())
                                targetPath.mkdir();
                        if (!(CaseInsensitiveFile.getCaseInsensitiveFile(targetPath,
                                        "ignore.txt").exists())) {
                                for (File fbsl : f.listFiles()) {
                                        if (fbsl.getName().toLowerCase().endsWith(".bsl")) {
                                                File targetFile = new File(targetPath, fbsl.getName());
                                                if (addon || !targetFile.exists()) {
                                                        try {
                                                                FileUtils.copyFile(fbsl, targetFile);
                                                        } catch (IOException e) {
                                                                e.printStackTrace();
                                                        }
                                                }
                                        }
                                }
                        }
                }
        }

        private static void applyPatches(
                        TreeMap<String, Vector<File>> oniLevelFolders,
                        List<File> patchFolders, InstallProgressListener listener,
                        Logger log) {
                log.println();
                log.println("Applying XML patches");
                listener.installProgressUpdate(0, 1, "Applying XML patches");

                long startMS = new Date().getTime();

                String tmpFolderName = "installrun_temp-"
                                + new SimpleDateFormat("yyyy_MM_dd-HH_mm_ss")
                                                .format(new Date());
                File tmpFolder = new File(Paths.getTempPath(), tmpFolderName);
                tmpFolder.mkdir();

                TreeMap<String, Vector<File>> patches = new TreeMap<String, Vector<File>>(
                                new NaturalOrderComparator());
                for (File patchFolder : patchFolders) {
                        for (File levelFolder : patchFolder.listFiles(dirFileFilter)) {
                                String lvlName = levelFolder.getName().toLowerCase();
                                for (File f : FileUtils.listFiles(levelFolder,
                                                new String[] { "oni-patch" }, true)) {
                                        if (!patches.containsKey(lvlName))
                                                patches.put(lvlName, new Vector<File>());
                                        patches.get(lvlName).add(f);
                                }
                        }
                }

                for (String level : patches.keySet()) {
                        File levelFolder = new File(tmpFolder, level);
                        levelFolder.mkdir();

                        log.println("\t\tPatches for " + level);

                        Vector<String> exportPatterns = new Vector<String>();
                        // Get files to be patched from vanilla.dat
                        for (File patch : patches.get(level)) {
                                String patternWildcard = patch.getName();
                                patternWildcard = patternWildcard.substring(0,
                                                patternWildcard.indexOf(".oni-patch"));
                                patternWildcard = patternWildcard.replace('-', '*');
                                exportPatterns.add(patternWildcard);
                        }
                        for (File srcFolder : oniLevelFolders.get(level)) {
                                if (srcFolder.isFile()) {
                                        if (srcFolder.getPath().toLowerCase().contains("vanilla")) {
                                                // Extract from .dat
                                                ApplicationInvocationResult res = OniSplit.export(
                                                                levelFolder, srcFolder, exportPatterns);
                                                log.logAppOutput(res, true);
                                        }
                                }
                        }

                        // Get files to be patched from packages
                        for (File patch : patches.get(level)) {
                                String patternWildcard = patch.getName();
                                patternWildcard = patternWildcard.substring(0,
                                                patternWildcard.indexOf(".oni-patch"));
                                patternWildcard = patternWildcard.replace('-', '*');
                                patternWildcard = patternWildcard + ".oni";
                                Vector<String> patterns = new Vector<String>();
                                patterns.add(patternWildcard);
                                final Pattern patternRegex = Pattern.compile(
                                                patternWildcard.replaceAll("\\*", ".\\*"),
                                                Pattern.CASE_INSENSITIVE);

                                for (File srcFolder : oniLevelFolders.get(level)) {
                                        if (srcFolder.isFile()) {
                                                if (!srcFolder.getPath().toLowerCase()
                                                                .contains("vanilla")) {
                                                        // Extract from .dat
                                                        ApplicationInvocationResult res = OniSplit.export(
                                                                        levelFolder, srcFolder, patterns);
                                                        log.logAppOutput(res, true);
                                                }
                                        } else {
                                                // Copy from folder with overwrite
                                                for (File f : FileUtils.listFiles(srcFolder,
                                                                new RegexFileFilter(patternRegex),
                                                                TrueFileFilter.TRUE)) {
                                                        try {
                                                                FileUtils.copyFileToDirectory(f, levelFolder);
                                                        } catch (IOException e) {
                                                                e.printStackTrace();
                                                        }
                                                }
                                        }
                                }
                        }

                        // Extract files to XML
                        File levelFolderXML = new File(levelFolder, "xml");
                        Vector<File> files = new Vector<File>();
                        files.add(new File(levelFolder, "*.oni"));
                        ApplicationInvocationResult res = OniSplit.convertOniToXML(
                                        levelFolderXML, files);
                        log.logAppOutput(res, true);

                        // Create masterpatch file (containing calls to all individual
                        // patches)
                        File masterpatch = new File(levelFolderXML, "masterpatch.oni-patch");
                        PrintWriter masterpatchWriter = null;
                        try {
                                masterpatchWriter = new PrintWriter(masterpatch);
                        } catch (FileNotFoundException e) {
                                e.printStackTrace();
                        }
                        for (File patch : patches.get(level)) {
                                String patternWildcard = patch.getName();
                                patternWildcard = patternWildcard.substring(0,
                                                patternWildcard.indexOf(".oni-patch"));
                                patternWildcard = patternWildcard + ".xml";
                                patternWildcard = patternWildcard.replace('-', '*');
                                File xmlFilePath = new File(levelFolderXML, patternWildcard);
                                masterpatchWriter
                                                .println(String
                                                                .format("@COMMAND patchfile -filename:\"%s\" -forceinfiles:\"%s\"",
                                                                                patch.getPath(), xmlFilePath.getPath()));
                        }
                        masterpatchWriter.close();
                        // Apply patches through masterpatch in levelFolderXML
                        res = XMLTools.patch(masterpatch, null);
                        log.logAppOutput(res, false);

                        // Create .oni files from XML
                        files.clear();
                        files.add(new File(levelFolderXML, "*.xml"));
                        res = OniSplit.convertXMLtoOni(levelFolder, files);
                        log.logAppOutput(res, true);

                        // Remove XML folder as import will only require .oni's
                        // try {
                        // FileUtils.deleteDirectory(levelFolderXML);
                        // } catch (IOException e) {
                        // e.printStackTrace();
                        // }

                        oniLevelFolders.get(level).add(levelFolder);
                }

                log.println("Applying XML patches took "
                                + (new Date().getTime() - startMS) + " ms");
        }

        private static void combineBinaryFiles(
                        TreeMap<String, Vector<File>> oniLevelFolders,
                        InstallProgressListener listener, Logger log) {
                long startMS = new Date().getTime();

                int totalSteps = oniLevelFolders.size() + 1;
                int stepsDone = 0;

                log.println();
                log.println("Importing levels");
                for (String l : oniLevelFolders.keySet()) {
                        log.println("\tLevel " + l);
                        listener.installProgressUpdate(stepsDone, totalSteps,
                                        "Installing level " + l);

                        ApplicationInvocationResult res = OniSplit.packLevel(
                                        oniLevelFolders.get(l), new File(Paths.getEditionGDF(),
                                                        sanitizeLevelName(l) + ".dat"));
                        log.logAppOutput(res, true);

                        stepsDone++;
                }

                log.println("Importing levels took " + (new Date().getTime() - startMS)
                                + " ms");
                log.println();
        }

        private static void copyVideos(Logger log) {
                log.println();
                if (SettingsManager.getInstance().get("copyintro", false)) {
                        File src = new File(Paths.getVanillaGDF(), "intro.bik");
                        log.println("Copying intro");
                        if (src.exists()) {
                                try {
                                        FileUtils.copyFileToDirectory(src, Paths.getEditionGDF());
                                } catch (IOException e) {
                                        e.printStackTrace();
                                }
                        }
                } else {
                        log.println("NOT copying intro");
                }
                if (SettingsManager.getInstance().get("copyoutro", true)) {
                        File src = new File(Paths.getVanillaGDF(), "outro.bik");
                        log.println("Copying outro");
                        if (src.exists()) {
                                try {
                                        FileUtils.copyFileToDirectory(src, Paths.getEditionGDF());
                                } catch (IOException e) {
                                        e.printStackTrace();
                                }
                        }
                } else {
                        log.println("NOT copying outro");
                }
        }

        private static void unlockLevels(TreeSet<Integer> unlockLevels, Logger log) {
                File dat = new File(Paths.getEditionBasePath(), "persist.dat");
                log.println();
                log.println("Unlocking levels: " + unlockLevels.toString());
                if (!dat.exists()) {
                        InputStream is = AEInstaller2.class
                                        .getResourceAsStream("/net/oni2/aeinstaller/resources/persist.dat");
                        try {
                                FileUtils.copyInputStreamToFile(is, dat);
                        } catch (IOException e) {
                                e.printStackTrace();
                        }
                }
                PersistDat save = new PersistDat(dat);
                HashSet<Integer> currentlyUnlocked = save.getUnlockedLevels();
                currentlyUnlocked.addAll(unlockLevels);
                save.setUnlockedLevels(currentlyUnlocked);
                save.close();
        }

        private static String sanitizeLevelName(String ln) {
                int ind = ln.indexOf("_");
                String res = ln.substring(0, ind + 1);
                res += ln.substring(ind + 1, ind + 2).toUpperCase();
                res += ln.substring(ind + 2);
                return res;
        }

}
Revision:	749
Committed:	Tue Mar 26 17:26:41 2013 UTC (12 years, 6 months ago) by alloc
Content type:	text/x-java
File size:	18946 byte(s)
Log Message:	AEI2 0.99x: - Refactoring
#	Content
1	package net.oni2.aeinstaller.backend.oni.management;
2
3	import java.io.File;
4	import java.io.FileFilter;
5	import java.io.FileNotFoundException;
6	import java.io.FilenameFilter;
7	import java.io.IOException;
8	import java.io.InputStream;
9	import java.io.PrintWriter;
10	import java.text.SimpleDateFormat;
11	import java.util.Date;
12	import java.util.HashMap;
13	import java.util.HashSet;
14	import java.util.List;
15	import java.util.TreeMap;
16	import java.util.TreeSet;
17	import java.util.Vector;
18	import java.util.regex.Pattern;
19
20	import net.oni2.SettingsManager;
21	import net.oni2.aeinstaller.AEInstaller2;
22	import net.oni2.aeinstaller.backend.CaseInsensitiveFile;
23	import net.oni2.aeinstaller.backend.Paths;
24	import net.oni2.aeinstaller.backend.oni.OniSplit;
25	import net.oni2.aeinstaller.backend.oni.PersistDat;
26	import net.oni2.aeinstaller.backend.oni.XMLTools;
27	import net.oni2.aeinstaller.backend.packages.EBSLInstallType;
28	import net.oni2.aeinstaller.backend.packages.Package;
29	import net.oni2.aeinstaller.backend.packages.PackageManager;
30	import net.oni2.platformtools.PlatformInformation;
31	import net.oni2.platformtools.PlatformInformation.Platform;
32	import net.oni2.platformtools.applicationinvoker.ApplicationInvocationResult;
33
34	import org.apache.commons.io.FileUtils;
35	import org.apache.commons.io.filefilter.RegexFileFilter;
36	import org.apache.commons.io.filefilter.TrueFileFilter;
37	import org.javabuilders.swing.SwingJavaBuilder;
38
39	import com.paour.NaturalOrderComparator;
40
41	/**
42	* @author Christian Illy
43	*/
44	public class Installer {
45	private static FileFilter dirFileFilter = new FileFilter() {
46	@Override
47	public boolean accept(File pathname) {
48	return pathname.isDirectory();
49	}
50	};
51
52	/**
53	* Verify that the Edition is within a subfolder to vanilla Oni
54	* (..../Oni/Edition/AEInstaller)
55	*
56	* @return true if GDF can be found in the parent's parent-path
57	*/
58	public static boolean verifyRunningDirectory() {
59	return Paths.getVanillaGDF().exists()
60	&& Paths.getVanillaGDF().isDirectory();
61	}
62
63	/**
64	* @return Is Edition Core initialized
65	*/
66	public static boolean isEditionInitialized() {
67	return Paths.getVanillaOnisPath().exists();
68	}
69
70	/**
71	* @return list of currently installed mods
72	*/
73	public static Vector<Integer> getInstalledMods() {
74	File installCfg = new File(Paths.getEditionGDF(), "installed_mods.xml");
75	return PackageManager.getInstance().loadModSelection(installCfg);
76	}
77
78	/**
79	* Install the given set of mods
80	*
81	* @param mods
82	* Mods to install
83	* @param listener
84	* Listener for install progress updates
85	*/
86	public static void install(TreeSet<Package> mods,
87	InstallProgressListener listener) {
88	File logFile = new File(Paths.getInstallerPath(), "Installation.log");
89	Logger log = null;
90	try {
91	log = new Logger(logFile);
92	} catch (FileNotFoundException e) {
93	e.printStackTrace();
94	}
95	SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
96	Date start = new Date();
97	log.println("Installation of mods started at " + sdf.format(start));
98
99	log.println();
100	log.println("AEI2 version: "
101	+ SwingJavaBuilder.getConfig().getResource("appversion"));
102	log.println("Installed tools:");
103	for (Package t : PackageManager.getInstance().getInstalledTools()) {
104	log.println(String.format(" - %s (%s)", t.getName(), t.getVersion()));
105	}
106	log.println("Installing mods:");
107	for (Package m : mods) {
108	log.println(String.format(" - %s (%s)", m.getName(), m.getVersion()));
109	}
110	log.println();
111
112	Paths.getEditionGDF().mkdirs();
113	for (File f : Paths.getEditionGDF().listFiles(new FilenameFilter() {
114	public boolean accept(File arg0, String arg1) {
115	String s = arg1.toLowerCase();
116	return s.endsWith(".dat")
117	\|\| s.endsWith(".raw")
118	\|\| s.endsWith(".sep")
119	\|\| (s.equals("intro.bik") && !SettingsManager
120	.getInstance().get("copyintro", false))
121	\|\| (s.equals("outro.bik") && !SettingsManager
122	.getInstance().get("copyoutro", false));
123	}
124	})) {
125	f.delete();
126	}
127	File IGMD = new File(Paths.getEditionGDF(), "IGMD");
128	if (IGMD.exists()) {
129	for (File f : IGMD.listFiles(new FileFilter() {
130	@Override
131	public boolean accept(File pathname) {
132	return pathname.isDirectory();
133	}
134	})) {
135	File ignore = CaseInsensitiveFile.getCaseInsensitiveFile(f,
136	"ignore.txt");
137	if (!ignore.exists()) {
138	try {
139	FileUtils.deleteDirectory(f);
140	} catch (IOException e) {
141	e.printStackTrace();
142	}
143	}
144	}
145	}
146
147	File installCfg = new File(Paths.getEditionGDF(), "installed_mods.xml");
148	PackageManager.getInstance().saveModSelection(installCfg, mods);
149
150	TreeSet<Integer> unlockLevels = new TreeSet<Integer>();
151
152	Vector<File> foldersOni = new Vector<File>();
153	foldersOni.add(Paths.getVanillaOnisPath());
154
155	Vector<File> foldersPatches = new Vector<File>();
156
157	for (Package m : mods) {
158	for (int lev : m.getUnlockLevels())
159	unlockLevels.add(lev);
160
161	File oni = CaseInsensitiveFile.getCaseInsensitiveFile(
162	m.getLocalPath(), "oni");
163	if (oni.exists()) {
164	if (m.hasSeparatePlatformDirs()) {
165	File oniCommon = CaseInsensitiveFile
166	.getCaseInsensitiveFile(oni, "common");
167	File oniMac = CaseInsensitiveFile.getCaseInsensitiveFile(
168	oni, "mac_only");
169	File oniWin = CaseInsensitiveFile.getCaseInsensitiveFile(
170	oni, "win_only");
171	if (oniCommon.exists())
172	foldersOni.add(oniCommon);
173	if (PlatformInformation.getPlatform() == Platform.MACOS
174	&& oniMac.exists())
175	foldersOni.add(oniMac);
176	else if (oniWin.exists())
177	foldersOni.add(oniWin);
178	} else {
179	foldersOni.add(oni);
180	}
181	}
182
183	File patches = CaseInsensitiveFile.getCaseInsensitiveFile(
184	m.getLocalPath(), "patches");
185	if (patches.exists()) {
186	if (m.hasSeparatePlatformDirs()) {
187	File patchesCommon = CaseInsensitiveFile
188	.getCaseInsensitiveFile(patches, "common");
189	File patchesMac = CaseInsensitiveFile
190	.getCaseInsensitiveFile(patches, "mac_only");
191	File patchesWin = CaseInsensitiveFile
192	.getCaseInsensitiveFile(patches, "win_only");
193	if (patchesCommon.exists())
194	foldersPatches.add(patchesCommon);
195	if (PlatformInformation.getPlatform() == Platform.MACOS
196	&& patchesMac.exists())
197	foldersPatches.add(patchesMac);
198	else if (patchesWin.exists())
199	foldersPatches.add(patchesWin);
200	} else {
201	foldersPatches.add(patches);
202	}
203	}
204	}
205
206	TreeMap<String, Vector<File>> levels = new TreeMap<String, Vector<File>>(
207	new NaturalOrderComparator());
208	for (File path : foldersOni) {
209	for (File levelF : path.listFiles()) {
210	String fn = levelF.getName().toLowerCase();
211	String levelN = null;
212	if (levelF.isDirectory()) {
213	levelN = fn;
214	} else if (fn.endsWith(".dat")) {
215	levelN = fn.substring(0, fn.lastIndexOf('.')).toLowerCase();
216	}
217	if (levelN != null) {
218	if (!levels.containsKey(levelN))
219	levels.put(levelN, new Vector<File>());
220	levels.get(levelN).add(levelF);
221	}
222	}
223	}
224
225	applyPatches(levels, foldersPatches, listener, log);
226
227	combineBinaryFiles(levels, listener, log);
228	combineBSLFolders(mods, listener, log);
229
230	copyVideos(log);
231
232	if (unlockLevels.size() > 0) {
233	unlockLevels(unlockLevels, log);
234	}
235
236	log.println();
237	Date end = new Date();
238	log.println("Initialization ended at " + sdf.format(end));
239	log.println("Process took "
240	+ ((end.getTime() - start.getTime()) / 1000) + " seconds");
241	log.close();
242	}
243
244	private static void combineBSLFolders(TreeSet<Package> mods,
245	InstallProgressListener listener, Logger log) {
246	listener.installProgressUpdate(95, 100, "Installing BSL files");
247	log.println();
248	log.println("Installing BSL files");
249
250	HashMap<EBSLInstallType, Vector<Package>> modsToInclude = new HashMap<EBSLInstallType, Vector<Package>>();
251	modsToInclude.put(EBSLInstallType.NORMAL, new Vector<Package>());
252	modsToInclude.put(EBSLInstallType.ADDON, new Vector<Package>());
253
254	for (Package m : mods.descendingSet()) {
255	File bsl = CaseInsensitiveFile.getCaseInsensitiveFile(
256	m.getLocalPath(), "bsl");
257	if (bsl.exists()) {
258	if (m.hasSeparatePlatformDirs()) {
259	File bslCommon = CaseInsensitiveFile
260	.getCaseInsensitiveFile(bsl, "common");
261	File bslMac = CaseInsensitiveFile.getCaseInsensitiveFile(
262	bsl, "mac_only");
263	File bslWin = CaseInsensitiveFile.getCaseInsensitiveFile(
264	bsl, "win_only");
265	if ((PlatformInformation.getPlatform() == Platform.MACOS && bslMac
266	.exists())
267	\|\| ((PlatformInformation.getPlatform() == Platform.WIN \|\| PlatformInformation
268	.getPlatform() == Platform.LINUX) && bslWin
269	.exists()) \|\| bslCommon.exists()) {
270	modsToInclude.get(m.getBSLInstallType()).add(m);
271	}
272	} else {
273	modsToInclude.get(m.getBSLInstallType()).add(m);
274	}
275	}
276	}
277
278	for (Package m : modsToInclude.get(EBSLInstallType.NORMAL)) {
279	copyBSL(m, false, log);
280	}
281	Vector<Package> addons = modsToInclude.get(EBSLInstallType.ADDON);
282	for (int i = addons.size() - 1; i >= 0; i--) {
283	copyBSL(addons.get(i), true, log);
284	}
285	}
286
287	private static void copyBSL(Package sourceMod, boolean addon, Logger log) {
288	File targetBaseFolder = new File(Paths.getEditionGDF(), "IGMD");
289	if (!targetBaseFolder.exists())
290	targetBaseFolder.mkdir();
291
292	Vector<File> sources = new Vector<File>();
293	File bsl = CaseInsensitiveFile.getCaseInsensitiveFile(
294	sourceMod.getLocalPath(), "bsl");
295	if (sourceMod.hasSeparatePlatformDirs()) {
296	File bslCommon = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
297	"common");
298	File bslMac = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
299	"mac_only");
300	File bslWin = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
301	"win_only");
302	if (PlatformInformation.getPlatform() == Platform.MACOS
303	&& bslMac.exists()) {
304	for (File f : bslMac.listFiles(dirFileFilter)) {
305	File targetBSL = new File(targetBaseFolder, f.getName());
306	if (addon \|\| !targetBSL.exists())
307	sources.add(f);
308	}
309	}
310	if ((PlatformInformation.getPlatform() == Platform.WIN \|\| PlatformInformation
311	.getPlatform() == Platform.LINUX) && bslWin.exists()) {
312	for (File f : bslWin.listFiles(dirFileFilter)) {
313	File targetBSL = new File(targetBaseFolder, f.getName());
314	if (addon \|\| !targetBSL.exists())
315	sources.add(f);
316	}
317	}
318	if (bslCommon.exists()) {
319	for (File f : bslCommon.listFiles(dirFileFilter)) {
320	File targetBSL = new File(targetBaseFolder, f.getName());
321	if (addon \|\| !targetBSL.exists())
322	sources.add(f);
323	}
324	}
325	} else {
326	for (File f : bsl.listFiles(dirFileFilter)) {
327	File targetBSL = new File(targetBaseFolder, f.getName());
328	if (addon \|\| !targetBSL.exists())
329	sources.add(f);
330	}
331	}
332
333	log.println("\tMod \"" + sourceMod.getName() + "\"");
334	for (File f : sources) {
335	log.println("\t\t" + f.getName());
336	File targetPath = new File(targetBaseFolder, f.getName());
337	if (!targetPath.exists())
338	targetPath.mkdir();
339	if (!(CaseInsensitiveFile.getCaseInsensitiveFile(targetPath,
340	"ignore.txt").exists())) {
341	for (File fbsl : f.listFiles()) {
342	if (fbsl.getName().toLowerCase().endsWith(".bsl")) {
343	File targetFile = new File(targetPath, fbsl.getName());
344	if (addon \|\| !targetFile.exists()) {
345	try {
346	FileUtils.copyFile(fbsl, targetFile);
347	} catch (IOException e) {
348	e.printStackTrace();
349	}
350	}
351	}
352	}
353	}
354	}
355	}
356
357	private static void applyPatches(
358	TreeMap<String, Vector<File>> oniLevelFolders,
359	List<File> patchFolders, InstallProgressListener listener,
360	Logger log) {
361	log.println();
362	log.println("Applying XML patches");
363	listener.installProgressUpdate(0, 1, "Applying XML patches");
364
365	long startMS = new Date().getTime();
366
367	String tmpFolderName = "installrun_temp-"
368	+ new SimpleDateFormat("yyyy_MM_dd-HH_mm_ss")
369	.format(new Date());
370	File tmpFolder = new File(Paths.getTempPath(), tmpFolderName);
371	tmpFolder.mkdir();
372
373	TreeMap<String, Vector<File>> patches = new TreeMap<String, Vector<File>>(
374	new NaturalOrderComparator());
375	for (File patchFolder : patchFolders) {
376	for (File levelFolder : patchFolder.listFiles(dirFileFilter)) {
377	String lvlName = levelFolder.getName().toLowerCase();
378	for (File f : FileUtils.listFiles(levelFolder,
379	new String[] { "oni-patch" }, true)) {
380	if (!patches.containsKey(lvlName))
381	patches.put(lvlName, new Vector<File>());
382	patches.get(lvlName).add(f);
383	}
384	}
385	}
386
387	for (String level : patches.keySet()) {
388	File levelFolder = new File(tmpFolder, level);
389	levelFolder.mkdir();
390
391	log.println("\t\tPatches for " + level);
392
393	Vector<String> exportPatterns = new Vector<String>();
394	// Get files to be patched from vanilla.dat
395	for (File patch : patches.get(level)) {
396	String patternWildcard = patch.getName();
397	patternWildcard = patternWildcard.substring(0,
398	patternWildcard.indexOf(".oni-patch"));
399	patternWildcard = patternWildcard.replace('-', '*');
400	exportPatterns.add(patternWildcard);
401	}
402	for (File srcFolder : oniLevelFolders.get(level)) {
403	if (srcFolder.isFile()) {
404	if (srcFolder.getPath().toLowerCase().contains("vanilla")) {
405	// Extract from .dat
406	ApplicationInvocationResult res = OniSplit.export(
407	levelFolder, srcFolder, exportPatterns);
408	log.logAppOutput(res, true);
409	}
410	}
411	}
412
413	// Get files to be patched from packages
414	for (File patch : patches.get(level)) {
415	String patternWildcard = patch.getName();
416	patternWildcard = patternWildcard.substring(0,
417	patternWildcard.indexOf(".oni-patch"));
418	patternWildcard = patternWildcard.replace('-', '*');
419	patternWildcard = patternWildcard + ".oni";
420	Vector<String> patterns = new Vector<String>();
421	patterns.add(patternWildcard);
422	final Pattern patternRegex = Pattern.compile(
423	patternWildcard.replaceAll("\\", ".\\"),
424	Pattern.CASE_INSENSITIVE);
425
426	for (File srcFolder : oniLevelFolders.get(level)) {
427	if (srcFolder.isFile()) {
428	if (!srcFolder.getPath().toLowerCase()
429	.contains("vanilla")) {
430	// Extract from .dat
431	ApplicationInvocationResult res = OniSplit.export(
432	levelFolder, srcFolder, patterns);
433	log.logAppOutput(res, true);
434	}
435	} else {
436	// Copy from folder with overwrite
437	for (File f : FileUtils.listFiles(srcFolder,
438	new RegexFileFilter(patternRegex),
439	TrueFileFilter.TRUE)) {
440	try {
441	FileUtils.copyFileToDirectory(f, levelFolder);
442	} catch (IOException e) {
443	e.printStackTrace();
444	}
445	}
446	}
447	}
448	}
449
450	// Extract files to XML
451	File levelFolderXML = new File(levelFolder, "xml");
452	Vector<File> files = new Vector<File>();
453	files.add(new File(levelFolder, "*.oni"));
454	ApplicationInvocationResult res = OniSplit.convertOniToXML(
455	levelFolderXML, files);
456	log.logAppOutput(res, true);
457
458	// Create masterpatch file (containing calls to all individual
459	// patches)
460	File masterpatch = new File(levelFolderXML, "masterpatch.oni-patch");
461	PrintWriter masterpatchWriter = null;
462	try {
463	masterpatchWriter = new PrintWriter(masterpatch);
464	} catch (FileNotFoundException e) {
465	e.printStackTrace();
466	}
467	for (File patch : patches.get(level)) {
468	String patternWildcard = patch.getName();
469	patternWildcard = patternWildcard.substring(0,
470	patternWildcard.indexOf(".oni-patch"));
471	patternWildcard = patternWildcard + ".xml";
472	patternWildcard = patternWildcard.replace('-', '*');
473	File xmlFilePath = new File(levelFolderXML, patternWildcard);
474	masterpatchWriter
475	.println(String
476	.format("@COMMAND patchfile -filename:\"%s\" -forceinfiles:\"%s\"",
477	patch.getPath(), xmlFilePath.getPath()));
478	}
479	masterpatchWriter.close();
480	// Apply patches through masterpatch in levelFolderXML
481	res = XMLTools.patch(masterpatch, null);
482	log.logAppOutput(res, false);
483
484	// Create .oni files from XML
485	files.clear();
486	files.add(new File(levelFolderXML, "*.xml"));
487	res = OniSplit.convertXMLtoOni(levelFolder, files);
488	log.logAppOutput(res, true);
489
490	// Remove XML folder as import will only require .oni's
491	// try {
492	// FileUtils.deleteDirectory(levelFolderXML);
493	// } catch (IOException e) {
494	// e.printStackTrace();
495	// }
496
497	oniLevelFolders.get(level).add(levelFolder);
498	}
499
500	log.println("Applying XML patches took "
501	+ (new Date().getTime() - startMS) + " ms");
502	}
503
504	private static void combineBinaryFiles(
505	TreeMap<String, Vector<File>> oniLevelFolders,
506	InstallProgressListener listener, Logger log) {
507	long startMS = new Date().getTime();
508
509	int totalSteps = oniLevelFolders.size() + 1;
510	int stepsDone = 0;
511
512	log.println();
513	log.println("Importing levels");
514	for (String l : oniLevelFolders.keySet()) {
515	log.println("\tLevel " + l);
516	listener.installProgressUpdate(stepsDone, totalSteps,
517	"Installing level " + l);
518
519	ApplicationInvocationResult res = OniSplit.packLevel(
520	oniLevelFolders.get(l), new File(Paths.getEditionGDF(),
521	sanitizeLevelName(l) + ".dat"));
522	log.logAppOutput(res, true);
523
524	stepsDone++;
525	}
526
527	log.println("Importing levels took " + (new Date().getTime() - startMS)
528	+ " ms");
529	log.println();
530	}
531
532	private static void copyVideos(Logger log) {
533	log.println();
534	if (SettingsManager.getInstance().get("copyintro", false)) {
535	File src = new File(Paths.getVanillaGDF(), "intro.bik");
536	log.println("Copying intro");
537	if (src.exists()) {
538	try {
539	FileUtils.copyFileToDirectory(src, Paths.getEditionGDF());
540	} catch (IOException e) {
541	e.printStackTrace();
542	}
543	}
544	} else {
545	log.println("NOT copying intro");
546	}
547	if (SettingsManager.getInstance().get("copyoutro", true)) {
548	File src = new File(Paths.getVanillaGDF(), "outro.bik");
549	log.println("Copying outro");
550	if (src.exists()) {
551	try {
552	FileUtils.copyFileToDirectory(src, Paths.getEditionGDF());
553	} catch (IOException e) {
554	e.printStackTrace();
555	}
556	}
557	} else {
558	log.println("NOT copying outro");
559	}
560	}
561
562	private static void unlockLevels(TreeSet<Integer> unlockLevels, Logger log) {
563	File dat = new File(Paths.getEditionBasePath(), "persist.dat");
564	log.println();
565	log.println("Unlocking levels: " + unlockLevels.toString());
566	if (!dat.exists()) {
567	InputStream is = AEInstaller2.class
568	.getResourceAsStream("/net/oni2/aeinstaller/resources/persist.dat");
569	try {
570	FileUtils.copyInputStreamToFile(is, dat);
571	} catch (IOException e) {
572	e.printStackTrace();
573	}
574	}
575	PersistDat save = new PersistDat(dat);
576	HashSet<Integer> currentlyUnlocked = save.getUnlockedLevels();
577	currentlyUnlocked.addAll(unlockLevels);
578	save.setUnlockedLevels(currentlyUnlocked);
579	save.close();
580	}
581
582	private static String sanitizeLevelName(String ln) {
583	int ind = ln.indexOf("_");
584	String res = ln.substring(0, ind + 1);
585	res += ln.substring(ind + 1, ind + 2).toUpperCase();
586	res += ln.substring(ind + 2);
587	return res;
588	}
589
590	}