package org.briarproject.briar.android.reporting;

import android.app.ActivityManager;
import android.bluetooth.BluetoothAdapter;
import android.content.Context;
import android.net.ConnectivityManager;
import android.net.NetworkInfo;
import android.net.wifi.WifiInfo;
import android.net.wifi.WifiManager;
import android.os.Build;
import android.os.Environment;
import android.os.Handler;
import android.os.Looper;
import android.provider.Settings;
import android.support.annotation.NonNull;

import org.acra.builder.ReportBuilder;
import org.acra.builder.ReportPrimer;
import org.briarproject.bramble.util.StringUtils;
import org.briarproject.briar.BuildConfig;

import java.io.File;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.FutureTask;

import static android.bluetooth.BluetoothAdapter.SCAN_MODE_CONNECTABLE;
import static android.bluetooth.BluetoothAdapter.SCAN_MODE_CONNECTABLE_DISCOVERABLE;
import static android.content.Context.ACTIVITY_SERVICE;
import static android.content.Context.CONNECTIVITY_SERVICE;
import static android.content.Context.WIFI_P2P_SERVICE;
import static android.content.Context.WIFI_SERVICE;
import static android.net.ConnectivityManager.TYPE_MOBILE;
import static android.net.ConnectivityManager.TYPE_WIFI;
import static android.net.wifi.WifiManager.WIFI_STATE_ENABLED;
import static org.briarproject.bramble.util.PrivacyUtils.scrubMacAddress;

public class BriarReportPrimer implements ReportPrimer {

	@Override
	public void primeReport(@NonNull Context ctx,
			@NonNull ReportBuilder builder) {
		CustomDataTask task = new CustomDataTask(ctx);
		FutureTask<Map<String, String>> futureTask = new FutureTask<>(task);
		// Use a new thread as the Android executor thread may have died
		new SingleShotAndroidExecutor(futureTask).start();
		try {
			builder.customData(futureTask.get());
		} catch (InterruptedException | ExecutionException e) {
			builder.customData("Custom data exception", e.toString());
		}
	}

	private static class CustomDataTask
			implements Callable<Map<String, String>> {

		private final Context ctx;

		private CustomDataTask(Context ctx) {
			this.ctx = ctx;
		}

		@Override
		public Map<String, String> call() {
			Map<String, String> customData = new LinkedHashMap<>();

			// System memory
			Object o = ctx.getSystemService(ACTIVITY_SERVICE);
			ActivityManager am = (ActivityManager) o;
			ActivityManager.MemoryInfo mem = new ActivityManager.MemoryInfo();
			am.getMemoryInfo(mem);
			String systemMemory;
			if (Build.VERSION.SDK_INT >= 16) {
				systemMemory = (mem.totalMem / 1024 / 1024) + " MiB total, "
						+ (mem.availMem / 1024 / 1204) + " MiB free, "
						+ (mem.threshold / 1024 / 1024) + " MiB threshold";
			} else {
				systemMemory = (mem.availMem / 1024 / 1204) + " MiB free, "
						+ (mem.threshold / 1024 / 1024) + " MiB threshold";
			}
			customData.put("System memory", systemMemory);

			// Virtual machine memory
			Runtime runtime = Runtime.getRuntime();
			long heap = runtime.totalMemory();
			long heapFree = runtime.freeMemory();
			long heapMax = runtime.maxMemory();
			String vmMemory = (heap / 1024 / 1024) + " MiB allocated, "
					+ (heapFree / 1024 / 1024) + " MiB free, "
					+ (heapMax / 1024 / 1024) + " MiB maximum";
			customData.put("Virtual machine memory", vmMemory);

			// Internal storage
			File root = Environment.getRootDirectory();
			long rootTotal = root.getTotalSpace();
			long rootFree = root.getFreeSpace();
			String internal = (rootTotal / 1024 / 1024) + " MiB total, "
					+ (rootFree / 1024 / 1024) + " MiB free";
			customData.put("Internal storage", internal);

			// External storage (SD card)
			File sd = Environment.getExternalStorageDirectory();
			long sdTotal = sd.getTotalSpace();
			long sdFree = sd.getFreeSpace();
			String external = (sdTotal / 1024 / 1024) + " MiB total, "
					+ (sdFree / 1024 / 1024) + " MiB free";
			customData.put("External storage", external);

			// Is mobile data available?
			o = ctx.getSystemService(CONNECTIVITY_SERVICE);
			ConnectivityManager cm = (ConnectivityManager) o;
			NetworkInfo mobile = cm.getNetworkInfo(TYPE_MOBILE);
			boolean mobileAvailable = mobile != null && mobile.isAvailable();
			// Is mobile data enabled?
			boolean mobileEnabled = false;
			try {
				Class<?> clazz = Class.forName(cm.getClass().getName());
				Method method = clazz.getDeclaredMethod("getMobileDataEnabled");
				method.setAccessible(true);
				mobileEnabled = (Boolean) method.invoke(cm);
			} catch (ClassNotFoundException
					| NoSuchMethodException
					| IllegalArgumentException
					| InvocationTargetException
					| IllegalAccessException e) {
				customData.put("Mobile data reflection exception",
						e.toString());
			}
			// Is mobile data connected ?
			boolean mobileConnected = mobile != null && mobile.isConnected();

			String mobileStatus;
			if (mobileAvailable) mobileStatus = "Available, ";
			else mobileStatus = "Not available, ";
			if (mobileEnabled) mobileStatus += "enabled, ";
			else mobileStatus += "not enabled, ";
			if (mobileConnected) mobileStatus += "connected";
			else mobileStatus += "not connected";
			customData.put("Mobile data status", mobileStatus);

			// Is wifi available?
			NetworkInfo wifi = cm.getNetworkInfo(TYPE_WIFI);
			boolean wifiAvailable = wifi != null && wifi.isAvailable();
			// Is wifi enabled?
			o = ctx.getApplicationContext().getSystemService(WIFI_SERVICE);
			WifiManager wm = (WifiManager) o;
			boolean wifiEnabled = wm != null &&
					wm.getWifiState() == WIFI_STATE_ENABLED;
			// Is wifi connected?
			boolean wifiConnected = wifi != null && wifi.isConnected();

			String wifiStatus;
			if (wifiAvailable) wifiStatus = "Available, ";
			else wifiStatus = "Not available, ";
			if (wifiEnabled) wifiStatus += "enabled, ";
			else wifiStatus += "not enabled, ";
			if (wifiConnected) wifiStatus += "connected";
			else wifiStatus += "not connected";
			customData.put("Wi-Fi status", wifiStatus);

			// Is wifi direct supported?
			String wifiDirectStatus = "Supported";
			if (ctx.getSystemService(WIFI_P2P_SERVICE) == null)
				wifiDirectStatus = "Not supported";
			customData.put("Wi-Fi Direct", wifiDirectStatus);

			if (wm != null) {
				WifiInfo wifiInfo = wm.getConnectionInfo();
				if (wifiInfo != null) {
					int ip = wifiInfo.getIpAddress(); // Nice API, Google
					int ip1 = ip & 0xFF;
					int ip2 = (ip >> 8) & 0xFF;
					int ip3 = (ip >> 16) & 0xFF;
					int ip4 = (ip >> 24) & 0xFF;
					String address = ip1 + "." + ip2 + "." + ip3 + "." + ip4;
					customData.put("Wi-Fi address", address);
				}
			}

			// Is Bluetooth available?
			BluetoothAdapter bt = BluetoothAdapter.getDefaultAdapter();
			boolean btAvailable = bt != null;
			// Is Bluetooth enabled?
			boolean btEnabled = bt != null && bt.isEnabled() &&
					!StringUtils.isNullOrEmpty(bt.getAddress());
			// Is Bluetooth connectable?
			boolean btConnectable = bt != null &&
					(bt.getScanMode() == SCAN_MODE_CONNECTABLE ||
							bt.getScanMode() ==
									SCAN_MODE_CONNECTABLE_DISCOVERABLE);
			// Is Bluetooth discoverable?
			boolean btDiscoverable = bt != null &&
					bt.getScanMode() == SCAN_MODE_CONNECTABLE_DISCOVERABLE;
			// Is Bluetooth LE scanning and advertising supported?
			boolean btLeApi = false, btLeScan = false, btLeAdvertise = false;
			if (bt != null && Build.VERSION.SDK_INT >= 21) {
				btLeApi = true;
				btLeScan = bt.getBluetoothLeScanner() != null;
				btLeAdvertise = bt.getBluetoothLeAdvertiser() != null;
			}

			String btStatus;
			if (btAvailable) btStatus = "Available, ";
			else btStatus = "Not available, ";
			if (btEnabled) btStatus += "enabled, ";
			else btStatus += "not enabled, ";
			if (btConnectable) btStatus += "connectable, ";
			else btStatus += "not connectable, ";
			if (btDiscoverable) btStatus += "discoverable";
			else btStatus += "not discoverable";
			customData.put("Bluetooth status", btStatus);
			if (btLeApi) {
				String btLeStatus;
				if (btLeScan) btLeStatus = "Scanning, ";
				else btLeStatus = "No scanning, ";
				if (btLeAdvertise) btLeStatus += "advertising";
				else btLeStatus += "no advertising";
				customData.put("Bluetooth LE status", btLeStatus);
			}

			if (bt != null)
				customData.put("Bluetooth address",
						scrubMacAddress(bt.getAddress()));
			String btSettingsAddr;
			try {
				btSettingsAddr = Settings.Secure.getString(
						ctx.getContentResolver(), "bluetooth_address");
			} catch (SecurityException e) {
				btSettingsAddr = "Could not get address from settings";
			}
			customData.put("Bluetooth address from settings",
					scrubMacAddress(btSettingsAddr));

			// Git commit ID
			customData.put("Commit ID", BuildConfig.GitHash);

			return Collections.unmodifiableMap(customData);
		}
	}

	private static class SingleShotAndroidExecutor extends Thread {

		private final Runnable runnable;

		private SingleShotAndroidExecutor(Runnable runnable) {
			this.runnable = runnable;
		}

		@Override
		public void run() {
			Looper.prepare();
			Handler handler = new Handler();
			handler.post(runnable);
			handler.post(() -> {
				Looper looper = Looper.myLooper();
				if (looper != null) looper.quit();
			});
			Looper.loop();
		}
	}
}
