From 8b069d3e6f878eb117ea3fd76a4a0c363bcf72aa Mon Sep 17 00:00:00 2001
From: coppro <coppro@0fc631ac-6414-0410-93d0-97cfa31319b6>
Date: Thu, 12 Jul 2007 22:48:20 +0000
Subject: [PATCH] Added installer code.

git-svn-id: svn://svn.code.sf.net/p/magicseteditor/code/trunk@561 0fc631ac-6414-0410-93d0-97cfa31319b6
---
 src/data/installer.cpp          | 100 ++++++++++++++++++++++++++++----
 src/data/installer.hpp          |   4 +-
 src/data/settings.cpp           |   2 +-
 src/gui/html_export_window.cpp  |   2 +-
 src/gui/thumbnail_thread.cpp    |   2 +-
 src/main.cpp                    |  20 ++++---
 src/util/io/package.cpp         |   8 +++
 src/util/io/package.hpp         |  11 +++-
 src/util/io/package_manager.cpp |  15 -----
 src/util/io/package_manager.hpp |   3 +
 src/util/platform.hpp           |   9 ---
 11 files changed, 127 insertions(+), 49 deletions(-)

diff --git a/src/data/installer.cpp b/src/data/installer.cpp
index 52bfda67..351210f7 100644
--- a/src/data/installer.cpp
+++ b/src/data/installer.cpp
@@ -7,11 +7,24 @@
 // ----------------------------------------------------------------------------- : Includes
 
 #include <data/installer.hpp>
+#include <data/locale.hpp>
+#include <data/game.hpp>
+#include <data/stylesheet.hpp>
+#include <data/symbol_font.hpp>
+#include <data/export_template.hpp>
 #include <util/io/package_manager.hpp>
+#include <util/platform.hpp>
 #include <script/to_value.hpp>
 #include <wx/filename.h>
+#include <wx/wfstream.h>
+#include <wx/zipstrm.h>
+#include <wx/stdpaths.h>
 
 DECLARE_TYPEOF_COLLECTION(String);
+DECLARE_TYPEOF_COLLECTION(PackageP);
+DECLARE_TYPEOF_COLLECTION(PackageDependencyP);
+DECLARE_POINTER_TYPE(wxFileInputStream);
+DECLARE_POINTER_TYPE(wxZipInputStream);
 
 // ----------------------------------------------------------------------------- : Installer
 
@@ -24,31 +37,96 @@ IMPLEMENT_REFLECTION(Installer) {
 
 // ----------------------------------------------------------------------------- : Installing
 
-void Installer::installFrom(const String& filename, bool message_on_success) {
+void Installer::installFrom(const String& filename, bool message_on_success, bool local) {
 	Installer i;
 	i.open(filename);
-	i.install();
+	try {
+		i.install(local);
+	}
+	catch (Error& e) {
+		handle_error(e);
+		return;
+	}
 	if (message_on_success) {
 		wxMessageBox(String::Format(_("'%s' successfully installed %d package%s."), i.name().c_str(), i.packages.size(), i.packages.size() == 1 ? _("") : _("s")),
 		             _("Magic Set Editor"), wxOK | wxICON_INFORMATION);
 	}
 }
 
-void Installer::install() {
-	// Walk over all files in this installer
+struct dependency_check : public unary_function<bool, PackagedP> {
+	dependency_check(PackageDependencyP dep) : dep (dep) {}
+	bool operator () (PackagedP package) {
+		return package->name() == dep->package && package->version >= dep->version;
+	}
+  private:
+	PackageDependencyP dep;
+};
+
+void Installer::install(bool local) {
+	// Destination directory
+	String install_dir = local ? ::packages.getLocalDataDir() : ::packages.getGlobalDataDir();
+	if (!wxDirExists(install_dir))
+		wxMkdir(install_dir, 0755);
+	
+	// All the packages we're installing.
+	vector<PackagedP> new_packages;
+
+	FOR_EACH(p, packages) {
+		if (wxDirExists(install_dir + _("/") + p) || wxFileExists(install_dir + _("/") + p))
+			throw PackageError(_("Package ") + p + _(" is already installed. Overwriting currently not supported."));
+		PackagedP pack;
+		wxString fn(wxFileName(p).GetExt());
+		if      (fn == _("mse-game"))            pack = new_intrusive<Game>();
+		else if (fn == _("mse-style"))           pack = new_intrusive<StyleSheet>();
+		else if (fn == _("mse-locale"))          pack = new_intrusive<Locale>();
+		else if (fn == _("mse-include"))         pack = new_intrusive<IncludePackage>();
+		else if (fn == _("mse-symbol-font"))     pack = new_intrusive<SymbolFont>();
+		else if (fn == _("mse-export-template")) pack = new_intrusive<ExportTemplate>();
+		else {
+			throw PackageError(_("Unrecognized package type: '") + fn + _("'\nwhile trying to install: ") + p);
+		}
+		Reader reader(openIn(p + _("/") + fn.Mid(4)));
+		pack->Packaged::reflect_impl(reader);
+		new_packages.push_back(pack);
+	}
+	
+
+	// Check dependencies for each and every package.
+	FOR_EACH(p, new_packages) {
+		FOR_EACH(d, p->dependencies) {
+			if (find_if(new_packages.begin(), new_packages.end(), dependency_check(d)) == new_packages.end() &&
+				!::packages.checkDependency(*d, false)) {
+				throw PackageError(_("Unmet dependency for package ") + p->relativeFilename() + _(": ") + d->package + _(", version ") + d->version.toString() + _(" or higher."));
+			}
+		}
+	}
+
 	const FileInfos& file_infos = getFileInfos();
 	for (FileInfos::const_iterator it = file_infos.begin() ; it != file_infos.end() ; ++it) {
-//		const String& filename = it->first;
-		//
-	}
-	// REMOVE?
-	FOR_EACH(p, packages) {
-		install(p);
+		String file = it->first;
+
+		wxFileName fn(file);
+		wxArrayString dirs = fn.GetDirs();
+
+		if (fn.IsDir() || !dirs.GetCount() || find(packages.begin(), packages.end(), dirs[0]) == packages.end())
+			continue;
+
+		String current_dir = install_dir;
+		for (size_t j = 0; j < dirs.GetCount(); ++j) {
+			current_dir += _("/") + dirs[j];
+			if (!wxDirExists(current_dir) && !wxMkdir(current_dir, 0755))
+				throw PackageError(_("Cannot create folder ") + current_dir + _(" for install. Warning: some packages may have been installed anyway, and some may only be partially installed."));
+		}
+
+		InputStreamP is = openIn(file);
+		wxFileOutputStream os (install_dir + _("/") + file);
+		if (!os.IsOk())
+			throw PackageError(_("Cannot create file ") + install_dir + _("/") + file + _(" for install. Warning: some packages may have been installed anyway, and some may only be partially installed."));
+		os.Write(*is);
 	}
 }
 
 void Installer::install(const String& package) {
-	// 1. Make sure the package is not loaded
 	// TODO
 }
 
diff --git a/src/data/installer.hpp b/src/data/installer.hpp
index 7374e987..d65f8732 100644
--- a/src/data/installer.hpp
+++ b/src/data/installer.hpp
@@ -21,9 +21,9 @@ class Installer : public Packaged {
 	vector<String> packages;	///< Packages to install
 	
 	/// Load an installer from a file, and run it
-	static void installFrom(const String& filename, bool message_on_success);
+	static void installFrom(const String& filename, bool message_on_success, bool local);
 	/// Install all the packages
-	void install();
+	void install(bool local);
 	/// Install a specific package
 	void install(const String& package);
 	
diff --git a/src/data/settings.cpp b/src/data/settings.cpp
index 5fa91868..55476bec 100644
--- a/src/data/settings.cpp
+++ b/src/data/settings.cpp
@@ -156,7 +156,7 @@ IndexMap<FieldP,ValueP>& Settings::exportOptionsFor(const ExportTemplate& export
 /// Retrieve the directory to use for settings and other data files
 String user_settings_dir() {
 	String dir = wxStandardPaths::Get().GetUserDataDir();
-	if (!wxDirExists(dir)) wxMkDir(dir);
+	if (!wxDirExists(dir)) wxMkdir(dir);
 	return dir + _("/");
 }
 
diff --git a/src/gui/html_export_window.cpp b/src/gui/html_export_window.cpp
index 288caba9..6d69aa81 100644
--- a/src/gui/html_export_window.cpp
+++ b/src/gui/html_export_window.cpp
@@ -64,7 +64,7 @@ void HtmlExportWindow::onOk(wxCommandEvent&) {
 		info.directory_relative = fn.GetName() + _("-files");
 		fn.SetFullName(info.directory_relative);
 		info.directory_absolute = fn.GetFullPath();
-		wxMkDir(info.directory_absolute);
+		wxMkdir(info.directory_absolute);
 	}
 	// run export script
 	Context& ctx = set->getContext();
diff --git a/src/gui/thumbnail_thread.cpp b/src/gui/thumbnail_thread.cpp
index eed080c5..754a59fd 100644
--- a/src/gui/thumbnail_thread.cpp
+++ b/src/gui/thumbnail_thread.cpp
@@ -19,7 +19,7 @@ DECLARE_TYPEOF_COLLECTION(pair_ThumbnailRequestP_Image);
 String user_settings_dir();
 String image_cache_dir() {
 	String dir = user_settings_dir() + _("/cache");
-	if (!wxDirExists(dir)) wxMkDir(dir);
+	if (!wxDirExists(dir)) wxMkdir(dir);
 	return dir + _("/");
 }
 
diff --git a/src/main.cpp b/src/main.cpp
index 70f71e9d..158f1958 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -94,9 +94,13 @@ bool MSE::OnInit() {
 					return true;
 				} else if (f.GetExt() == _("mse-installer")) {
 					// Installer; install it
-					Installer::installFrom(argv[1], true);
-					packages.destroy();
-					return false;
+					bool local = false;
+					if (argc > 2) {
+						String arg2 = argv[2];
+						local = arg2 == _("--local");
+					}
+					Installer::installFrom(argv[1], true, local);
+					return true;
 				} else if (arg == _("--symbol-editor")) {
 					Window* wnd = new SymbolWindow(nullptr);
 					wnd->Show();
@@ -112,8 +116,7 @@ bool MSE::OnInit() {
 					} else {
 						inst.saveAs(inst.prefered_filename, false);
 					}
-					packages.destroy();
-					return false;
+					return true;
 				} else if (arg == _("--help") || arg == _("-?")) {
 					// command line help
 					write_stdout( String(_("Magic Set Editor\n\n"))
@@ -128,17 +131,18 @@ bool MSE::OnInit() {
 					            + _("  -v --version      \tShow version information.\n")
 					            + _("  --create-installer\n")
 					            + _("      FILE [FILE]...\tCreate an instaler named FILE, containing the listed packges.\n") );
-					return false;
+					return true;
 				} else if (arg == _("--version") || arg == _("-v")) {
 					// dump version
 					write_stdout( _("Magic Set Editor\nVersion ") + app_version.toString() + version_suffix );
-					packages.destroy();
-					return false;
+					return true;
 				} else {
 					handle_error(_("Invalid command line argument:\n") + String(argv[1]));
 				}
 			} catch (const Error& e) {
 				handle_error(e);
+				OnExit();
+				return false;
 			}
 		}
 		
diff --git a/src/util/io/package.cpp b/src/util/io/package.cpp
index e501a76f..3c971f08 100644
--- a/src/util/io/package.cpp
+++ b/src/util/io/package.cpp
@@ -497,3 +497,11 @@ void Packaged::validate(Version) {
 		packages.checkDependency(*dep, true);
 	}
 }
+
+// ----------------------------------------------------------------------------- : IncludePackage
+
+String IncludePackage::typeName() const { return _("include"); }
+
+IMPLEMENT_REFLECTION(IncludePackage) {
+	REFLECT_BASE(Packaged);
+}
\ No newline at end of file
diff --git a/src/util/io/package.hpp b/src/util/io/package.hpp
index 28ca8d59..949f9dc4 100644
--- a/src/util/io/package.hpp
+++ b/src/util/io/package.hpp
@@ -171,8 +171,8 @@ class Package : public IntrusivePtrVirtualBase {
 	void openDirectory();
 	void openSubdir(const String&);
 	void openZipfile();
-	void saveToDirectory(const String&, bool);
 	void saveToZipfile(const String&, bool);
+	void saveToDirectory(const String&, bool);
 	FileInfos::iterator addFile(const String& file);
 	static String toStandardName(const String& file);
 };
@@ -228,5 +228,14 @@ class Packaged : public Package {
 	friend struct JustAsPackageProxy;
 };
 
+// ----------------------------------------------------------------------------- : IncludePackage
+
+/// A package that just contains a bunch of files that are used from other packages
+class IncludePackage : public Packaged {
+  protected:
+	String typeName() const;
+	DECLARE_REFLECTION();
+};
+
 // ----------------------------------------------------------------------------- : EOF
 #endif
diff --git a/src/util/io/package_manager.cpp b/src/util/io/package_manager.cpp
index 62895ed4..4e625350 100644
--- a/src/util/io/package_manager.cpp
+++ b/src/util/io/package_manager.cpp
@@ -15,21 +15,6 @@
 #include <data/export_template.hpp>
 #include <wx/stdpaths.h>
 
-// ----------------------------------------------------------------------------- : IncludePackage
-
-/// A package that just contains a bunch of files that are used from other packages
-class IncludePackage : public Packaged {
-  protected:
-	String typeName() const;
-	DECLARE_REFLECTION();
-};
-
-String IncludePackage::typeName() const { return _("include"); }
-
-IMPLEMENT_REFLECTION(IncludePackage) {
-	REFLECT_BASE(Packaged);
-}
-
 // ----------------------------------------------------------------------------- : PackageManager
 
 PackageManager packages;
diff --git a/src/util/io/package_manager.hpp b/src/util/io/package_manager.hpp
index 641b3790..7ce7521f 100644
--- a/src/util/io/package_manager.hpp
+++ b/src/util/io/package_manager.hpp
@@ -70,6 +70,9 @@ class PackageManager {
 	
 	/// Check if the given dependency is currently installed
 	bool checkDependency(const PackageDependency& dep, bool report_errors = false);
+
+	inline String getGlobalDataDir() const { return global_data_directory; }
+	inline String getLocalDataDir() const { return local_data_directory; }
 	
   private:
 	map<String, PackagedP> loaded_packages;
diff --git a/src/util/platform.hpp b/src/util/platform.hpp
index 7f52ce67..9e8030db 100644
--- a/src/util/platform.hpp
+++ b/src/util/platform.hpp
@@ -21,15 +21,6 @@
 
 // ----------------------------------------------------------------------------- : Linux
 
-#ifdef __linux__
-	
-	/// wxMkDir as documented
-	inline void wxMkDir(const String& dir) {
-		wxMkDir(wxConvLocal.cWX2MB(dir), 0777);
-	}
-	
-#endif
-
 // ----------------------------------------------------------------------------- : GCC
 
 #ifdef __GNUC__