Added WhatsApp-protocol

git-svn-id: http://svn.miranda-ng.org/main/trunk@4861 1316c22d-e87f-b044-9b9b-93d7a3e3ba9c

1 files changed, 446 insertions, 0 deletions

diff --git a/protocols/WhatsApp/src/proto.cpp b/protocols/WhatsApp/src/proto.cpp
new file mode 100644
index 0000000000..99a903226f
--- /dev/null
+++ b/protocols/WhatsApp/src/proto.cpp
@@ -0,0 +1,446 @@
+#include "common.h"

+

+WhatsAppProto::WhatsAppProto(const char* proto_name, const TCHAR* username)

+{

+   ProtoConstructor(this, proto_name, username);

+

+   this->challenge = new std::vector<unsigned char>;

+   this->msgId = 0;

+   this->msgIdHeader = time(NULL);

+

+   update_loop_lock_ = CreateEvent(NULL, false, false, NULL);

+   FMessage::generating_lock = new Mutex();

+

+   CreateProtoService(m_szModuleName, PS_CREATEACCMGRUI, &WhatsAppProto::SvcCreateAccMgrUI, this);

+	CreateProtoService(m_szModuleName, PS_JOINCHAT,  &WhatsAppProto::OnJoinChat,  this);

+	CreateProtoService(m_szModuleName, PS_LEAVECHAT, &WhatsAppProto::OnLeaveChat, this);

+

+   HookProtoEvent(ME_CLIST_PREBUILDSTATUSMENU, &WhatsAppProto::OnBuildStatusMenu, this);

+   HookProtoEvent(ME_GC_EVENT,                 &WhatsAppProto::OnChatOutgoing,    this);

+

+   this->InitContactMenus();

+

+   // Create standard network connection

+   TCHAR descr[512];

+   NETLIBUSER nlu = {sizeof(nlu)};

+   nlu.flags = NUF_INCOMING | NUF_OUTGOING | NUF_HTTPCONNS;

+	char module[512];

+	mir_snprintf(module, SIZEOF(module), "%s", m_szModuleName);

+	nlu.szSettingsModule = module;

+	mir_sntprintf(descr, SIZEOF(descr), TranslateT("%s server connection"), m_tszUserName);

+	nlu.ptszDescriptiveName = descr;

+   m_hNetlibUser = (HANDLE) CallService(MS_NETLIB_REGISTERUSER, 0, (LPARAM) &nlu);

+	if (m_hNetlibUser == NULL)

+		MessageBox(NULL, TranslateT("Unable to get Netlib connection for WhatsApp"), m_tszUserName, MB_OK);

+   

+   WASocketConnection::initNetwork(m_hNetlibUser);

+

+	TCHAR *profile = Utils_ReplaceVarsT( _T("%miranda_avatarcache%"));

+	def_avatar_folder_ = std::tstring(profile) + _T("\\") + m_tszUserName;

+	mir_free(profile);

+	hAvatarFolder_ = FoldersRegisterCustomPathT(m_szModuleName, "Avatars", def_avatar_folder_.c_str());

+

+   // Register group chat

+	GCREGISTER gcr = {0};

+	gcr.cbSize = sizeof(GCREGISTER);

+	gcr.dwFlags = GC_TYPNOTIF | GC_CHANMGR | GC_TCHAR;

+	gcr.iMaxText = 0;

+	gcr.nColors = 0;

+	gcr.pColors = NULL; //(COLORREF*)crCols;

+	gcr.ptszModuleDispName = m_tszUserName;

+	gcr.pszModule = m_szModuleName;

+	CallServiceSync(MS_GC_REGISTER, 0, (LPARAM)&gcr);

+

+	//HookProtoEvent(ME_GC_EVENT, &CMsnProto::MSN_GCEventHook);

+	//HookProtoEvent(ME_GC_BUILDMENU, &CMsnProto::MSN_GCMenuHook);

+

+   SetAllContactStatuses(ID_STATUS_OFFLINE, true);

+}

+

+WhatsAppProto::~WhatsAppProto()

+{

+   CloseHandle(update_loop_lock_);

+

+   if (this->challenge != NULL)

+      delete this->challenge;

+

+   ProtoDestructor(this);

+}

+

+DWORD_PTR WhatsAppProto::GetCaps( int type, HANDLE hContact )

+{

+	switch(type)

+	{

+	   case PFLAGNUM_1:

+	   {

+		   DWORD_PTR flags = PF1_IM | PF1_CHAT | PF1_BASICSEARCH | PF1_ADDSEARCHRES;

+         return flags | PF1_MODEMSGRECV; // #TODO

+	   }

+	   case PFLAGNUM_2:

+		   return PF2_ONLINE | PF2_INVISIBLE;

+	   case PFLAGNUM_3:

+		   return 0;

+	   case PFLAGNUM_4:

+         return PF4_NOCUSTOMAUTH | PF4_IMSENDUTF | PF4_FORCEADDED  | PF4_NOAUTHDENYREASON | PF4_IMSENDOFFLINE | PF4_NOAUTHDENYREASON | PF4_SUPPORTTYPING | PF4_AVATARS;

+	   case PFLAGNUM_5:

+		   return 0;

+	   case PFLAG_MAXLENOFMESSAGE:

+         return 500; // #TODO

+	   case PFLAG_UNIQUEIDTEXT:

+		   return (DWORD_PTR) "WhatsApp ID";

+	   case PFLAG_UNIQUEIDSETTING:

+		   return (DWORD_PTR) "ID";

+	}

+	return 0;

+}

+

+int WhatsAppProto::SetStatus( int new_status )

+{

+	LOG("===== Beginning SetStatus process");

+

+	// Routing statuses not supported by WhatsApp

+	switch ( new_status )

+	{

+	case ID_STATUS_INVISIBLE:

+	case ID_STATUS_OFFLINE:

+		m_iDesiredStatus = new_status;

+		break;

+

+	/*

+	case ID_STATUS_CONNECTING:

+		m_iDesiredStatus = ID_STATUS_OFFLINE;

+		break;

+   */

+

+	case ID_STATUS_IDLE:

+	default:

+		m_iDesiredStatus = ID_STATUS_INVISIBLE;

+		if (db_get_b(NULL, m_szModuleName, WHATSAPP_KEY_MAP_STATUSES, DEFAULT_MAP_STATUSES))

+			break;

+	case ID_STATUS_ONLINE:

+	case ID_STATUS_FREECHAT:

+		m_iDesiredStatus = ID_STATUS_ONLINE;

+		break;

+	}

+

+	if (m_iStatus == ID_STATUS_CONNECTING)

+	{

+		LOG("===== Status is connecting, no change");

+		return 0;

+	}

+

+	if (m_iStatus == m_iDesiredStatus)

+	{

+		LOG("===== Statuses are same, no change");

+		return 0;

+	}

+

+	ForkThread( &WhatsAppProto::ChangeStatus, this );

+

+	return 0;

+}

+

+HANDLE WhatsAppProto::AddToList( int flags, PROTOSEARCHRESULT* psr )

+{

+   return NULL;

+}

+

+int WhatsAppProto::AuthRequest(HANDLE hContact,const PROTOCHAR *message)

+{

+   return this->RequestFriendship((WPARAM)hContact, NULL);

+}

+

+int WhatsAppProto::Authorize(HANDLE hDbEvent)

+{

+	return 1;

+}

+

+HANDLE WhatsAppProto::SearchBasic( const PROTOCHAR* id )

+{

+	if (isOffline())

+		return 0;

+

+	TCHAR* email = mir_tstrdup(id);

+	ForkThread(&WhatsAppProto::SearchAckThread, this, (void*)email);

+

+	return email;

+}

+

+void WhatsAppProto::RequestCode()

+{

+   std::string cc, number, idx;

+   bool error;

+   DBVARIANT dbv;

+

+   if ( WASocketConnection::hNetlibUser == NULL)

+   {

+      NotifyEvent(m_tszUserName,TranslateT("Network-connection error."),NULL,WHATSAPP_EVENT_CLIENT);

+      return;

+   }

+

+   if ( !db_get_s(NULL,m_szModuleName,WHATSAPP_KEY_IDX,&dbv,DBVT_ASCIIZ))

+   {

+      idx = dbv.pszVal;

+      if (idx.empty())

+      {

+         std::stringstream tm;

+         tm << time(NULL);

+         unsigned char* idxBuf = new unsigned char[16];

+         MD5((unsigned char*) tm.str().c_str(), tm.str().length(), idxBuf);

+         idx = std::string((const char*) idxBuf, 16);

+         db_set_s(0, m_szModuleName,WHATSAPP_KEY_IDX, idx.c_str());

+         delete idxBuf;

+      }

+   }

+	if ( !db_get_s(NULL,m_szModuleName,WHATSAPP_KEY_CC,&dbv, DBVT_ASCIIZ))

+	{

+		cc = dbv.pszVal;

+		db_free(&dbv);

+		if (cc.empty())

+      {

+         NotifyEvent(m_tszUserName,TranslateT("Please enter a country-code."),NULL,WHATSAPP_EVENT_CLIENT);

+         return;

+      }

+	}

+	if ( !db_get_s(NULL,m_szModuleName,WHATSAPP_KEY_LOGIN,&dbv, DBVT_ASCIIZ))

+	{

+		number = dbv.pszVal;

+		db_free(&dbv);

+		if (number.empty())

+      {

+         NotifyEvent(m_tszUserName,TranslateT("Please enter a phone-number without country-code."),NULL,WHATSAPP_EVENT_CLIENT);

+         return;

+      }

+	}

+

+	std::string token(Utilities::md5String(std::string(ACCOUNT_TOKEN_PREFIX1) + ACCOUNT_TOKEN_PREFIX2 + number));

+

+   NETLIBHTTPHEADER agentHdr;

+   agentHdr.szName = "User-Agent";

+   agentHdr.szValue = ACCOUNT_USER_AGENT_REGISTRATION;

+

+   NETLIBHTTPHEADER acceptHdr;

+   acceptHdr.szName = "Accept";

+   acceptHdr.szValue = "text/json";

+

+   NETLIBHTTPHEADER ctypeHdr;

+   ctypeHdr.szName = "Content-Type";

+   ctypeHdr.szValue = "application/x-www-form-urlencoded";

+

+   NETLIBHTTPHEADER headers[] = { agentHdr, acceptHdr, ctypeHdr };

+

+   NETLIBHTTPREQUEST nlhr = {sizeof(NETLIBHTTPREQUEST)};

+   nlhr.requestType = REQUEST_POST;

+   nlhr.szUrl = (char*) (std::string(ACCOUNT_URL_CODEREQUESTV2) + "?cc="+ cc + "&in="+ number +

+      "lc=US&lg=en&mcc=000&mnc=000&method=sms&id=" + idx + "&token="+ token).c_str();

+   nlhr.headers = &headers[0];

+   nlhr.headersCount = 3;

+   nlhr.flags = NLHRF_HTTP11 | NLHRF_GENERATEHOST | NLHRF_REMOVEHOST;

+

+   return;

+

+   NETLIBHTTPREQUEST* pnlhr = (NETLIBHTTPREQUEST*) CallService(MS_NETLIB_HTTPTRANSACTION,

+      (WPARAM) WASocketConnection::hNetlibUser, (LPARAM)&nlhr);

+

+   // #TODO

+}

+

+void WhatsAppProto::RegisterCode(const std::string& code)

+{

+}

+

+//////////////////////////////////////////////////////////////////////////////

+// EVENTS

+

+int WhatsAppProto::SvcCreateAccMgrUI(WPARAM wParam,LPARAM lParam)

+{

+	return (int)CreateDialogParam(g_hInstance, MAKEINTRESOURCE(IDD_WHATSAPPACCOUNT),

+		 (HWND)lParam, WhatsAppAccountProc, (LPARAM)this );

+}

+

+int WhatsAppProto::RefreshBuddyList(WPARAM, LPARAM )

+{

+   LOG("");

+   if (!isOffline())

+   {

+      //ForkThread(

+   }

+   return 0;

+}

+

+int WhatsAppProto::RequestFriendship(WPARAM wParam, LPARAM lParam)

+{

+	if (wParam == NULL || isOffline())

+		return 0;

+

+	HANDLE hContact = reinterpret_cast<HANDLE>(wParam);

+

+	DBVARIANT dbv;

+	if ( !db_get_s(hContact,m_szModuleName,WHATSAPP_KEY_ID,&dbv, DBVT_ASCIIZ))

+	{

+		std::string id(dbv.pszVal);

+      this->connection->sendQueryLastOnline(id);

+      this->connection->sendPresenceSubscriptionRequest(id);

+		db_free(&dbv);

+	}

+

+	return 0;

+}

+

+std::tstring WhatsAppProto::GetAvatarFolder()

+{

+	TCHAR path[MAX_PATH];

+	if ( hAvatarFolder_ && FoldersGetCustomPathT(hAvatarFolder_, path, SIZEOF(path), _T("")) == 0 )

+		return path;

+	else

+		return def_avatar_folder_;

+}

+

+int WhatsAppProto::Log(const char* fn, const char *fmt,...)

+{

+#if !defined(_DEBUG)

+	if (!getByte(WHATSAPP_KEY_LOGGING_ENABLE, 0))

+		return EXIT_SUCCESS;

+#endif

+

+	va_list va;

+	char text[65535];

+	ScopedLock s(log_lock_);

+

+	va_start(va,fmt);

+	mir_vsnprintf(text,sizeof(text),fmt,va);

+	va_end(va);

+

+	// Write into network log

+	//CallService(MS_NETLIB_LOG, (WPARAM)m_hNetlibUser, (LPARAM)text);

+

+	// Write into log file

+	return utils::debug::log(m_szModuleName, std::string(fn).append(" - ").append(text));

+}

+

+LRESULT CALLBACK PopupDlgProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)

+{

+	switch(message)

+	{

+	case WM_COMMAND:

+	{

+		// After a click, destroy popup

+		PUDeletePopup(hwnd);

+		} break;

+

+	case WM_CONTEXTMENU:

+		PUDeletePopup(hwnd);

+		break;

+

+	case UM_FREEPLUGINDATA:

+	{

+		// After close, free

+		TCHAR* url = (TCHAR*)PUGetPluginData(hwnd);

+		if (url != NULL)

+			mir_free(url);

+	} return FALSE;

+

+	default:

+		break;

+	}

+

+	return DefWindowProc(hwnd, message, wParam, lParam);

+};

+

+void WhatsAppProto::NotifyEvent(const string& title, const string& info, HANDLE contact, DWORD flags, TCHAR* url)

+{

+   TCHAR* rawTitle = mir_a2t_cp(title.c_str(), CP_UTF8);

+   TCHAR* rawInfo = mir_a2t_cp(info.c_str(), CP_UTF8);

+   this->NotifyEvent(rawTitle, rawInfo, contact, flags, url);

+   mir_free(rawTitle);

+   mir_free(rawInfo);

+}

+

+void WhatsAppProto::NotifyEvent(TCHAR* title, TCHAR* info, HANDLE contact, DWORD flags, TCHAR* szUrl)

+{

+	int ret; int timeout; COLORREF colorBack = 0; COLORREF colorText = 0;

+

+	switch ( flags )

+	{

+	   case WHATSAPP_EVENT_CLIENT:

+		   if ( !getByte( WHATSAPP_KEY_EVENT_CLIENT_ENABLE, DEFAULT_EVENT_CLIENT_ENABLE ))

+			   goto exit;

+		   if ( !getByte( WHATSAPP_KEY_EVENT_CLIENT_DEFAULT, 0 ))

+		   {

+			   colorBack = getDword( WHATSAPP_KEY_EVENT_CLIENT_COLBACK, DEFAULT_EVENT_COLBACK );

+			   colorText = getDword( WHATSAPP_KEY_EVENT_CLIENT_COLTEXT, DEFAULT_EVENT_COLTEXT );

+		   }

+		   timeout = getDword( WHATSAPP_KEY_EVENT_CLIENT_TIMEOUT, 0 );

+		   flags |= NIIF_WARNING;

+		   break;

+

+	   case WHATSAPP_EVENT_OTHER:

+		   if ( !getByte( WHATSAPP_KEY_EVENT_OTHER_ENABLE, DEFAULT_EVENT_OTHER_ENABLE ))

+			   goto exit;

+		   if ( !getByte( WHATSAPP_KEY_EVENT_OTHER_DEFAULT, 0 ))

+		   {

+			   colorBack = getDword( WHATSAPP_KEY_EVENT_OTHER_COLBACK, DEFAULT_EVENT_COLBACK );

+			   colorText = getDword( WHATSAPP_KEY_EVENT_OTHER_COLTEXT, DEFAULT_EVENT_COLTEXT );

+		   }

+		   timeout = getDword( WHATSAPP_KEY_EVENT_OTHER_TIMEOUT, -1 );

+		   SkinPlaySound( "OtherEvent" );

+		   flags |= NIIF_INFO;

+		   break;

+   }

+

+	if ( !getByte(WHATSAPP_KEY_SYSTRAY_NOTIFY,DEFAULT_SYSTRAY_NOTIFY))

+	{

+		if (ServiceExists(MS_POPUP_ADDPOPUP))

+		{

+			POPUPDATAT pd;

+			pd.colorBack = colorBack;

+			pd.colorText = colorText;

+			pd.iSeconds = timeout;

+			pd.lchContact = contact;

+			pd.lchIcon = Skin_GetIconByHandle(m_hProtoIcon); // TODO: Icon test

+			pd.PluginData = szUrl;

+			pd.PluginWindowProc = (WNDPROC)PopupDlgProc;

+			lstrcpy(pd.lptzContactName, title);

+			lstrcpy(pd.lptzText, info);

+			ret = PUAddPopupT(&pd);

+

+			if (ret == 0)

+				return;

+		}

+	} else {

+		if (ServiceExists(MS_CLIST_SYSTRAY_NOTIFY))

+		{

+			MIRANDASYSTRAYNOTIFY err;

+			int niif_flags = flags;

+			REMOVE_FLAG( niif_flags, WHATSAPP_EVENT_CLIENT |

+			                         WHATSAPP_EVENT_NOTIFICATION |

+			                         WHATSAPP_EVENT_OTHER );

+			err.szProto = m_szModuleName;

+			err.cbSize = sizeof(err);

+			err.dwInfoFlags = NIIF_INTERN_TCHAR | niif_flags;

+			err.tszInfoTitle = title;

+			err.tszInfo = info;

+			err.uTimeout = 1000 * timeout;

+			ret = CallService(MS_CLIST_SYSTRAY_NOTIFY, 0, (LPARAM) & err);

+

+			if (ret == 0)

+				goto exit;

+		}

+	}

+

+	if (FLAG_CONTAINS(flags, WHATSAPP_EVENT_CLIENT))

+		MessageBox(NULL, info, title, MB_OK | MB_ICONINFORMATION);

+

+exit:

+	if (szUrl != NULL)

+		mir_free(szUrl);

+}

+

+string WhatsAppProto::TranslateStr(const char* str, ...)

+{

+   va_list ap;

+   va_start(ap, str);

+   string ret = Utilities::string_format(Translate(str), ap);

+   va_end(ap);

+   return ret;

+}
\ No newline at end of file