diff options
Diffstat (limited to 'apps/generate.cpp')
| -rw-r--r-- | apps/generate.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/apps/generate.cpp b/apps/generate.cpp index f0bf0fd..ca6f8de 100644 --- a/apps/generate.cpp +++ b/apps/generate.cpp @@ -18,7 +18,9 @@ int main(int argc, char *argv[]) { } out << "#pragma once" << endl + << endl << "#include <string>" << endl + << "#include <boost/algorithm/string/predicate.hpp>" << endl << endl; vector<pair<string, string>> apps; @@ -45,7 +47,7 @@ int main(int argc, char *argv[]) { bool first = true; out << "template<typename App>" << endl - << "int launch_app(int argc, const char *argv[]);" + << "int launch_app(int argc, const char *argv[]);" << endl << endl; out << "int launch(const std::string app_name, int argc, const char *argv[]) {" << endl; @@ -58,7 +60,7 @@ int main(int argc, char *argv[]) { first = false; } - out << "if (app_name == \"" << pair.first << "\") {" << endl + out << "if (boost::ends_with(app_name, \"" << pair.first << "\")) {" << endl << " return launch_app<" << pair.second << ">(argc, argv);" << endl; }); |