1 files changed, 262 insertions, 0 deletions
diff --git a/plugins/AVS/src/contact_ava.cpp b/plugins/AVS/src/contact_ava.cpp
new file mode 100644
index 0000000000..9a4b935a49
--- /dev/null
+++ b/plugins/AVS/src/contact_ava.cpp
@@ -0,0 +1,262 @@
+/*
+Copyright (C) 2012-22 Miranda NG team (https://miranda-ng.org)
+
+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 version 2
+of the License.
+
+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"
+
+class CContactAvatarDlg : public CDlgBase
+{
+	MCONTACT m_hContact;
+	HANDLE   m_hHook = 0;
+
+	UI_MESSAGE_MAP(CContactAvatarDlg, CDlgBase);
+		UI_MESSAGE(WM_DRAWITEM, OnDrawItem);
+		UI_MESSAGE(DM_AVATARCHANGED, OnAvatarChanged);
+	UI_MESSAGE_MAP_END();
+
+	INT_PTR OnAvatarChanged(UINT, WPARAM, LPARAM)
+	{
+		InvalidateRect(GetDlgItem(m_hwnd, IDC_PROTOPIC), nullptr, TRUE);
+	}
+
+	INT_PTR OnDrawItem(UINT, WPARAM, LPARAM lParam)
+	{
+		LPDRAWITEMSTRUCT dis = (LPDRAWITEMSTRUCT)lParam;
+		if (dis->CtlType == ODT_BUTTON && dis->CtlID == IDC_PROTOPIC) {
+			AVATARDRAWREQUEST avdrq = { 0 };
+			GetClientRect(GetDlgItem(m_hwnd, IDC_PROTOPIC), &avdrq.rcDraw);
+
+			FillRect(dis->hDC, &avdrq.rcDraw, GetSysColorBrush(COLOR_BTNFACE));
+
+			avdrq.hContact = m_hContact;
+			avdrq.hTargetDC = dis->hDC;
+			avdrq.dwFlags |= AVDRQ_DRAWBORDER;
+			avdrq.clrBorder = GetSysColor(COLOR_BTNTEXT);
+			avdrq.radius = 6;
+			if (!CallService(MS_AV_DRAWAVATAR, 0, (LPARAM)&avdrq)) {
+				// Get text rectangle
+				RECT rc = avdrq.rcDraw;
+				rc.top += 10;
+				rc.bottom -= 10;
+				rc.left += 10;
+				rc.right -= 10;
+
+				// Calc text size
+				RECT rc_ret = rc;
+				DrawText(dis->hDC, TranslateT("Contact has no avatar"), -1, &rc_ret,
+					DT_WORDBREAK | DT_NOPREFIX | DT_CENTER | DT_CALCRECT);
+
+				// Calc needed size
+				rc.top += ((rc.bottom - rc.top) - (rc_ret.bottom - rc_ret.top)) / 2;
+				rc.bottom = rc.top + (rc_ret.bottom - rc_ret.top);
+				DrawText(dis->hDC, TranslateT("Contact has no avatar"), -1, &rc,
+					DT_WORDBREAK | DT_NOPREFIX | DT_CENTER);
+			}
+
+			FrameRect(dis->hDC, &avdrq.rcDraw, GetSysColorBrush(COLOR_BTNSHADOW));
+		}
+		return TRUE;
+	}
+
+	void ReloadAvatar()
+	{
+		SaveTransparentData(m_hwnd, m_hContact, chkProtect.IsChecked());
+		ChangeAvatar(m_hContact, true);
+		OnAvatarChanged(0, 0, 0);
+	}
+
+	void SetAvatarName()
+	{
+		uint8_t is_locked = db_get_b(m_hContact, "ContactPhoto", "Locked", 0);
+
+		wchar_t szFinalName[MAX_PATH];
+		szFinalName[0] = 0;
+
+		DBVARIANT dbv = {};
+		if (is_locked && !db_get_ws(m_hContact, "ContactPhoto", "Backup", &dbv)) {
+			MyPathToAbsolute(dbv.pwszVal, szFinalName);
+			db_free(&dbv);
+		}
+		else if (!db_get_ws(m_hContact, "ContactPhoto", "RFile", &dbv)) {
+			MyPathToAbsolute(dbv.pwszVal, szFinalName);
+			db_free(&dbv);
+		}
+		else if (!db_get_ws(m_hContact, "ContactPhoto", "File", &dbv)) {
+			MyPathToAbsolute(dbv.pwszVal, szFinalName);
+			db_free(&dbv);
+		}
+		szFinalName[MAX_PATH - 1] = 0;
+		SetDlgItemText(m_hwnd, IDC_AVATARNAME, szFinalName);
+	}
+
+	CCtrlSpin spin1, spin2;
+	CCtrlCheck chkProtect, chkMakeTrans, chkHide;
+	CCtrlButton btnUseDefault, btnChange, btnReset, btnDefault;
+
+public:
+	CContactAvatarDlg(MCONTACT hContact) :
+		CDlgBase(g_plugin, IDD_AVATAROPTIONS),
+		m_hContact(hContact),
+		spin1(this, IDC_BKG_NUM_POINTS_SPIN, 8, 2),
+		spin2(this, IDC_BKG_COLOR_DIFFERENCE_SPIN, 100),
+		btnReset(this, IDC_RESET),
+		btnChange(this, IDC_CHANGE),
+		btnDefault(this, IDC_DELETE),
+		btnUseDefault(this, ID_USE_DEFAULTS),
+		chkHide(this, IDC_HIDEAVATAR),
+		chkProtect(this, IDC_PROTECTAVATAR),
+		chkMakeTrans(this, IDC_MAKETRANSPBKG)
+	{
+		btnReset.OnClick = Callback(this, &CContactAvatarDlg::onClick_Reset);
+		btnChange.OnClick = Callback(this, &CContactAvatarDlg::onClick_Change);
+		btnDefault.OnClick = Callback(this, &CContactAvatarDlg::onClick_Default);
+		btnUseDefault.OnClick = Callback(this, &CContactAvatarDlg::onClick_UseDefault);
+
+		chkProtect.OnChange = Callback(this, &CContactAvatarDlg::onChange_Protect);
+		chkMakeTrans.OnChange = Callback(this, &CContactAvatarDlg::onChange_MakeTrans);
+	}
+	
+	bool OnInitDialog() override
+	{
+		m_hHook = HookEventMessage(ME_AV_AVATARCHANGED, m_hwnd, DM_AVATARCHANGED);
+
+		if (m_hContact) {
+			wchar_t szTitle[512];
+			mir_snwprintf(szTitle, TranslateT("Set avatar options for %s"), Clist_GetContactDisplayName(m_hContact));
+			SetWindowText(m_hwnd, szTitle);
+		}
+
+		SetAvatarName();
+		ShowWindow(m_hwnd, SW_SHOWNORMAL);
+		OnAvatarChanged(0, 0, 0);
+		chkProtect.SetState(db_get_b(m_hContact, "ContactPhoto", "Locked", 0));
+		chkHide.SetState(Contact::IsHidden(m_hContact));
+
+		LoadTransparentData(m_hwnd, GetContactThatHaveTheAvatar(m_hContact));
+		SendMessage(m_hwnd, WM_SETICON, IMAGE_ICON, (LPARAM)g_plugin.getIcon(IDI_AVATAR));
+		return true;
+	}
+
+	bool OnApply() override
+	{
+		bool locked = chkProtect.IsChecked();
+		bool hidden = chkHide.IsChecked();
+
+		SetAvatarAttribute(m_hContact, AVS_HIDEONCLIST, hidden);
+		if (hidden != Contact::IsHidden(m_hContact))
+			Contact::Hide(m_hContact, hidden);
+
+		if (!locked && db_get_b(m_hContact, "ContactPhoto", "NeedUpdate", 0))
+			QueueAdd(m_hContact);
+		return true;
+	}
+
+	void OnDestroy() override
+	{
+		UnhookEvent(m_hHook);
+	}
+
+	void onClick_UseDefault(CCtrlButton *)
+	{
+		MCONTACT hContact = GetContactThatHaveTheAvatar(m_hContact);
+
+		db_unset(hContact, "ContactPhoto", "MakeTransparentBkg");
+		db_unset(hContact, "ContactPhoto", "TranspBkgNumPoints");
+		db_unset(hContact, "ContactPhoto", "TranspBkgColorDiff");
+
+		LoadTransparentData(m_hwnd, hContact);
+		ReloadAvatar();
+	}
+	
+	void onClick_Change(CCtrlButton *)
+	{
+		SetAvatar(m_hContact, 0);
+		SetAvatarName();
+		chkProtect.SetState(db_get_b(m_hContact, "ContactPhoto", "Locked"));
+	}
+
+	void onClick_Reset(CCtrlButton *)
+	{
+		ProtectAvatar(m_hContact, 0);
+		if (MessageBox(nullptr, TranslateT("Delete picture file from disk (may be necessary to force a reload, but will delete local pictures)?"), TranslateT("Reset contact picture"), MB_YESNO) == IDYES) {
+			DBVARIANT dbv = { 0 };
+			if (!db_get_ws(m_hContact, "ContactPhoto", "File", &dbv)) {
+				DeleteFileW(dbv.pwszVal);
+				db_free(&dbv);
+			}
+		}
+		db_unset(m_hContact, "ContactPhoto", "Locked");
+		db_unset(m_hContact, "ContactPhoto", "Backup");
+		db_unset(m_hContact, "ContactPhoto", "RFile");
+		db_unset(m_hContact, "ContactPhoto", "File");
+		db_unset(m_hContact, "ContactPhoto", "Format");
+
+		db_unset(m_hContact, Proto_GetBaseAccountName(m_hContact), "AvatarHash");
+		DeleteAvatarFromCache(m_hContact, FALSE);
+
+		QueueAdd(m_hContact);
+		DestroyWindow(m_hwnd);
+	}
+
+	void onClick_Default(CCtrlButton *)
+	{
+		if (MessageBoxW(nullptr, TranslateT("Delete picture file from disk (may be necessary to force a reload, but will delete local pictures)?"), TranslateT("Reset contact picture"), MB_YESNO) == IDYES) {
+			DBVARIANT dbv = { 0 };
+			ProtectAvatar(m_hContact, 0);
+			if (!db_get_ws(m_hContact, "ContactPhoto", "File", &dbv)) {
+				DeleteFileW(dbv.pwszVal);
+				db_free(&dbv);
+			}
+		}
+		db_unset(m_hContact, "ContactPhoto", "Locked");
+		db_unset(m_hContact, "ContactPhoto", "Backup");
+		db_unset(m_hContact, "ContactPhoto", "RFile");
+		db_unset(m_hContact, "ContactPhoto", "File");
+		db_unset(m_hContact, "ContactPhoto", "Format");
+		DeleteAvatarFromCache(m_hContact, FALSE);
+		SetAvatarName();
+		OnAvatarChanged(0, 0, 0);
+	}
+
+	void onChange_Protect(CCtrlCheck *)
+	{
+		ProtectAvatar(m_hContact, chkProtect.IsChecked());
+	}
+
+	void onChange_MakeTrans(CCtrlCheck *)
+	{
+		bool enable = chkMakeTrans.IsChecked();
+		EnableWindow(GetDlgItem(m_hwnd, IDC_BKG_NUM_POINTS_L), enable);
+		EnableWindow(GetDlgItem(m_hwnd, IDC_BKG_NUM_POINTS_SPIN), enable);
+		EnableWindow(GetDlgItem(m_hwnd, IDC_BKG_NUM_POINTS), enable);
+		EnableWindow(GetDlgItem(m_hwnd, IDC_BKG_COLOR_DIFFERENCE_L), enable);
+		EnableWindow(GetDlgItem(m_hwnd, IDC_BKG_COLOR_DIFFERENCE_SPIN), enable);
+		EnableWindow(GetDlgItem(m_hwnd, IDC_BKG_COLOR_DIFFERENCE), enable);
+
+		ReloadAvatar();
+	}
+};
+
+INT_PTR ContactOptions(WPARAM wParam, LPARAM lParam)
+{
+	if (wParam) {
+		auto *pDlg = new CContactAvatarDlg(wParam);
+		if (lParam)
+			pDlg->SetParent((HWND)lParam);
+		pDlg->Create();
+	}
+	return 0;
+}