source: AE/installer2/src/net/oni2/aeinstaller/backend/oni/Installer.java@ 701

Last change on this file since 701 was 700, checked in by alloc, 12 years ago

AEI2 0.99t:

  • Removed a System.exit call where it didn't belong ;)
File size: 27.4 KB
RevLine 
[596]1package net.oni2.aeinstaller.backend.oni;
2
3import java.io.File;
[597]4import java.io.FileFilter;
[624]5import java.io.FileInputStream;
[600]6import java.io.FileNotFoundException;
[624]7import java.io.FileOutputStream;
[596]8import java.io.FilenameFilter;
9import java.io.IOException;
[631]10import java.io.InputStream;
[600]11import java.io.PrintWriter;
12import java.text.SimpleDateFormat;
13import java.util.Date;
[608]14import java.util.HashMap;
[624]15import java.util.HashSet;
[598]16import java.util.List;
[596]17import java.util.Scanner;
[606]18import java.util.TreeMap;
[598]19import java.util.TreeSet;
20import java.util.Vector;
[698]21import java.util.regex.Pattern;
[596]22
[631]23import net.oni2.aeinstaller.AEInstaller2;
[699]24import net.oni2.aeinstaller.backend.CaseInsensitiveFile;
[596]25import net.oni2.aeinstaller.backend.Paths;
[597]26import net.oni2.aeinstaller.backend.Settings;
27import net.oni2.aeinstaller.backend.Settings.Platform;
[648]28import net.oni2.aeinstaller.backend.packages.EBSLInstallType;
29import net.oni2.aeinstaller.backend.packages.Package;
30import net.oni2.aeinstaller.backend.packages.PackageManager;
[596]31
32import org.apache.commons.io.FileUtils;
[698]33import org.apache.commons.io.filefilter.RegexFileFilter;
34import org.apache.commons.io.filefilter.TrueFileFilter;
[654]35import org.javabuilders.swing.SwingJavaBuilder;
[596]36
[624]37import com.thoughtworks.xstream.XStream;
38import com.thoughtworks.xstream.io.xml.StaxDriver;
39
[596]40/**
41 * @author Christian Illy
42 */
43public class Installer {
[608]44 private static FileFilter dirFileFilter = new FileFilter() {
45 @Override
46 public boolean accept(File pathname) {
47 return pathname.isDirectory();
48 }
49 };
50
[596]51 /**
52 * @return Is Edition Core initialized
53 */
54 public static boolean isEditionInitialized() {
[597]55 return Paths.getVanillaOnisPath().exists();
[596]56 }
57
58 private static void createEmptyPath(File path) throws IOException {
59 if (path.exists())
60 FileUtils.deleteDirectory(path);
61 path.mkdirs();
62 }
63
[602]64 /**
[604]65 * @return list of currently installed mods
66 */
67 public static Vector<Integer> getInstalledMods() {
68 File installCfg = new File(Paths.getEditionGDF(), "installed_mods.xml");
[648]69 return PackageManager.getInstance().loadModSelection(installCfg);
[604]70 }
71
72 /**
[624]73 * @return Currently installed tools
74 */
75 @SuppressWarnings("unchecked")
76 public static TreeSet<Integer> getInstalledTools() {
77 File installCfg = new File(Paths.getInstallerPath(),
78 "installed_tools.xml");
79 TreeSet<Integer> res = new TreeSet<Integer>();
80 try {
81 if (installCfg.exists()) {
82 FileInputStream fis = new FileInputStream(installCfg);
83 XStream xs = new XStream(new StaxDriver());
84 Object obj = xs.fromXML(fis);
85 if (obj instanceof TreeSet<?>)
86 res = (TreeSet<Integer>) obj;
87 fis.close();
88 }
89 } catch (FileNotFoundException e) {
90 e.printStackTrace();
91 } catch (IOException e) {
92 e.printStackTrace();
93 }
94 return res;
95 }
96
97 private static void writeInstalledTools(TreeSet<Integer> tools) {
98 File installCfg = new File(Paths.getInstallerPath(),
99 "installed_tools.xml");
100 try {
101 FileOutputStream fos = new FileOutputStream(installCfg);
102 XStream xs = new XStream(new StaxDriver());
103 xs.toXML(tools, fos);
104 fos.close();
105 } catch (FileNotFoundException e) {
106 e.printStackTrace();
107 } catch (IOException e) {
108 e.printStackTrace();
109 }
110 }
111
112 /**
[604]113 * @param tools
[699]114 * Tools to (un)install
115 * @param uninstall
116 * Uninstall tools or install?
[604]117 */
[699]118 public static void installTools(TreeSet<Package> tools, boolean uninstall) {
[624]119 TreeSet<Integer> installed = getInstalledTools();
[648]120 for (Package m : tools) {
[699]121 if (!uninstall || installed.contains(m.getPackageNumber())) {
122 File plain = CaseInsensitiveFile.getCaseInsensitiveFile(
123 m.getLocalPath(), "plain");
[624]124 if (plain.exists()) {
125 if (m.hasSeparatePlatformDirs()) {
[699]126 File plainCommon = CaseInsensitiveFile
127 .getCaseInsensitiveFile(plain, "common");
128 File plainMac = CaseInsensitiveFile
129 .getCaseInsensitiveFile(plain, "mac_only");
130 File plainWin = CaseInsensitiveFile
131 .getCaseInsensitiveFile(plain, "win_only");
[624]132 if (plainCommon.exists())
[699]133 copyRemoveToolsFiles(plainCommon,
134 Paths.getEditionBasePath(), uninstall);
[624]135 if (Settings.getPlatform() == Platform.MACOS
136 && plainMac.exists())
[699]137 copyRemoveToolsFiles(plainMac,
138 Paths.getEditionBasePath(), uninstall);
[624]139 else if (plainWin.exists())
[699]140 copyRemoveToolsFiles(plainWin,
141 Paths.getEditionBasePath(), uninstall);
[624]142 } else {
[699]143 copyRemoveToolsFiles(plain, Paths.getEditionBasePath(),
144 uninstall);
[624]145 }
146 }
147 }
[699]148 if (uninstall)
149 installed.remove(m.getPackageNumber());
150 else
151 installed.add(m.getPackageNumber());
[624]152 }
153 writeInstalledTools(installed);
[604]154 }
155
[699]156 private static void copyRemoveToolsFiles(File srcFolder, File targetFolder,
157 boolean remove) {
[604]158 for (File f : srcFolder.listFiles()) {
159 try {
160 if (f.isDirectory())
[699]161 copyRemoveToolsFiles(f,
162 CaseInsensitiveFile.getCaseInsensitiveFile(
163 targetFolder, f.getName()), remove);
164 else {
165 File targetFile = CaseInsensitiveFile
166 .getCaseInsensitiveFile(targetFolder,
167 f.getName());
168 if (remove) {
169 if (targetFile.exists())
170 targetFile.delete();
171 } else {
172 if (!targetFile.getName().equals(f.getName()))
173 targetFile.delete();
174 FileUtils.copyFileToDirectory(f, targetFolder);
175 }
176 }
[604]177 } catch (IOException e) {
178 e.printStackTrace();
179 }
180 }
[699]181 if (remove)
182 if (targetFolder.list().length == 0)
183 targetFolder.delete();
[604]184 }
[608]185
[604]186 /**
[602]187 * Install the given set of mods
188 *
189 * @param mods
190 * Mods to install
191 * @param listener
192 * Listener for install progress updates
193 */
[648]194 public static void install(TreeSet<Package> mods,
[600]195 InstallProgressListener listener) {
[653]196 File logFile = new File(Paths.getInstallerPath(), "Installation.log");
197 PrintWriter log = null;
198 try {
199 log = new PrintWriter(logFile);
200 } catch (FileNotFoundException e) {
201 e.printStackTrace();
202 }
203 SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
204 Date start = new Date();
205 log.println("Installation of mods started at " + sdf.format(start));
206
207 log.println();
[663]208 log.println("AEI2 version: "
209 + SwingJavaBuilder.getConfig().getResource("appversion"));
[653]210 log.println("Installed tools:");
211 for (Package t : PackageManager.getInstance().getInstalledTools()) {
212 log.println(String.format(" - %s (%s)", t.getName(), t.getVersion()));
213 }
214 log.println("Installing mods:");
215 for (Package m : mods) {
216 log.println(String.format(" - %s (%s)", m.getName(), m.getVersion()));
217 }
218 log.println();
219
[663]220 Paths.getEditionGDF().mkdirs();
221 for (File f : Paths.getEditionGDF().listFiles(new FilenameFilter() {
222 public boolean accept(File arg0, String arg1) {
223 String s = arg1.toLowerCase();
224 return s.endsWith(".dat")
225 || s.endsWith(".raw")
226 || s.endsWith(".sep")
227 || (s.equals("intro.bik") && !Settings.getInstance()
228 .get("copyintro", false))
229 || (s.equals("outro.bik") && !Settings.getInstance()
230 .get("copyoutro", false));
231 }
232 })) {
233 f.delete();
234 }
235 File IGMD = new File(Paths.getEditionGDF(), "IGMD");
236 if (IGMD.exists()) {
237 for (File f : IGMD.listFiles(new FileFilter() {
238 @Override
239 public boolean accept(File pathname) {
240 return pathname.isDirectory();
241 }
242 })) {
[699]243 File ignore = CaseInsensitiveFile.getCaseInsensitiveFile(f,
244 "ignore.txt");
[663]245 if (!ignore.exists()) {
246 try {
247 FileUtils.deleteDirectory(f);
248 } catch (IOException e) {
249 e.printStackTrace();
250 }
251 }
252 }
253 }
254
[604]255 File installCfg = new File(Paths.getEditionGDF(), "installed_mods.xml");
[648]256 PackageManager.getInstance().saveModSelection(installCfg, mods);
[604]257
[653]258 TreeSet<Integer> unlockLevels = new TreeSet<Integer>();
[624]259
[608]260 Vector<File> foldersOni = new Vector<File>();
261 foldersOni.add(Paths.getVanillaOnisPath());
[598]262
[698]263 Vector<File> foldersPatches = new Vector<File>();
264
[648]265 for (Package m : mods) {
[624]266 for (int lev : m.getUnlockLevels())
267 unlockLevels.add(lev);
268
[699]269 File oni = CaseInsensitiveFile.getCaseInsensitiveFile(
270 m.getLocalPath(), "oni");
[598]271 if (oni.exists()) {
272 if (m.hasSeparatePlatformDirs()) {
[699]273 File oniCommon = CaseInsensitiveFile
274 .getCaseInsensitiveFile(oni, "common");
275 File oniMac = CaseInsensitiveFile.getCaseInsensitiveFile(
276 oni, "mac_only");
277 File oniWin = CaseInsensitiveFile.getCaseInsensitiveFile(
278 oni, "win_only");
[598]279 if (oniCommon.exists())
[608]280 foldersOni.add(oniCommon);
[598]281 if (Settings.getPlatform() == Platform.MACOS
282 && oniMac.exists())
[608]283 foldersOni.add(oniMac);
[598]284 else if (oniWin.exists())
[608]285 foldersOni.add(oniWin);
[598]286 } else {
[608]287 foldersOni.add(oni);
[598]288 }
289 }
[698]290
[699]291 File patches = CaseInsensitiveFile.getCaseInsensitiveFile(
292 m.getLocalPath(), "patches");
[698]293 if (patches.exists()) {
294 if (m.hasSeparatePlatformDirs()) {
[699]295 File patchesCommon = CaseInsensitiveFile
296 .getCaseInsensitiveFile(patches, "common");
297 File patchesMac = CaseInsensitiveFile
298 .getCaseInsensitiveFile(patches, "mac_only");
299 File patchesWin = CaseInsensitiveFile
300 .getCaseInsensitiveFile(patches, "win_only");
[698]301 if (patchesCommon.exists())
302 foldersPatches.add(patchesCommon);
303 if (Settings.getPlatform() == Platform.MACOS
304 && patchesMac.exists())
305 foldersPatches.add(patchesMac);
306 else if (patchesWin.exists())
307 foldersPatches.add(patchesWin);
308 } else {
309 foldersPatches.add(patches);
310 }
311 }
[598]312 }
[663]313
[698]314 TreeMap<String, Vector<File>> levels = new TreeMap<String, Vector<File>>();
315 for (File path : foldersOni) {
316 for (File levelF : path.listFiles()) {
317 String fn = levelF.getName().toLowerCase();
318 String levelN = null;
319 if (levelF.isDirectory()) {
320 levelN = fn;
321 } else if (fn.endsWith(".dat")) {
322 levelN = fn.substring(0, fn.lastIndexOf('.')).toLowerCase();
323 }
324 if (levelN != null) {
325 if (!levels.containsKey(levelN))
326 levels.put(levelN, new Vector<File>());
327 levels.get(levelN).add(levelF);
328 }
329 }
330 }
331
332 applyPatches(levels, foldersPatches, listener, log);
333
334 combineBinaryFiles(levels, listener, log);
[653]335 combineBSLFolders(mods, listener, log);
[624]336
[653]337 copyVideos(log);
[625]338
[624]339 if (unlockLevels.size() > 0) {
[653]340 unlockLevels(unlockLevels, log);
[624]341 }
[653]342
[655]343 log.println();
[653]344 Date end = new Date();
345 log.println("Initialization ended at " + sdf.format(end));
346 log.println("Process took "
347 + ((end.getTime() - start.getTime()) / 1000) + " seconds");
348 log.close();
[608]349 }
[598]350
[648]351 private static void combineBSLFolders(TreeSet<Package> mods,
[653]352 InstallProgressListener listener, PrintWriter log) {
[608]353 listener.installProgressUpdate(95, 100, "Installing BSL files");
[653]354 log.println("Installing BSL files");
[598]355
[648]356 HashMap<EBSLInstallType, Vector<Package>> modsToInclude = new HashMap<EBSLInstallType, Vector<Package>>();
357 modsToInclude.put(EBSLInstallType.NORMAL, new Vector<Package>());
358 modsToInclude.put(EBSLInstallType.ADDON, new Vector<Package>());
[608]359
[648]360 for (Package m : mods.descendingSet()) {
[699]361 File bsl = CaseInsensitiveFile.getCaseInsensitiveFile(
362 m.getLocalPath(), "bsl");
[608]363 if (bsl.exists()) {
364 if (m.hasSeparatePlatformDirs()) {
[699]365 File bslCommon = CaseInsensitiveFile
366 .getCaseInsensitiveFile(bsl, "common");
367 File bslMac = CaseInsensitiveFile.getCaseInsensitiveFile(
368 bsl, "mac_only");
369 File bslWin = CaseInsensitiveFile.getCaseInsensitiveFile(
370 bsl, "win_only");
[608]371 if ((Settings.getPlatform() == Platform.MACOS && bslMac
372 .exists())
373 || ((Settings.getPlatform() == Platform.WIN || Settings
374 .getPlatform() == Platform.LINUX) && bslWin
375 .exists()) || bslCommon.exists()) {
376 modsToInclude.get(m.getBSLInstallType()).add(m);
377 }
378 } else {
379 modsToInclude.get(m.getBSLInstallType()).add(m);
380 }
381 }
382 }
383
[648]384 for (Package m : modsToInclude.get(EBSLInstallType.NORMAL)) {
[608]385 copyBSL(m, false);
386 }
[648]387 Vector<Package> addons = modsToInclude.get(EBSLInstallType.ADDON);
[610]388 for (int i = addons.size() - 1; i >= 0; i--) {
389 copyBSL(addons.get(i), true);
[608]390 }
[598]391 }
392
[648]393 private static void copyBSL(Package sourceMod, boolean addon) {
[608]394 File targetBaseFolder = new File(Paths.getEditionGDF(), "IGMD");
395 if (!targetBaseFolder.exists())
396 targetBaseFolder.mkdir();
397
398 Vector<File> sources = new Vector<File>();
[699]399 File bsl = CaseInsensitiveFile.getCaseInsensitiveFile(
400 sourceMod.getLocalPath(), "bsl");
[608]401 if (sourceMod.hasSeparatePlatformDirs()) {
[699]402 File bslCommon = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
403 "common");
404 File bslMac = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
405 "mac_only");
406 File bslWin = CaseInsensitiveFile.getCaseInsensitiveFile(bsl,
407 "win_only");
[608]408 if (Settings.getPlatform() == Platform.MACOS && bslMac.exists()) {
409 for (File f : bslMac.listFiles(dirFileFilter)) {
410 File targetBSL = new File(targetBaseFolder, f.getName());
411 if (addon || !targetBSL.exists())
412 sources.add(f);
413 }
414 }
415 if ((Settings.getPlatform() == Platform.WIN || Settings
416 .getPlatform() == Platform.LINUX) && bslWin.exists()) {
417 for (File f : bslWin.listFiles(dirFileFilter)) {
418 File targetBSL = new File(targetBaseFolder, f.getName());
419 if (addon || !targetBSL.exists())
420 sources.add(f);
421 }
422 }
423 if (bslCommon.exists()) {
424 for (File f : bslCommon.listFiles(dirFileFilter)) {
425 File targetBSL = new File(targetBaseFolder, f.getName());
426 if (addon || !targetBSL.exists())
427 sources.add(f);
428 }
429 }
430 } else {
431 for (File f : bsl.listFiles(dirFileFilter)) {
432 File targetBSL = new File(targetBaseFolder, f.getName());
433 if (addon || !targetBSL.exists())
434 sources.add(f);
435 }
436 }
437
438 System.out.println("For mod: " + sourceMod.getName()
439 + " install BSL folders: " + sources.toString());
440 for (File f : sources) {
441 File targetPath = new File(targetBaseFolder, f.getName());
442 if (!targetPath.exists())
443 targetPath.mkdir();
[699]444 if (!(CaseInsensitiveFile.getCaseInsensitiveFile(targetPath,
445 "ignore.txt").exists())) {
[663]446 for (File fbsl : f.listFiles()) {
[664]447 if (fbsl.getName().toLowerCase().endsWith(".bsl")) {
448 File targetFile = new File(targetPath, fbsl.getName());
449 if (addon || !targetFile.exists()) {
450 try {
451 FileUtils.copyFile(fbsl, targetFile);
452 } catch (IOException e) {
453 e.printStackTrace();
454 }
[663]455 }
[608]456 }
457 }
458 }
459 }
460 }
461
[698]462 private static void applyPatches(
463 TreeMap<String, Vector<File>> oniLevelFolders,
464 List<File> patchFolders, InstallProgressListener listener,
465 PrintWriter log) {
466 // TODO: REMOVE
467 long startMS = new Date().getTime();
468 // TODO: Implement this
469 String tmpFolderName = "installrun_temp-"
470 + new SimpleDateFormat("yyyy_MM_dd-HH_mm_ss")
471 .format(new Date());
472 File tmpFolder = new File(Paths.getTempPath(), tmpFolderName);
[598]473
[698]474 HashMap<String, Vector<File>> patches = new HashMap<String, Vector<File>>();
475 for (File patchFolder : patchFolders) {
476 for (File levelFolder : patchFolder.listFiles(dirFileFilter)) {
477 String lvlName = levelFolder.getName().toLowerCase();
478 for (File f : FileUtils.listFiles(levelFolder,
479 new String[] { "oni-patch" }, true)) {
480 if (!patches.containsKey(lvlName))
481 patches.put(lvlName, new Vector<File>());
482 patches.get(lvlName).add(f);
[598]483 }
[698]484 }
485 }
486
487 for (String level : patches.keySet()) {
488 File levelFolder = new File(tmpFolder, level);
489 levelFolder.mkdir();
490
491 // Get files to be patched from vanilla.dat / packages
492 for (File patch : patches.get(level)) {
493 String patternWildcard = patch.getName();
494 patternWildcard = patternWildcard.substring(0,
495 patternWildcard.indexOf(".oni-patch"));
496 patternWildcard = patternWildcard.replace('-', '*');
497 final Pattern patternRegex = Pattern.compile(
498 patternWildcard.replaceAll("\\*", ".\\*"),
499 Pattern.CASE_INSENSITIVE);
500
501 for (File srcFolder : oniLevelFolders.get(level)) {
502 if (srcFolder.isFile()) {
503 // Extract from .dat
504 OniSplit.export(levelFolder, srcFolder, patternWildcard);
505 } else {
506 // Copy from folder with overwrite
507 for (File f : FileUtils.listFiles(srcFolder,
508 new RegexFileFilter(patternRegex),
509 TrueFileFilter.TRUE)) {
510 try {
511 FileUtils.copyFileToDirectory(f, levelFolder);
512 } catch (IOException e) {
513 e.printStackTrace();
514 }
515 }
516 }
[604]517 }
[598]518 }
[698]519
520 // Extract files to XML
521 File levelFolderXML = new File(levelFolder, "xml");
522 Vector<File> files = new Vector<File>();
523 files.add(new File(levelFolder, "*.oni"));
524 OniSplit.convertOniToXML(levelFolderXML, files);
525
526 // Apply patches in levelFolderXML
527 for (File patch : patches.get(level)) {
528 String patternWildcard = patch.getName();
529 patternWildcard = patternWildcard.substring(0,
530 patternWildcard.indexOf(".oni-patch"));
531 patternWildcard = patternWildcard.replace('-', '*');
532 final Pattern patternRegex = Pattern.compile(
533 patternWildcard.replaceAll("\\*", ".\\*"),
534 Pattern.CASE_INSENSITIVE);
535
536 for (File toPatch : FileUtils.listFiles(levelFolderXML,
537 new RegexFileFilter(patternRegex), null)) {
538 // Apply patch "patch" to "toPatch"
539 XMLTools.patch(patch, toPatch);
540 }
541 }
542
543 // Create .oni files from XML
544 files.clear();
545 files.add(new File(levelFolderXML, "*.xml"));
546 OniSplit.convertXMLtoOni(levelFolder, files);
547
548 // Remove XML folder as import will only require .oni's
549 try {
550 FileUtils.deleteDirectory(levelFolderXML);
551 } catch (IOException e) {
552 e.printStackTrace();
553 }
[699]554
[698]555 oniLevelFolders.get(level).add(levelFolder);
[604]556 }
[598]557
[698]558 System.out.println("Took: " + (new Date().getTime() - startMS)
559 + " msec");
560 }
561
562 private static void combineBinaryFiles(
563 TreeMap<String, Vector<File>> oniLevelFolders,
564 InstallProgressListener listener, PrintWriter log) {
[604]565 int totalSteps = 0;
566 int stepsDone = 0;
[600]567
[604]568 for (@SuppressWarnings("unused")
[698]569 String s : oniLevelFolders.keySet())
[604]570 totalSteps++;
[608]571 totalSteps++;
[600]572
[604]573 log.println("Importing levels");
[698]574 for (String l : oniLevelFolders.keySet()) {
[604]575 log.println("\tLevel " + l);
[600]576 listener.installProgressUpdate(stepsDone, totalSteps,
[604]577 "Installing level " + l);
[698]578 for (File f : oniLevelFolders.get(l)) {
[604]579 log.println("\t\t\t" + f.getPath());
580 }
[600]581
[698]582 Vector<String> res = OniSplit.packLevel(oniLevelFolders.get(l),
583 new File(Paths.getEditionGDF(), sanitizeLevelName(l)
584 + ".dat"));
[604]585 if (res != null && res.size() > 0) {
586 for (String s : res)
587 log.println("\t\t" + s);
[598]588 }
[602]589
[604]590 log.println();
[608]591 stepsDone++;
[598]592 }
593 }
594
[653]595 private static void copyVideos(PrintWriter log) {
[631]596 if (Settings.getInstance().get("copyintro", false)) {
597 File src = new File(Paths.getVanillaGDF(), "intro.bik");
[653]598 log.println("Copying intro");
[631]599 if (src.exists()) {
600 try {
601 FileUtils.copyFileToDirectory(src, Paths.getEditionGDF());
602 } catch (IOException e) {
603 e.printStackTrace();
604 }
605 }
606 }
607 if (Settings.getInstance().get("copyoutro", true)) {
608 File src = new File(Paths.getVanillaGDF(), "outro.bik");
[653]609 log.println("Copying outro");
[631]610 if (src.exists()) {
611 try {
612 FileUtils.copyFileToDirectory(src, Paths.getEditionGDF());
613 } catch (IOException e) {
614 e.printStackTrace();
615 }
616 }
617 }
618 }
619
[653]620 private static void unlockLevels(TreeSet<Integer> unlockLevels,
621 PrintWriter log) {
[631]622 File dat = new File(Paths.getEditionBasePath(), "persist.dat");
[653]623 log.println("Unlocking levels: " + unlockLevels.toString());
[631]624 if (!dat.exists()) {
[653]625 InputStream is = AEInstaller2.class
626 .getResourceAsStream("/net/oni2/aeinstaller/resources/persist.dat");
[631]627 try {
628 FileUtils.copyInputStreamToFile(is, dat);
629 } catch (IOException e) {
630 e.printStackTrace();
631 }
632 }
633 PersistDat save = new PersistDat(dat);
634 HashSet<Integer> currentlyUnlocked = save.getUnlockedLevels();
635 currentlyUnlocked.addAll(unlockLevels);
636 save.setUnlockedLevels(currentlyUnlocked);
637 save.close();
638 }
639
[596]640 /**
641 * Initializes the Edition core
[600]642 *
643 * @param listener
644 * Listener for status updates
[596]645 */
[600]646 public static void initializeEdition(InstallProgressListener listener) {
[596]647 File init = new File(Paths.getTempPath(), "init");
[600]648
649 int totalSteps = 0;
650 int stepsDone = 0;
651
652 SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
653
654 for (@SuppressWarnings("unused")
655 File f : Paths.getVanillaGDF().listFiles(new FilenameFilter() {
656 @Override
657 public boolean accept(File dir, String name) {
658 return name.endsWith(".dat");
659 }
660 })) {
661 totalSteps++;
662 }
663 totalSteps = totalSteps * 2 + 2;
664
[596]665 try {
[600]666 File logFile = new File(Paths.getInstallerPath(),
667 "Initialization.log");
668 PrintWriter log = new PrintWriter(logFile);
669
670 Date start = new Date();
671 log.println("Initialization of Edition core started at "
672 + sdf.format(start));
673 log.println("Cleaning directories");
674
675 listener.installProgressUpdate(stepsDone, totalSteps,
676 "Cleaning up directories");
[596]677 createEmptyPath(Paths.getVanillaOnisPath());
678 createEmptyPath(init);
[597]679 File level0Folder = new File(init, "level0_Final");
680 createEmptyPath(level0Folder);
[596]681
[600]682 stepsDone++;
683
684 log.println("Exporting levels and moving files to level0");
685
[596]686 for (File f : Paths.getVanillaGDF().listFiles(new FilenameFilter() {
687 @Override
688 public boolean accept(File dir, String name) {
689 return name.endsWith(".dat");
690 }
691 })) {
692 String levelName = f.getName().substring(0,
693 f.getName().indexOf('.'));
694 Scanner fi = new Scanner(levelName);
[597]695 int levelNumber = Integer.parseInt(fi.findInLine("[0-9]+"));
[596]696
[600]697 log.println("\t" + levelName + ":");
698 log.println("\t\tExporting");
699 listener.installProgressUpdate(stepsDone, totalSteps,
700 "Exporting vanilla level " + levelNumber);
701
[597]702 // Edition/GameDataFolder/level*_Final/
703 File tempLevelFolder = new File(init, levelName);
704
705 // Export Vanilla-Level-Dat -> Temp/Level
[600]706 Vector<String> res = OniSplit.export(tempLevelFolder, f);
707 if (res != null && res.size() > 0) {
708 for (String s : res)
709 log.println("\t\t\t" + s);
710 }
[597]711
[600]712 log.println("\t\tMoving files");
[597]713 handleFileGlobalisation(tempLevelFolder, level0Folder,
[600]714 levelNumber);
715 stepsDone++;
716 log.println();
[596]717 }
718
[600]719 log.println("Reimporting levels");
720
[597]721 for (File f : init.listFiles()) {
722 String levelName = f.getName();
723
[600]724 log.println("\t" + levelName);
725 listener.installProgressUpdate(stepsDone, totalSteps,
726 "Creating globalized " + levelName);
[597]727
[598]728 Vector<File> folders = new Vector<File>();
729 folders.add(f);
730
[600]731 Vector<String> res = OniSplit.importLevel(folders, new File(
732 Paths.getVanillaOnisPath(), levelName + ".dat"));
733 if (res != null && res.size() > 0) {
734 for (String s : res)
735 log.println("\t\t" + s);
736 }
737
738 log.println();
739 stepsDone++;
[597]740 }
741
[600]742 listener.installProgressUpdate(stepsDone, totalSteps,
743 "Copying basic files");
[597]744 // Copy Oni-configs
745 File persistVanilla = new File(Paths.getOniBasePath(),
746 "persist.dat");
747 File persistEdition = new File(Paths.getEditionBasePath(),
748 "persist.dat");
749 File keyConfVanilla = new File(Paths.getOniBasePath(),
750 "key_config.txt");
751 File keyConfEdition = new File(Paths.getEditionBasePath(),
752 "key_config.txt");
753 if (persistVanilla.exists() && !persistEdition.exists())
754 FileUtils.copyFile(persistVanilla, persistEdition);
755 if (keyConfVanilla.exists() && !keyConfEdition.exists())
756 FileUtils.copyFile(keyConfVanilla, keyConfEdition);
757
[603]758 FileUtils.deleteDirectory(init);
[604]759
[600]760 Date end = new Date();
761 log.println("Initialization ended at " + sdf.format(end));
762 log.println("Process took "
763 + ((end.getTime() - start.getTime()) / 1000) + " seconds");
764 log.close();
[596]765 } catch (IOException e) {
766 e.printStackTrace();
767 }
768 }
[597]769
770 private static void moveFileToTargetOrDelete(File source, File target) {
771 if (source.equals(target))
772 return;
773 if (!target.exists()) {
774 if (!source.renameTo(target)) {
775 System.err.println("File " + source.getPath() + " not moved!");
776 }
777 } else if (!source.delete()) {
778 System.err.println("File " + source.getPath() + " not deleted!");
779 }
780 }
781
782 private static void handleFileGlobalisation(File tempFolder,
[600]783 File level0Folder, int levelNumber) {
[597]784 // Move AKEV and related files to subfolder so they're not globalized:
785 if (levelNumber != 0) {
786 File akevFolder = new File(tempFolder, "AKEV");
787 akevFolder.mkdir();
788 OniSplit.move(akevFolder, tempFolder.getPath() + "/AKEV*.oni",
789 "overwrite");
790 }
791
792 for (File f : tempFolder.listFiles(new FileFilter() {
793 @Override
794 public boolean accept(File pathname) {
795 return pathname.isFile();
796 }
797 })) {
798 // Move matching files to subfolder NoGlobal:
799 if (f.getName().startsWith("TXMPfail")
800 || f.getName().startsWith("TXMPlevel")
801 || (f.getName().startsWith("TXMP") && f.getName().contains(
802 "intro"))
803 || f.getName().startsWith("TXMB")
804 || f.getName().equals("M3GMpowerup_lsi.oni")
805 || f.getName().equals("TXMPlsi_icon.oni")
806 || (f.getName().startsWith("TXMB") && f.getName().contains(
807 "splash_screen.oni"))) {
808 File noGlobal = new File(tempFolder, "NoGlobal");
809 noGlobal.mkdir();
810 File noGlobalFile = new File(noGlobal, f.getName());
811 moveFileToTargetOrDelete(f, noGlobalFile);
812 }
813 // Move matching files to level0_Animations/level0_TRAC
814 else if (f.getName().startsWith("TRAC")) {
815 File level0File = new File(level0Folder, f.getName());
816 moveFileToTargetOrDelete(f, level0File);
817 }
818 // Move matching files to level0_Animations/level0_TRAM
819 else if (f.getName().startsWith("TRAM")) {
820 File level0File = new File(level0Folder, f.getName());
821 moveFileToTargetOrDelete(f, level0File);
822 }
823 // Move matching files to level0_Textures
824 else if (f.getName().startsWith("ONSK")
825 || f.getName().startsWith("TXMP")) {
826 File level0File = new File(level0Folder, f.getName());
[598]827 moveFileToTargetOrDelete(f, level0File);
[597]828 }
829 // Move matching files to *VANILLA*/level0_Characters
830 else if (f.getName().startsWith("ONCC")
831 || f.getName().startsWith("TRBS")
832 || f.getName().startsWith("ONCV")
833 || f.getName().startsWith("ONVL")
834 || f.getName().startsWith("TRMA")
835 || f.getName().startsWith("TRSC")
836 || f.getName().startsWith("TRAS")) {
[598]837 File level0File = new File(level0Folder, f.getName());
838 moveFileToTargetOrDelete(f, level0File);
[597]839 }
840 // Move matching files to level0_Sounds
841 else if (f.getName().startsWith("OSBD")
842 || f.getName().startsWith("SNDD")) {
843 File level0File = new File(level0Folder, f.getName());
844 moveFileToTargetOrDelete(f, level0File);
845 }
846 // Move matching files to level0_Particles
847 else if (f.getName().startsWith("BINA3")
848 || f.getName().startsWith("M3GMdebris")
849 || f.getName().equals("M3GMtoxic_bubble.oni")
850 || f.getName().startsWith("M3GMelec")
851 || f.getName().startsWith("M3GMrat")
852 || f.getName().startsWith("M3GMjet")
853 || f.getName().startsWith("M3GMbomb_")
854 || f.getName().equals("M3GMbarab_swave.oni")
855 || f.getName().equals("M3GMbloodyfoot.oni")) {
856 File level0File = new File(level0Folder, f.getName());
857 moveFileToTargetOrDelete(f, level0File);
858 }
859 // Move matching files to Archive (aka delete them)
860 else if (f.getName().startsWith("AGDB")
861 || f.getName().startsWith("TRCM")) {
862 f.delete();
863 }
[599]864 // Move matching files to /level0_Final/
[597]865 else if (f.getName().startsWith("ONWC")) {
[598]866 File level0File = new File(level0Folder, f.getName());
867 moveFileToTargetOrDelete(f, level0File);
[597]868 }
869 }
870 }
[603]871
[619]872 private static String sanitizeLevelName(String ln) {
873 int ind = ln.indexOf("_");
874 String res = ln.substring(0, ind + 1);
875 res += ln.substring(ind + 1, ind + 2).toUpperCase();
876 res += ln.substring(ind + 2);
877 return res;
878 }
879
[603]880 /**
[604]881 * Verify that the Edition is within a subfolder to vanilla Oni
882 * (..../Oni/Edition/AEInstaller)
883 *
[603]884 * @return true if GDF can be found in the parent's parent-path
885 */
886 public static boolean verifyRunningDirectory() {
[604]887 return Paths.getVanillaGDF().exists()
888 && Paths.getVanillaGDF().isDirectory();
[603]889 }
[596]890}
Note: See TracBrowser for help on using the repository browser.