// Copyright 2005-2019 The Mumble Developers. All rights reserved.
// Use of this source code is governed by a BSD-style license
// that can be found in the LICENSE file at the root of the
// Mumble source tree or at <https://www.mumble.info/LICENSE>.

#include "../mumble_plugin_win32.h" // Include standard plugin header.
#include "../mumble_plugin_utils.h" // Include plugin header for special functions, like "escape".

// Variable to contain module's addresses
procptr_t RendDX9 = 0;


static int fetch(float *avatar_pos, float *avatar_front, float *avatar_top, float *camera_pos, float *camera_front, float *camera_top, std::string &context, std::wstring &identity) {
	bool ok;
	char server_name[100], soldier_name[100];
	BYTE team_id, squad_id, is_commander, is_squad_leader, target_squad_id, on_voip, on_voip_com;

	for (int i=0;i<3;i++)
		avatar_pos[i] = avatar_front[i] = avatar_top[i] = camera_pos[i] = camera_front[i] = camera_top[i] = 0.0f;


	// Server name pointers
	// Doubles as an in-server state variable (NULL if not in a game)
	procptr_t server_offset_0 = peekProcPtr(pModule + 0x61B5D8);
	if (!server_offset_0) {
		context.clear(); // Clear context
		identity.clear(); // Clear identity

		return true; // This tells Mumble to ignore all vectors.
	}

	procptr_t server_offset_1 = peekProcPtr(server_offset_0 + 0x108);
	if (!server_offset_1) return false;

	// Team ID pointers
	procptr_t team_offset_0 = peekProcPtr(pModule + 0x624430);
	if (!team_offset_0) return false;
	procptr_t team_offset_1 = peekProcPtr(team_offset_0 + 0x1B4);
	if (!team_offset_1) return false;

	// Commander, SquadLeader, SquadID base pointers
	procptr_t role_offset_0 = peekProcPtr(pModule + 0x685228);
	if (!role_offset_0) return false;
	procptr_t role_offset_1 = peekProcPtr(role_offset_0 + 0x6C);
	if (!role_offset_1) return false;

	// Tactical pointers
	procptr_t tactical_offset_0 = peekProcPtr(pModule + 0x72D148);
	if (!tactical_offset_0) return false;
	procptr_t tactical_offset_1 = peekProcPtr(tactical_offset_0 + 0xE4);
	if (!tactical_offset_1) return false;

	// VoiP pointers
	procptr_t voip_offset_0 = peekProcPtr(pModule + 0x620D30);
	if (!voip_offset_0) return false;

	// Peekproc and assign game addresses to our containers, so we can retrieve positional data
	ok = peekProc(pModule + 0x6697C8      , avatar_pos, 12) &&   // Avatar position values (X, Y and Z).
	     peekProc(RendDX9 + 0x20E9F4      , camera_pos, 12) &&   // Camera position values (X, Y and Z).
	     peekProc(RendDX9 + 0x2139D4      , camera_front, 12) && // Avatar front vector values (X, Y and Z).
	     peekProc(RendDX9 + 0x2139C4      , camera_top, 12) &&   // Avatar top vector values (X, Y and Z).
	     peekProc(server_offset_1         , server_name) &&      // Server name.
	     peekProc(pModule + 0x6228B9      , soldier_name) &&     // Soldier name.
	     peekProc(team_offset_1 + 0xA0    , team_id) &&          // Team ID (0 for PAC, 1 for EU).
	     peekProc(role_offset_1 + 0x108   , squad_id);           // Squad ID (0 for none, 1-9 for squads).
	     peekProc(role_offset_1 + 0x10C   , is_commander) &&     // Whether Commander.
	     peekProc(role_offset_1 + 0x10D   , is_squad_leader) &&  // Whether Squad Leader.
	     peekProc(tactical_offset_1 + 0xCC, target_squad_id) &&  // Commander's selected squad.
	     peekProc(voip_offset_0 + 0xA1    , on_voip) &&          // Whether VoiP is active.
	     peekProc(voip_offset_0 + 0xA0    , on_voip_com);        // Whether VoiP to/from commander is active.

	// This prevents the plugin from linking to the game in case something goes wrong during values retrieval from memory addresses.
	if (!ok)
		return false;


	// Begin context
	std::ostringstream ocontext;

	// Server name
	escape(server_name, sizeof(server_name));

	if (strcmp(server_name, "") != 0)
		ocontext << " {\"ipport\": \"" << server_name << "\"}";

	context = ocontext.str();
	// End context

	// Begin identity
	std::wostringstream oidentity;

	// Soldier name
	escape(soldier_name, sizeof(soldier_name));

	oidentity << "{"
	          << "\"ipport\": \"" << server_name << "\", "
	          << "\"team_id\": " << team_id << ", "
	          << "\"squad_id\": " << squad_id << ", "
	          << "\"is_commander\": " << (is_commander ? "true" : "false") << ", "
	          << "\"is_squad_leader\": " << (is_squad_leader ? "true" : "false") << ", "
	          << "\"target_squad_id\": " << target_squad_id << ", "
	          << "\"on_voip\": " << (on_voip ? "true" : "false") << ", "
	          << "\"on_voip_com\": " << (on_voip_com ? "true" : "false") << ", "
	          << "\"name\": \"" << soldier_name << "\""
	          << "}";

	identity = oidentity.str();
	// End identity

	// Copy camera direction vectors to avatar
	for (int i=0; i<3; i++) {
		avatar_front[i] = camera_front[i];
		avatar_top[i] = camera_top[i];
	}

	return true;
}

static int trylock(const std::multimap<std::wstring, unsigned long long int> &pids) {

	if (!initialize(pids, L"BF2142.exe")) // Retrieve game executable's memory address
		return false;

	RendDX9 = getModuleAddr(L"RendDX9.dll"); // Retrieve "RendDX9.dll" module's memory address
	// This prevents the plugin from linking to the game in case something goes wrong during module's memory address retrieval.
	if (!RendDX9)
		return false;

	// Check if we can get meaningful data from it
	float apos[3], afront[3], atop[3], cpos[3], cfront[3], ctop[3];
	std::wstring sidentity;
	std::string scontext;

	if (fetch(apos, afront, atop, cpos, cfront, ctop, scontext, sidentity)) {
		return true;
	} else {
		generic_unlock();
		return false;
	}
}

static const std::wstring longdesc() {
	return std::wstring(L"Supports Battlefield 2142 version 1.51."); // Plugin long description
}

static std::wstring description(L"Battlefield 2142 1.51"); // Plugin short description
static std::wstring shortname(L"Battlefield 2142"); // Plugin short name

static int trylock1() {
	return trylock(std::multimap<std::wstring, unsigned long long int>());
}

static MumblePlugin bf2142plug = {
	MUMBLE_PLUGIN_MAGIC,
	description,
	shortname,
	NULL,
	NULL,
	trylock1,
	generic_unlock,
	longdesc,
	fetch
};

static MumblePlugin2 bf2142plug2 = {
	MUMBLE_PLUGIN_MAGIC_2,
	MUMBLE_PLUGIN_VERSION,
	trylock
};

extern "C" MUMBLE_PLUGIN_EXPORT MumblePlugin *getMumblePlugin() {
	return &bf2142plug;
}

extern "C" MUMBLE_PLUGIN_EXPORT MumblePlugin2 *getMumblePlugin2() {
	return &bf2142plug2;
}
