#pragma once
#include <vector>
#include <map>
#include <windows.h>
#include <thread>
#include <chrono>
using namespace std::chrono_literals;

// https://github.com/erayarslan/WriteProcessMemory-Example
// http://stackoverflow.com/q/32798185
// http://stackoverflow.com/q/36018838
// http://stackoverflow.com/q/1387064
class Memory
{
public:
	Memory(const std::string& processName);
	~Memory();

	template<class T>
	std::vector<T> ReadData(const std::vector<int>& offsets, int numItems) {
		std::vector<T> data;
		data.resize(numItems);
		for (int i=0; i<5; i++) {
			if (ReadProcessMemory(_handle, (LPVOID)ComputeOffset(offsets), &data[0], sizeof(T) * numItems, NULL))
			{
				return data;
			}
			// std::this_thread::sleep_for(10ms);
		}
		ThrowError();
		return {};
	}

	template <class T>
	void WriteData(const std::vector<int>& offsets, const std::vector<T>& data) {
		for (int i=0; i<5; i++) {
			if (WriteProcessMemory(_handle, (LPVOID)ComputeOffset(offsets), &data[0], sizeof(T) * data.size(), NULL)) {
				return;
			}
			// std::this_thread::sleep_for(10ms);
		}
		ThrowError();
	}
	
private:
	void ThrowError();

	uintptr_t ComputeOffset(std::vector<int> offsets);

	std::map<uintptr_t, uintptr_t> _computedAddresses;
	uintptr_t _baseAddress;
	HANDLE _handle;
};