/*

Omegle plugin for Miranda Instant Messenger
_____________________________________________

Copyright © 2011-17 Robert Pösel, 2017-18 Miranda NG team

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 "stdafx.h"

void OmegleProto::UpdateChat(const wchar_t *name, const wchar_t *message, bool addtolog)
{
	// replace % to %% to not interfere with chat color codes
	std::wstring smessage = message;
	utils::text::treplace_all(&smessage, L"%", L"%%");

	GCEVENT gce = { m_szModuleName, m_tszUserName, GC_EVENT_MESSAGE };
	gce.time = ::time(nullptr);
	gce.ptszText = smessage.c_str();

	if (name == nullptr) {
		gce.iType = GC_EVENT_INFORMATION;
		name = TranslateT("Server");
		gce.bIsMe = false;
	}
	else gce.bIsMe = !mir_wstrcmp(name, this->facy.nick_);

	if (addtolog)
		gce.dwFlags |= GCEF_ADDTOLOG;

	gce.ptszNick = name;
	gce.ptszUID = gce.ptszNick;
	Chat_Event(&gce);
}

int OmegleProto::OnChatEvent(WPARAM, LPARAM lParam)
{
	GCHOOK *hook = reinterpret_cast<GCHOOK*>(lParam);

	if (mir_strcmp(hook->pszModule, m_szModuleName))
		return 0;

	switch (hook->iType)
	{
	case GC_USER_MESSAGE:
	{
		std::string text = mir_u2a_cp(hook->ptszText, CP_UTF8);

		// replace %% back to %, because chat automatically does this to sent messages
		utils::text::replace_all(&text, "%%", "%");

		if (text.empty())
			break;

		if (text.substr(0, 1) == "/")
		{ // Process commands

			std::string command = "";
			std::string params = "";

			std::string::size_type pos = 0;
			if ((pos = text.find(" ")) != std::string::npos) {
				command = text.substr(1, pos - 1);
				params = text.substr(pos + 1);
			}
			else {
				command = text.substr(1);
			}

			if (!mir_strcmpi(command.c_str(), "new"))
			{
				facy.spy_mode_ = false;
				facy.question_.clear();

				ForkThread(&OmegleProto::NewChatWorker, nullptr);
				break;
			}
			else if (!mir_strcmpi(command.c_str(), "quit"))
			{
				ForkThread(&OmegleProto::StopChatWorker, nullptr);
				break;
			}
			else if (!mir_strcmpi(command.c_str(), "spy"))
			{
				facy.spy_mode_ = true;
				facy.question_.clear();

				ForkThread(&OmegleProto::NewChatWorker, nullptr);
				break;
			}
			else if (!mir_strcmpi(command.c_str(), "ask"))
			{
				if (params.empty()) {
					// Load last question
					DBVARIANT dbv;
					if (!getU8String(OMEGLE_KEY_LAST_QUESTION, &dbv)) {
						params = dbv.pszVal;
						db_free(&dbv);
					}

					if (params.empty()) {
						UpdateChat(nullptr, TranslateT("Last question is empty."), false);
						break;
					}
				}
				else {
					// Save actual question as last question
					if (params.length() >= OMEGLE_QUESTION_MIN_LENGTH)
					{
						setU8String(OMEGLE_KEY_LAST_QUESTION, params.c_str());
					}
				}

				if (params.length() < OMEGLE_QUESTION_MIN_LENGTH)
				{
					UpdateChat(nullptr, TranslateT("Your question is too short."), false);
					break;
				}

				facy.spy_mode_ = true;
				facy.question_ = params;
				ForkThread(&OmegleProto::NewChatWorker, nullptr);
				break;
			}
			else if (!mir_strcmpi(command.c_str(), "asl"))
			{
				DBVARIANT dbv;
				if (!getU8String(OMEGLE_KEY_ASL, &dbv)) {
					text = dbv.pszVal;
					db_free(&dbv);

					SendChatMessage(text);
				}
				else {
					UpdateChat(nullptr, TranslateT("Your '/asl' setting is empty."), false);
					break;
				}
			}
			else if (!mir_strcmpi(command.c_str(), "help"))
			{
				UpdateChat(nullptr, TranslateT("There are three different modes of chatting:\
\n1) Standard mode\t - You chat with random stranger privately\
\n2) Question mode\t - You ask two strangers a question and see how they discuss it (you can't join their conversation, only watch)\
\n3) Spy mode\t - You and stranger got a question to discuss from third stranger (he can't join your conversation, only watch)\
\n\nSend '/commands' for available commands."), false);
			}
			else if (!mir_strcmpi(command.c_str(), "commands"))
			{
				UpdateChat(nullptr, TranslateT("You can use different commands:\
\n/help\t - show info about chat modes\
\n/new\t - start standard mode\
\n/ask <question> - start question mode with your question\
\n/ask\t - start question mode with your last asked question\
\n/spy\t - start spy mode\
\n/quit\t - disconnect from stranger or stop connecting\
\n/asl\t - send your predefined ASL message\
\n\nNote: You can reconnect to different stranger without disconnecting from current one."), false);
				break;
			}
			else
			{
				UpdateChat(nullptr, TranslateT("Unknown command. Send '/commands' for list."), false);
				break;
			}

		}
		else {
			// Outgoing message
			SendChatMessage(text);
		}

		break;
	}

	case GC_USER_LEAVE:
	case GC_SESSION_TERMINATE:
		facy.nick_ = nullptr;
		ForkThread(&OmegleProto::StopChatWorker, nullptr);
		break;
	}

	return 0;
}

void OmegleProto::SendChatMessage(std::string text)
{
	switch (facy.state_) {
	case STATE_ACTIVE:
		debugLogA("**Chat - Outgoing message: %s", text.c_str());
		ForkThread(&OmegleProto::SendMsgWorker, new std::string(text));
		break;

	case STATE_INACTIVE:
		UpdateChat(nullptr, TranslateT("You aren't connected to any stranger. Send '/help' or '/commands' for help."), false);
		break;

	case STATE_SPY:
		UpdateChat(nullptr, TranslateT("You can't send messages in question mode."), false);
		break;

		//case STATE_WAITING:
		//case STATE_DISCONNECTING:
	default:
		break;
	}
}

/*void OmegleProto::SendChatEvent(int type)
{
GCEVENT gce = { m_szModuleName, m_tszUserName, GC_EVENT_CONTROL };
Chat_Event(WINDOW_CLEARLOG,&gce);
}*/

void OmegleProto::AddChatContact(const wchar_t *name)
{
	GCEVENT gce = { m_szModuleName, m_tszUserName, GC_EVENT_JOIN };
	gce.time = DWORD(time(nullptr));
	gce.dwFlags = GCEF_ADDTOLOG;
	gce.ptszNick = name;
	gce.ptszUID = gce.ptszNick;

	if (name == nullptr)
		gce.bIsMe = false;
	else
		gce.bIsMe = mir_wstrcmp(name, this->facy.nick_);

	if (gce.bIsMe)
		gce.ptszStatus = L"Admin";
	else
		gce.ptszStatus = L"Normal";

	Chat_Event(&gce);
}

void OmegleProto::DeleteChatContact(const wchar_t *name)
{
	GCEVENT gce = { m_szModuleName, m_tszUserName, GC_EVENT_PART };
	gce.dwFlags = GCEF_ADDTOLOG;
	gce.ptszNick = name;
	gce.ptszUID = gce.ptszNick;
	gce.time = DWORD(time(nullptr));
	if (name == nullptr)
		gce.bIsMe = false;
	else
		gce.bIsMe = mir_wstrcmp(name, this->facy.nick_);

	Chat_Event(&gce);
}

INT_PTR OmegleProto::OnJoinChat(WPARAM, LPARAM suppress)
{
	// Create the group chat session
	Chat_NewSession(GCW_PRIVMESS, m_szModuleName, m_tszUserName, m_tszUserName);

	if (m_iStatus == ID_STATUS_OFFLINE)
		return 0;

	// Create a group
	Chat_AddGroup(m_szModuleName, m_tszUserName, TranslateT("Admin"));
	Chat_AddGroup(m_szModuleName, m_tszUserName, TranslateT("Normal"));

	SetTopic();

	// Note: Initialization will finish up in SetChatStatus, called separately
	if (!suppress)
		SetChatStatus(m_iStatus);

	return 0;
}

void OmegleProto::SetTopic(const wchar_t *topic)
{
	GCEVENT gce = { m_szModuleName, m_tszUserName, GC_EVENT_TOPIC };
	gce.time = ::time(nullptr);

	if (topic == nullptr)
		gce.ptszText = TranslateT("Omegle is a great way of meeting new friends!");
	else
		gce.ptszText = topic;

	Chat_Event(&gce);
}

INT_PTR OmegleProto::OnLeaveChat(WPARAM, LPARAM)
{
	Chat_Control(m_szModuleName, m_tszUserName, SESSION_OFFLINE);
	Chat_Terminate(m_szModuleName, m_tszUserName);
	return 0;
}

void OmegleProto::SetChatStatus(int status)
{
	if (status == ID_STATUS_ONLINE)
	{
		// Load actual name from database
		facy.nick_ = db_get_wsa(NULL, m_szModuleName, OMEGLE_KEY_NAME);
		if (facy.nick_ == NULL) {
			facy.nick_ = mir_wstrdup(TranslateT("You"));
			db_set_ws(NULL, m_szModuleName, OMEGLE_KEY_NAME, facy.nick_);
		}

		// Add self contact
		AddChatContact(facy.nick_);

		Chat_Control(m_szModuleName, m_tszUserName, SESSION_INITDONE);
		Chat_Control(m_szModuleName, m_tszUserName, SESSION_ONLINE);
	}
	else
	{
		Chat_Control(m_szModuleName, m_tszUserName, SESSION_OFFLINE);
	}
}

void OmegleProto::ClearChat()
{
	if (!getByte(OMEGLE_KEY_NO_CLEAR, 0))
		Chat_Control(m_szModuleName, m_tszUserName, WINDOW_CLEARLOG);
}

// TODO: Could this be done better?
MCONTACT OmegleProto::GetChatHandle()
{
	/*if (facy.chatHandle_ != NULL)
		return facy.chatHandle_;

		for (auto &hContact : AccContacts()) {
		if (db_get_b(hContact, m_szModuleName, "ChatRoom", 0) > 0) {
		ptrA id = db_get_sa(hContact, m_szModuleName, "ChatRoomId");
		if (id != NULL && !mir_strcmp(id, m_szModuleName))
		return hContact;
		}
		}

		return NULL;*/

	GC_INFO gci = { 0 };
	gci.Flags = GCF_HCONTACT;
	gci.pszModule = m_szModuleName;
	gci.pszID = m_tszUserName;
	Chat_GetInfo(&gci);

	return gci.hContact;
}