kepka/Telegram/SourceFiles/codegen/numbers/generator.cpp

75 lines
2.3 KiB
C++

/*
This file is part of Telegram Desktop,
the official desktop version of Telegram messaging app, see https://telegram.org
Telegram Desktop is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
It 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 General Public License for more details.
In addition, as a special exception, the copyright holders give permission
to link the code of portions of this program with the OpenSSL library.
Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
Copyright (c) 2014-2017 John Preston, https://desktop.telegram.org
*/
#include "codegen/numbers/generator.h"
#include <QtCore/QDir>
#include <QtCore/QSet>
#include <functional>
namespace codegen {
namespace numbers {
namespace {} // namespace
Generator::Generator(const Rules &rules, const QString &destBasePath, const common::ProjectInfo &project)
: rules_(rules)
, basePath_(destBasePath)
, project_(project) {}
bool Generator::writeHeader() {
header_ = std::make_unique<common::CppFile>(basePath_ + ".h", project_);
header_->include("QString", true);
header_->stream() << "QVector<int> phoneNumberParse(const QString &number);\n";
return header_->finalize();
}
bool Generator::writeSource() {
source_ = std::make_unique<common::CppFile>(basePath_ + ".cpp", project_);
source_->include("map", true);
source_->include("QVector", true);
source_->stream() << R"code(const std::map<QString, QVector<int>> RulesMap = {
)code";
for (auto rule = rules_.data.cbegin(), e = rules_.data.cend(); rule != e; ++rule) {
auto k = rule.key();
source_->stream() << "{\"" << k << "\",{";
for (auto &c : rule.key()) {
source_->stream() << c.toLatin1() << ",";
}
source_->stream() << "}},\n";
}
source_->stream() << "\
};";
source_->stream() << R"code(
QVector<int> phoneNumberParse(const QString &number) {
return RulesMap.find(number) != RulesMap.end() ? RulesMap.at(number) : QVector<int>();
}
)code";
return source_->finalize();
}
} // namespace numbers
} // namespace codegen