ICQ: added messages' forwarding + "Add to favorites" NewStory menu item

1 files changed, 136 insertions, 36 deletions

diff --git a/protocols/ICQ-WIM/src/menus.cpp b/protocols/ICQ-WIM/src/menus.cpp
index b78b308f43..5c7fca94ad 100644
--- a/protocols/ICQ-WIM/src/menus.cpp
+++ b/protocols/ICQ-WIM/src/menus.cpp
@@ -19,6 +19,22 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
 
 #define MenuExecService "/NSExecMenu"
 
+int CIcqProto::OnPrebuildMenu(WPARAM hContact, LPARAM lParam)
+{
+	if (!Proto_IsProtoOnContact(hContact, m_szModuleName)) {
+		Menu_ShowItem(hmiForward, false);
+		Menu_ShowItem(hmiConvert, false);
+	}
+	else {
+		auto *dbei = (DB::EventInfo *)lParam;
+		Menu_ShowItem(hmiForward, dbei->eventType == EVENTTYPE_MESSAGE || dbei->eventType == EVENTTYPE_FILE);
+
+		ptrW wszText(DbEvent_GetTextW(dbei, CP_UTF8));
+		Menu_ShowItem(hmiConvert, fileText2url(wszText.get()));
+	}
+	return 0;
+}
+
 void CIcqProto::InitMenus()
 {
 	if (!HookProtoEvent(ME_NS_PREBUILDMENU, &CIcqProto::OnPrebuildMenu))
@@ -31,50 +47,134 @@ void CIcqProto::InitMenus()
 	mi.pszService = szServiceName;
 
 	mi.position = 1000000;
+	mi.hIcolibItem = g_plugin.getIconHandle(IDI_FORWARD);
+	mi.name.a = LPGEN("Forward");
+	hmiForward = Menu_AddNewStoryMenuItem(&mi, 1);
+
+	mi.position++;
+	mi.hIcolibItem = Skin_GetIconHandle(SKINICON_OTHER_ADDCONTACT);
+	mi.name.a = LPGEN("Add to faforites");
+	hmiConvert = Menu_AddNewStoryMenuItem(&mi, 2);
+
+	mi.position++;
 	mi.hIcolibItem = Skin_GetIconHandle(SKINICON_EVENT_FILE);
 	mi.name.a = LPGEN("Convert a message into a file transfer");
-	hmiConvert = Menu_AddNewStoryMenuItem(&mi, 1);
+	hmiConvert = Menu_AddNewStoryMenuItem(&mi, 3);
 }
 
-INT_PTR CIcqProto::SvcExecMenu(WPARAM iCommand, LPARAM pHandle)
+/////////////////////////////////////////////////////////////////////////////////////////
+// Dialog for message forwarding
+
+class CForwardDlg : public CIcqDlgBase
 {
-	// convert a message into a file transfer
-	if (iCommand == 1) {
-		if (MEVENT hEvent = NS_GetCurrent(HANDLE(pHandle))) {
-			DB::EventInfo dbei(hEvent);
-			if (!dbei)
-				return 0;
-
-			IcqFileInfo *pFileInfo = nullptr;
-			CMStringW wszText(ptrW(DbEvent_GetTextW(&dbei, CP_UTF8)));
-			if (CheckFile(dbei.hContact, wszText, pFileInfo)) {
-				if (!pFileInfo || pFileInfo->bIsSticker) {
-					// sticker is a simple text message prcoessed by SmileyAdd
-					T2Utf szBody(wszText);
-					mir_free(dbei.pBlob);
-					dbei.cbBlob = (int)mir_strlen(szBody.get());
-					dbei.pBlob = szBody.detach();
-				}
-				else {
-					// create the offline file event
-					dbei.eventType = EVENTTYPE_FILE;
-
-					DB::FILE_BLOB blob(pFileInfo->wszDescr, wszText);
-					blob.setUrl(pFileInfo->szOrigUrl);
-					blob.setSize(pFileInfo->dwFileSize);
-					blob.write(dbei);
-				}
-				db_event_edit(hEvent, &dbei);
-			}
-		}
+	CCtrlClc m_clist;
+	MEVENT m_hEvent;
+
+	void FilterList(CCtrlClc *)
+	{
+		for (auto &hContact : Contacts())
+			if (!Proto_IsProtoOnContact(hContact, m_proto->m_szModuleName))
+				if (HANDLE hItem = m_clist.FindContact(hContact))
+					m_clist.DeleteItem(hItem);
 	}
-	return 0;
+
+	void ResetListOptions(CCtrlClc *)
+	{
+		m_clist.SetHideEmptyGroups(true);
+		m_clist.SetHideOfflineRoot(true);
+	}
+
+public:
+	CForwardDlg(CIcqProto *ppro, MEVENT hEvent) :
+		CIcqDlgBase(ppro, IDD_FORWARD),
+		m_hEvent(hEvent),
+		m_clist(this, IDC_CLIST)
+	{
+		m_clist.OnNewContact =
+			m_clist.OnListRebuilt = Callback(this, &CForwardDlg::FilterList);
+		m_clist.OnOptionsChanged = Callback(this, &CForwardDlg::ResetListOptions);
+	}
+
+	bool OnInitDialog() override
+	{
+		SetWindowLongPtr(m_clist.GetHwnd(), GWL_STYLE,
+			GetWindowLongPtr(m_clist.GetHwnd(), GWL_STYLE) | CLS_SHOWHIDDEN | CLS_HIDEOFFLINE | CLS_CHECKBOXES | CLS_HIDEEMPTYGROUPS | CLS_USEGROUPS | CLS_GREYALTERNATE | CLS_GROUPCHECKBOXES);
+		m_clist.SendMsg(CLM_SETEXSTYLE, CLS_EX_DISABLEDRAGDROP | CLS_EX_TRACKSELECT, 0);
+		ResetListOptions(&m_clist);
+		FilterList(&m_clist);
+		return true;
+	}
+
+	bool OnApply() override
+	{
+		for (auto &hContact : m_proto->AccContacts())
+			if (HANDLE hItem = m_clist.FindContact(hContact))
+				if (m_clist.GetCheck(hItem))
+					m_proto->ForwardMessage(m_hEvent, hContact);
+
+		return true;
+	}
+};
+
+void CIcqProto::ForwardMessage(MEVENT hEvent, MCONTACT to)
+{
+	DB::EventInfo dbei(hEvent);
+	if (!dbei || !dbei.szId || mir_strcmp(dbei.szModule, m_szModuleName))
+		return;
+
+	CMStringW wszId(GetUserId(dbei.hContact));
+	ptrW wszText(DbEvent_GetTextW(&dbei, CP_UTF8));
+
+	JSONNode parts(JSON_ARRAY);
+	JSONNode msgText; msgText << CHAR_PARAM("mediaType", "forward") << WCHAR_PARAM("sn", wszId) << INT_PARAM("time", dbei.timestamp)
+		<< CHAR_PARAM("msgId", dbei.szId) << WCHAR_PARAM("text", wszText);
+	parts.push_back(msgText);
+
+	SendMessageParts(to, parts);
 }
 
-int CIcqProto::OnPrebuildMenu(WPARAM, LPARAM lParam)
+/////////////////////////////////////////////////////////////////////////////////////////
+
+INT_PTR CIcqProto::SvcExecMenu(WPARAM iCommand, LPARAM pHandle)
 {
-	auto *dbei = (DB::EventInfo *)lParam;
-	ptrW wszText(DbEvent_GetTextW(dbei, CP_UTF8));
-	Menu_ShowItem(hmiConvert, fileText2url(wszText.get()));
+	MEVENT hEvent = NS_GetCurrent(HANDLE(pHandle));
+	if (!hEvent)
+		return 0;
+
+	switch (iCommand) {
+	case 1: // forward message
+		CForwardDlg(this, hEvent).DoModal();
+		break;
+
+	case 2: // Add to favorites
+		ForwardMessage(hEvent, m_hFavContact);
+
+	case 3: // convert a message into a file transfer
+		DB::EventInfo dbei(hEvent);
+		if (!dbei)
+			return 0;
+
+		IcqFileInfo *pFileInfo = nullptr;
+		CMStringW wszText(ptrW(DbEvent_GetTextW(&dbei, CP_UTF8)));
+		if (CheckFile(dbei.hContact, wszText, pFileInfo)) {
+			if (!pFileInfo || pFileInfo->bIsSticker) {
+				// sticker is a simple text message prcoessed by SmileyAdd
+				T2Utf szBody(wszText);
+				mir_free(dbei.pBlob);
+				dbei.cbBlob = (int)mir_strlen(szBody.get());
+				dbei.pBlob = szBody.detach();
+			}
+			else {
+				// create the offline file event
+				dbei.eventType = EVENTTYPE_FILE;
+
+				DB::FILE_BLOB blob(pFileInfo->wszDescr, wszText);
+				blob.setUrl(pFileInfo->szOrigUrl);
+				blob.setSize(pFileInfo->dwFileSize);
+				blob.write(dbei);
+			}
+			db_event_edit(hEvent, &dbei);
+		}
+	}
 	return 0;
 }