mirror of https://github.com/AxioDL/metaforce.git
320 lines
8.6 KiB
C++
320 lines
8.6 KiB
C++
#if _WIN32
|
|
#ifndef NOMINMAX
|
|
#define NOMINMAX
|
|
#endif
|
|
#define WIN_PAUSE 0
|
|
#include <objbase.h>
|
|
#endif
|
|
|
|
#include <clocale>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <stdarg.h>
|
|
#include <signal.h>
|
|
#include <regex>
|
|
#include <list>
|
|
#include "hecl/Database.hpp"
|
|
#include "hecl/Blender/BlenderConnection.hpp"
|
|
#include "logvisor/logvisor.hpp"
|
|
|
|
logvisor::Module LogModule("hecl::Driver");
|
|
|
|
#include "ToolBase.hpp"
|
|
#include "ToolInit.hpp"
|
|
#include "ToolSpec.hpp"
|
|
#include "ToolExtract.hpp"
|
|
#include "ToolAdd.hpp"
|
|
#include "ToolRemove.hpp"
|
|
#include "ToolGroup.hpp"
|
|
#include "ToolCook.hpp"
|
|
#include "ToolClean.hpp"
|
|
#include "ToolPackage.hpp"
|
|
#include "ToolHelp.hpp"
|
|
|
|
/* Static reference to dataspec additions
|
|
* (used by MSVC to definitively link DataSpecs) */
|
|
#include "../DataSpecRegistry.hpp"
|
|
|
|
bool XTERM_COLOR = false;
|
|
|
|
|
|
/*
|
|
#define HECL_GIT 1234567
|
|
#define HECL_GIT_S "1234567"
|
|
#define HECL_BRANCH master
|
|
#define HECL_BRANCH_S "master"
|
|
*/
|
|
|
|
/* Main usage message */
|
|
static void printHelp(const hecl::SystemChar* pname)
|
|
{
|
|
if (XTERM_COLOR)
|
|
hecl::Printf(_S("" BOLD "HECL" NORMAL ""));
|
|
else
|
|
hecl::Printf(_S("HECL"));
|
|
#if HECL_GIT
|
|
hecl::Printf(_S(" Commit " HECL_GIT_S " " HECL_BRANCH_S "\nUsage: %s extract|init|add|remove|group|cook|clean|package|help\n"), pname);
|
|
#elif HECL_VER
|
|
hecl::Printf(_S(" Version " HECL_VER_S "\nUsage: %s extract|init|add|remove|group|cook|clean|package|help\n"), pname);
|
|
#else
|
|
hecl::Printf(_S("\nUsage: %s extract|init|add|remove|group|cook|clean|package|help\n"), pname);
|
|
#endif
|
|
}
|
|
|
|
/* Regex patterns */
|
|
static const hecl::SystemRegex regOPEN(_S("-o([^\"]*|\\S*)"), std::regex::ECMAScript|std::regex::optimize);
|
|
|
|
/* SIGINT will gracefully close blender connections and delete blends in progress */
|
|
static void SIGINTHandler(int sig)
|
|
{
|
|
hecl::BlenderConnection::Shutdown();
|
|
exit(1);
|
|
}
|
|
|
|
/* SIGWINCH should do nothing */
|
|
static void SIGWINCHHandler(int sig) {}
|
|
|
|
static logvisor::Module AthenaLog("Athena");
|
|
static void AthenaExc(athena::error::Level level, const char* file,
|
|
const char*, int line, const char* fmt, ...)
|
|
{
|
|
va_list ap;
|
|
va_start(ap, fmt);
|
|
AthenaLog.report(logvisor::Level(level), fmt, ap);
|
|
va_end(ap);
|
|
}
|
|
|
|
static hecl::SystemChar cwdbuf[1024];
|
|
hecl::SystemString ExeDir;
|
|
|
|
#if _WIN32
|
|
int wmain(int argc, const wchar_t** argv)
|
|
#else
|
|
int main(int argc, const char** argv)
|
|
#endif
|
|
{
|
|
#if _WIN32
|
|
CoInitializeEx(nullptr, COINIT_MULTITHREADED | COINIT_DISABLE_OLE1DDE);
|
|
#else
|
|
std::setlocale(LC_ALL, "en-US.UTF-8");
|
|
#endif
|
|
|
|
/* Xterm check */
|
|
#if _WIN32
|
|
const char* conemuANSI = getenv("ConEmuANSI");
|
|
if (conemuANSI && !strcmp(conemuANSI, "ON"))
|
|
XTERM_COLOR = true;
|
|
#else
|
|
const char* term = getenv("TERM");
|
|
if (term && !strncmp(term, "xterm", 5))
|
|
XTERM_COLOR = true;
|
|
signal(SIGWINCH, SIGWINCHHandler);
|
|
#endif
|
|
signal(SIGINT, SIGINTHandler);
|
|
|
|
logvisor::RegisterStandardExceptions();
|
|
logvisor::RegisterConsoleLogger();
|
|
atSetExceptionHandler(AthenaExc);
|
|
|
|
/* Basic usage check */
|
|
if (argc == 1)
|
|
{
|
|
printHelp(argv[0]);
|
|
#if WIN_PAUSE
|
|
system("PAUSE");
|
|
#endif
|
|
return 0;
|
|
}
|
|
else if (argc == 0)
|
|
{
|
|
printHelp(_S("hecl"));
|
|
#if WIN_PAUSE
|
|
system("PAUSE");
|
|
#endif
|
|
return 0;
|
|
}
|
|
|
|
/* Prepare DataSpecs */
|
|
HECLRegisterDataSpecs();
|
|
|
|
/* Assemble common tool pass info */
|
|
ToolPassInfo info;
|
|
info.pname = argv[0];
|
|
if (hecl::Getcwd(cwdbuf, 1024))
|
|
{
|
|
info.cwd = cwdbuf;
|
|
if (info.cwd.size() && info.cwd.back() != _S('/') && info.cwd.back() != _S('\\'))
|
|
#if _WIN32
|
|
info.cwd += _S('\\');
|
|
#else
|
|
info.cwd += _S('/');
|
|
#endif
|
|
|
|
if (hecl::PathRelative(argv[0]))
|
|
ExeDir = hecl::SystemString(cwdbuf) + _S('/');
|
|
hecl::SystemString Argv0(argv[0]);
|
|
hecl::SystemString::size_type lastIdx = Argv0.find_last_of(_S("/\\"));
|
|
if (lastIdx != hecl::SystemString::npos)
|
|
ExeDir.insert(ExeDir.end(), Argv0.begin(), Argv0.begin() + lastIdx);
|
|
}
|
|
|
|
/* Concatenate args */
|
|
std::vector<hecl::SystemString> args;
|
|
args.reserve(argc-2);
|
|
for (int i=2 ; i<argc ; ++i)
|
|
args.push_back(hecl::SystemString(argv[i]));
|
|
|
|
if (!args.empty())
|
|
{
|
|
/* Extract output argument */
|
|
for (auto it = args.cbegin() ; it != args.cend() ;)
|
|
{
|
|
const hecl::SystemString& arg = *it;
|
|
hecl::SystemRegexMatch oMatch;
|
|
if (std::regex_search(arg, oMatch, regOPEN))
|
|
{
|
|
const hecl::SystemString& token = oMatch[1].str();
|
|
if (token.size())
|
|
{
|
|
if (info.output.empty())
|
|
info.output = oMatch[1].str();
|
|
it = args.erase(it);
|
|
}
|
|
else
|
|
{
|
|
it = args.erase(it);
|
|
if (it == args.end())
|
|
break;
|
|
if (info.output.empty())
|
|
info.output = *it;
|
|
it = args.erase(it);
|
|
}
|
|
continue;
|
|
}
|
|
++it;
|
|
}
|
|
|
|
/* Iterate flags */
|
|
for (auto it = args.cbegin() ; it != args.cend() ;)
|
|
{
|
|
const hecl::SystemString& arg = *it;
|
|
if (arg.size() < 2 || arg[0] != _S('-') || arg[1] == _S('-'))
|
|
{
|
|
++it;
|
|
continue;
|
|
}
|
|
|
|
for (auto chit = arg.cbegin() + 1 ; chit != arg.cend() ; ++chit)
|
|
{
|
|
if (*chit == _S('v'))
|
|
++info.verbosityLevel;
|
|
else if (*chit == _S('f'))
|
|
info.force = true;
|
|
else if (*chit == _S('y'))
|
|
info.yes = true;
|
|
else
|
|
info.flags.push_back(*chit);
|
|
}
|
|
|
|
it = args.erase(it);
|
|
}
|
|
|
|
/* Gather remaining args */
|
|
info.args.reserve(args.size());
|
|
for (const hecl::SystemString& arg : args)
|
|
info.args.push_back(arg);
|
|
}
|
|
|
|
/* Attempt to find hecl project */
|
|
hecl::ProjectRootPath rootPath = hecl::SearchForProject(info.cwd);
|
|
std::unique_ptr<hecl::Database::Project> project;
|
|
if (rootPath)
|
|
{
|
|
size_t ErrorRef = logvisor::ErrorCount;
|
|
hecl::Database::Project* newProj = new hecl::Database::Project(rootPath);
|
|
if (logvisor::ErrorCount > ErrorRef)
|
|
{
|
|
#if WIN_PAUSE
|
|
system("PAUSE");
|
|
#endif
|
|
delete newProj;
|
|
return -1;
|
|
}
|
|
project.reset(newProj);
|
|
info.project = newProj;
|
|
}
|
|
|
|
/* Construct selected tool */
|
|
hecl::SystemString toolName(argv[1]);
|
|
hecl::ToLower(toolName);
|
|
std::unique_ptr<ToolBase> tool;
|
|
|
|
size_t ErrorRef = logvisor::ErrorCount;
|
|
if (toolName == _S("init"))
|
|
tool.reset(new ToolInit(info));
|
|
else if (toolName == _S("spec"))
|
|
tool.reset(new ToolSpec(info));
|
|
else if (toolName == _S("extract"))
|
|
tool.reset(new ToolExtract(info));
|
|
else if (toolName == _S("add"))
|
|
tool.reset(new ToolAdd(info));
|
|
else if (toolName == _S("remove") || toolName == _S("rm"))
|
|
tool.reset(new ToolRemove(info));
|
|
else if (toolName == _S("group"))
|
|
tool.reset(new ToolGroup(info));
|
|
else if (toolName == _S("cook"))
|
|
tool.reset(new ToolCook(info));
|
|
else if (toolName == _S("clean"))
|
|
tool.reset(new ToolClean(info));
|
|
else if (toolName == _S("package") || toolName == _S("pack"))
|
|
tool.reset(new ToolPackage(info));
|
|
else if (toolName == _S("help"))
|
|
tool.reset(new ToolHelp(info));
|
|
else
|
|
{
|
|
FILE* fp = hecl::Fopen(argv[1], _S("rb"));
|
|
if (!fp)
|
|
LogModule.report(logvisor::Error, _S("unrecognized tool '%s'"), toolName.c_str());
|
|
else
|
|
{
|
|
/* Shortcut-case: implicit extract */
|
|
fclose(fp);
|
|
info.args.insert(info.args.begin(), argv[1]);
|
|
tool.reset(new ToolExtract(info));
|
|
}
|
|
}
|
|
|
|
if (logvisor::ErrorCount > ErrorRef)
|
|
{
|
|
#if WIN_PAUSE
|
|
system("PAUSE");
|
|
#endif
|
|
return -1;
|
|
}
|
|
|
|
if (info.verbosityLevel)
|
|
LogModule.report(logvisor::Info, _S("Constructed tool '%s' %d\n"),
|
|
tool->toolName().c_str(), info.verbosityLevel);
|
|
|
|
/* Run tool */
|
|
ErrorRef = logvisor::ErrorCount;
|
|
int retval = tool->run();
|
|
if (logvisor::ErrorCount > ErrorRef)
|
|
{
|
|
hecl::BlenderConnection::Shutdown();
|
|
#if WIN_PAUSE
|
|
system("PAUSE");
|
|
#endif
|
|
return -1;
|
|
}
|
|
|
|
hecl::BlenderConnection::Shutdown();
|
|
#if WIN_PAUSE
|
|
system("PAUSE");
|
|
#endif
|
|
return retval;
|
|
}
|
|
|
|
|