// This file is under GNU General Public License 3.0

 // see LICENSE.txt

 #include <stdlib.h>

 #include <string>

 #include <iostream>

 #include <vector>

 #include <cstring> // for strcmp()

 #include "keymanagement.h"

 #include "message_api.h"

 #include "mime.h"

 #include "test_util.h"

 #include "pEpEngine.h"

 #include <cpptest.h>

 #include "EngineTestSessionSuite.h"

 #include "LeastColorGroupTests.h"

 using namespace std;

 LeastColorGroupTests::LeastColorGroupTests(string suitename, string test_home_dir) :

     EngineTestSessionSuite::EngineTestSessionSuite(suitename, test_home_dir) {

     add_test_to_suite(std::pair<std::string, void (Test::Suite::*)()>(string("LeastColorGroupTests::check_least_color_group"),

                                                                       static_cast<Func>(&LeastColorGroupTests::check_least_color_group)));

 }

 void LeastColorGroupTests::check_least_color_group() {

     const char* mailfile = "test_mails/color_test.eml";

     const std::vector<const char*> keynames = {

                               "test_keys/priv/pep.color.test.P-0x3EBE215C_priv.asc",

                               "test_keys/pub/pep.color.test.H-0xD17E598E_pub.asc",

                               "test_keys/pub/pep.color.test.L-0xE9CDB4CE_pub.asc",

                               "test_keys/pub/pep.color.test.P-0x3EBE215C_pub.asc",

                               "test_keys/pub/pep.color.test.V-0x71FC6D28_pub.asc"

                           };

     for (auto name : keynames) {

         cout << "\t read keyfile \"" << name << "\"..." << std::endl;

         const string keytextkey = slurp(name);

         PEP_STATUS statuskey = import_key(session, keytextkey.c_str(), keytextkey.length(), NULL);

         TEST_ASSERT_MSG((statuskey == PEP_KEY_IMPORTED), "statuskey == PEP_STATUS_OK");

     }

     cout << "\t read keyfile mailfile \"" << mailfile << "\"..." << std::endl;

     const string mailtext = slurp(mailfile);

     cout << "\t All files read successfully." << std::endl;

     pEp_identity * me1 = new_identity("pep.color.test.P@kgrothoff.org", 

                                       "7EE6C60C68851954E1797F81EA59715E3EBE215C", 

                                       PEP_OWN_USERID, "Pep Color Test P (recip)");

     me1->me = true;

     PEP_STATUS status = myself(session, me1);

     pEp_identity * sender1 = new_identity("pep.color.test.V@kgrothoff.org",

                                           NULL, "TOFU_pep.color.test.V@kgrothoff.org",

                                           "Pep Color Test V (sender)");

     status = update_identity(session, sender1);

     trust_personal_key(session, sender1);

     status = update_identity(session, sender1);

     message* msg_ptr = nullptr;

     message* dest_msg = nullptr;

     message* final_ptr = nullptr;

     stringlist_t* keylist = nullptr;

     PEP_rating rating;

     PEP_decrypt_flags_t flags;

     status = mime_decode_message(mailtext.c_str(), mailtext.length(), &msg_ptr);

     TEST_ASSERT_MSG((status == PEP_STATUS_OK), "status == PEP_STATUS_OK");

     TEST_ASSERT_MSG((msg_ptr), "msg_ptr");

     final_ptr = msg_ptr;

     flags = 0;

     status = decrypt_message(session, msg_ptr, &dest_msg, &keylist, &rating, &flags);

     final_ptr = dest_msg ? dest_msg : msg_ptr;

     cout << "shortmsg: " << final_ptr->shortmsg << endl << endl;

     cout << "longmsg: " << final_ptr->longmsg << endl << endl;

     cout << "longmsg_formatted: " << (final_ptr->longmsg_formatted ? final_ptr->longmsg_formatted : "(empty)") << endl << endl;

     cout << "rating: " << rating << endl << endl;

     cout << "keys used: " << endl;

     int i = 0;

     for (stringlist_t* k = keylist; k; k = k->next) {

         if (i == 0)

             cout << "\t Signer (key 0):\t" << k->value << endl;

         else

             cout << "\t #" << i << ":\t" << k->value << endl;

         i++;

     }

 //    free_identity(me1);

     if (final_ptr == dest_msg)

     	free_message(dest_msg);

     free_message(msg_ptr);

     free_stringlist(keylist);    

 }