Open source Star Ruler 2 source code!
This commit is contained in:
Lucas de Vries
@@ -0,0 +1,531 @@
|
||||
#include "compat/misc.h"
|
||||
#include "files.h"
|
||||
#include "mods/mod_manager.h"
|
||||
#include "main/logging.h"
|
||||
#include "str_util.h"
|
||||
#include "main/references.h"
|
||||
#include <string>
|
||||
|
||||
namespace mods {
|
||||
|
||||
static const unsigned CUR_COMPATIBILITY = 200;
|
||||
|
||||
Mod::Mod() : parent(0), isBase(false), version(1), listed(true), enabled(false), isNew(false), forced(false), compatibility(0), forCurrentVersion(false) {
|
||||
}
|
||||
|
||||
bool Mod::resolve(std::string& path) const {
|
||||
const Mod* mod = this;
|
||||
while(mod) {
|
||||
std::string modPath = path_join(mod->abspath, path);
|
||||
if(fileExists(modPath)) {
|
||||
path = modPath;
|
||||
return true;
|
||||
}
|
||||
mod = mod->parent;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
void Mod::resolve(const std::string& path, std::vector<std::string>& check) const {
|
||||
const Mod* mod = this;
|
||||
while(mod) {
|
||||
std::string modPath = path_join(mod->abspath, path);
|
||||
if(fileExists(modPath))
|
||||
check.push_back(modPath);
|
||||
mod = mod->parent;
|
||||
}
|
||||
}
|
||||
|
||||
bool Mod::hasOverride(const std::string& path, bool isDirectory) const {
|
||||
foreach(it, override_patterns) {
|
||||
if(match(path.c_str(), *it))
|
||||
return true;
|
||||
}
|
||||
|
||||
if(isDirectory)
|
||||
return hasOverride(path+"/");
|
||||
return false;
|
||||
}
|
||||
|
||||
void Mod::listFiles(const std::string& dir, std::map<std::string, std::string>& out,
|
||||
const char* filter, bool recurse, bool inherit) const {
|
||||
const Mod* mod = this;
|
||||
|
||||
CompiledPattern filterReqs;
|
||||
compile_pattern(filter, filterReqs);
|
||||
|
||||
std::function<void(const std::string&, const std::string&, const std::string&)> readDir
|
||||
= [&](const std::string& relpath, const std::string& abspath, const std::string& dirpath) {
|
||||
std::vector<std::string> filenames;
|
||||
listDirectory(abspath, filenames);
|
||||
|
||||
foreach(it, filenames) {
|
||||
std::string absfile = path_join(abspath, *it);
|
||||
std::string relfile = path_join(relpath, *it);
|
||||
std::string dirfile = path_join(dirpath, relfile);
|
||||
bool isDir = isDirectory(absfile);
|
||||
|
||||
//if(!dirpath.empty() && hasOverride(dirfile, isDir))
|
||||
// continue;
|
||||
|
||||
//Recurse into directories
|
||||
if(recurse && isDir) {
|
||||
readDir(relfile, absfile, dirpath);
|
||||
continue;
|
||||
}
|
||||
|
||||
//Check if it matches the pattern
|
||||
if(!match(it->c_str(), filterReqs))
|
||||
continue;
|
||||
|
||||
auto fnd = out.find(relfile);
|
||||
if(fnd == out.end())
|
||||
out[relfile] = absfile;
|
||||
}
|
||||
};
|
||||
|
||||
while(mod) {
|
||||
std::string abspath = path_join(mod->abspath, dir);
|
||||
if(isDirectory(abspath))
|
||||
readDir("", abspath, mod == this ? "" : dir);
|
||||
|
||||
if(mod == this && (!inherit || hasOverride(dir, true)))
|
||||
break;
|
||||
mod = mod->parent;
|
||||
}
|
||||
}
|
||||
|
||||
void Mod::resolveDirectory(const std::string& dir, std::vector<std::string>& out) const {
|
||||
const Mod* mod = this;
|
||||
while(mod) {
|
||||
std::string path = path_join(mod->abspath, dir);
|
||||
if(isDirectory(path))
|
||||
out.push_back(path);
|
||||
mod = mod->parent;
|
||||
}
|
||||
}
|
||||
|
||||
const Mod* Mod::getFallback(unsigned short v) const {
|
||||
if(v > version)
|
||||
return nullptr;
|
||||
if(v == version)
|
||||
return this;
|
||||
//Look for a version in [v,version)
|
||||
for(unsigned short i = v; i < version; ++i) {
|
||||
auto fb = fallbacks.find(i);
|
||||
if(fb != fallbacks.end())
|
||||
return devices.mods.getMod(fb->second);
|
||||
}
|
||||
return this;
|
||||
}
|
||||
|
||||
void Mod::loadFileList() {
|
||||
containedFiles.clear();
|
||||
|
||||
std::function<void(const std::string&, const std::string&)> readDir
|
||||
= [&](const std::string& relpath, const std::string& abspath) {
|
||||
std::vector<std::string> filenames;
|
||||
listDirectory(abspath, filenames);
|
||||
|
||||
foreach(it, filenames) {
|
||||
std::string absfile = path_join(abspath, *it);
|
||||
std::string relfile = path_join(relpath, *it);
|
||||
bool isDir = isDirectory(absfile);
|
||||
|
||||
if(isDir) {
|
||||
readDir(relfile, absfile);
|
||||
}
|
||||
else if(!relpath.empty()){
|
||||
containedFiles.insert(relfile);
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
readDir("", abspath);
|
||||
}
|
||||
|
||||
void Mod::getConflicts(const Mod* other, std::vector<std::string>& conflicts) const {
|
||||
foreach(it, containedFiles) {
|
||||
if(other->containedFiles.find(*it) != other->containedFiles.end())
|
||||
conflicts.push_back(*it);
|
||||
}
|
||||
}
|
||||
|
||||
bool Mod::isCompatible(const Mod* other) const {
|
||||
if(isBase || other->isBase)
|
||||
return !isBase || !other->isBase;
|
||||
foreach(it, containedFiles) {
|
||||
if(other->containedFiles.find(*it) != other->containedFiles.end())
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
std::string Mod::getProfile() const {
|
||||
std::string dir = getProfileRoot();
|
||||
|
||||
//Make sure the root profile dir is created
|
||||
if(!isDirectory(dir))
|
||||
makeDirectory(dir);
|
||||
|
||||
//Create any intermediate directories
|
||||
std::vector<std::string> dirs;
|
||||
|
||||
if(dirname.empty())
|
||||
dirs.push_back("base");
|
||||
else
|
||||
dirs.push_back(dirname);
|
||||
|
||||
foreach(it, dirs) {
|
||||
dir = path_join(dir, *it);
|
||||
|
||||
if(!isDirectory(dir))
|
||||
makeDirectory(dir);
|
||||
}
|
||||
|
||||
return dir;
|
||||
}
|
||||
|
||||
std::string Mod::getProfile(const std::string& path) const {
|
||||
std::string dir = getProfileRoot();
|
||||
|
||||
//Make sure the root profile dir is created
|
||||
if(!isDirectory(dir))
|
||||
makeDirectory(dir);
|
||||
|
||||
//Create any intermediate directories
|
||||
std::vector<std::string> dirs;
|
||||
|
||||
if(dirname.empty())
|
||||
dirs.push_back("base");
|
||||
else
|
||||
dirs.push_back(dirname);
|
||||
|
||||
if(!path.empty())
|
||||
path_split(path, dirs);
|
||||
|
||||
foreach(it, dirs) {
|
||||
dir = path_join(dir, *it);
|
||||
|
||||
if(!isDirectory(dir))
|
||||
makeDirectory(dir);
|
||||
}
|
||||
|
||||
return dir;
|
||||
}
|
||||
|
||||
Manager::Manager() {
|
||||
baseMod = new Mod();
|
||||
baseMod->ident = "base";
|
||||
baseMod->name = "base";
|
||||
baseMod->abspath = getAbsolutePath(".");
|
||||
baseMod->isBase = true;
|
||||
baseMod->listed = false;
|
||||
baseMod->version = 3;
|
||||
baseMod->fallbacks[0] = "r3444";
|
||||
baseMod->fallbacks[1] = "r4257";
|
||||
baseMod->fallbacks[2] = "r4676";
|
||||
baseMod->compatibility = 200;
|
||||
baseMod->forCurrentVersion = true;
|
||||
|
||||
modNames["base"] = baseMod;
|
||||
mods.push_back(baseMod);
|
||||
}
|
||||
|
||||
Manager::~Manager() {
|
||||
foreach(it, mods)
|
||||
delete *it;
|
||||
}
|
||||
|
||||
void Manager::registerMod(const std::string& folder, const std::string& ident) {
|
||||
std::string infopath = path_join(folder, "modinfo.txt");
|
||||
if(!fileExists(infopath))
|
||||
return;
|
||||
|
||||
Mod& mod = *new Mod();
|
||||
mod.ident = ident;
|
||||
mod.dirname = ident;
|
||||
mod.abspath = getAbsolutePath(folder);
|
||||
|
||||
std::string key, value;
|
||||
DataReader datafile(infopath);
|
||||
while(datafile++) {
|
||||
key = datafile.key;
|
||||
value = datafile.value;
|
||||
if(key == "Name") {
|
||||
mod.name = value;
|
||||
}
|
||||
else if(key == "Description") {
|
||||
mod.description = value;
|
||||
}
|
||||
else if(key == "Override") {
|
||||
mod.overrides.push_back(value);
|
||||
|
||||
CompiledPattern compiled;
|
||||
compile_pattern(value.c_str(), compiled);
|
||||
|
||||
mod.override_patterns.push_back(compiled);
|
||||
}
|
||||
else if(key == "Derives From") {
|
||||
if(value == "-")
|
||||
mod.parentname = "";
|
||||
else
|
||||
mod.parentname = value;
|
||||
}
|
||||
else if(key == "Base Mod") {
|
||||
mod.isBase = toBool(value);
|
||||
}
|
||||
else if(key == "Listed") {
|
||||
mod.listed = toBool(value);
|
||||
}
|
||||
else if(key == "Version") {
|
||||
mod.version = toNumber<unsigned short>(value);
|
||||
}
|
||||
else if(key == "Compatibility") {
|
||||
mod.compatibility = toNumber<unsigned>(value);
|
||||
mod.forCurrentVersion = mod.compatibility >= CUR_COMPATIBILITY;
|
||||
}
|
||||
else if(key == "Fallback") {
|
||||
std::string m, v;
|
||||
splitKeyValue(value, m, v, "=");
|
||||
mod.fallbacks[toNumber<unsigned short>(v)] = m;
|
||||
}
|
||||
}
|
||||
|
||||
if(!mod.name.empty()) {
|
||||
mods.push_back(&mod);
|
||||
if(mod.parentname.empty() && !mod.overrides.empty())
|
||||
warn("WARNING: Mod '%s' has overrides declared, but does not derive from any mod.", mod.name.c_str());
|
||||
while(modNames.find(mod.name) != modNames.end()) {
|
||||
error("Duplicate mod named '%s'", mod.name.c_str());
|
||||
mod.name += " (Copy)";
|
||||
}
|
||||
while(modNames.find(mod.ident) != modNames.end()) {
|
||||
error("Duplicate mod ident '%s'", mod.ident.c_str());
|
||||
mod.ident += " (Copy)";
|
||||
}
|
||||
modNames[mod.name] = &mod;
|
||||
modNames[mod.ident] = &mod;
|
||||
}
|
||||
else {
|
||||
delete &mod;
|
||||
}
|
||||
}
|
||||
|
||||
void Manager::registerDirectory(const std::string& dirname) {
|
||||
std::vector<std::string> files;
|
||||
listDirectory(dirname, files);
|
||||
|
||||
foreach(it, files) {
|
||||
std::string abspath = path_join(dirname, *it);
|
||||
std::string infopath = path_join(abspath, "modinfo.txt");
|
||||
|
||||
if(isDirectory(abspath) && fileExists(infopath))
|
||||
registerMod(abspath, *it);
|
||||
}
|
||||
}
|
||||
|
||||
void Manager::finalize() {
|
||||
foreach(it, mods) {
|
||||
auto* mod = *it;
|
||||
if(!mod->parentname.empty()) {
|
||||
auto fnd = modNames.find(mod->parentname);
|
||||
if(fnd != modNames.end()) {
|
||||
mod->parent = fnd->second;
|
||||
if(!mod->isBase && fnd->second->isBase)
|
||||
mod->isBase = true;
|
||||
}
|
||||
}
|
||||
if(mod->listed)
|
||||
mod->isNew = true;
|
||||
}
|
||||
|
||||
//Load enabled and disabled mod files
|
||||
std::string filename = path_join(getProfileRoot(), "mods.txt");
|
||||
DataReader datafile(filename);
|
||||
while(datafile++) {
|
||||
auto it = modNames.find(datafile.value);
|
||||
if(it != modNames.end()) {
|
||||
it->second->enabled = datafile.key == "Enabled" || datafile.key == "Forced";
|
||||
it->second->forced = datafile.key == "Forced";
|
||||
it->second->isNew = false;
|
||||
|
||||
if(it->second->enabled && it->second->compatibility < CUR_COMPATIBILITY) {
|
||||
if(!it->second->forced) {
|
||||
it->second->enabled = false;
|
||||
it->second->isNew = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//Disable conflicted mods
|
||||
for(size_t i = 0, cnt = mods.size(); i < cnt; ++i) {
|
||||
auto* mod = mods[i];
|
||||
mod->loadFileList();
|
||||
if(!mod->enabled)
|
||||
continue;
|
||||
for(size_t j = 0; j < i; ++j) {
|
||||
auto* othermod = mods[j];
|
||||
if(!othermod->enabled)
|
||||
continue;
|
||||
if(!mod->isCompatible(othermod)) {
|
||||
error("ERROR: Mod '%s' is not compatible with previously enabled mod '%s'. Disabling.",
|
||||
othermod->name.c_str(), mod->name.c_str());
|
||||
othermod->enabled = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void Manager::saveState() {
|
||||
std::string filename = path_join(getProfileRoot(), "mods.txt");
|
||||
std::ofstream file(filename);
|
||||
|
||||
for(auto it = mods.begin(), end = mods.end(); it != end; ++it) {
|
||||
auto* mod = *it;
|
||||
if(!mod->listed)
|
||||
continue;
|
||||
|
||||
mod->isNew = false;
|
||||
if(mod->enabled) {
|
||||
if(mod->forced && mod->compatibility < CUR_COMPATIBILITY)
|
||||
file << "Forced: ";
|
||||
else
|
||||
file << "Enabled: ";
|
||||
}
|
||||
else
|
||||
file << "Disabled: ";
|
||||
file << mod->ident << "\n";
|
||||
}
|
||||
|
||||
file.close();
|
||||
}
|
||||
|
||||
void Manager::clearMods() {
|
||||
activeMods.clear();
|
||||
activeMods.push_back(baseMod);
|
||||
currentMod = baseMod;
|
||||
}
|
||||
|
||||
bool Manager::enableMod(const std::string& name) {
|
||||
auto it = modNames.find(name);
|
||||
if(it == modNames.end()) {
|
||||
error("Could not find mod: %s", name.c_str());
|
||||
return false;
|
||||
}
|
||||
|
||||
Mod* mod = it->second;
|
||||
foreach(chk, activeMods) {
|
||||
if(*chk == mod)
|
||||
return true;
|
||||
if(mod->isBase && (*chk)->isBase) {
|
||||
if(*chk == baseMod) {
|
||||
activeMods[0] = mod;
|
||||
currentMod = mod;
|
||||
return true;
|
||||
}
|
||||
error("Cannot enable mod: %s. A base mod is already loaded.", name.c_str());
|
||||
return false;
|
||||
}
|
||||
if(!mod->isCompatible(*chk)) {
|
||||
error("Cannot enable mod: %s. Incompatible with mod %s.", name.c_str(), (*chk)->name.c_str());
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
activeMods.push_back(mod);
|
||||
return true;
|
||||
}
|
||||
|
||||
const Mod* Manager::getMod(const std::string& name) const {
|
||||
auto it = modNames.find(name);
|
||||
if(it == modNames.end())
|
||||
return 0;
|
||||
return it->second;
|
||||
}
|
||||
|
||||
std::string Manager::resolve(const std::string& path) const {
|
||||
if(path[0] == '~')
|
||||
return path.substr(1, path.size() - 1);
|
||||
std::string result = path;
|
||||
for(int i = (int)activeMods.size() - 1; i >= 0; --i) {
|
||||
if(activeMods[i]->resolve(result))
|
||||
return result;
|
||||
}
|
||||
return path;
|
||||
}
|
||||
|
||||
std::string Manager::resolve(const std::string& path, const std::string& indir) const {
|
||||
if(path[0] == '~') {
|
||||
if(path[1] == '/')
|
||||
return path.substr(2, path.size() - 2);
|
||||
else
|
||||
return path_join(indir, path.substr(1, path.size() - 1));
|
||||
}
|
||||
if(path[0] == '/')
|
||||
return resolve(path.substr(1, path.size() - 1));
|
||||
return resolve(path_join(indir, path));
|
||||
}
|
||||
|
||||
void Manager::resolve(const std::string& path, std::vector<std::string>& check) const {
|
||||
if(path[0] == '~') {
|
||||
check.push_back(path.substr(1, path.size() - 1));
|
||||
}
|
||||
else {
|
||||
for(int i = (int)activeMods.size() - 1; i >= 0; --i)
|
||||
activeMods[i]->resolve(path, check);
|
||||
}
|
||||
}
|
||||
|
||||
void Manager::listFiles(const std::string& dir, std::map<std::string, std::string>& out,
|
||||
const char* filter, bool recurse) const {
|
||||
for(int i = (int)activeMods.size() - 1; i >= 0; --i)
|
||||
activeMods[i]->listFiles(dir, out, filter, recurse);
|
||||
}
|
||||
|
||||
void Manager::listFiles(const std::string& dir, const char* filter,
|
||||
std::function<void(const std::string&)> cb, bool recurse) const {
|
||||
std::map<std::string, std::string> result;
|
||||
listFiles(dir, result, filter, recurse);
|
||||
|
||||
foreach(it, result)
|
||||
cb(it->second);
|
||||
}
|
||||
|
||||
void Manager::resolveDirectory(const std::string& dir, std::vector<std::string>& out) const {
|
||||
currentMod->resolveDirectory(dir, out);
|
||||
}
|
||||
|
||||
std::string Manager::getProfile(const std::string& path) const {
|
||||
return currentMod->getProfile(path);
|
||||
}
|
||||
|
||||
std::string Manager::getProfile() const {
|
||||
return currentMod->getProfile();
|
||||
}
|
||||
|
||||
std::string Manager::getGlobalProfile(const std::string& path) const {
|
||||
std::string dir = getProfileRoot();
|
||||
|
||||
//Make sure the root profile dir is created
|
||||
if(!isDirectory(dir))
|
||||
makeDirectory(dir);
|
||||
|
||||
//Create any intermediate directories
|
||||
std::vector<std::string> dirs;
|
||||
|
||||
if(!path.empty())
|
||||
path_split(path, dirs);
|
||||
|
||||
foreach(it, dirs) {
|
||||
dir = path_join(dir, *it);
|
||||
|
||||
if(!isDirectory(dir))
|
||||
makeDirectory(dir);
|
||||
}
|
||||
|
||||
return dir;
|
||||
}
|
||||
|
||||
};
|
||||
Reference in New Issue
Block a user