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
Line 
1package net.oni2.aeinstaller.backend.oni;
2
3import java.io.File;
4import java.io.FileFilter;
5import java.io.FileInputStream;
6import java.io.FileNotFoundException;
7import java.io.FileOutputStream;
8import java.io.FilenameFilter;
9import java.io.IOException;
10import java.io.InputStream;
11import java.io.PrintWriter;
12import java.text.SimpleDateFormat;
13import java.util.Date;
14import java.util.HashMap;
15import java.util.HashSet;
16import java.util.List;
17import java.util.Scanner;
18import java.util.TreeMap;
19import java.util.TreeSet;
20import java.util.Vector;
21import java.util.regex.Pattern;
22
23import net.oni2.aeinstaller.AEInstaller2;
24import net.oni2.aeinstaller.backend.CaseInsensitiveFile;
25import net.oni2.aeinstaller.backend.Paths;
26import net.oni2.aeinstaller.backend.Settings;
27import net.oni2.aeinstaller.backend.Settings.Platform;
28import net.oni2.aeinstaller.backend.packages.EBSLInstallType;
29import net.oni2.aeinstaller.backend.packages.Package;
30import net.oni2.aeinstaller.backend.packages.PackageManager;
31
32import org.apache.commons.io.FileUtils;
33import org.apache.commons.io.filefilter.RegexFileFilter;
34import org.apache.commons.io.filefilter.TrueFileFilter;
35import org.javabuilders.swing.SwingJavaBuilder;
36
37import com.thoughtworks.xstream.XStream;
38import com.thoughtworks.xstream.io.xml.StaxDriver;
39
40/**
41 * @author Christian Illy
42 */
43public class Installer {
44 private static FileFilter dirFileFilter = new FileFilter() {
45 @Override
46 public boolean accept(File pathname) {
47 return pathname.isDirectory();
48 }
49 };
50
51 /**
52 * @return Is Edition Core initialized
53 */
54 public static boolean isEditionInitialized() {
55 return Paths.getVanillaOnisPath().exists();
56 }
57
58 private static void createEmptyPath(File path) throws IOException {
59 if (path.exists())
60 FileUtils.deleteDirectory(path);
61 path.mkdirs();
62 }
63
64 /**
65 * @return list of currently installed mods
66 */
67 public static Vector<Integer> getInstalledMods() {
68 File installCfg = new File(Paths.getEditionGDF(), "installed_mods.xml");
69 return PackageManager.getInstance().loadModSelection(installCfg);
70 }
71
72 /**
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 /**
113 * @param tools
114 * Tools to (un)install
115 * @param uninstall
116 * Uninstall tools or install?
117 */
118 public static void installTools(TreeSet<Package> tools, boolean uninstall) {
119 TreeSet<Integer> installed = getInstalledTools();
120 for (Package m : tools) {
121 if (!uninstall || installed.contains(m.getPackageNumber())) {
122 File plain = CaseInsensitiveFile.getCaseInsensitiveFile(
123 m.getLocalPath(), "plain");
124 if (plain.exists()) {
125 if (m.hasSeparatePlatformDirs()) {
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");
132 if (plainCommon.exists())
133 copyRemoveToolsFiles(plainCommon,
134 Paths.getEditionBasePath(), uninstall);
135 if (Settings.getPlatform() == Platform.MACOS
136 && plainMac.exists())
137 copyRemoveToolsFiles(plainMac,
138 Paths.getEditionBasePath(), uninstall);
139 else if (plainWin.exists())
140 copyRemoveToolsFiles(plainWin,
141 Paths.getEditionBasePath(), uninstall);
142 } else {
143 copyRemoveToolsFiles(plain, Paths.getEditionBasePath(),
144 uninstall);
145 }
146 }
147 }
148 if (uninstall)
149 installed.remove(m.getPackageNumber());
150 else
151 installed.add(m.getPackageNumber());
152 }
153 writeInstalledTools(installed);
154 }
155
156 private static void copyRemoveToolsFiles(File srcFolder, File targetFolder,
157 boolean remove) {
158 for (File f : srcFolder.listFiles()) {
159 try {
160 if (f.isDirectory())
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 }
177 } catch (IOException e) {
178 e.printStackTrace();
179 }
180 }
181 if (remove)
182 if (targetFolder.list().length == 0)
183 targetFolder.delete();
184 }
185
186 /**
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 */
194 public static void install(TreeSet<Package> mods,
195 InstallProgressListener listener) {
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();
208 log.println("AEI2 version: "
209 + SwingJavaBuilder.getConfig().getResource("appversion"));
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
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 })) {
243 File ignore = CaseInsensitiveFile.getCaseInsensitiveFile(f,
244 "ignore.txt");
245 if (!ignore.exists()) {
246 try {
247 FileUtils.deleteDirectory(f);
248 } catch (IOException e) {
249 e.printStackTrace();
250 }
251 }
252 }
253 }
254
255 File installCfg = new File(Paths.getEditionGDF(), "installed_mods.xml");
256 PackageManager.getInstance().saveModSelection(installCfg, mods);
257
258 TreeSet<Integer> unlockLevels = new TreeSet<Integer>();
259
260 Vector<File> foldersOni = new Vector<File>();
261 foldersOni.add(Paths.getVanillaOnisPath());
262
263 Vector<File> foldersPatches = new Vector<File>();
264
265 for (Package m : mods) {
266 for (int lev : m.getUnlockLevels())
267 unlockLevels.add(lev);
268
269 File oni = CaseInsensitiveFile.getCaseInsensitiveFile(
270 m.getLocalPath(), "oni");
271 if (oni.exists()) {
272 if (m.hasSeparatePlatformDirs()) {
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");
279 if (oniCommon.exists())
280 foldersOni.add(oniCommon);
281 if (Settings.getPlatform() == Platform.MACOS
282 && oniMac.exists())
283 foldersOni.add(oniMac);
284 else if (oniWin.exists())
285 foldersOni.add(oniWin);
286 } else {
287 foldersOni.add(oni);
288 }
289 }
290
291 File patches = CaseInsensitiveFile.getCaseInsensitiveFile(
292 m.getLocalPath(), "patches");
293 if (patches.exists()) {
294 if (m.hasSeparatePlatformDirs()) {
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");
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 }
312 }
313
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);
335 combineBSLFolders(mods, listener, log);
336
337 copyVideos(log);
338
339 if (unlockLevels.size() > 0) {
340 unlockLevels(unlockLevels, log);
341 }
342
343 log.println();
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();
349 }
350
351 private static void combineBSLFolders(TreeSet<Package> mods,
352 InstallProgressListener listener, PrintWriter log) {
353 listener.installProgressUpdate(95, 100, "Installing BSL files");
354 log.println("Installing BSL files");
355
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>());
359
360 for (Package m : mods.descendingSet()) {
361 File bsl = CaseInsensitiveFile.getCaseInsensitiveFile(
362 m.getLocalPath(), "bsl");
363 if (bsl.exists()) {
364 if (m.hasSeparatePlatformDirs()) {
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");
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
384 for (Package m : modsToInclude.get(EBSLInstallType.NORMAL)) {
385 copyBSL(m, false);
386 }
387 Vector<Package> addons = modsToInclude.get(EBSLInstallType.ADDON);
388 for (int i = addons.size() - 1; i >= 0; i--) {
389 copyBSL(addons.get(i), true);
390 }
391 }
392
393 private static void copyBSL(Package sourceMod, boolean addon) {
394 File targetBaseFolder = new File(Paths.getEditionGDF(), "IGMD");
395 if (!targetBaseFolder.exists())
396 targetBaseFolder.mkdir();
397
398 Vector<File> sources = new Vector<File>();
399 File bsl = CaseInsensitiveFile.getCaseInsensitiveFile(
400 sourceMod.getLocalPath(), "bsl");
401 if (sourceMod.hasSeparatePlatformDirs()) {
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");
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();
444 if (!(CaseInsensitiveFile.getCaseInsensitiveFile(targetPath,
445 "ignore.txt").exists())) {
446 for (File fbsl : f.listFiles()) {
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 }
455 }
456 }
457 }
458 }
459 }
460 }
461
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);
473
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);
483 }
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 }
517 }
518 }
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 }
554
555 oniLevelFolders.get(level).add(levelFolder);
556 }
557
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) {
565 int totalSteps = 0;
566 int stepsDone = 0;
567
568 for (@SuppressWarnings("unused")
569 String s : oniLevelFolders.keySet())
570 totalSteps++;
571 totalSteps++;
572
573 log.println("Importing levels");
574 for (String l : oniLevelFolders.keySet()) {
575 log.println("\tLevel " + l);
576 listener.installProgressUpdate(stepsDone, totalSteps,
577 "Installing level " + l);
578 for (File f : oniLevelFolders.get(l)) {
579 log.println("\t\t\t" + f.getPath());
580 }
581
582 Vector<String> res = OniSplit.packLevel(oniLevelFolders.get(l),
583 new File(Paths.getEditionGDF(), sanitizeLevelName(l)
584 + ".dat"));
585 if (res != null && res.size() > 0) {
586 for (String s : res)
587 log.println("\t\t" + s);
588 }
589
590 log.println();
591 stepsDone++;
592 }
593 }
594
595 private static void copyVideos(PrintWriter log) {
596 if (Settings.getInstance().get("copyintro", false)) {
597 File src = new File(Paths.getVanillaGDF(), "intro.bik");
598 log.println("Copying intro");
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");
609 log.println("Copying outro");
610 if (src.exists()) {
611 try {
612 FileUtils.copyFileToDirectory(src, Paths.getEditionGDF());
613 } catch (IOException e) {
614 e.printStackTrace();
615 }
616 }
617 }
618 }
619
620 private static void unlockLevels(TreeSet<Integer> unlockLevels,
621 PrintWriter log) {
622 File dat = new File(Paths.getEditionBasePath(), "persist.dat");
623 log.println("Unlocking levels: " + unlockLevels.toString());
624 if (!dat.exists()) {
625 InputStream is = AEInstaller2.class
626 .getResourceAsStream("/net/oni2/aeinstaller/resources/persist.dat");
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
640 /**
641 * Initializes the Edition core
642 *
643 * @param listener
644 * Listener for status updates
645 */
646 public static void initializeEdition(InstallProgressListener listener) {
647 File init = new File(Paths.getTempPath(), "init");
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
665 try {
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");
677 createEmptyPath(Paths.getVanillaOnisPath());
678 createEmptyPath(init);
679 File level0Folder = new File(init, "level0_Final");
680 createEmptyPath(level0Folder);
681
682 stepsDone++;
683
684 log.println("Exporting levels and moving files to level0");
685
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);
695 int levelNumber = Integer.parseInt(fi.findInLine("[0-9]+"));
696
697 log.println("\t" + levelName + ":");
698 log.println("\t\tExporting");
699 listener.installProgressUpdate(stepsDone, totalSteps,
700 "Exporting vanilla level " + levelNumber);
701
702 // Edition/GameDataFolder/level*_Final/
703 File tempLevelFolder = new File(init, levelName);
704
705 // Export Vanilla-Level-Dat -> Temp/Level
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 }
711
712 log.println("\t\tMoving files");
713 handleFileGlobalisation(tempLevelFolder, level0Folder,
714 levelNumber);
715 stepsDone++;
716 log.println();
717 }
718
719 log.println("Reimporting levels");
720
721 for (File f : init.listFiles()) {
722 String levelName = f.getName();
723
724 log.println("\t" + levelName);
725 listener.installProgressUpdate(stepsDone, totalSteps,
726 "Creating globalized " + levelName);
727
728 Vector<File> folders = new Vector<File>();
729 folders.add(f);
730
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++;
740 }
741
742 listener.installProgressUpdate(stepsDone, totalSteps,
743 "Copying basic files");
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
758 FileUtils.deleteDirectory(init);
759
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();
765 } catch (IOException e) {
766 e.printStackTrace();
767 }
768 }
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,
783 File level0Folder, int levelNumber) {
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());
827 moveFileToTargetOrDelete(f, level0File);
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")) {
837 File level0File = new File(level0Folder, f.getName());
838 moveFileToTargetOrDelete(f, level0File);
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 }
864 // Move matching files to /level0_Final/
865 else if (f.getName().startsWith("ONWC")) {
866 File level0File = new File(level0Folder, f.getName());
867 moveFileToTargetOrDelete(f, level0File);
868 }
869 }
870 }
871
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
880 /**
881 * Verify that the Edition is within a subfolder to vanilla Oni
882 * (..../Oni/Edition/AEInstaller)
883 *
884 * @return true if GDF can be found in the parent's parent-path
885 */
886 public static boolean verifyRunningDirectory() {
887 return Paths.getVanillaGDF().exists()
888 && Paths.getVanillaGDF().isDirectory();
889 }
890}
Note: See TracBrowser for help on using the repository browser.