Differential D1142 Diff 6395 ps/trunk/source/ps/ModInstaller.cpp

Changeset View

Standalone View

ps/trunk/source/ps/ModInstaller.cpp

Property	Old Value	New Value
svn:eol-style	null	native \ No newline at end of property

				/* Copyright (C) 2018 Wildfire Games.
				* This file is part of 0 A.D.
				*
				* 0 A.D. is free software: you can redistribute it and/or modify
				* it under the terms of the GNU General Public License as published by
				* the Free Software Foundation, either version 2 of the License, or
				* (at your option) any later version.
				*
				* 0 A.D. is distributed in the hope that it will be useful,
				* but WITHOUT ANY WARRANTY; without even the implied warranty of
				* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
				* GNU General Public License for more details.
				*
				* You should have received a copy of the GNU General Public License
				* along with 0 A.D. If not, see <http://www.gnu.org/licenses/>.
				*/

				#include "precompiled.h"

				#include "ModInstaller.h"

				#include "lib/file/vfs/vfs_util.h"
				#include "ps/Filesystem.h"
				#include "ps/XML/Xeromyces.h"

				#include <fstream>

				CModInstaller::CModInstaller(const OsPath& modsdir, const OsPath& tempdir) :
				m_ModsDir(modsdir), m_TempDir(tempdir / "_modscache"), m_CacheDir("cache/")
				{
				m_VFS = CreateVfs();
				CreateDirectories(m_TempDir, 0700);
				}

				CModInstaller::~CModInstaller()
				{
				m_VFS.reset();
				DeleteDirectory(m_TempDir);
				}

				CModInstaller::ModInstallationResult CModInstaller::Install(
				const OsPath& mod,
				const std::shared_ptr<ScriptRuntime>& scriptRuntime,
				bool deleteAfterInstall)
				{
				const OsPath modTemp = m_TempDir / mod.Basename() / mod.Filename().ChangeExtension(L".zip");
				CreateDirectories(modTemp.Parent(), 0700);

				CopyFile(mod, modTemp, true);

				// Load the mod to VFS
				if (m_VFS->Mount(m_CacheDir, m_TempDir / "") != INFO::OK)
				return FAIL_ON_VFS_MOUNT;
				CVFSFile modinfo;
				PSRETURN modinfo_status = modinfo.Load(m_VFS, m_CacheDir / modTemp.Basename() / "mod.json", false);
				m_VFS->Clear();
				if (modinfo_status != PSRETURN_OK)
				return FAIL_ON_MOD_LOAD;

				// Extract the name of the mod
				ScriptInterface scriptInterface("Engine", "ModInstaller", scriptRuntime);
				JSContext* cx = scriptInterface.GetContext();
				JS::RootedValue json_val(cx);
				if (!scriptInterface.ParseJSON(modinfo.GetAsString(), &json_val))
				return FAIL_ON_PARSE_JSON;
				JS::RootedObject json_obj(cx, json_val.toObjectOrNull());
				JS::RootedValue name_val(cx);
				if (!JS_GetProperty(cx, json_obj, "name", &name_val))
				return FAIL_ON_EXTRACT_NAME;
				CStr modName;
				ScriptInterface::FromJSVal(cx, name_val, modName);
				if (modName.empty())
				return FAIL_ON_EXTRACT_NAME;

				const OsPath modDir = m_ModsDir / modName;
				const OsPath modPath = modDir / (modName + ".zip");

				// Create a directory with the following structure:
				// mod-name/
				// mod-name.zip
				CreateDirectories(modDir, 0700);
				if (wrename(modTemp, modPath) != 0)
				return FAIL_ON_MOD_MOVE;
				ffffffffUnsubmitted Not Done Inline Actions fail here ffffffff: fail here
				DeleteDirectory(modTemp.Parent());

				#ifdef OS_WIN
				// On Windows, write the contents of mod.json to a separate file next to the archive:
				// mod-name/
				// mod-name.zip
				// mod.json
				std::ofstream mod_json((modDir / "mod.json").string8());
				if (mod_json.good())
				{
				mod_json << modinfo.GetAsString();
				mod_json.close();
				}
				#endif // OS_WIN

				// Remove the original file if requested
				if (deleteAfterInstall)
				wunlink(mod);

				m_InstalledMods.emplace_back(modName);

				return SUCCESS;
				}

				const std::vector<CStr>& CModInstaller::GetInstalledMods() const
				{
				return m_InstalledMods;
				}