From 6717e588e71d0586613053521b2dc591405410ba Mon Sep 17 00:00:00 2001
From: Oleg Morozenkov <omorozenkov@gmail.com>
Date: Sun, 5 Jul 2015 01:17:15 +0300
Subject: First version

---
 src/tgbot/Parser.cpp | 485 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 485 insertions(+)
 create mode 100644 src/tgbot/Parser.cpp

(limited to 'src/tgbot/Parser.cpp')

diff --git a/src/tgbot/Parser.cpp b/src/tgbot/Parser.cpp
new file mode 100644
index 0000000..3730607
--- /dev/null
+++ b/src/tgbot/Parser.cpp
@@ -0,0 +1,485 @@
+/*
+ * Copyright (c) 2015 Oleg Morozenkov
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software", "to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+#include "Parser.h"
+
+using namespace std;
+using namespace boost::property_tree;
+
+namespace TgBot {
+
+User::Ptr Parser::parseUser(const ptree& data) const {
+    User::Ptr result(new User);
+    result->id = data.get<int32_t>("id");
+    result->firstName = data.get<string>("first_name");
+    result->lastName = data.get("last_name", "");
+    result->username = data.get("username", "");
+    return result;
+}
+
+string Parser::parseUser(const User::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "id", object->id);
+    appendToJson(result, "first_name", object->firstName);
+    appendToJson(result, "last_name", object->lastName);
+    appendToJson(result, "username", object->username);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+GroupChat::Ptr Parser::parseGroupChat(const ptree& data) const {
+    GroupChat::Ptr result(new GroupChat);
+    result->id = data.get<int32_t>("id");
+    result->title = data.get<string>("title");
+    return result;
+}
+
+string Parser::parseGroupChat(const GroupChat::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "id", object->id);
+    appendToJson(result, "title", object->title);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+Message::Ptr Parser::parseMessage(const ptree& data) const {
+    Message::Ptr result(new Message);
+    result->messageId = data.get<int32_t>("message_id");
+    result->from = parseUser(data.find("from")->second);
+    result->date = data.get<int32_t>("date");
+    result->chat = parseGenericChat(data.find("chat")->second);
+    result->forwardFrom = tryParse<User>(parseUser, data, "forward_from");
+    result->forwardDate = data.get("forward_date", 0);
+    result->replyToMessage = tryParse<Message>(parseMessage, data, "reply_to_message");
+    result->text = data.get("text", "");
+    result->audio = tryParse<Audio>(parseAudio, data, "audio");
+    result->document = tryParse<Document>(parseDocument, data, "document");
+    result->photo = parseArray<PhotoSize>(parsePhotoSize, data, "photo");
+    result->sticker = tryParse<Sticker>(parseSticker, data, "sticker");
+    result->video = tryParse<Video>(parseVideo, data, "video");
+    result->contact = tryParse<Contact>(parseContact, data, "contact");
+    result->location = tryParse<Location>(parseLocation, data, "location");
+    result->newChatParticipant = tryParse<User>(parseUser, data, "new_chat_participant");
+    result->leftChatParticipant = tryParse<User>(parseUser, data, "left_chat_participant");
+    result->newChatTitle = data.get("new_chat_title", "");
+    result->newChatPhoto = parseArray<PhotoSize>(parsePhotoSize, data, "new_chat_photo");
+    result->deleteChatPhoto = data.get("delete_chat_photo", false);
+    result->groupChatCreated = data.get("group_chat_created", false);
+    return result;
+}
+
+string Parser::parseMessage(const Message::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "message_id", object->messageId);
+    appendToJson(result, "from", parseUser(object->from));
+    appendToJson(result, "date", object->date);
+    appendToJson(result, "chat", parseGenericChat(object->chat));
+    appendToJson(result, "forward_from", parseUser(object->forwardFrom));
+    appendToJson(result, "forward_date", object->forwardDate);
+    appendToJson(result, "reply_to_message", parseMessage(object->replyToMessage));
+    appendToJson(result, "text", object->text);
+    appendToJson(result, "audio", parseAudio(object->audio));
+    appendToJson(result, "document", parseDocument(object->document));
+    appendToJson(result, "photo", parseArray(parsePhotoSize, object->photo));
+    appendToJson(result, "sticker", parseSticker(object->sticker));
+    appendToJson(result, "video", parseVideo(object->video));
+    appendToJson(result, "contact", parseContact(object->contact));
+    appendToJson(result, "location", parseLocation(object->location));
+    appendToJson(result, "new_chat_participant", parseUser(object->newChatParticipant));
+    appendToJson(result, "left_chat_participant", parseUser(object->leftChatParticipant));
+    appendToJson(result, "new_chat_title", object->newChatTitle);
+    appendToJson(result, "new_chat_photo", parseArray(parsePhotoSize, object->newChatPhoto));
+    appendToJson(result, "delete_chat_photo", object->deleteChatPhoto);
+    appendToJson(result, "group_chat_created", object->groupChatCreated);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+PhotoSize::Ptr Parser::parsePhotoSize(const ptree& data) const {
+    PhotoSize::Ptr result(new PhotoSize);
+    result->fileId = data.get<string>("file_id");
+    result->width = data.get<int32_t>("width");
+    result->height = data.get<int32_t>("height");
+    result->fileSize = data.get("file_size", 0);
+    return result;
+}
+
+string Parser::parsePhotoSize(const PhotoSize::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "file_id", object->fileId);
+    appendToJson(result, "width", object->width);
+    appendToJson(result, "height", object->height);
+    appendToJson(result, "file_size", object->fileSize);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+Audio::Ptr Parser::parseAudio(const ptree& data) const {
+    Audio::Ptr result(new Audio);
+    result->fileId = data.get<string>("file_id");
+    result->duration = data.get<int32_t>("duration");
+    result->mimeType = data.get("mime_type", "");
+    result->fileSize = data.get("file_size", 0);
+    return result;
+}
+
+string Parser::parseAudio(const Audio::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "file_id", object->fileId);
+    appendToJson(result, "duration", object->duration);
+    appendToJson(result, "mime_type", object->mimeType);
+    appendToJson(result, "file_size", object->fileSize);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+Document::Ptr Parser::parseDocument(const ptree& data) const {
+    Document::Ptr result(new Document);
+    result->fileId = data.get<string>("file_id");
+    result->thumb = parsePhotoSize(data.find("thumb")->second);
+    result->fileName = data.get("file_name", "");
+    result->mimeType = data.get("mime_type", "");
+    result->fileSize = data.get("file_size", 0);
+    return result;
+}
+
+string Parser::parseDocument(const Document::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "file_id", object->fileId);
+    appendToJson(result, "thumb", parsePhotoSize(object->thumb));
+    appendToJson(result, "file_name", object->fileName);
+    appendToJson(result, "mime_type", object->mimeType);
+    appendToJson(result, "file_size", object->fileSize);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+Sticker::Ptr Parser::parseSticker(const ptree& data) const {
+    Sticker::Ptr result(new Sticker);
+    result->fileId = data.get<string>("file_id");
+    result->width = data.get<int32_t>("width");
+    result->height = data.get<int32_t>("height");
+    result->thumb = parsePhotoSize(data.find("thumb")->second);
+    result->fileSize = data.get("file_size", 0);
+    return result;
+}
+
+string Parser::parseSticker(const Sticker::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "file_id", object->fileId);
+    appendToJson(result, "width", object->width);
+    appendToJson(result, "height", object->height);
+    appendToJson(result, "thumb", parsePhotoSize(object->thumb));
+    appendToJson(result, "file_size", object->fileSize);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+Video::Ptr Parser::parseVideo(const ptree& data) const {
+    Video::Ptr result(new Video);
+    result->fileId = data.get<string>("file_id");
+    result->width = data.get<int32_t>("width");
+    result->height = data.get<int32_t>("height");
+    result->duration = data.get<int32_t>("duration");
+    result->thumb = parsePhotoSize(data.find("thumb")->second);
+    result->mimeType = data.get("mime_type", "");
+    result->fileSize = data.get("file_size", 0);
+    result->caption = data.get("caption", "");
+    return result;
+}
+
+string Parser::parseVideo(const Video::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "file_id", object->fileId);
+    appendToJson(result, "width", object->width);
+    appendToJson(result, "height", object->height);
+    appendToJson(result, "duration", object->duration);
+    appendToJson(result, "thumb", parsePhotoSize(object->thumb));
+    appendToJson(result, "mime_type", object->mimeType);
+    appendToJson(result, "file_size", object->fileSize);
+    appendToJson(result, "caption", object->caption);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+Contact::Ptr Parser::parseContact(const ptree& data) const {
+    Contact::Ptr result(new Contact);
+    result->phoneNumber = data.get<string>("phone_number");
+    result->firstName = data.get<string>("first_name");
+    result->lastName = data.get("last_name", "");
+    result->userId = data.get("user_id", "");
+    return result;
+}
+
+string Parser::parseContact(const Contact::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "phone_number", object->phoneNumber);
+    appendToJson(result, "first_name", object->firstName);
+    appendToJson(result, "last_name", object->lastName);
+    appendToJson(result, "user_id", object->userId);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+Location::Ptr Parser::parseLocation(const ptree& data) const {
+    Location::Ptr result(new Location);
+    result->longitude = data.get<float>("longitude");
+    result->latitude = data.get<float>("latitude");
+    return result;
+}
+
+string Parser::parseLocation(const Location::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "longitude", object->longitude);
+    appendToJson(result, "latitude", object->latitude);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+Update::Ptr Parser::parseUpdate(const ptree& data) const {
+    Update::Ptr result(new Update);
+    result->updateId = data.get<int32_t>("update_id");
+    result->message = parseMessage(data.find("message")->second);
+    return result;
+}
+
+string Parser::parseUpdate(const Update::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "update_id", object->updateId);
+    appendToJson(result, "message", parseMessage(object->message));
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+UserProfilePhotos::Ptr Parser::parseUserProfilePhotos(const ptree& data) const {
+    UserProfilePhotos::Ptr result(new UserProfilePhotos);
+    result->totalCount = data.get<int32_t>("total_count");
+    result->photos = parse2DArray<PhotoSize>(parsePhotoSize, data, "photos");
+    return result;
+}
+
+string Parser::parseUserProfilePhotos(const UserProfilePhotos::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "total_count", object->totalCount);
+    appendToJson(result, "photos", parse2DArray(parsePhotoSize, object->photos));
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+ReplyKeyboardMarkup::Ptr Parser::parseReplyKeyboardMarkup(const boost::property_tree::ptree& data) const {
+    ReplyKeyboardMarkup::Ptr result(new ReplyKeyboardMarkup);
+    for (const pair<const string, ptree>& item : data.find("keyboard")->second) {
+        vector<string> array;
+        for (const pair<const string, ptree>& innerItem : item.second) {
+            array.push_back(innerItem.second.data());
+        }
+        result->keyboard.push_back(array);
+    }
+    result->resizeKeyboard = data.get<bool>("resize_keyboard");
+    result->oneTimeKeyboard = data.get<bool>("one_time_keyboard");
+    result->selective = data.get<bool>("selective");
+    return result;
+}
+
+std::string Parser::parseReplyKeyboardMarkup(const ReplyKeyboardMarkup::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    result += "\"keyboard\":[";
+    for (vector<string>& item : object->keyboard) {
+        result += '[';
+        for (string& innerItem : item) {
+            result += '"';
+            result += innerItem;
+            result += "\",";
+        }
+        result.erase(result.length() - 1);
+        result += "],";
+    }
+    result.erase(result.length() - 1);
+    result += "],";
+    appendToJson(result, "resize_keyboard", object->resizeKeyboard);
+    appendToJson(result, "one_time_keyboard", object->oneTimeKeyboard);
+    appendToJson(result, "selective", object->selective);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+ReplyKeyboardHide::Ptr Parser::parseReplyKeyboardHide(const boost::property_tree::ptree& data) const {
+    ReplyKeyboardHide::Ptr result(new ReplyKeyboardHide);
+    result->selective = data.get<bool>("selective");
+    return result;
+}
+
+std::string Parser::parseReplyKeyboardHide(const ReplyKeyboardHide::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "selective", object->selective);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+ForceReply::Ptr Parser::parseForceReply(const boost::property_tree::ptree& data) const {
+    ForceReply::Ptr result(new ForceReply);
+    result->selective = data.get<bool>("selective");
+    return result;
+}
+
+std::string Parser::parseForceReply(const ForceReply::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    string result;
+    result += '{';
+    appendToJson(result, "selective", object->selective);
+    result.erase(result.length() - 1);
+    result += '}';
+    return result;
+}
+
+GenericChat::Ptr Parser::parseGenericChat(const ptree& data) const {
+    if (data.find("first_name") == data.not_found()) {
+        return static_pointer_cast<GenericChat>(parseGroupChat(data));
+    } else {
+        return static_pointer_cast<GenericChat>(parseUser(data));
+    }
+}
+
+string Parser::parseGenericChat(const GenericChat::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    if (dynamic_pointer_cast<User>(object) == nullptr) {
+        return parseGroupChat(static_pointer_cast<GroupChat>(object));
+    } else {
+        return parseUser(static_pointer_cast<User>(object));
+    }
+}
+
+GenericReply::Ptr Parser::parseGenericReply(const boost::property_tree::ptree& data) const {
+    if (data.find("force_reply") != data.not_found()) {
+        return static_pointer_cast<GenericReply>(parseForceReply(data));
+    } else if (data.find("hide_keyboard") != data.not_found()) {
+        return static_pointer_cast<GenericReply>(parseReplyKeyboardHide(data));
+    } else {
+        return static_pointer_cast<GenericReply>(parseReplyKeyboardMarkup(data));
+    }
+}
+
+std::string Parser::parseGenericReply(const GenericReply::Ptr& object) const {
+    if (!object) {
+        return "";
+    }
+    if (dynamic_pointer_cast<ForceReply>(object) != nullptr) {
+        return parseForceReply(static_pointer_cast<ForceReply>(object));
+    } else if (dynamic_pointer_cast<ReplyKeyboardHide>(object) != nullptr) {
+        return parseReplyKeyboardHide(static_pointer_cast<ReplyKeyboardHide>(object));
+    } else {
+        return parseReplyKeyboardMarkup(static_pointer_cast<ReplyKeyboardMarkup>(object));
+    }
+}
+
+void Parser::appendToJson(string& json, const string& varName, const string& value) const {
+    if (value.empty()) {
+        return;
+    }
+    json += '"';
+    json += varName;
+    json += "\":";
+    if (value.front() != '{') {
+        json += '"';
+    }
+    json += value;
+    if (value.back() != '}') {
+        json += '"';
+    }
+    json += ',';
+}
+
+}
\ No newline at end of file
-- 
cgit v1.2.3