org.eclipse.om2m.sdt/org.eclipse.om2m.sdt.home.lifx/src/main/java/org/eclipse/om2m/sdt/home/lifx/impl/cloud/DiscoveryCloud.java

/*******************************************************************************
 * Copyright (c) 2014, 2016 Orange.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *******************************************************************************/
package org.eclipse.om2m.sdt.home.lifx.impl.cloud;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Timer;
import java.util.TimerTask;

import org.eclipse.om2m.sdt.home.lifx.impl.Discovery;
import org.eclipse.om2m.sdt.home.lifx.impl.Logger;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.parser.JSONParser;

public class DiscoveryCloud extends Discovery {

	private static final String LIGHT_URL = "https://api.lifx.com/v1/lights/";
	private static final String ALL_LIGHTS = "https://api.lifx.com/v1/lights/all";
	private static URL ALL_LIGHTS_URL;
	static {
		try {
			ALL_LIGHTS_URL = new URL(ALL_LIGHTS);
		} catch (MalformedURLException e) {
		}
	}

	private final String authenticationToken;

	private Map<String, LIFXDeviceCloud> devices = new HashMap<>();;

	private Timer timer;
	private TimerTask timerTask;

	public DiscoveryCloud(final String pAuthenticationToken) {
		authenticationToken = pAuthenticationToken;
		timerTask = new TimerTask() {
			@Override
			public void run() {
				retrieveLIFXDevices();
			}
		};
	}

	@SuppressWarnings("rawtypes")
	public void retrieveLIFXDevices() {
		JSONArray jsonArray = retrieveLIFXDevice(null, authenticationToken);
		if (jsonArray != null) {
			for (Iterator it = jsonArray.iterator(); it.hasNext();) {
				JSONObject jsonLifxDevice = (JSONObject) it.next();
				LIFXDeviceCloud lifxDeviceCloud = LIFXDeviceCloud.fromJson(jsonLifxDevice, this.authenticationToken);
				addOrRemoveLIFXDevice(lifxDeviceCloud);
			}
		}
	}

	/**
	 * 
	 * @param url
	 *            if null, retrieve all devices
	 * @param lifxDevice
	 */
	public static JSONArray retrieveLIFXDevice(String url, String authenticationToken) {
		try {
			HttpURLConnection httpUrlConnection = null;
			if (url != null) {
				httpUrlConnection = (HttpURLConnection) new URL(url).openConnection();
			} else {
				httpUrlConnection = (HttpURLConnection) ALL_LIGHTS_URL.openConnection();
			}
			httpUrlConnection.setRequestMethod("GET");
			httpUrlConnection.setRequestProperty("Authorization", "Bearer " + authenticationToken);
			httpUrlConnection.setDoOutput(false);
			httpUrlConnection.setDoInput(true);

			httpUrlConnection.connect();

			if (httpUrlConnection.getResponseCode() == 200) {
				// OK
				InputStream is = httpUrlConnection.getInputStream();
				BufferedReader br = new BufferedReader(new InputStreamReader(is));
				String line = null;
				String finalLine = "";
				while ((line = br.readLine()) != null) {
					finalLine += line;
				}
				return (JSONArray) new JSONParser().parse(finalLine);
			}
		} catch (Exception e) {
			Logger.getInstance().warning(DiscoveryCloud.class, e.toString());
		}
		return null;
	}

	private void addOrRemoveLIFXDevice(LIFXDeviceCloud device) {
		if (device.isConnected()) {
			boolean deviceAdded = false;
			synchronized (devices) {
				if (!devices.containsKey(device.getId())) {
					devices.put(device.getId(), device);
					deviceAdded = true;
				}
			}
			if (deviceAdded) {
				notifyAllListeners_DeviceArrived(device);
			}
		} else {
			// device disconnected
			boolean deviceRemoved = false;
			// 2016 11 28 - BONNARDEL Gregory
			// do not remove LIFX device
			// synchronized (devices) {
			// if (devices.containsKey(device.getId())) {
			// devices.remove(device.getId());
			// deviceRemoved = true;
			// }
			// }

			if (deviceRemoved) {
				notifyAllListeners_DeviceLeft(device);
			}
		}
	}

	@Override
	public void startDiscoveryTask() {
		Logger.getInstance().info(DiscoveryCloud.class, "Start discovery");
		timer = new Timer();
		timer.schedule(timerTask, 0, 30000);
	}

	@Override
	public void stopDiscoveryTask() {
		Logger.getInstance().info(DiscoveryCloud.class, "Stop discovery");
		if (timer != null) {
			timer.cancel();
			timer = null;
		}
	}

	public static void updateLightState(LIFXDeviceCloud lifxDevice) {
		JSONArray jsonArray = retrieveLIFXDevice(LIGHT_URL + lifxDevice.getId(), lifxDevice.getAuthenticationToken());
		if (jsonArray != null) {
			lifxDevice.updateLightState((JSONObject) jsonArray.get(0));
		}
	}

	public static void setLightPower(LIFXDeviceCloud lifxDevice, String power, Double hue, Double saturation,
			Double kelvin, Double brightness, Double duration) throws MalformedURLException, IOException {
		HttpURLConnection httpUrlConnection = null;
		httpUrlConnection = (HttpURLConnection) new URL(LIGHT_URL + lifxDevice.getId() + "/state").openConnection();
		httpUrlConnection.setRequestMethod("PUT");
		httpUrlConnection.setRequestProperty("Authorization", "Bearer " + lifxDevice.getAuthenticationToken());
		httpUrlConnection.setRequestProperty("Content-Type", "application/json");

		String data = "{";
		if (power != null) {
			data += "\"power\":\"" + power + "\",";
		}

		if (brightness != null) {
			data += "\"brightness\":" + brightness + ",";
		}
		if ((hue != null) || (saturation != null) || (kelvin != null)) {
			data += "\"color\":\"";
			if (hue != null) {
				data += "hue:" + hue + " ";
			}
			if (kelvin != null) {
				data += "kelvin:" + kelvin + " ";
			}
			if (saturation != null) {
				data += "saturation:" + saturation;
			}

			data += "\",";
		}
		if (duration != null) {
			data += "\"duration\":" + duration + ",";
		}
		if (data.endsWith(",")) {
			data = data.substring(0, data.length() - 1);
		}
		data += "}";

		httpUrlConnection.setDoOutput(true);
		httpUrlConnection.setDoInput(true);

		httpUrlConnection.getOutputStream().write(data.getBytes());
		httpUrlConnection.connect();
		System.out.println(httpUrlConnection.getResponseMessage());

		if (httpUrlConnection.getResponseCode() == 207) {

		}
	}

	public static void setLightPower(LIFXDeviceCloud lifxDevice, String power, Integer red, Integer green, Integer blue,
			Integer duration) throws MalformedURLException, IOException {
		HttpURLConnection httpUrlConnection = null;
		httpUrlConnection = (HttpURLConnection) new URL(LIGHT_URL + lifxDevice.getId() + "/state").openConnection();
		httpUrlConnection.setRequestMethod("PUT");
		httpUrlConnection.setRequestProperty("Authorization", "Bearer " + lifxDevice.getAuthenticationToken());
		httpUrlConnection.setRequestProperty("Content-Type", "application/json");

		String data = "{";
		if (power != null) {
			data += "\"power\":\"" + power + "\",";
		}

		if ((red != null) || (green != null) || (blue != null)) {
			data += "\"color\":\"rgb:";
			data += red;
			data += ",";
			data += green;
			data += ",";
			data += blue;
			data += "\",";
		}
		if (duration != null) {
			data += "\"duration\":" + duration + ",";
		}
		if (data.endsWith(",")) {
			data = data.substring(0, data.length() - 1);
		}
		data += "}";

		httpUrlConnection.setDoOutput(true);
		httpUrlConnection.setDoInput(true);

		httpUrlConnection.getOutputStream().write(data.getBytes());
		httpUrlConnection.connect();
		System.out.println(httpUrlConnection.getResponseMessage());

		if (httpUrlConnection.getResponseCode() == 207) {

		}
	}

	public List<LIFXDeviceCloud> getLIFXDeviceCloud() {
		List<LIFXDeviceCloud> toBeReturned = new ArrayList<>();
		synchronized (devices) {
			toBeReturned.addAll(devices.values());
		}
		return toBeReturned;
	}

}