CModHandler.cpp 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857
  1. #include "StdInc.h"
  2. #include "CModHandler.h"
  3. #include "CObjectClassesHandler.h"
  4. #include "JsonNode.h"
  5. #include "filesystem/Filesystem.h"
  6. #include "filesystem/AdapterLoaders.h"
  7. #include "filesystem/CFilesystemLoader.h"
  8. #include "CCreatureHandler.h"
  9. #include "CArtHandler.h"
  10. #include "CTownHandler.h"
  11. #include "CHeroHandler.h"
  12. #include "CObjectHandler.h"
  13. #include "StringConstants.h"
  14. #include "CStopWatch.h"
  15. #include "IHandlerBase.h"
  16. #include "CSpellHandler.h"
  17. /*
  18. * CModHandler.cpp, part of VCMI engine
  19. *
  20. * Authors: listed in file AUTHORS in main folder
  21. *
  22. * License: GNU General Public License v2.0 or later
  23. * Full text of license available in license.txt file, in main folder
  24. *
  25. */
  26. void CIdentifierStorage::checkIdentifier(std::string & ID)
  27. {
  28. if (boost::algorithm::ends_with(ID, "."))
  29. logGlobal->warnStream() << "BIG WARNING: identifier " << ID << " seems to be broken!";
  30. else
  31. {
  32. size_t pos = 0;
  33. do
  34. {
  35. if (std::tolower(ID[pos]) != ID[pos] ) //Not in camelCase
  36. {
  37. logGlobal->warnStream() << "Warning: identifier " << ID << " is not in camelCase!";
  38. ID[pos] = std::tolower(ID[pos]);// Try to fix the ID
  39. }
  40. pos = ID.find('.', pos);
  41. }
  42. while(pos++ != std::string::npos);
  43. }
  44. }
  45. CIdentifierStorage::ObjectCallback::ObjectCallback(std::string localScope, std::string remoteScope, std::string type,
  46. std::string name, const std::function<void(si32)> & callback, bool optional):
  47. localScope(localScope),
  48. remoteScope(remoteScope),
  49. type(type),
  50. name(name),
  51. callback(callback),
  52. optional(optional)
  53. {}
  54. static std::pair<std::string, std::string> splitString(std::string input, char separator)
  55. {
  56. std::pair<std::string, std::string> ret;
  57. size_t splitPos = input.find(separator);
  58. if (splitPos == std::string::npos)
  59. {
  60. ret.first.clear();
  61. ret.second = input;
  62. }
  63. else
  64. {
  65. ret.first = input.substr(0, splitPos);
  66. ret.second = input.substr(splitPos + 1);
  67. }
  68. return ret;
  69. }
  70. void CIdentifierStorage::requestIdentifier(ObjectCallback callback)
  71. {
  72. checkIdentifier(callback.type);
  73. checkIdentifier(callback.name);
  74. assert(!callback.localScope.empty());
  75. scheduledRequests.push_back(callback);
  76. }
  77. void CIdentifierStorage::requestIdentifier(std::string scope, std::string type, std::string name, const std::function<void(si32)> & callback)
  78. {
  79. auto pair = splitString(name, ':'); // remoteScope:name
  80. requestIdentifier(ObjectCallback(scope, pair.first, type, pair.second, callback, false));
  81. }
  82. void CIdentifierStorage::requestIdentifier(std::string type, const JsonNode & name, const std::function<void(si32)> & callback)
  83. {
  84. auto pair = splitString(name.String(), ':'); // remoteScope:name
  85. requestIdentifier(ObjectCallback(name.meta, pair.first, type, pair.second, callback, false));
  86. }
  87. void CIdentifierStorage::requestIdentifier(const JsonNode & name, const std::function<void(si32)> & callback)
  88. {
  89. auto pair = splitString(name.String(), ':'); // remoteScope:<type.name>
  90. auto pair2 = splitString(pair.second, '.'); // type.name
  91. requestIdentifier(ObjectCallback(name.meta, pair.first, pair2.first, pair2.second, callback, false));
  92. }
  93. void CIdentifierStorage::tryRequestIdentifier(std::string scope, std::string type, std::string name, const std::function<void(si32)> & callback)
  94. {
  95. auto pair = splitString(name, ':'); // remoteScope:name
  96. requestIdentifier(ObjectCallback(scope, pair.first, type, pair.second, callback, true));
  97. }
  98. void CIdentifierStorage::tryRequestIdentifier(std::string type, const JsonNode & name, const std::function<void(si32)> & callback)
  99. {
  100. auto pair = splitString(name.String(), ':'); // remoteScope:name
  101. requestIdentifier(ObjectCallback(name.meta, pair.first, type, pair.second, callback, true));
  102. }
  103. boost::optional<si32> CIdentifierStorage::getIdentifier(std::string scope, std::string type, std::string name, bool silent)
  104. {
  105. auto pair = splitString(name, ':'); // remoteScope:name
  106. auto idList = getPossibleIdentifiers(ObjectCallback(scope, pair.first, type, pair.second, std::function<void(si32)>(), silent));
  107. if (idList.size() == 1)
  108. return idList.front().id;
  109. if (!silent)
  110. logGlobal->errorStream() << "Failed to resolve identifier " << name << " from mod " << scope;
  111. return boost::optional<si32>();
  112. }
  113. boost::optional<si32> CIdentifierStorage::getIdentifier(std::string type, const JsonNode & name, bool silent)
  114. {
  115. auto pair = splitString(name.String(), ':'); // remoteScope:name
  116. auto idList = getPossibleIdentifiers(ObjectCallback(name.meta, pair.first, type, pair.second, std::function<void(si32)>(), silent));
  117. if (idList.size() == 1)
  118. return idList.front().id;
  119. if (!silent)
  120. logGlobal->errorStream() << "Failed to resolve identifier " << name.String() << " from mod " << type;
  121. return boost::optional<si32>();
  122. }
  123. boost::optional<si32> CIdentifierStorage::getIdentifier(const JsonNode & name, bool silent)
  124. {
  125. auto pair = splitString(name.String(), ':'); // remoteScope:<type.name>
  126. auto pair2 = splitString(pair.second, '.'); // type.name
  127. auto idList = getPossibleIdentifiers(ObjectCallback(name.meta, pair.first, pair2.first, pair2.second, std::function<void(si32)>(), silent));
  128. if (idList.size() == 1)
  129. return idList.front().id;
  130. if (!silent)
  131. logGlobal->errorStream() << "Failed to resolve identifier " << name.String() << " from mod " << name.meta;
  132. return boost::optional<si32>();
  133. }
  134. void CIdentifierStorage::registerObject(std::string scope, std::string type, std::string name, si32 identifier)
  135. {
  136. ObjectData data;
  137. data.scope = scope;
  138. data.id = identifier;
  139. std::string fullID = type + '.' + name;
  140. checkIdentifier(fullID);
  141. registeredObjects.insert(std::make_pair(fullID, data));
  142. }
  143. std::vector<CIdentifierStorage::ObjectData> CIdentifierStorage::getPossibleIdentifiers(const ObjectCallback & request)
  144. {
  145. std::set<std::string> allowedScopes;
  146. if (request.remoteScope.empty())
  147. {
  148. // normally ID's from all required mods, own mod and virtual "core" mod are allowed
  149. if (request.localScope != "core" && request.localScope != "")
  150. allowedScopes = VLC->modh->getModData(request.localScope).dependencies;
  151. allowedScopes.insert(request.localScope);
  152. allowedScopes.insert("core");
  153. }
  154. else
  155. {
  156. //...unless destination mod was specified explicitly
  157. auto myDeps = VLC->modh->getModData(request.localScope).dependencies;
  158. if (request.remoteScope == "core" || // allow only available to all core mod
  159. myDeps.count(request.remoteScope)) // or dependencies
  160. allowedScopes.insert(request.remoteScope);
  161. }
  162. std::string fullID = request.type + '.' + request.name;
  163. auto entries = registeredObjects.equal_range(fullID);
  164. if (entries.first != entries.second)
  165. {
  166. std::vector<ObjectData> locatedIDs;
  167. for (auto it = entries.first; it != entries.second; it++)
  168. {
  169. if (vstd::contains(allowedScopes, it->second.scope))
  170. {
  171. locatedIDs.push_back(it->second);
  172. }
  173. }
  174. return locatedIDs;
  175. }
  176. return std::vector<ObjectData>();
  177. }
  178. bool CIdentifierStorage::resolveIdentifier(const ObjectCallback & request)
  179. {
  180. auto identifiers = getPossibleIdentifiers(request);
  181. if (identifiers.size() == 1) // normally resolved ID
  182. {
  183. request.callback(identifiers.front().id);
  184. return true;
  185. }
  186. if (request.optional && identifiers.empty()) // failed to resolve optinal ID
  187. return true;
  188. // error found. Try to generate some debug info
  189. if (identifiers.size() == 0)
  190. logGlobal->errorStream() << "Unknown identifier!";
  191. else
  192. logGlobal->errorStream() << "Ambiguous identifier request!";
  193. logGlobal->errorStream() << "Request for " << request.type << "." << request.name << " from mod " << request.localScope;
  194. for (auto id : identifiers)
  195. {
  196. logGlobal->errorStream() << "\tID is available in mod " << id.scope;
  197. }
  198. return false;
  199. }
  200. void CIdentifierStorage::finalize()
  201. {
  202. bool errorsFound = false;
  203. for(const ObjectCallback & request : scheduledRequests)
  204. {
  205. errorsFound |= !resolveIdentifier(request);
  206. }
  207. if (errorsFound)
  208. {
  209. for(auto object : registeredObjects)
  210. {
  211. logGlobal->traceStream() << object.first << " -> " << object.second.id;
  212. }
  213. logGlobal->errorStream() << "All known identifiers were dumped into log file";
  214. }
  215. assert(errorsFound == false);
  216. }
  217. CContentHandler::ContentTypeHandler::ContentTypeHandler(IHandlerBase * handler, std::string objectName):
  218. handler(handler),
  219. objectName(objectName),
  220. originalData(handler->loadLegacyData(VLC->modh->settings.data["textData"][objectName].Float()))
  221. {
  222. for(auto & node : originalData)
  223. {
  224. node.setMeta("core");
  225. }
  226. }
  227. bool CContentHandler::ContentTypeHandler::preloadModData(std::string modName, std::vector<std::string> fileList, bool validate)
  228. {
  229. bool result;
  230. JsonNode data = JsonUtils::assembleFromFiles(fileList, result);
  231. data.setMeta(modName);
  232. ModInfo & modInfo = modData[modName];
  233. for(auto entry : data.Struct())
  234. {
  235. size_t colon = entry.first.find(':');
  236. if (colon == std::string::npos)
  237. {
  238. // normal object, local to this mod
  239. modInfo.modData[entry.first].swap(entry.second);
  240. }
  241. else
  242. {
  243. std::string remoteName = entry.first.substr(0, colon);
  244. std::string objectName = entry.first.substr(colon + 1);
  245. // patching this mod? Send warning and continue - this situation can be handled normally
  246. if (remoteName == modName)
  247. logGlobal->warnStream() << "Redundant namespace definition for " << objectName;
  248. JsonNode & remoteConf = modData[remoteName].patches[objectName];
  249. JsonUtils::merge(remoteConf, entry.second);
  250. }
  251. }
  252. return result;
  253. }
  254. bool CContentHandler::ContentTypeHandler::loadMod(std::string modName, bool validate)
  255. {
  256. ModInfo & modInfo = modData[modName];
  257. bool result = true;
  258. // apply patches
  259. if (!modInfo.patches.isNull())
  260. JsonUtils::merge(modInfo.modData, modInfo.patches);
  261. for(auto & entry : modInfo.modData.Struct())
  262. {
  263. const std::string & name = entry.first;
  264. JsonNode & data = entry.second;
  265. if (vstd::contains(data.Struct(), "index") && !data["index"].isNull())
  266. {
  267. // try to add H3 object data
  268. size_t index = data["index"].Float();
  269. if (originalData.size() > index)
  270. {
  271. JsonUtils::merge(originalData[index], data);
  272. if (validate)
  273. result &= JsonUtils::validate(originalData[index], "vcmi:" + objectName, name);
  274. handler->loadObject(modName, name, originalData[index], index);
  275. originalData[index].clear(); // do not use same data twice (same ID)
  276. continue;
  277. }
  278. }
  279. // normal new object or one with index bigger that data size
  280. if (validate)
  281. result &= JsonUtils::validate(data, "vcmi:" + objectName, name);
  282. handler->loadObject(modName, name, data);
  283. }
  284. return result;
  285. }
  286. void CContentHandler::ContentTypeHandler::afterLoadFinalization()
  287. {
  288. handler->afterLoadFinalization();
  289. }
  290. CContentHandler::CContentHandler()
  291. {
  292. handlers.insert(std::make_pair("heroClasses", ContentTypeHandler(&VLC->heroh->classes, "heroClass")));
  293. handlers.insert(std::make_pair("artifacts", ContentTypeHandler(VLC->arth, "artifact")));
  294. handlers.insert(std::make_pair("creatures", ContentTypeHandler(VLC->creh, "creature")));
  295. handlers.insert(std::make_pair("factions", ContentTypeHandler(VLC->townh, "faction")));
  296. handlers.insert(std::make_pair("objects", ContentTypeHandler(VLC->objtypeh, "object")));
  297. handlers.insert(std::make_pair("heroes", ContentTypeHandler(VLC->heroh, "hero")));
  298. handlers.insert(std::make_pair("spells", ContentTypeHandler(VLC->spellh, "spell")));
  299. //TODO: any other types of moddables?
  300. }
  301. bool CContentHandler::preloadModData(std::string modName, JsonNode modConfig, bool validate)
  302. {
  303. bool result = true;
  304. for(auto & handler : handlers)
  305. {
  306. result &= handler.second.preloadModData(modName, modConfig[handler.first].convertTo<std::vector<std::string> >(), validate);
  307. }
  308. return result;
  309. }
  310. bool CContentHandler::loadMod(std::string modName, bool validate)
  311. {
  312. bool result = true;
  313. for(auto & handler : handlers)
  314. {
  315. result &= handler.second.loadMod(modName, validate);
  316. }
  317. return result;
  318. }
  319. void CContentHandler::afterLoadFinalization()
  320. {
  321. for(auto & handler : handlers)
  322. {
  323. handler.second.afterLoadFinalization();
  324. }
  325. }
  326. void CContentHandler::preloadData(CModInfo & mod)
  327. {
  328. bool validate = (mod.validation != CModInfo::PASSED);
  329. // print message in format [<8-symbols checksum>] <modname>
  330. logGlobal->infoStream() << "\t\t[" << std::noshowbase << std::hex << std::setw(8) << std::setfill('0')
  331. << mod.checksum << "] " << mod.name;
  332. if (validate && mod.identifier != "core")
  333. {
  334. if (!JsonUtils::validate(mod.config, "vcmi:mod", mod.identifier))
  335. mod.validation = CModInfo::FAILED;
  336. }
  337. if (!preloadModData(mod.identifier, mod.config, validate))
  338. mod.validation = CModInfo::FAILED;
  339. }
  340. void CContentHandler::load(CModInfo & mod)
  341. {
  342. bool validate = (mod.validation != CModInfo::PASSED);
  343. if (!loadMod(mod.identifier, validate))
  344. mod.validation = CModInfo::FAILED;
  345. if (validate)
  346. {
  347. if (mod.validation != CModInfo::FAILED)
  348. logGlobal->infoStream() << "\t\t[DONE] " << mod.name;
  349. else
  350. logGlobal->errorStream() << "\t\t[FAIL] " << mod.name;
  351. }
  352. else
  353. logGlobal->infoStream() << "\t\t[SKIP] " << mod.name;
  354. }
  355. static JsonNode loadModSettings(std::string path)
  356. {
  357. if (CResourceHandler::get("local")->existsResource(ResourceID(path)))
  358. {
  359. return JsonNode(ResourceID(path, EResType::TEXT));
  360. }
  361. // Probably new install. Create initial configuration
  362. CResourceHandler::get("local")->createResource(path);
  363. return JsonNode();
  364. }
  365. JsonNode addMeta(JsonNode config, std::string meta)
  366. {
  367. config.setMeta(meta);
  368. return std::move(config);
  369. }
  370. CModInfo::CModInfo(std::string identifier,const JsonNode & local, const JsonNode & config):
  371. identifier(identifier),
  372. name(config["name"].String()),
  373. description(config["description"].String()),
  374. dependencies(config["depends"].convertTo<std::set<std::string> >()),
  375. conflicts(config["conflicts"].convertTo<std::set<std::string> >()),
  376. validation(PENDING),
  377. config(addMeta(config, identifier))
  378. {
  379. loadLocalData(local);
  380. }
  381. JsonNode CModInfo::saveLocalData() const
  382. {
  383. std::ostringstream stream;
  384. stream << std::noshowbase << std::hex << std::setw(8) << std::setfill('0') << checksum;
  385. JsonNode conf;
  386. conf["active"].Bool() = enabled;
  387. conf["validated"].Bool() = validation != FAILED;
  388. conf["checksum"].String() = stream.str();
  389. return conf;
  390. }
  391. std::string CModInfo::getModDir(std::string name)
  392. {
  393. return "MODS/" + boost::algorithm::replace_all_copy(name, ".", "/MODS/");
  394. }
  395. std::string CModInfo::getModFile(std::string name)
  396. {
  397. return getModDir(name) + "/mod.json";
  398. }
  399. void CModInfo::updateChecksum(ui32 newChecksum)
  400. {
  401. // comment-out next line to force validation of all mods ignoring checksum
  402. if (newChecksum != checksum)
  403. {
  404. checksum = newChecksum;
  405. validation = PENDING;
  406. }
  407. }
  408. void CModInfo::loadLocalData(const JsonNode & data)
  409. {
  410. bool validated = false;
  411. enabled = true;
  412. checksum = 0;
  413. if (data.getType() == JsonNode::DATA_BOOL)
  414. {
  415. enabled = data.Bool();
  416. }
  417. if (data.getType() == JsonNode::DATA_STRUCT)
  418. {
  419. enabled = data["active"].Bool();
  420. validated = data["validated"].Bool();
  421. checksum = strtol(data["checksum"].String().c_str(), nullptr, 16);
  422. }
  423. if (enabled)
  424. validation = validated ? PASSED : PENDING;
  425. else
  426. validation = validated ? PASSED : FAILED;
  427. }
  428. CModHandler::CModHandler()
  429. {
  430. for (int i = 0; i < GameConstants::RESOURCE_QUANTITY; ++i)
  431. {
  432. identifiers.registerObject("core", "resource", GameConstants::RESOURCE_NAMES[i], i);
  433. }
  434. for(int i=0; i<GameConstants::PRIMARY_SKILLS; ++i)
  435. identifiers.registerObject("core", "primSkill", PrimarySkill::names[i], i);
  436. }
  437. void CModHandler::loadConfigFromFile (std::string name)
  438. {
  439. settings.data = JsonUtils::assembleFromFiles("config/" + name);
  440. const JsonNode & hardcodedFeatures = settings.data["hardcodedFeatures"];
  441. settings.CREEP_SIZE = hardcodedFeatures["CREEP_SIZE"].Float();
  442. settings.WEEKLY_GROWTH = hardcodedFeatures["WEEKLY_GROWTH_PERCENT"].Float();
  443. settings.NEUTRAL_STACK_EXP = hardcodedFeatures["NEUTRAL_STACK_EXP_DAILY"].Float();
  444. settings.MAX_BUILDING_PER_TURN = hardcodedFeatures["MAX_BUILDING_PER_TURN"].Float();
  445. settings.DWELLINGS_ACCUMULATE_CREATURES = hardcodedFeatures["DWELLINGS_ACCUMULATE_CREATURES"].Bool();
  446. settings.ALL_CREATURES_GET_DOUBLE_MONTHS = hardcodedFeatures["ALL_CREATURES_GET_DOUBLE_MONTHS"].Bool();
  447. const JsonNode & gameModules = settings.data["modules"];
  448. modules.STACK_EXP = gameModules["STACK_EXPERIENCE"].Bool();
  449. modules.STACK_ARTIFACT = gameModules["STACK_ARTIFACTS"].Bool();
  450. modules.COMMANDERS = gameModules["COMMANDERS"].Bool();
  451. modules.MITHRIL = gameModules["MITHRIL"].Bool();
  452. }
  453. // currentList is passed by value to get current list of depending mods
  454. bool CModHandler::hasCircularDependency(TModID modID, std::set <TModID> currentList) const
  455. {
  456. const CModInfo & mod = allMods.at(modID);
  457. // Mod already present? We found a loop
  458. if (vstd::contains(currentList, modID))
  459. {
  460. logGlobal->errorStream() << "Error: Circular dependency detected! Printing dependency list:";
  461. logGlobal->errorStream() << "\t" << mod.name << " -> ";
  462. return true;
  463. }
  464. currentList.insert(modID);
  465. // recursively check every dependency of this mod
  466. for(const TModID & dependency : mod.dependencies)
  467. {
  468. if (hasCircularDependency(dependency, currentList))
  469. {
  470. logGlobal->errorStream() << "\t" << mod.name << " ->\n"; // conflict detected, print dependency list
  471. return true;
  472. }
  473. }
  474. return false;
  475. }
  476. bool CModHandler::checkDependencies(const std::vector <TModID> & input) const
  477. {
  478. for(const TModID & id : input)
  479. {
  480. const CModInfo & mod = allMods.at(id);
  481. for(const TModID & dep : mod.dependencies)
  482. {
  483. if (!vstd::contains(input, dep))
  484. {
  485. logGlobal->errorStream() << "Error: Mod " << mod.name << " requires missing " << dep << "!";
  486. return false;
  487. }
  488. }
  489. for(const TModID & conflicting : mod.conflicts)
  490. {
  491. if (vstd::contains(input, conflicting))
  492. {
  493. logGlobal->errorStream() << "Error: Mod " << mod.name << " conflicts with " << allMods.at(conflicting).name << "!";
  494. return false;
  495. }
  496. }
  497. if (hasCircularDependency(id))
  498. return false;
  499. }
  500. return true;
  501. }
  502. std::vector <TModID> CModHandler::resolveDependencies(std::vector <TModID> input) const
  503. {
  504. // Topological sort algorithm
  505. // May not be the fastest one but VCMI does not needs any speed here
  506. // Unless user have dozens of mods with complex dependencies this code should be fine
  507. // first - sort input to have input strictly based on name (and not on hashmap or anything else)
  508. boost::range::sort(input);
  509. std::vector <TModID> output;
  510. output.reserve(input.size());
  511. std::set <TModID> resolvedMods;
  512. // Check if all mod dependencies are resolved (moved to resolvedMods)
  513. auto isResolved = [&](const CModInfo mod) -> bool
  514. {
  515. for(const TModID & dependency : mod.dependencies)
  516. {
  517. if (!vstd::contains(resolvedMods, dependency))
  518. return false;
  519. }
  520. return true;
  521. };
  522. while (!input.empty())
  523. {
  524. std::set <TModID> toResolve; // list of mods resolved on this iteration
  525. for (auto it = input.begin(); it != input.end();)
  526. {
  527. if (isResolved(allMods.at(*it)))
  528. {
  529. toResolve.insert(*it);
  530. output.push_back(*it);
  531. it = input.erase(it);
  532. continue;
  533. }
  534. it++;
  535. }
  536. resolvedMods.insert(toResolve.begin(), toResolve.end());
  537. }
  538. return output;
  539. }
  540. std::vector<std::string> CModHandler::getModList(std::string path)
  541. {
  542. std::string modDir = boost::to_upper_copy(path + "MODS/");
  543. size_t depth = boost::range::count(modDir, '/');
  544. auto list = CResourceHandler::get("initial")->getFilteredFiles([&](const ResourceID & id) -> bool
  545. {
  546. if (id.getType() != EResType::DIRECTORY)
  547. return false;
  548. if (!boost::algorithm::starts_with(id.getName(), modDir))
  549. return false;
  550. if (boost::range::count(id.getName(), '/') != depth )
  551. return false;
  552. return true;
  553. });
  554. //storage for found mods
  555. std::vector<std::string> foundMods;
  556. for (auto & entry : list)
  557. {
  558. std::string name = entry.getName();
  559. name.erase(0, modDir.size()); //Remove path prefix
  560. // check if wog is actually present. Hack-ish but better than crash
  561. // TODO: remove soon (hopefully - before 0.96)
  562. if (name == "WOG")
  563. {
  564. if (!CResourceHandler::get("initial")->existsResource(ResourceID("DATA/ZVS", EResType::DIRECTORY)) &&
  565. !CResourceHandler::get("initial")->existsResource(ResourceID("MODS/WOG/DATA/ZVS", EResType::DIRECTORY)))
  566. {
  567. continue;
  568. }
  569. }
  570. if (!name.empty())
  571. foundMods.push_back(name);
  572. }
  573. return foundMods;
  574. }
  575. void CModHandler::loadMods(std::string path, std::string parent, const JsonNode & modSettings, bool enableMods)
  576. {
  577. for (std::string modName : getModList(path))
  578. {
  579. boost::to_lower(modName);
  580. std::string modFullName = parent.empty() ? modName : parent + '.' + modName;
  581. if (CResourceHandler::get("initial")->existsResource(ResourceID(CModInfo::getModFile(modFullName))))
  582. {
  583. CModInfo mod(modFullName, modSettings[modName], JsonNode(ResourceID(CModInfo::getModFile(modFullName))));
  584. if (!parent.empty()) // this is submod, add parent to dependecies
  585. mod.dependencies.insert(parent);
  586. allMods[modFullName] = mod;
  587. if (mod.enabled && enableMods)
  588. activeMods.push_back(modFullName);
  589. loadMods(CModInfo::getModDir(modFullName) + '/', modFullName, modSettings[modName]["mods"], enableMods && mod.enabled);
  590. }
  591. }
  592. }
  593. void CModHandler::loadMods()
  594. {
  595. const JsonNode modConfig = loadModSettings("config/modSettings.json");
  596. loadMods("", "", modConfig["activeMods"], true);
  597. coreMod = CModInfo("core", modConfig["core"], JsonNode(ResourceID("config/gameConfig.json")));
  598. coreMod.name = "Original game files";
  599. }
  600. std::vector<std::string> CModHandler::getAllMods()
  601. {
  602. std::vector<std::string> modlist;
  603. for (auto & entry : allMods)
  604. modlist.push_back(entry.first);
  605. return modlist;
  606. }
  607. std::vector<std::string> CModHandler::getActiveMods()
  608. {
  609. return activeMods;
  610. }
  611. static JsonNode genDefaultFS()
  612. {
  613. // default FS config for mods: directory "Content" that acts as H3 root directory
  614. JsonNode defaultFS;
  615. defaultFS[""].Vector().resize(2);
  616. defaultFS[""].Vector()[0]["type"].String() = "zip";
  617. defaultFS[""].Vector()[0]["path"].String() = "/Content.zip";
  618. defaultFS[""].Vector()[1]["type"].String() = "dir";
  619. defaultFS[""].Vector()[1]["path"].String() = "/Content";
  620. return defaultFS;
  621. }
  622. static ISimpleResourceLoader * genModFilesystem(const std::string & modName, const JsonNode & conf)
  623. {
  624. static const JsonNode defaultFS = genDefaultFS();
  625. if (!conf["filesystem"].isNull())
  626. return CResourceHandler::createFileSystem(CModInfo::getModDir(modName), conf["filesystem"]);
  627. else
  628. return CResourceHandler::createFileSystem(CModInfo::getModDir(modName), defaultFS);
  629. }
  630. static ui32 calculateModChecksum(const std::string modName, ISimpleResourceLoader * filesystem)
  631. {
  632. boost::crc_32_type modChecksum;
  633. // first - add current VCMI version into checksum to force re-validation on VCMI updates
  634. modChecksum.process_bytes(reinterpret_cast<const void*>(GameConstants::VCMI_VERSION.data()), GameConstants::VCMI_VERSION.size());
  635. // second - add mod.json into checksum because filesystem does not contains this file
  636. // FIXME: remove workaround for core mod
  637. if (modName != "core")
  638. {
  639. ResourceID modConfFile(CModInfo::getModFile(modName), EResType::TEXT);
  640. ui32 configChecksum = CResourceHandler::get("initial")->load(modConfFile)->calculateCRC32();
  641. modChecksum.process_bytes(reinterpret_cast<const void *>(&configChecksum), sizeof(configChecksum));
  642. }
  643. // third - add all detected text files from this mod into checksum
  644. auto files = filesystem->getFilteredFiles([](const ResourceID & resID)
  645. {
  646. return resID.getType() == EResType::TEXT &&
  647. ( boost::starts_with(resID.getName(), "DATA") ||
  648. boost::starts_with(resID.getName(), "CONFIG"));
  649. });
  650. for (const ResourceID & file : files)
  651. {
  652. ui32 fileChecksum = filesystem->load(file)->calculateCRC32();
  653. modChecksum.process_bytes(reinterpret_cast<const void *>(&fileChecksum), sizeof(fileChecksum));
  654. }
  655. return modChecksum.checksum();
  656. }
  657. void CModHandler::loadModFilesystems()
  658. {
  659. activeMods = resolveDependencies(activeMods);
  660. coreMod.updateChecksum(calculateModChecksum("core", CResourceHandler::get("core")));
  661. for(std::string & modName : activeMods)
  662. {
  663. CModInfo & mod = allMods[modName];
  664. CResourceHandler::addFilesystem("data", modName, genModFilesystem(modName, mod.config));
  665. }
  666. }
  667. CModInfo & CModHandler::getModData(TModID modId)
  668. {
  669. CModInfo & mod = allMods.at(modId);
  670. assert(vstd::contains(activeMods, modId)); // not really necessary but won't hurt
  671. return mod;
  672. }
  673. void CModHandler::initializeConfig()
  674. {
  675. loadConfigFromFile("defaultMods.json");
  676. }
  677. void CModHandler::load()
  678. {
  679. CStopWatch totalTime, timer;
  680. CContentHandler content;
  681. logGlobal->infoStream() << "\tInitializing content handler: " << timer.getDiff() << " ms";
  682. for(const TModID & modName : activeMods)
  683. {
  684. logGlobal->traceStream() << "Generating checksum for " << modName;
  685. allMods[modName].updateChecksum(calculateModChecksum(modName, CResourceHandler::get(modName)));
  686. }
  687. // first - load virtual "core" mod that contains all data
  688. // TODO? move all data into real mods? RoE, AB, SoD, WoG
  689. content.preloadData(coreMod);
  690. for(const TModID & modName : activeMods)
  691. content.preloadData(allMods[modName]);
  692. logGlobal->infoStream() << "\tParsing mod data: " << timer.getDiff() << " ms";
  693. content.load(coreMod);
  694. for(const TModID & modName : activeMods)
  695. content.load(allMods[modName]);
  696. logGlobal->infoStream() << "\tLoading mod data: " << timer.getDiff() << "ms";
  697. VLC->creh->loadCrExpBon();
  698. VLC->creh->buildBonusTreeForTiers(); //do that after all new creatures are loaded
  699. identifiers.finalize();
  700. logGlobal->infoStream() << "\tResolving identifiers: " << timer.getDiff() << " ms";
  701. content.afterLoadFinalization();
  702. logGlobal->infoStream() << "\tHandlers post-load finalization: " << timer.getDiff() << " ms";
  703. logGlobal->infoStream() << "\tAll game content loaded in " << totalTime.getDiff() << " ms";
  704. }
  705. void CModHandler::afterLoad()
  706. {
  707. JsonNode modSettings;
  708. for (auto & modEntry : allMods)
  709. {
  710. std::string pointer = "/" + boost::algorithm::replace_all_copy(modEntry.first, ".", "/mods/");
  711. modSettings["activeMods"].resolvePointer(pointer) = modEntry.second.saveLocalData();
  712. }
  713. modSettings["core"] = coreMod.saveLocalData();
  714. std::ofstream file(*CResourceHandler::get()->getResourceName(ResourceID("config/modSettings.json")), std::ofstream::trunc);
  715. file << modSettings;
  716. }