/*

Facebook plugin for Miranda Instant Messenger
_____________________________________________

Copyright © 2009-11 Michal Zelinka, 2011-13 Robert Pösel

This program 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 2 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

*/

#include "common.h"
#include "JSON_CAJUN/reader.h"
#include "JSON_CAJUN/writer.h"
#include "JSON_CAJUN/elements.h"

int facebook_json_parser::parse_buddy_list(void* data, List::List< facebook_user >* buddy_list)
{
	using namespace json;

	try
	{
		facebook_user* current = NULL;
		std::string buddyData = static_cast< std::string* >(data)->substr(9);
		std::istringstream sDocument(buddyData);
		Object objDocument;
		Reader::Read(objDocument, sDocument);

		const Object& objRoot = objDocument;
/*		const Array& wasAvailableIDs = objRoot["payload"]["buddy_list"]["wasAvailableIDs"];

		for (Array::const_iterator itWasAvailable(wasAvailableIDs.Begin());
			itWasAvailable != wasAvailableIDs.End(); ++itWasAvailable)
		{
			const Number& member = *itWasAvailable;
			char was_id[32];
			lltoa(member.Value(), was_id, 10);

			current = buddy_list->find(std::string(was_id));
			if (current != NULL)
				current->status_id = ID_STATUS_OFFLINE;
		}*/ // Facebook removed support for "wasAvailableIDs"

		// Set all contacts in map to offline
		for (List::Item< facebook_user >* i = buddy_list->begin(); i != NULL; i = i->next) {
			i->data->status_id = ID_STATUS_OFFLINE;
		}

		// Load last active times
		const Object& lastActive = objRoot["payload"]["buddy_list"]["last_active_times"];
		for (Object::const_iterator itLastActive(lastActive.Begin()); itLastActive != lastActive.End(); ++itLastActive)
		{
			const Object::Member& member = *itLastActive;

			current = buddy_list->find(member.name);
			if (current == NULL) {
				buddy_list->insert(std::make_pair(member.name, new facebook_user()));
				current = buddy_list->find(member.name);
				current->user_id = member.name;
			}

			const Number& timestamp = member.element;
			current->last_active = timestamp.Value();
		}
		
		// Find mobile friends
		const Array& mobileFriends = objRoot["payload"]["buddy_list"]["mobile_friends"];
		for (Array::const_iterator buddy(mobileFriends.Begin()); buddy != mobileFriends.End(); ++buddy) {
			const Number& member = *buddy;
			char was_id[32];
			lltoa(member.Value(), was_id, 10);

			std::string id = was_id;
			if (!id.empty()) {
				current = buddy_list->find(id);
									
				if (current == NULL) {
					buddy_list->insert(std::make_pair(id, new facebook_user()));
					current = buddy_list->find(id);
					current->user_id = id;
				}

				current->status_id = ID_STATUS_ONTHEPHONE;
			}
		}

		// Find now awailable contacts
		const Object& nowAvailableList = objRoot["payload"]["buddy_list"]["nowAvailableList"];
		for (Object::const_iterator itAvailable(nowAvailableList.Begin());
			itAvailable != nowAvailableList.End(); ++itAvailable)
		{
			const Object::Member& member = *itAvailable;
			const Object& objMember = member.element;
			const Boolean idle = objMember["i"]; // In new version of Facebook "i" means "offline"

			current = buddy_list->find(member.name);
			if (current == NULL) {
				if (idle) continue; // Just little optimalization

				buddy_list->insert(std::make_pair(member.name, new facebook_user()));
				current = buddy_list->find(member.name);
				current->user_id = current->real_name = member.name;	
			}

			current->status_id = (idle ? ID_STATUS_OFFLINE : ID_STATUS_ONLINE);
		}

		// Get aditional informations about contacts (if available)
		const Object& userInfosList = objRoot["payload"]["buddy_list"]["userInfos"];
		for (Object::const_iterator itUserInfo(userInfosList.Begin());
			itUserInfo != userInfosList.End(); ++itUserInfo)
		{
			const Object::Member& member = *itUserInfo;

			current = buddy_list->find(member.name);
			if (current == NULL)
				continue;

			const Object& objMember = member.element;
			const String& realName = objMember["name"];
			const String& imageUrl = objMember["thumbSrc"];

			current->real_name = utils::text::slashu_to_utf8(
			    utils::text::special_expressions_decode(realName.Value()));
			current->image_url = utils::text::slashu_to_utf8(
			    utils::text::special_expressions_decode(imageUrl.Value()));
		}

	}
	catch (Reader::ParseException& e)
	{
		proto->Log("!!!!! Caught json::ParseException: %s", e.what());
		proto->Log("      Line/offset: %d/%d", e.m_locTokenBegin.m_nLine + 1, e.m_locTokenBegin.m_nLineOffset + 1);
	}
	catch (const Exception& e)
	{
		proto->Log("!!!!! Caught json::Exception: %s", e.what());
	}
	catch (const std::exception& e)
	{
		proto->Log("!!!!! Caught std::exception: %s", e.what());
	}

	return EXIT_SUCCESS;
}

int facebook_json_parser::parse_friends(void* data, std::map< std::string, facebook_user* >* friends)
{
	using namespace json;

	try
	{
		std::string buddyData = static_cast< std::string* >(data)->substr(9);
		std::istringstream sDocument(buddyData);
		Object objDocument;
		Reader::Read(objDocument, sDocument);

		const Object& objRoot = objDocument;
		const Object& payload = objRoot["payload"];

		for (Object::const_iterator payload_item(payload.Begin()); payload_item != payload.End(); ++payload_item)
		{
			const Object::Member& member = *payload_item;

			const Object& objMember = member.element;

			const String& realName = objMember["name"];
			const String& imageUrl = objMember["thumbSrc"];
			//const String& vanity = objMember["vanity"];
			//const String& uri = objMember["uri"];
			const Number& gender = objMember["gender"];
			//const Boolean& isFriend = objMember["is_friend"];
			//const String& type = objMember["type"]; // "friend" = existing classic contact, "user" = contact with disabled/deleted account
			
			facebook_user *fbu = new facebook_user();

			fbu->user_id = member.name;
			fbu->real_name = utils::text::slashu_to_utf8(
			    utils::text::special_expressions_decode(realName.Value()));
			fbu->image_url = utils::text::slashu_to_utf8(
			    utils::text::special_expressions_decode(imageUrl.Value()));

			if (gender.Value() == 1) {
				fbu->gender = 70; // female
			} else if (gender.Value() == 2) {
				fbu->gender = 77; // male
			}

			friends->insert(std::make_pair(member.name, fbu));
		}
	}
	catch (Reader::ParseException& e)
	{
		proto->Log("!!!!! Caught json::ParseException: %s", e.what());
		proto->Log("      Line/offset: %d/%d", e.m_locTokenBegin.m_nLine + 1, e.m_locTokenBegin.m_nLineOffset + 1);
	}
	catch (const Exception& e)
	{
		proto->Log("!!!!! Caught json::Exception: %s", e.what());
	}
	catch (const std::exception& e)
	{
		proto->Log("!!!!! Caught std::exception: %s", e.what());
	}

	return EXIT_SUCCESS;
}


int facebook_json_parser::parse_notifications(void *data, std::vector< facebook_notification* > *notifications) 
{
	using namespace json;

	try
	{
		std::string notificationsData = static_cast< std::string* >(data)->substr(9);
		std::istringstream sDocument(notificationsData);
		Object objDocument;
		Reader::Read(objDocument, sDocument);

		const Object& objRoot = objDocument;
		const Object& payload = objRoot["payload"]["notifications"];

		for (Object::const_iterator payload_item(payload.Begin()); payload_item != payload.End(); ++payload_item)
		{
			const Object::Member& member = *payload_item;
			
			const Object& objMember = member.element;
			const String& content = objMember["markup"];
			const Number& unread = objMember["unread"];
			
			if (unread.Value() == 0) // ignore old notifications
				continue;
			
			std::string text = utils::text::slashu_to_utf8(
								utils::text::special_expressions_decode(content.Value()));

			facebook_notification* notification = new facebook_notification();
			
			notification->text = utils::text::remove_html(utils::text::source_get_value(&text, 1, "<abbr"));
			notification->link = utils::text::source_get_value(&text, 3, "<a ", "href=\"", "\"");
			notification->id = member.name;

			notifications->push_back(notification);
		}

	}
	catch (Reader::ParseException& e)
	{
		proto->Log("!!!!! Caught json::ParseException: %s", e.what());
		proto->Log("      Line/offset: %d/%d", e.m_locTokenBegin.m_nLine + 1, e.m_locTokenBegin.m_nLineOffset + 1);
	}
	catch (const Exception& e)
	{
		proto->Log("!!!!! Caught json::Exception: %s", e.what());
	}
	catch (const std::exception& e)
	{
		proto->Log("!!!!! Caught std::exception: %s", e.what());
	}

	return EXIT_SUCCESS;
}

bool ignore_duplicits(FacebookProto *proto, std::string mid, std::string text) {
	std::map<std::string, bool>::iterator it = proto->facy.messages_ignore.find(mid);
	if (it != proto->facy.messages_ignore.end()) {
		std::string msg = "????? Ignoring duplicit/sent message\n" + text;
		proto->Log(msg.c_str());

		it->second = true; // mark to delete it at the end
		return true;
	}
					
	// remember this id to ignore duplicits
	proto->facy.messages_ignore.insert(std::make_pair(mid, true));
	return false;
}

int facebook_json_parser::parse_messages(void* data, std::vector< facebook_message* >* messages, std::vector< facebook_notification* >* notifications)
{
	using namespace json;

	try
	{
		std::string messageData = static_cast< std::string* >(data)->substr(9);
		std::istringstream sDocument(messageData);
		Object objDocument;
		Reader::Read(objDocument, sDocument);

		const Object& objRoot = objDocument;
		const Array& messagesArray = objRoot["ms"];

		for (Array::const_iterator itMessage(messagesArray.Begin()); itMessage != messagesArray.End(); ++itMessage)
		{
			const Object& objMember = *itMessage;

			const String& type = objMember["type"];

			if (type.Value() == "msg" || type.Value() == "offline_msg") // direct message
			{
				const Number& from = objMember["from"];
				char was_id[32];
				lltoa(from.Value(), was_id, 10);
				
				const Object& messageContent = objMember["msg"];
				const String& text = messageContent["text"];
				const String& message_id = messageContent["messageId"];
				//"tab_type":"friend",     objMember["tab_type"]
        
				const Number& time_sent = messageContent["time"];

				// ignore duplicits or messages sent from miranda
				if (ignore_duplicits(proto, message_id.Value(), text.Value()))
					continue;

				std::string message_text = utils::text::trim(utils::text::special_expressions_decode(utils::text::slashu_to_utf8(text.Value())), true);
				if (message_text.empty())
					continue;

				if (was_id == proto->facy.self_.user_id) {
					const Number& to = objMember["to"];
					char to_id[32];
					lltoa(to.Value(), to_id, 10);

					HANDLE hContact = proto->ContactIDToHContact(to_id);
					if (!hContact)
						continue;

					DBEVENTINFO dbei = {0};
					dbei.cbSize = sizeof(dbei);
					dbei.eventType = EVENTTYPE_MESSAGE;
					dbei.flags = DBEF_SENT | DBEF_UTF;
					dbei.szModule = proto->m_szModuleName;

					bool local_time = db_get_b(NULL, proto->m_szModuleName, FACEBOOK_KEY_LOCAL_TIMESTAMP, 0) != 0;
					dbei.timestamp = local_time ? ::time(NULL) : utils::time::fix_timestamp(time_sent.Value());

					dbei.cbBlob = (DWORD)message_text.length() + 1;
					dbei.pBlob = (PBYTE)message_text.c_str();
					db_event_add(hContact, &dbei);

					continue;
				}

				if ((messageContent.Find("truncated") != messageContent.End())
					&& (((const Number &)messageContent["truncated"]).Value() == 1)) {
					// If we got truncated message, we can ignore it, because we should get it again as "messaging" type
					std::string msg = "????? We got truncated message so we ignore it\n";
					msg += utils::text::special_expressions_decode(utils::text::slashu_to_utf8(text.Value()));
					proto->Log(msg.c_str());
				} else {
  					facebook_message* message = new facebook_message();
					message->message_text = message_text;
					message->time = utils::time::fix_timestamp(time_sent.Value());
					message->user_id = was_id;
					message->message_id = message_id;

					messages->push_back(message);
				}
			}
			else if (type.Value() == "messaging") 
			{				
				const String& type = objMember["event"];

				if (type.Value() == "read_receipt") {
					// user read message
					const Number& reader = objMember["reader"];
					const Number& time = objMember["time"];

					char user_id[32];
					lltoa(reader.Value(), user_id, 10);

					// TODO: add check for chat contacts
					HANDLE hContact = proto->ContactIDToHContact(user_id);
					if (hContact) {
						TCHAR ttime[64], tstr[100];
						_tcsftime(ttime, SIZEOF(ttime), _T("%X"), utils::conversion::fbtime_to_timeinfo(time.Value()));
						mir_sntprintf(tstr, SIZEOF(tstr), TranslateT("Message read: %s"), ttime);

						CallService(MS_MSG_SETSTATUSTEXT, (WPARAM)hContact, (LPARAM)tstr);
					}

				} else if (type.Value() == "deliver") {
					// inbox message (multiuser or direct)
					const Object& messageContent = objMember["message"];

					const Number& sender_fbid = messageContent["sender_fbid"];
					const String& sender_name = messageContent["sender_name"];
					const String& text = messageContent["body"];
					const String& tid = messageContent["tid"];
					const String& mid = messageContent["mid"];
					const Number& time_sent = messageContent["timestamp"];

					char was_id[32];
					lltoa(sender_fbid.Value(), was_id, 10);

					// ignore duplicits or messages sent from miranda
					if (ignore_duplicits(proto, mid.Value(), text.Value()))
						continue;

					// Ignore messages from myself
					if (was_id == proto->facy.self_.user_id)
						continue;

					std::string message_text = utils::text::trim(utils::text::special_expressions_decode(utils::text::slashu_to_utf8(text.Value())), true);
					if (message_text.empty())
						continue;

					facebook_message* message = new facebook_message();
					message->message_text = message_text;
					message->sender_name = utils::text::special_expressions_decode(utils::text::slashu_to_utf8(sender_name.Value()));
					message->time = utils::time::fix_timestamp(time_sent.Value());
					message->user_id = was_id; // TODO: Check if we have contact with this ID in friendlist and otherwise do something different?
					message->message_id = mid.Value();

					messages->push_back(message);
				}
			}
			else if (type.Value() == "thread_msg") // multiuser message
			{
				const String& from_name = objMember["from_name"];
				const String& to_name = objMember["to_name"]["__html"];
				const String& to_id = objMember["to"];

				const Number& from = objMember["from"];
				char was_id[32];
				lltoa(from.Value(), was_id, 10);

				const Object& messageContent = objMember["msg"];
  				const String& text = messageContent["text"];
				const String& mid = messageContent["messageId"];

				// ignore duplicits or messages sent from miranda
				if (ignore_duplicits(proto, mid.Value(), text.Value()))
					continue;

				// Ignore messages from myself
				if (was_id == proto->facy.self_.user_id)
					continue;

				std::string message_text = utils::text::trim(utils::text::special_expressions_decode(utils::text::slashu_to_utf8(text.Value())), true);
				if (message_text.empty())
					continue;

				std::string title = utils::text::special_expressions_decode(utils::text::slashu_to_utf8(to_name.Value()));
				std::string url = "/?action=read&sk=inbox&page&query&tid=" + to_id.Value();
				std::string popup_text = utils::text::special_expressions_decode(utils::text::slashu_to_utf8(from_name.Value()));
				popup_text += ": " + message_text;

				proto->Log("      Got multichat message");
		    
				TCHAR* szTitle = mir_utf8decodeT(title.c_str());
				TCHAR* szText = mir_utf8decodeT(popup_text.c_str());
				proto->NotifyEvent(szTitle, szText, NULL, FACEBOOK_EVENT_OTHER, &url);
				mir_free(szTitle);
				mir_free(szText);
			}
			else if (type.Value() == "notification_json") // event notification
			{
				if (!db_get_b(NULL, proto->m_szModuleName, FACEBOOK_KEY_EVENT_NOTIFICATIONS_ENABLE, DEFAULT_EVENT_NOTIFICATIONS_ENABLE))
					continue;

				const Array& notificationsArray = objMember["nodes"];

				for (Array::const_iterator itNotification(notificationsArray.Begin()); itNotification != notificationsArray.End(); ++itNotification)
				{
					const Object& objNotification = *itNotification;

					//const String& text = objNotification["title"]["text"];
					const String& text = objNotification["unaggregatedTitle"]["text"];
					const String& link = objNotification["url"];
					const String& id = objNotification["alert_id"];

					const Number& time_sent = objNotification["timestamp"]["time"];        
					if (time_sent.Value() > proto->facy.last_notification_time_) // Check against duplicit notifications
					{
						proto->facy.last_notification_time_ = time_sent.Value();

						facebook_notification* notification = new facebook_notification();
						notification->text = utils::text::slashu_to_utf8(text.Value());
  						notification->link = utils::text::special_expressions_decode(link.Value());
						
						std::string::size_type pos = id.Value().find(":");
						notification->id = (pos != std::string::npos) ? id.Value().substr(pos+1) : id.Value();

						notifications->push_back(notification);
					}
				}
			}
			else if (type.Value() == "typ") // chat typing notification
			{
				const Number& from = objMember["from"];
				char user_id[32];
				lltoa(from.Value(), user_id, 10);

				facebook_user fbu;
				fbu.user_id = user_id;

				HANDLE hContact = proto->AddToContactList(&fbu, FACEBOOK_CONTACT_FRIEND);
				
				if (db_get_w(hContact, proto->m_szModuleName, "Status", 0) == ID_STATUS_OFFLINE)
					db_set_w(hContact, proto->m_szModuleName, "Status", ID_STATUS_ONLINE);

				const Number& state = objMember["st"];
				if (state.Value() == 1)
					CallService(MS_PROTO_CONTACTISTYPING, (WPARAM)hContact, (LPARAM)60);
				else
					CallService(MS_PROTO_CONTACTISTYPING, (WPARAM)hContact, (LPARAM)PROTOTYPE_CONTACTTYPING_OFF);
			}
			else if (type.Value() == "privacy_changed")
			{
				const String& event_type = objMember["event"];
				const Object& event_data = objMember["data"];

				if (event_type.Value() == "visibility_update")
				{ // change of chat status
					const Boolean visibility = event_data["visibility"];
					proto->Log("      Requested chat switch to %s", visibility ? "Online" : "Offline");
					proto->SetStatus(visibility ? ID_STATUS_ONLINE : ID_STATUS_INVISIBLE);
				}				
			}
			else
				continue;
		}

		// remove received messages from map
		for (std::map<std::string, bool>::iterator it = proto->facy.messages_ignore.begin(); it != proto->facy.messages_ignore.end(); ++it) {
			if (it->second)
				proto->facy.messages_ignore.erase(it);
		}
	}
	catch (Reader::ParseException& e)
	{
		proto->Log("!!!!! Caught json::ParseException: %s", e.what());
		proto->Log("      Line/offset: %d/%d", e.m_locTokenBegin.m_nLine + 1, e.m_locTokenBegin.m_nLineOffset + 1);
	}
	catch (const Exception& e)
	{
		proto->Log ("!!!!! Caught json::Exception: %s", e.what());
	}

	return EXIT_SUCCESS;
}