bundles/org.eclipse.e4.languages.javascript.debug/src/org/eclipse/e4/languages/javascript/rhino/jsdi/event/EventQueueImpl.java

package org.eclipse.e4.languages.javascript.rhino.jsdi.event;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.e4.languages.javascript.debug.JSONConstants;
import org.eclipse.e4.languages.javascript.debug.connect.DisconnectException;
import org.eclipse.e4.languages.javascript.debug.connect.EventPacket;
import org.eclipse.e4.languages.javascript.debug.connect.TimeoutException;
import org.eclipse.e4.languages.javascript.jsdi.Location;
import org.eclipse.e4.languages.javascript.jsdi.ThreadReference;
import org.eclipse.e4.languages.javascript.jsdi.event.EventQueue;
import org.eclipse.e4.languages.javascript.jsdi.event.EventSet;
import org.eclipse.e4.languages.javascript.jsdi.request.ExceptionRequest;
import org.eclipse.e4.languages.javascript.jsdi.request.ScriptLoadRequest;
import org.eclipse.e4.languages.javascript.jsdi.request.ThreadEnterRequest;
import org.eclipse.e4.languages.javascript.jsdi.request.ThreadExitRequest;
import org.eclipse.e4.languages.javascript.rhino.jsdi.ScriptReferenceImpl;
import org.eclipse.e4.languages.javascript.rhino.jsdi.ThreadReferenceImpl;
import org.eclipse.e4.languages.javascript.rhino.jsdi.VirtualMachineImpl;
import org.eclipse.e4.languages.javascript.rhino.jsdi.request.BreakpointRequestImpl;
import org.eclipse.e4.languages.javascript.rhino.jsdi.request.EventRequestManagerImpl;
import org.eclipse.e4.languages.javascript.rhino.jsdi.request.StepRequestImpl;
import org.eclipse.e4.languages.javascript.rhino.jsdi.request.SuspendRequestImpl;

/**
 * Default implementation of {@link EventQueue}
 * 
 * @since 1.0
 */
public class EventQueueImpl implements EventQueue {

	private VirtualMachineImpl vm;
	private EventRequestManagerImpl eventRequestManager;

	/**
	 * Constructor
	 * 
	 * @param vm
	 * @param eventRequestManager
	 */
	public EventQueueImpl(VirtualMachineImpl vm, EventRequestManagerImpl eventRequestManager) {
		this.vm = vm;
		this.eventRequestManager = eventRequestManager;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.e4.languages.javascript.jsdi.event.EventQueue#remove()
	 */
	public EventSet remove() {
		return remove(-1);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.e4.languages.javascript.jsdi.event.EventQueue#remove(int)
	 */
	public EventSet remove(int timeout) {
		try {
			while (true) {
				EventPacket event = vm.receiveEvent(timeout);
				String eventName = event.getEvent();
				EventSetImpl eventSet = new EventSetImpl(vm);
				if (eventName.equals(JSONConstants.SCRIPT)) {
					Long threadId = new Long(((Number) event.getBody().get(JSONConstants.THREAD_ID)).longValue());
					ThreadReferenceImpl thread = vm.getThread(threadId);
					eventSet.setThread(thread);

					Long scriptId = new Long(((Number) event.getBody().get(JSONConstants.SCRIPT_ID)).longValue());
					ScriptReferenceImpl script = vm.addScript(scriptId);

					List scriptLoadRequests = eventRequestManager.scriptLoadRequests();
					for (Iterator iterator = scriptLoadRequests.iterator(); iterator.hasNext();) {
						ScriptLoadRequest request = (ScriptLoadRequest) iterator.next();
						if (request.isEnabled())
							eventSet.add(new ScriptLoadEventImpl(vm, thread, script, request));
					}
				} else if (eventName.equals(JSONConstants.BREAK)) {
					Long threadId = new Long(((Number) event.getBody().get(JSONConstants.THREAD_ID)).longValue());
					ThreadReferenceImpl thread = vm.getThread(threadId);
					eventSet.setThread(thread);

					Long scriptId = new Long(((Number) event.getBody().get(JSONConstants.SCRIPT_ID)).longValue());
					ScriptReferenceImpl script = vm.getScript(scriptId);

					int lineNumber = ((Number) event.getBody().get(JSONConstants.LINE_NUMBER)).intValue();
					Location location = script.lineLocation(lineNumber);

					List jsonBreakpoints = (List) event.getBody().get(JSONConstants.BREAKPOINTS);
					if (jsonBreakpoints != null) {
						List breakpoints = new ArrayList(jsonBreakpoints.size());
						for (Iterator iterator = jsonBreakpoints.iterator(); iterator.hasNext();) {
							Number breakpointId = (Number) iterator.next();
							breakpoints.add(new Long(breakpointId.longValue()));
						}

						List breakpointRequests = eventRequestManager.breakpointRequests();
						for (Iterator iterator = breakpointRequests.iterator(); iterator.hasNext();) {
							BreakpointRequestImpl request = (BreakpointRequestImpl) iterator.next();
							ThreadReference requestThread = request.thread();
							if (request.isEnabled() && (requestThread == null || thread == requestThread) && breakpoints.contains(request.breakpointId()))
								eventSet.add(new BreakpointEventImpl(vm, thread, location, request));
						}
					}

					String stepType = (String) event.getBody().get(JSONConstants.STEP_TYPE);
					if (JSONConstants.SUSPENDED.equals(stepType)) {
						List suspendRequests = eventRequestManager.suspendRequests();
						for (Iterator iterator = suspendRequests.iterator(); iterator.hasNext();) {
							SuspendRequestImpl request = (SuspendRequestImpl) iterator.next();
							ThreadReference requestThread = request.thread();
							if (request.isEnabled() && (requestThread == null || thread == requestThread))
								eventSet.add(new SuspendEventImpl(vm, thread, location, request));
						}
					} else if (stepType != null) {
						List stepRequests = eventRequestManager.suspendRequests();
						for (Iterator iterator = stepRequests.iterator(); iterator.hasNext();) {
							StepRequestImpl request = (StepRequestImpl) iterator.next();
							ThreadReference requestThread = request.thread();
							if (request.isEnabled() && (requestThread == null || thread == requestThread))
								eventSet.add(new StepEventImpl(vm, thread, location, request));
						}
					}
					if (eventSet.isEmpty()) {
						// debugger case
						thread.markSuspended(false);
						continue;
					}

				} else if (eventName.equals(JSONConstants.EXCEPTION)) {
					Long threadId = new Long(((Number) event.getBody().get(JSONConstants.THREAD_ID)).longValue());
					ThreadReferenceImpl thread = vm.getThread(threadId);
					eventSet.setThread(thread);

					Long scriptId = new Long(((Number) event.getBody().get(JSONConstants.SCRIPT_ID)).longValue());
					ScriptReferenceImpl script = vm.getScript(scriptId);

					int lineNumber = ((Number) event.getBody().get(JSONConstants.LINE_NUMBER)).intValue();
					Location location = script.lineLocation(lineNumber);

					String message = (String) event.getBody().get(JSONConstants.MESSAGE);

					List exceptionRequests = eventRequestManager.scriptLoadRequests();
					for (Iterator iterator = exceptionRequests.iterator(); iterator.hasNext();) {
						ExceptionRequest request = (ExceptionRequest) iterator.next();
						eventSet.add(new ExceptionEventImpl(vm, thread, location, message, request));
					}
				} else if (eventName.equals(JSONConstants.THREAD)) {
					Long threadId = new Long(((Number) event.getBody().get(JSONConstants.THREAD_ID)).longValue());

					String type = (String) event.getBody().get(JSONConstants.TYPE);

					if (JSONConstants.ENTER.equals(type)) {
						ThreadReferenceImpl thread = vm.addThread(threadId);
						eventSet.setThread(thread);
						List threadEnterRequests = eventRequestManager.threadEnterRequests();
						for (Iterator iterator = threadEnterRequests.iterator(); iterator.hasNext();) {
							ThreadEnterRequest request = (ThreadEnterRequest) iterator.next();
							eventSet.add(new ThreadEnterEventImpl(vm, thread, request));
						}
					} else if (JSONConstants.EXIT.equals(type)) {
						ThreadReferenceImpl thread = vm.removeThread(threadId);
						List threadExitRequests = eventRequestManager.threadExitRequests();
						for (Iterator iterator = threadExitRequests.iterator(); iterator.hasNext();) {
							ThreadExitRequest request = (ThreadExitRequest) iterator.next();
							eventSet.add(new ThreadExitEventImpl(vm, thread, request));
						}
					}
				}
				if (eventSet.isEmpty()) {
					eventSet.resume();
					continue;
				}
				return eventSet;
			}
		} catch (TimeoutException e) {
			e.printStackTrace();
		} catch (DisconnectException e) {
			e.printStackTrace();
		}
		return null;
	}
}