1 files changed, 40 insertions, 2 deletions
diff --git a/srmm.cpp b/srmm.cpp
index 6d00040..d5f6061 100644
--- a/srmm.cpp
+++ b/srmm.cpp
@@ -41,7 +41,45 @@ int __cdecl onIconPressed(WPARAM wParam, LPARAM lParam) {
 
 	StatusIconClickData *sicd = (StatusIconClickData *)lParam;
 	if(strcmp(sicd->szModule, szGPGModuleName) !=  0) 
-		return 0; // not our event
-
+		return 0; // not our event
+	
+	void setSrmmIcon(HANDLE);
+	void setClistIcon(HANDLE);
+	TCHAR *key = UniGetContactSettingUtf(hContact, szGPGModuleName, "GPGPubKey", _T(""));
+	BYTE enc = DBGetContactSettingByte(hContact, szGPGModuleName, "GPGEncryption", 0);
+	if(enc)
+	{
+		DBWriteContactSettingByte(hContact, szGPGModuleName, "GPGEncryption", 0);
+		setSrmmIcon(hContact);
+		setClistIcon(hContact);
+	}
+	else if(!enc)
+	{
+		if(_tcslen(key) < 1)
+		{
+			void ShowLoadPublicKeyDialog();
+			extern map<int, HANDLE> user_data;
+			extern int item_num;
+			item_num = 0;		 //black magic here
+			user_data[1] = hContact;
+			ShowLoadPublicKeyDialog();
+		}
+		else
+		{
+			DBWriteContactSettingByte(hContact, szGPGModuleName, "GPGEncryption", 1);
+			setSrmmIcon(hContact);
+			setClistIcon(hContact);
+			mir_free(key);
+			return 0;
+		}
+		mir_free(key);
+		key = UniGetContactSettingUtf(hContact, szGPGModuleName, "GPGPubKey", _T(""));
+		if(_tcslen(key) > 0)
+		{
+			DBWriteContactSettingByte(hContact, szGPGModuleName, "GPGEncryption", 1);
+			setSrmmIcon(hContact);
+			setClistIcon(hContact);
+		}
+	}
 	return 0;
 }