Continue Refactoring

src/main/java/com/bwssystems/HABridge/plugins/hass/HassHome.java

@@ -0,0 +1,164 @@
+package com.bwssystems.HABridge.plugins.hass;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.bwssystems.HABridge.BridgeSettingsDescriptor;
+import com.bwssystems.HABridge.Home;
+import com.bwssystems.HABridge.NamedIP;
+import com.bwssystems.HABridge.api.CallItem;
+import com.bwssystems.HABridge.api.hue.DeviceState;
+import com.bwssystems.HABridge.api.hue.StateChangeBody;
+import com.bwssystems.HABridge.dao.DeviceDescriptor;
+import com.bwssystems.HABridge.hue.BrightnessDecode;
+import com.bwssystems.HABridge.hue.MultiCommandUtil;
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+
+public class HassHome implements Home {
+    private static final Logger log = LoggerFactory.getLogger(HassHome.class);
+	private Map<String, HomeAssistant> hassMap;
+	private Boolean validHass;
+	private Gson aGsonHandler;
+	
+	public HassHome(BridgeSettingsDescriptor bridgeSettings) {
+		super();
+		createHome(bridgeSettings);
+	}
+
+	@Override
+	public Home createHome(BridgeSettingsDescriptor bridgeSettings) {
+		hassMap = null;
+		aGsonHandler = null;
+		validHass = bridgeSettings.isValidHass();
+		if(!validHass){
+        	log.debug("not a valid hass");
+        } else {
+			hassMap = new HashMap<String,HomeAssistant>();
+			aGsonHandler =
+					new GsonBuilder()
+					.create();
+			Iterator<NamedIP> theList = bridgeSettings.getHassaddress().getDevices().iterator();
+			while(theList.hasNext() && validHass) {
+				NamedIP aHass = theList.next();
+		      	try {
+		      		hassMap.put(aHass.getName(), new HomeAssistant(aHass));
+				} catch (Exception e) {
+			        log.error("Cannot get hass (" + aHass.getName() + ") setup, Exiting with message: " + e.getMessage(), e);
+			        validHass = false;
+				}
+			}
+        }
+		return this;
+	}
+
+	public HomeAssistant getHomeAssistant(String aName) {
+		if(!validHass)
+			return null;
+		HomeAssistant aHomeAssistant;
+		if(aName == null || aName.equals("")) {
+			aHomeAssistant = null;
+			log.debug("Cannot get HomeAssistant for name as it is empty.");
+		}
+		else {
+			aHomeAssistant = hassMap.get(aName);
+			log.debug("Retrieved a HomeAssistant for name: " + aName);
+		}
+		return aHomeAssistant;
+	}
+	
+	@Override
+	public Object getItems(String type) {
+		log.debug("consolidating devices for hass");
+		if(!validHass)
+			return null;
+		List<State> theResponse = null;
+		Iterator<String> keys = hassMap.keySet().iterator();
+		List<HassDevice> deviceList = new ArrayList<HassDevice>();
+		while(keys.hasNext()) {
+			String key = keys.next();
+			theResponse = hassMap.get(key).getDevices();
+			if(theResponse != null)
+				addHassDevices(deviceList, theResponse, key);
+			else {
+				log.warn("Cannot get devices for HomeAssistant with name: " + key + ", skipping this Hass.");
+				continue;
+			}
+		}
+		return deviceList;
+	}
+
+	private Boolean addHassDevices(List<HassDevice> theDeviceList, List<State> theSourceList, String theKey) {
+		Iterator<State> devices = theSourceList.iterator();
+		while(devices.hasNext()) {
+			State theDevice = devices.next();
+			HassDevice aNewHassDevice = new HassDevice();
+			aNewHassDevice.setDeviceState(theDevice);
+			aNewHassDevice.setHassaddress(hassMap.get(theKey).getHassAddress().getIp());
+			aNewHassDevice.setHassname(theKey);
+			aNewHassDevice.setDeviceName(theDevice.getAttributes().get("friendly_name").getAsString());
+			aNewHassDevice.setDomain(theDevice.getEntityId().substring(0, theDevice.getEntityId().indexOf(".")));
+			theDeviceList.add(aNewHassDevice);
+		}
+		return true;
+	}
+	
+	@Override
+	public String deviceHandler(CallItem anItem, MultiCommandUtil aMultiUtil, String lightId, int iterationCount, DeviceState state,
+			StateChangeBody theStateChanges, boolean stateHasBri, boolean stateHasBriInc, DeviceDescriptor device, String body) {
+		String theReturn = null;
+		log.debug("executing HUE api request to send message to HomeAssistant: " + anItem.getItem().toString());
+		if(!validHass) {
+			log.warn("Should not get here, no HomeAssistant clients configured");
+			theReturn = "[{\"error\":{\"type\": 6, \"address\": \"/lights/" + lightId
+					+ "\",\"description\": \"Should not get here, no HomeAssistants configured\", \"parameter\": \"/lights/"
+					+ lightId + "state\"}}]";
+
+		} else {
+			HassCommand hassCommand = aGsonHandler.fromJson(anItem.getItem(), HassCommand.class);
+			hassCommand.setBri(BrightnessDecode.replaceIntensityValue(hassCommand.getBri(),
+					BrightnessDecode.calculateIntensity(state, theStateChanges, stateHasBri, stateHasBriInc), false));
+			HomeAssistant homeAssistant = getHomeAssistant(hassCommand.getHassName());
+			if (homeAssistant == null) {
+				log.warn("Should not get here, no HomeAssistants available");
+				theReturn = "[{\"error\":{\"type\": 6, \"address\": \"/lights/" + lightId
+						+ "\",\"description\": \"Should not get here, no HiomeAssistant clients available\", \"parameter\": \"/lights/"
+						+ lightId + "state\"}}]";
+			} else {
+				for (int x = 0; x < aMultiUtil.getSetCount(); x++) {
+					if (x > 0 || iterationCount > 0) {
+						try {
+							Thread.sleep(aMultiUtil.getTheDelay());
+						} catch (InterruptedException e) {
+							// ignore
+						}
+					}
+					if (anItem.getDelay() != null && anItem.getDelay() > 0)
+						aMultiUtil.setTheDelay(anItem.getDelay());
+					else
+						aMultiUtil.setTheDelay(aMultiUtil.getDelayDefault());
+					log.debug("calling HomeAssistant: " + hassCommand.getHassName() + " - "
+							+ hassCommand.getEntityId() + " - " + hassCommand.getState() + " - " + hassCommand.getBri()
+							+ " - iteration: " + String.valueOf(iterationCount) + " - count: " + String.valueOf(x));
+					homeAssistant.callCommand(hassCommand);
+				}
+			}
+		}
+		return theReturn;
+	}
+
+	@Override
+	public void closeHome() {
+		Iterator<String> keys = hassMap.keySet().iterator();
+		while(keys.hasNext()) {
+			String key = keys.next();
+			hassMap.get(key).closeClient();
+		}
+	}
+}