409 lines
11 KiB
C++
409 lines
11 KiB
C++
// Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
|
|
// Copyright (C) 2010 Winch Gate Property Limited
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as
|
|
// published by the Free Software Foundation, either version 3 of the
|
|
// License, or (at your option) any later version.
|
|
//
|
|
// This program 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 Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
#include "nel/misc/types_nl.h"
|
|
#include "nel/misc/config_file.h"
|
|
#include "nel/misc/path.h"
|
|
#include "nel/ligo/primitive.h"
|
|
#include "nel/ligo/ligo_config.h"
|
|
#include "nel/ligo/primitive_utils.h"
|
|
#include "nel/misc/algo.h"
|
|
|
|
using namespace std;
|
|
using namespace NLMISC;
|
|
using namespace NLLIGO;
|
|
|
|
CLigoConfig LigoConfig;
|
|
|
|
vector<string> Filters;
|
|
|
|
|
|
// always true predicate
|
|
struct TAllPrimitivePredicate : public std::unary_function<IPrimitive*, bool>
|
|
{
|
|
bool operator () (IPrimitive *prim)
|
|
{
|
|
|
|
return true;
|
|
}
|
|
|
|
};
|
|
|
|
bool isFiltered(const std::string &path)
|
|
{
|
|
for (uint i=0; i<Filters.size(); ++i)
|
|
{
|
|
if (path.find(Filters[i]) != string::npos)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void syncFile(const string &srcPath, const string &dstPath)
|
|
{
|
|
if (isFiltered(srcPath))
|
|
return;
|
|
|
|
nlinfo("Synchronizing file '%s'", CFile::getFilename(srcPath).c_str());
|
|
|
|
bool modified = false;
|
|
|
|
CPrimitives srcDoc;
|
|
CPrimitives dstDoc;
|
|
|
|
// load the src primitive
|
|
CPrimitiveContext::instance().CurrentPrimitive = &srcDoc;
|
|
loadXmlPrimitiveFile(srcDoc, srcPath, LigoConfig);
|
|
|
|
// load the dst primitive
|
|
CPrimitiveContext::instance().CurrentPrimitive = &dstDoc;
|
|
loadXmlPrimitiveFile(dstDoc, dstPath, LigoConfig);
|
|
CPrimitiveContext::instance().CurrentPrimitive = NULL;
|
|
|
|
// retrieve ALL the alias node in the src doc
|
|
TPrimitiveClassPredicate pred("alias");
|
|
TPrimitiveSet primAlias;
|
|
CPrimitiveSet<TPrimitiveClassPredicate> setBuilder;
|
|
|
|
setBuilder.buildSet(srcDoc.RootNode, pred, primAlias);
|
|
|
|
// look at each matched node
|
|
for (uint i=0; i<primAlias.size(); ++i)
|
|
{
|
|
CPrimAlias *srcPa = dynamic_cast<CPrimAlias*>(primAlias[i]);
|
|
if (srcPa == NULL)
|
|
continue;
|
|
|
|
// build a ascii path to the prim
|
|
string primPath = buildPrimPath(srcPa);
|
|
|
|
// look in the dst doc for a node with this path
|
|
TPrimitiveSet dstAlias;
|
|
selectPrimByPath(dstDoc.RootNode, primPath, dstAlias);
|
|
|
|
if (dstAlias.size() == 1)
|
|
{
|
|
// yeah ! we found a match
|
|
CPrimAlias *pa = dynamic_cast<CPrimAlias*>(dstAlias.front());
|
|
if (pa != NULL && pa->getAlias() != srcPa->getAlias())
|
|
{
|
|
// nldebug("%s : forcing alias %u", primPath.c_str(), srcPa->getAlias());
|
|
dstDoc.forceAlias(pa, srcPa->getAlias());
|
|
modified = true;
|
|
}
|
|
}
|
|
else if (dstAlias.size() == 0)
|
|
{
|
|
// try to go up one level and add a prim alias node
|
|
while (!primPath.empty() && primPath[primPath.size()-1] != '.')
|
|
primPath.resize(primPath.size()-1);
|
|
|
|
if (primPath.empty())
|
|
continue;
|
|
|
|
// remove the '.'
|
|
primPath.resize(primPath.size()-1);
|
|
if (primPath.empty())
|
|
continue;
|
|
|
|
// ok, retry the primitive select now
|
|
selectPrimByPath(dstDoc.RootNode, primPath, dstAlias);
|
|
|
|
if (dstAlias.size() == 1)
|
|
{
|
|
IPrimitive *parent = dstAlias.front();
|
|
|
|
CPrimAlias *pa = static_cast<CPrimAlias*> (CClassRegistry::create ("CPrimAlias"));
|
|
CPropertyString *pname = new CPropertyString ("alias");
|
|
pa->addPropertyByName("name", pname);
|
|
CPropertyString *pclass = new CPropertyString ("alias");
|
|
pa->addPropertyByName("class", pclass);
|
|
|
|
// nldebug("%s : creating new CPrimAlias for %u", primPath.c_str(), srcPa->getAlias());
|
|
// insert the prim alias at first pos
|
|
CPrimitiveContext::instance().CurrentPrimitive = &dstDoc;
|
|
parent->insertChild(pa, 0);
|
|
dstDoc.forceAlias(pa, srcPa->getAlias());
|
|
|
|
modified = true;
|
|
}
|
|
else
|
|
{
|
|
// multiple node match !
|
|
// try to affect the alias if src an dst have the same number of match
|
|
TPrimitiveSet srcAlias;
|
|
selectPrimByPath(srcDoc.RootNode, primPath, srcAlias);
|
|
|
|
if (dstAlias.size() == srcAlias.size())
|
|
{
|
|
// ok, we can match
|
|
for (uint i=0; i<srcAlias.size(); ++i)
|
|
{
|
|
IPrimitive *parent = dstAlias[i];
|
|
TPrimitiveClassPredicate pred("alias");
|
|
CPrimAlias *srcPa = static_cast<CPrimAlias*>(getPrimitiveChild(srcAlias[i], pred));
|
|
|
|
CPrimAlias *pa = static_cast<CPrimAlias*> (CClassRegistry::create ("CPrimAlias"));
|
|
CPropertyString *pname = new CPropertyString ("alias");
|
|
pa->addPropertyByName("name", pname);
|
|
CPropertyString *pclass = new CPropertyString ("alias");
|
|
pa->addPropertyByName("class", pclass);
|
|
|
|
// nldebug("%s : creating new CPrimAlias for %u", primPath.c_str(), srcPa->getAlias());
|
|
// insert the prim alias at first pos
|
|
CPrimitiveContext::instance().CurrentPrimitive = &dstDoc;
|
|
parent->insertChild(pa, 0);
|
|
dstDoc.forceAlias(pa, srcPa->getAlias());
|
|
|
|
modified = true;
|
|
|
|
}
|
|
}
|
|
else if (!dstAlias.empty())
|
|
{
|
|
nlwarning("In '%s', the path '%s' match to a different multiple node set, no alias restored !",
|
|
CFile::getFilename(srcPath).c_str(),
|
|
primPath.c_str());
|
|
}
|
|
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// multiple dst node mapping
|
|
// try to affect the alias if src an dst have the same number of match
|
|
TPrimitiveSet srcAlias;
|
|
selectPrimByPath(srcDoc.RootNode, primPath, srcAlias);
|
|
|
|
if (dstAlias.size() == srcAlias.size())
|
|
{
|
|
// ok, we can match
|
|
for (uint i=0; i<srcAlias.size(); ++i)
|
|
{
|
|
CPrimAlias *pa = dynamic_cast<CPrimAlias*>(dstAlias[i]);
|
|
CPrimAlias *srcPa = dynamic_cast<CPrimAlias*>(srcAlias[i]);
|
|
if (pa != NULL && pa->getAlias() != srcPa->getAlias())
|
|
{
|
|
// nldebug("%s : forcing alias %u", primPath.c_str(), srcPa->getAlias());
|
|
dstDoc.forceAlias(pa, srcPa->getAlias());
|
|
modified = true;
|
|
}
|
|
|
|
}
|
|
}
|
|
else if (!dstAlias.empty())
|
|
{
|
|
nlwarning("In '%s', the path '%s' match to a different multiple node set, no alias restored !",
|
|
CFile::getFilename(srcPath).c_str(),
|
|
primPath.c_str());
|
|
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
// second loop : for each dst node, look if it need a missing CPrimAlias node and generated it if needed
|
|
{
|
|
TAllPrimitivePredicate pred;
|
|
CPrimitiveEnumerator<TAllPrimitivePredicate> pe(dstDoc.RootNode, pred);
|
|
IPrimitive *prim;
|
|
|
|
while ((prim = pe.getNextMatch()) != NULL)
|
|
{
|
|
const CPrimitiveClass *pc = LigoConfig.getPrimitiveClass(*prim);
|
|
if (pc)
|
|
{
|
|
// check all the static childrens
|
|
for (uint i=0; i<pc->StaticChildren.size(); ++i)
|
|
{
|
|
const CPrimitiveClass::CChild &cc = pc->StaticChildren[i];
|
|
|
|
if (cc.ClassName == "alias" && cc.Name == "alias")
|
|
{
|
|
bool found = false;
|
|
// ok, this node need an alias, check if it is present
|
|
for (uint i=0; i<prim->getNumChildren(); ++i)
|
|
{
|
|
IPrimitive *child;
|
|
if (prim->getChild(child, i))
|
|
{
|
|
string className;
|
|
if (child->getPropertyByName("class", className) && className == "alias")
|
|
{
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!found)
|
|
{
|
|
// ok, we need to add the alias class
|
|
CPrimAlias *pa = static_cast<CPrimAlias*> (CClassRegistry::create ("CPrimAlias"));
|
|
CPropertyString *pname = new CPropertyString ("alias");
|
|
pa->addPropertyByName("name", pname);
|
|
CPropertyString *pclass = new CPropertyString ("alias");
|
|
pa->addPropertyByName("class", pclass);
|
|
|
|
// nldebug("%s : creating new CPrimAlias for %u", primPath.c_str(), srcPa->getAlias());
|
|
// insert the prim alias at first pos
|
|
CPrimitiveContext::instance().CurrentPrimitive = &dstDoc;
|
|
prim->insertChild(pa, 0);
|
|
|
|
modified = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (modified)
|
|
{
|
|
saveXmlPrimitiveFile(dstDoc, dstPath);
|
|
}
|
|
}
|
|
|
|
|
|
struct TSetFileName
|
|
{
|
|
void operator () (std::string &str)
|
|
{
|
|
str = CFile::getFilename(str);
|
|
}
|
|
};
|
|
|
|
struct TSetLastFolder
|
|
{
|
|
void operator () (std::string &str)
|
|
{
|
|
vector<string> folders;
|
|
str = CPath::standardizePath(str, false);
|
|
#ifdef NL_OS_WINDOWS
|
|
explode(str, std::string("/"), folders, true);
|
|
#else // NL_OS_WINDOW
|
|
NLMISC::splitString(str, "/", folders);
|
|
#endif // NL_OS_WINDOWS
|
|
if (!folders.empty())
|
|
str = folders.back();
|
|
else
|
|
str = "";
|
|
}
|
|
};
|
|
|
|
void syncFolder(const string &srcPath, const string &dstPath)
|
|
{
|
|
string tmp(srcPath);
|
|
TSetLastFolder slf;
|
|
slf.operator()(tmp);
|
|
nlinfo("Synchronizing folder '%s'", tmp.c_str());
|
|
|
|
// check all the file in the current folder
|
|
vector<string> srcFiles;
|
|
CPath::getPathContent(srcPath, false, false, true, srcFiles);
|
|
for_each(srcFiles.begin(), srcFiles.end(), TSetFileName());
|
|
sort(srcFiles.begin(), srcFiles.end());
|
|
|
|
vector<string> dstFiles;
|
|
CPath::getPathContent(dstPath, false, false, true, dstFiles);
|
|
for_each(dstFiles.begin(), dstFiles.end(), TSetFileName());
|
|
sort(dstFiles.begin(), dstFiles.end());
|
|
|
|
uint si=0; uint di=0;
|
|
for (; si < srcFiles.size() && di < dstFiles.size(); ++si, ++di)
|
|
{
|
|
if (srcFiles[si] < dstFiles[di])
|
|
--di;
|
|
else if (srcFiles[si] > dstFiles[di])
|
|
--si;
|
|
else if (srcFiles[si].find(".primitive") != string::npos)
|
|
{
|
|
// two file identical, synch the alias inside
|
|
syncFile(srcPath+"/"+srcFiles[si], dstPath+"/"+dstFiles[di]);
|
|
}
|
|
}
|
|
|
|
// check all the folders in the current folder
|
|
vector<string> srcFolders;
|
|
CPath::getPathContent(srcPath, false, true, false, srcFolders);
|
|
for_each(srcFolders.begin(), srcFolders.end(), TSetLastFolder());
|
|
sort(srcFolders.begin(), srcFolders.end());
|
|
|
|
vector<string> dstFolders;
|
|
CPath::getPathContent(dstPath, false, true, false, dstFolders);
|
|
for_each(dstFolders.begin(), dstFolders.end(), TSetLastFolder());
|
|
sort(dstFolders.begin(), dstFolders.end());
|
|
|
|
si=0; di=0;
|
|
for (; si < srcFolders.size() && di < dstFolders.size(); ++si, ++di)
|
|
{
|
|
if (srcFolders[si] < dstFolders[di])
|
|
--di;
|
|
else if (srcFolders[si] > dstFolders[di])
|
|
--si;
|
|
else
|
|
{
|
|
// two folder identical, synch the files inside
|
|
syncFolder(srcPath+"/"+srcFolders[si], dstPath+"/"+dstFolders[di]);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
int main()
|
|
{
|
|
new NLMISC::CApplicationContext;
|
|
|
|
CConfigFile cf;
|
|
|
|
// register ligo primitives
|
|
Register();
|
|
|
|
cf.load("alias_synchronizer.cfg");
|
|
|
|
CConfigFile::CVar &paths = cf.getVar("Paths");
|
|
CConfigFile::CVar &srcPath = cf.getVar("SrcPath");
|
|
CConfigFile::CVar &dstPath = cf.getVar("DstPath");
|
|
CConfigFile::CVar &filters = cf.getVar("Filters");
|
|
|
|
// store the filters
|
|
for (uint i=0; i<filters.size(); ++i)
|
|
{
|
|
Filters.push_back(filters.asString(i));
|
|
}
|
|
|
|
// add the search paths
|
|
for (uint i=0; i<paths.size(); ++i)
|
|
{
|
|
CPath::addSearchPath(NLMISC::expandEnvironmentVariables(paths.asString(i)), true, false);
|
|
}
|
|
|
|
// init ligo
|
|
LigoConfig.readPrimitiveClass("world_editor_classes.xml", false);
|
|
CPrimitiveContext::instance().CurrentLigoConfig = &LigoConfig;
|
|
|
|
nlinfo("Synchronizing folder '%s' to '%s'",
|
|
srcPath.asString().c_str(),
|
|
dstPath.asString().c_str());
|
|
|
|
// do a recursive scan of the src and dst folder
|
|
syncFolder(srcPath.asString(), dstPath.asString());
|
|
|
|
return 0;
|
|
}
|
|
|