package geniusweb.partiesserver.websocket;

import java.io.EOFException;
import java.io.IOException;
import java.net.URI;
import java.util.Date;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.logging.Level;

import javax.websocket.CloseReason;
import javax.websocket.CloseReason.CloseCodes;
import javax.websocket.OnClose;
import javax.websocket.OnError;
import javax.websocket.OnMessage;
import javax.websocket.OnOpen;
import javax.websocket.Session;
import javax.websocket.server.PathParam;
import javax.websocket.server.ServerEndpoint;

import com.fasterxml.jackson.core.JsonParseException;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.ObjectMapper;

import geniusweb.actions.Action;
import geniusweb.actions.PartyId;
import geniusweb.connection.ConnectionEnd;
import geniusweb.partiesserver.repository.RunningPartiesRepo;
import geniusweb.partiesserver.repository.RunningParty;
import geniusweb.party.Party;
import geniusweb.party.inform.Inform;
import geniusweb.party.inform.Settings;
import geniusweb.references.Reference;
import tudelft.utilities.listener.Listener;
import tudelft.utilities.logging.ReportToLogger;
import tudelft.utilities.logging.Reporter;

/**
 * Returns a websocket that communicates with the {@link RunningParty}. We close
 * the socket when we detect the party is gone (eg, due to a bug in the party or
 * a time-out). If the socket breaks, we close the party as it becomes isolated.
 */
@ServerEndpoint("/party/{party}")
public class PartySocket implements ConnectionEnd<Inform, Action> {
	private final static ObjectMapper jackson = new ObjectMapper();
	private final Reporter log;
	private final RunningPartiesRepo runningparties;
	private final List<Listener<Inform>> listeners = new CopyOnWriteArrayList<Listener<Inform>>();
	private static final int MAX_MESSAGE_SIZE = 20 * 1024 * 1024;

	// should all be final, except that we can only set them when start is
	// called...
	private Session session;

	private PartyId partyID;

	private SendBuffer outstream;
	private Throwable error = null;

	public PartySocket() {
		this(RunningPartiesRepo.instance(),
				new ReportToLogger("partiesserver"));
	}

	public PartySocket(RunningPartiesRepo parties, Reporter reporter) {
		this.runningparties = parties;
		this.log = reporter;
	}

	@OnOpen
	public void start(Session session,
			@PathParam("party") final String partyidname) throws IOException {
		if (partyidname == null) {
			throw new IllegalArgumentException("party can't be null");
		}
		session.setMaxTextMessageBufferSize(MAX_MESSAGE_SIZE);
		session.setMaxBinaryMessageBufferSize(MAX_MESSAGE_SIZE);

		this.session = session;
		this.partyID = new PartyId(partyidname);
		this.outstream = new SendBuffer(session.getBasicRemote(), log);

		// listen to parties repo, so that we can act if party is terminated
		runningparties.addListener(new Listener<PartyId>() {
			@Override
			public void notifyChange(PartyId data) {
				if (runningparties.get(partyID) == null) {
					// party was removed, probably due to time out.
					try {
						log.log(Level.INFO,
								"Party " + partyID + " was terminated ");
						runningparties.removeListener(this);
						session.close(new CloseReason(CloseCodes.GOING_AWAY,
								"detected that party vanished"));
					} catch (IOException e) {
						log.log(Level.WARNING,
								"failed to close the socket for " + partyID, e);
					}
				}
			}
		});
		RunningParty runningparty = runningparties.get(partyID);

		if (runningparty == null) {
			session.close(new CloseReason(CloseReason.CloseCodes.CANNOT_ACCEPT,
					"No such party: " + partyID));
		} else {
			runningparties.replace(runningparty.withConnection(this));
		}

		// we do not listen for changes on the Parties factory as we can't
		// replace the running party anyway
	}

	/**
	 * Called when a message comes in on the server socket that needs to be
	 * passed into the {@link Party}. We also sniff the message for termination
	 * indicators.
	 * 
	 * @param informmessage the incoming string, supposedly a JSON-fornatted
	 *                      {@link Inform}
	 * @param session       the session. We already have this info.
	 */
	@OnMessage
	public void onMessage(String informmessage, Session session)
			throws JsonParseException, JsonMappingException, IOException {
		if (this.session != session) {
			throw new IllegalStateException("Unexpected change of session ID");
		}
		Inform info = jackson.readValue(informmessage, Inform.class);
		RunningParty party = runningparties.get(partyID);
		if (party != null) {
			// first sniff, to ensure the deadlines are updated first.
			sniff(info, party);
			log.log(Level.FINE, "Inform " + partyID + ": " + info);
			try {
				party.inform(info);
			} catch (Throwable e) {
				log.log(Level.WARNING, "Party failed on inform.", e);
				e.printStackTrace();
				// severe as someone wants to debug that. Not severe for us
				// we don't use jackson.writeValueAsString(e) here because
				// CloseReason has 123 char limit.
				// The error will be too large to fit and will be truncated
				// therefore we send a plain string and hope that enough
				// will arrive at the other side (the protocol)
				session.close(new CloseReason(
						CloseReason.CloseCodes.CLOSED_ABNORMALLY,
						"party threw exception: " + collectReasons(e)));
			}
		} // else party is dead but that's handled in the listener above
	}

	/**
	 * 
	 * @param e an exception
	 * @return collect all reasons and sub-reasons
	 */
	private String collectReasons(Throwable e) {
		String reason = "";
		do {
			reason = reason + " : " + e.getClass().getSimpleName() + ":"
					+ e.getMessage();
			e = e.getCause();
		} while (e != null);
		return reason;
	}

	/**
	 * Sniffs the Inform for termination-related information
	 * 
	 * @param info
	 * @param party the party we're handling
	 * @throws IOException
	 */
	private void sniff(Inform info, RunningParty party) throws IOException {
		// do NOT sniff Finished messages and close sockets. Parties and their
		// sockets need to stay running till all messages are handled
		if (info instanceof Settings) {
			Date end = ((Settings) info).getProgress().getTerminationTime();
			runningparties.replace(party.withEndTime(end));
		}
	}

	@OnClose
	public void onClose() throws IOException {
		log.log(Level.INFO, "socket closed to " + partyID);
		/*
		 * for now just kill the party. Maybe we can do something with reconnect
		 * but that will be complex.
		 */
		runningparties.remove(partyID);
		outstream.stop();
	}

	@OnError
	public void onError(Throwable t) throws Throwable {
		if (t instanceof EOFException) {
			// This is a standard exception from Apache when a socket closes.
			// Just ignore
			// it...
			log.log(Level.FINEST,
					"apache reported EOF from (probably closed) socket, ignoring");
			return;
		}
		log.log(Level.WARNING, "Unhandled exception was reported by Tomcat", t);
	}

	@Override
	public void send(Action action) throws IOException {
		log.log(Level.FINE, "Action " + partyID + ": " + action);
		outstream.send(jackson.writeValueAsString(action));
	}

	@Override
	public Reference getReference() {
		return null;
	}

	@Override
	public URI getRemoteURI() {
		// In serverendpoints we apparently can't see who was contacting us.
		throw new UnsupportedOperationException();
	}

	@Override
	public void close() {
		try {
			session.close();
		} catch (IOException e) {
			log.log(Level.SEVERE, "failed to close " + partyID, e);
		}
	}

	@Override
	public Throwable getError() {
		return error;
	}

	@Override
	public void addListener(Listener<Inform> l) {
		listeners.add(l);
	}

	@Override
	public void removeListener(Listener<Inform> l) {
		listeners.remove(l);
	}

	/**
	 * This should only be called by the owner of the listenable, not by
	 * listeners or others. Avoid calling this from synchronized blocks as a
	 * notified listener might immediately make more calls to you.
	 * <p>
	 * Any listeners that throw an exception will be intercepted and their
	 * stacktrace is printed.
	 * 
	 * @param data information about the change.
	 */
	public void notifyListeners(Inform data) {
		for (Listener<Inform> l : listeners) {
			l.notifyChange(data);
		}
	}

}