Web4Rail/src/main/java/de/srsoftware/web4rail/tiles/Turnout.java

package de.srsoftware.web4rail.tiles;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeoutException;

import org.json.JSONObject;

import de.srsoftware.tools.Tag;
import de.srsoftware.web4rail.Command;
import de.srsoftware.web4rail.Command.Reply;
import de.srsoftware.web4rail.Device;
import de.srsoftware.web4rail.Protocol;
import de.srsoftware.web4rail.tags.Fieldset;
import de.srsoftware.web4rail.tags.Form;
import de.srsoftware.web4rail.tags.Input;
import de.srsoftware.web4rail.tags.Label;
import de.srsoftware.web4rail.tags.Radio;

/**
 * Base class for Turnouts
 * @author Stephan Richter, SRSoftware
 *
 */
public abstract class Turnout extends Tile implements Device{
	private   static final String PORT_A   = "port_a";
	private   static final String PORT_B   = "port_b";	
	public    static final String STATE    = "state";
	protected static final String STRAIGHT = "straight";

	protected int      address     = 0;
	protected int      delay       = 400;
	protected boolean  error       = false;
	protected boolean  initialized = false;
	private   Protocol protocol    = Protocol.DCC128;
	protected int      portA       = 0, portB = 1;
	protected State    state       = State.STRAIGHT;
	
	public enum State{
		LEFT,STRAIGHT,RIGHT,UNDEF;
	}
	
	@Override
	public Object click() throws IOException {
		LOG.debug(getClass().getSimpleName()+".click()");
		init();
		return super.click();
	}

	protected abstract String commandFor(State newState);
	
	public void error(Reply reply) {
		this.error = true;
		try {
			plan.stream(tag(null).toString());
		} catch (IOException e) {
			LOG.error("Was not able to stream: ",e);
		}
		throw new RuntimeException(reply.message()); 
	}
	
	protected Reply init() {
		if (address == 0) return new Reply(200,"OK");
		if (!initialized) {
			Command command = new Command("INIT {} GA "+address+" "+proto()) {

				@Override
				public void onSuccess() {
					super.onSuccess();
					initialized = true;
				}

				@Override
				public void onFailure(Reply r) {
					super.onSuccess();
					initialized = false;					
				}
				
			};			
			try {
				return plan.queue(command).reply();
			} catch (TimeoutException e) {
				LOG.warn(e.getMessage());
			}
		}
		return new Reply(200, "OK");
	}
	
	@Override
	public JSONObject json() {
		JSONObject json = super.json();
		if (portA != 0) json.put(PORT_A, portA);
		if (portB != 1) json.put(PORT_B, portB);
		if (address != 0) json.put(ADDRESS, address);
		json.put(PROTOCOL, protocol);
		json.put(STATE, state);
		return json;
	}
	
	@Override
	protected Tile load(JSONObject json) throws IOException {
		if (json.has(ADDRESS)) address = json.getInt(ADDRESS);
		if (json.has(PORT_A)) portA = json.getInt(PORT_A);
		if (json.has(PORT_B)) portB = json.getInt(PORT_B);
		if (json.has(PROTOCOL)) protocol = Protocol.valueOf(json.getString(PROTOCOL));
		if (json.has(STATE))    state = State.valueOf(json.getString(STATE));
		return super.load(json);
	}
	
	@Override
	public Form propForm(String id) {
		Form form = super.propForm(id);
		Fieldset fieldset = new Fieldset(t("Decoder settings"));
		Label protocol = new Label(t("Protocol:"));
		for (Protocol proto : Protocol.values()) {
			new Radio(PROTOCOL, proto.toString(), t(proto.toString()), proto == this.protocol).addTo(protocol);
		}
		protocol.addTo(fieldset).addTo(form);
		new Input(ADDRESS, address).numeric().addTo(new Label(t("Address:")+NBSP)).addTo(fieldset);
		return form;
	}
	
	private char proto() {
		switch (protocol) {
		case DCC14:
		case DCC27:
		case DCC28:
		case DCC128:
			return 'N';
		case MOTO:
			return 'M';
		case SELECTRIX:
			return 'S';
		default:
			return 'P';
		}		
	}
	
	public State state() {
		return state;
	}
	
	public Reply state(State newState) throws IOException {
		if (train != null && newState != state) return new Reply(415, t("{} locked by {}!",this,train));
		Reply reply = init();
		if (reply != null && !reply.succeeded()) return reply;
		if (address == 0) { 
			state = newState;
			plan.place(this);
			return new Reply(200,"OK");
		}
		try {
			return plan.queue(new Command(commandFor(newState)) {
				
				@Override
				public void onSuccess() {
					super.onSuccess();
					Turnout.this.state = newState;
					plan.place(Turnout.this);
				}

				@Override
				protected void onFailure(Reply reply) {
					super.onFailure(reply);
					plan.stream(t("Unable to switch \"{}\": {}",Turnout.this,reply.message()));
				}
				
			}).reply();
		} catch (TimeoutException e) {
			LOG.warn(e.getMessage());			
		}
		return new Reply(417,t("Timeout while trying to switch {}.",this));
	
		
	}
	
	public void success() {
		this.error = false;
		try {
			plan.stream(tag(null).toString());
		} catch (IOException e) {
			throw new RuntimeException(e);
		}
	}

	@Override
	public Tag tag(Map<String, Object> replacements) throws IOException {
		Tag tag = super.tag(replacements);
		tag.clazz(tag.get("class")+(" "+state).toLowerCase()+(error?" error":""));
		return tag;
	}
	
	@Override
	public String title() {
		return getClass().getSimpleName()+t("(Address: {}, Ports {} and {})",address,portA,portB);
	}
	
	@Override
	public Tile update(HashMap<String, String> params) throws IOException {
		if (params.containsKey(PROTOCOL)) protocol = Protocol.valueOf(params.get(PROTOCOL));
		if (params.containsKey(ADDRESS)) address = Integer.parseInt(params.get(ADDRESS));
		if (params.containsKey(PORT_A)) portA = Integer.parseInt(params.get(PORT_A));
		if (params.containsKey(PORT_B)) portB = Integer.parseInt(params.get(PORT_B));
		return super.update(params);
	}
}