#pragma once

#include "src/common.h"

// Socket includes
#include <cstring>

#ifdef WIN32
	// Define this so I actually get functions out of the windows header file
	#undef _WIN32_WINNT
	#define _WIN32_WINNT 0x0501
	#include <winsock2.h>
	#include <ws2tcpip.h>
#else
	#include <sys/types.h>
	#include <sys/socket.h>
	#include <netdb.h>
#endif

#include <stdio.h>

// libssh2 includes
#include <libssh2.h>
#include <libssh2_sftp.h>

class SFTP {
public:
	SFTP(const char *host, const char *user, const char *pass, const char *base_path="~", const char *port="22");
	~SFTP();
	string getFingerprint();
	string getFile(const char *path);
	bool createDir(const char *path);
	bool writeFile(const char *path, const char *data);
	//StringVector getDirList(const char *path);
private:
	struct addrinfo host_info;
	struct addrinfo *host_info_list = nullptr;
	const char *fingerprint;
	int sock;
	LIBSSH2_SESSION *session = nullptr;
	LIBSSH2_SFTP *sftp_session = nullptr;
	void create_socket(const char *host, const char *port);
	void destroy_socket();
	void ssh_connect();
	void sftp_connect();
};