modified: main.cpp

	modified:   messages.cpp
	modified:   new_gpg.rc
	modified:   new_gpg.vcproj

1 files changed, 192 insertions, 47 deletions

diff --git a/main.cpp b/main.cpp
index c6b899c..1cefa44 100644
--- a/main.cpp
+++ b/main.cpp
@@ -845,7 +845,7 @@ static BOOL CALLBACK DlgProcKeyGenDialog(HWND hwndDlg, UINT msg, WPARAM wParam,
 				  f<<"\n";
 				  tmp = new TCHAR [64]; //i hope this is enough for password
 				  GetDlgItemText(hwndDlg, IDC_KEY_PASSWD, tmp, 64);
-				  if(_tcslen(tmp) > 0)
+				  if(!tmp[0])
 				  {
 					  f<<"Passphrase: ";
 					  tmp2 = mir_utf8encodeW(tmp);
@@ -864,7 +864,7 @@ static BOOL CALLBACK DlgProcKeyGenDialog(HWND hwndDlg, UINT msg, WPARAM wParam,
 				  f<<"\n";
 				  tmp = new TCHAR [512];
 				  GetDlgItemText(hwndDlg, IDC_KEY_COMMENT, tmp, 512);
-				  if(_tcslen(tmp) > 0)
+				  if(!tmp[0])
 				  {
 					  tmp2 = mir_utf8encodeW(tmp);
 					  f<<"Name-Comment: ";
@@ -1428,7 +1428,29 @@ void ImportKey()
 	extern boost::mutex new_key_hcnt_mutex;
 	HANDLE hContact = new_key_hcnt;
 	new_key_hcnt_mutex.unlock();
-	DBWriteContactSettingTString(hContact, szGPGModuleName, "GPGPubKey", new_key.c_str());
+	bool for_all_sub = false;
+	if(metaIsProtoMetaContacts(hContact))
+		if(MessageBox(0, _T("Do you want load key for all subcontacts ?"), _T("Metacontact detected"), MB_YESNO) == IDYES)
+			for_all_sub = true;
+
+	if(metaIsProtoMetaContacts(hContact))
+	{
+		HANDLE hcnt = NULL;
+		if(for_all_sub)
+		{
+			int count = metaGetContactsNum(hContact);
+			for(int i = 0; i < count; i++)
+			{
+				hcnt = metaGetSubcontact(hContact, i);
+				if(hcnt)
+					DBWriteContactSettingTString(hcnt, szGPGModuleName, "GPGPubKey", new_key.c_str());
+			}
+		}
+		else
+			DBWriteContactSettingTString(metaGetCurrent(hContact), szGPGModuleName, "GPGPubKey", new_key.c_str());
+	}
+	else
+		DBWriteContactSettingTString(hContact, szGPGModuleName, "GPGPubKey", new_key.c_str());
 	new_key.clear();
 	{ //gpg execute block
 		wstring cmd;
@@ -1473,57 +1495,180 @@ void ImportKey()
 			MessageBox(0, _T("Set path to gpg.exe first!"), _T("Warning"), MB_OK);				  
 			return;
 		}
-		DBDeleteContactSetting(hContact, szGPGModuleName, "bAlwatsTrust");
 		{
-			char *tmp = NULL;
-			string::size_type s = output.find("gpg: key ") + strlen("gpg: key ");
-			string::size_type s2 = output.find(":", s);
-			DBWriteContactSettingString(hContact, szGPGModuleName, "KeyID", output.substr(s,s2-s).c_str());
-			s2+=2;
-			s = output.find("“", s2);
-			if(s == string::npos)
+			if(metaIsProtoMetaContacts(hContact))
 			{
-				s = output.find("\"", s2);
-				s += 1;
-			}
-			else
-				s += 3;
-			if((s2 = output.find("(", s)) == string::npos)
-				s2 = output.find("<", s);
-			else if(s2 > output.find("<", s))
-				s2 = output.find("<", s);
-			tmp = new char [output.substr(s,s2-s-1).length()+1];
-			strcpy(tmp, output.substr(s,s2-s-1).c_str());
-			mir_utf8decode(tmp, 0);
-			DBWriteContactSettingString(hContact, szGPGModuleName, "KeyMainName", tmp);
-			mir_free(tmp);
-			if((s = output.find(")", s2)) == string::npos)
-				s = output.find(">", s2);
-			else if(s > output.find(">", s2))
-				s = output.find(">", s2);
-			s2++;
-			if(output[s] == ')')
-			{
-				tmp = new char [output.substr(s2,s-s2).length()+1];
-				strcpy(tmp, output.substr(s2,s-s2).c_str());
-				mir_utf8decode(tmp, 0);
-				DBWriteContactSettingString(hContact, szGPGModuleName, "KeyComment", tmp);
-				mir_free(tmp);
-				s+=3;
-				s2 = output.find(">", s);
-				tmp = new char [output.substr(s,s2-s).length()+1];
-				strcpy(tmp, output.substr(s,s2-s).c_str());
-				mir_utf8decode(tmp, 0);
-				DBWriteContactSettingString(hContact, szGPGModuleName, "KeyMainEmail", tmp);
-				mir_free(tmp);
+				HANDLE hcnt = NULL;
+				if(for_all_sub)
+				{
+					int count = metaGetContactsNum(hContact);
+					for(int i = 0; i < count; i++)
+					{
+						hcnt = metaGetSubcontact(hContact, i);
+						if(hcnt)
+						{
+							char *tmp = NULL;
+							string::size_type s = output.find("gpg: key ") + strlen("gpg: key ");
+							string::size_type s2 = output.find(":", s);
+							DBWriteContactSettingString(hcnt, szGPGModuleName, "KeyID", output.substr(s,s2-s).c_str());
+							s2+=2;
+							s = output.find("“", s2);
+							if(s == string::npos)
+							{
+								s = output.find("\"", s2);
+								s += 1;
+							}
+							else
+								s += 3;
+							if((s2 = output.find("(", s)) == string::npos)
+								s2 = output.find("<", s);
+							else if(s2 > output.find("<", s))
+								s2 = output.find("<", s);
+							tmp = new char [output.substr(s,s2-s-1).length()+1];
+							strcpy(tmp, output.substr(s,s2-s-1).c_str());
+							mir_utf8decode(tmp, 0);
+							DBWriteContactSettingString(hcnt, szGPGModuleName, "KeyMainName", tmp);
+							mir_free(tmp);
+							if((s = output.find(")", s2)) == string::npos)
+								s = output.find(">", s2);
+							else if(s > output.find(">", s2))
+								s = output.find(">", s2);
+							s2++;
+							if(output[s] == ')')
+							{
+								tmp = new char [output.substr(s2,s-s2).length()+1];
+								strcpy(tmp, output.substr(s2,s-s2).c_str());
+								mir_utf8decode(tmp, 0);
+								DBWriteContactSettingString(hcnt, szGPGModuleName, "KeyComment", tmp);
+								mir_free(tmp);
+								s+=3;
+								s2 = output.find(">", s);
+								tmp = new char [output.substr(s,s2-s).length()+1];
+								strcpy(tmp, output.substr(s,s2-s).c_str());
+								mir_utf8decode(tmp, 0);
+								DBWriteContactSettingString(hcnt, szGPGModuleName, "KeyMainEmail", tmp);
+								mir_free(tmp);
+							}
+							else
+							{
+								tmp = new char [output.substr(s2,s-s2).length()+1];
+								strcpy(tmp, output.substr(s2,s-s2).c_str());
+								mir_utf8decode(tmp, 0);
+								DBWriteContactSettingString(hcnt, szGPGModuleName, "KeyMainEmail", output.substr(s2,s-s2).c_str());
+								mir_free(tmp);
+							}
+							DBDeleteContactSetting(hcnt, szGPGModuleName, "bAlwatsTrust");
+						}
+					}
+				}
+				else
+				{
+					char *tmp = NULL;
+					string::size_type s = output.find("gpg: key ") + strlen("gpg: key ");
+					string::size_type s2 = output.find(":", s);
+					DBWriteContactSettingString(metaGetCurrent(hContact), szGPGModuleName, "KeyID", output.substr(s,s2-s).c_str());
+					s2+=2;
+					s = output.find("“", s2);
+					if(s == string::npos)
+					{
+						s = output.find("\"", s2);
+						s += 1;
+					}
+					else
+						s += 3;
+					if((s2 = output.find("(", s)) == string::npos)
+						s2 = output.find("<", s);
+					else if(s2 > output.find("<", s))
+						s2 = output.find("<", s);
+					tmp = new char [output.substr(s,s2-s-1).length()+1];
+					strcpy(tmp, output.substr(s,s2-s-1).c_str());
+					mir_utf8decode(tmp, 0);
+					DBWriteContactSettingString(metaGetCurrent(hContact), szGPGModuleName, "KeyMainName", tmp);
+					mir_free(tmp);
+					if((s = output.find(")", s2)) == string::npos)
+						s = output.find(">", s2);
+					else if(s > output.find(">", s2))
+						s = output.find(">", s2);
+					s2++;
+					if(output[s] == ')')
+					{
+						tmp = new char [output.substr(s2,s-s2).length()+1];
+						strcpy(tmp, output.substr(s2,s-s2).c_str());
+						mir_utf8decode(tmp, 0);
+						DBWriteContactSettingString(metaGetCurrent(hContact), szGPGModuleName, "KeyComment", tmp);
+						mir_free(tmp);
+						s+=3;
+						s2 = output.find(">", s);
+						tmp = new char [output.substr(s,s2-s).length()+1];
+						strcpy(tmp, output.substr(s,s2-s).c_str());
+						mir_utf8decode(tmp, 0);
+						DBWriteContactSettingString(metaGetCurrent(hContact), szGPGModuleName, "KeyMainEmail", tmp);
+						mir_free(tmp);
+					}
+					else
+					{
+						tmp = new char [output.substr(s2,s-s2).length()+1];
+						strcpy(tmp, output.substr(s2,s-s2).c_str());
+						mir_utf8decode(tmp, 0);
+						DBWriteContactSettingString(metaGetCurrent(hContact), szGPGModuleName, "KeyMainEmail", output.substr(s2,s-s2).c_str());
+						mir_free(tmp);
+					}
+					DBDeleteContactSetting(metaGetCurrent(hContact), szGPGModuleName, "bAlwatsTrust");
+				}
 			}
 			else
 			{
-				tmp = new char [output.substr(s2,s-s2).length()+1];
-				strcpy(tmp, output.substr(s2,s-s2).c_str());
+				char *tmp = NULL;
+				string::size_type s = output.find("gpg: key ") + strlen("gpg: key ");
+				string::size_type s2 = output.find(":", s);
+				DBWriteContactSettingString(hContact, szGPGModuleName, "KeyID", output.substr(s,s2-s).c_str());
+				s2+=2;
+				s = output.find("“", s2);
+				if(s == string::npos)
+				{
+					s = output.find("\"", s2);
+					s += 1;
+				}
+				else
+					s += 3;
+				if((s2 = output.find("(", s)) == string::npos)
+					s2 = output.find("<", s);
+				else if(s2 > output.find("<", s))
+					s2 = output.find("<", s);
+				tmp = new char [output.substr(s,s2-s-1).length()+1];
+				strcpy(tmp, output.substr(s,s2-s-1).c_str());
 				mir_utf8decode(tmp, 0);
-				DBWriteContactSettingString(hContact, szGPGModuleName, "KeyMainEmail", output.substr(s2,s-s2).c_str());
+				DBWriteContactSettingString(hContact, szGPGModuleName, "KeyMainName", tmp);
 				mir_free(tmp);
+				if((s = output.find(")", s2)) == string::npos)
+					s = output.find(">", s2);
+				else if(s > output.find(">", s2))
+					s = output.find(">", s2);
+				s2++;
+				if(output[s] == ')')
+				{
+					tmp = new char [output.substr(s2,s-s2).length()+1];
+					strcpy(tmp, output.substr(s2,s-s2).c_str());
+					mir_utf8decode(tmp, 0);
+					DBWriteContactSettingString(hContact, szGPGModuleName, "KeyComment", tmp);
+					mir_free(tmp);
+					s+=3;
+					s2 = output.find(">", s);
+					tmp = new char [output.substr(s,s2-s).length()+1];
+					strcpy(tmp, output.substr(s,s2-s).c_str());
+					mir_utf8decode(tmp, 0);
+					DBWriteContactSettingString(hContact, szGPGModuleName, "KeyMainEmail", tmp);
+					mir_free(tmp);
+				}
+				else
+				{
+					tmp = new char [output.substr(s2,s-s2).length()+1];
+					strcpy(tmp, output.substr(s2,s-s2).c_str());
+					mir_utf8decode(tmp, 0);
+					DBWriteContactSettingString(hContact, szGPGModuleName, "KeyMainEmail", output.substr(s2,s-s2).c_str());
+					mir_free(tmp);
+				}
+				DBDeleteContactSetting(hContact, szGPGModuleName, "bAlwatsTrust");
 			}
 		}
 		ptmp = mir_utf8decodeW(output.c_str());