org.coweb.bots.transport.LocalTransport Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of coweb-server Show documentation
There is a newer version: 1.0
/**
 * Copyright (c) The Dojo Foundation 2011. All Rights Reserved.
 * Copyright (c) IBM Corporation 2008, 2011. All Rights Reserved.
 */
package org.coweb.bots.transport;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.ArrayList;
import java.util.logging.Logger;

import org.cometd.bayeux.Message;
import org.cometd.bayeux.server.ConfigurableServerChannel;
import org.cometd.bayeux.server.ServerChannel;
import org.cometd.bayeux.server.ServerSession;

import org.coweb.SessionHandler;
import org.coweb.bots.Bot;
import org.coweb.bots.Proxy;

public class LocalTransport extends Transport implements Proxy {
	private static final Logger log = Logger.getLogger(LocalTransport.class
			.getName());

	private Bot bot = null;
	private Map clients = new HashMap();

	private ArrayList subscribers = new ArrayList();

	public LocalTransport() {
		super();
	}

	public void init() {
		return;
	}

	public boolean subscribeUser(ServerSession client, Message message,
			boolean pub) throws IOException {

		if (this.bot == null) {
			this.bot = this.getBotInstance();
			if (this.bot == null)
				throw new IOException("unable to locate bot "
						+ this.serviceName);
		}

		if (pub) {
			this.bot.onSubscribe((String) client.getAttribute("username"));
			this.subscribers.add(client);
		}

		return true;
	}

	@Override
	public boolean unsubscribeUser(ServerSession client, Message message,
			boolean pub) throws IOException {

		log.fine("LocalTransport::unSubscribeUser");

		if (this.bot == null) {
			this.bot = this.getBotInstance();
			if (this.bot == null)
				throw new IOException("unable to locate bot "
						+ this.serviceName);
		}

		if (pub) {
			this.bot.onUnsubscribe((String) client.getAttribute("username"));
			this.subscribers.remove(client);
		}

		return true;

	}

	@Override
	public void shutdown() {
		log.fine("LocalTransport::shutdown");
		this.clients.clear();
		this.subscribers.clear();
		if (this.bot != null)
			this.bot.onShutdown();
	}

	@Override
	public boolean userRequest(ServerSession client, Message message)
			throws IOException {
		log.fine("LocalTransport::userRequest");
		log.fine("message = " + message);

		Map data = message.getDataAsMap();
		@SuppressWarnings("unchecked")
		Map params = (Map) data.get("value");
		String replyToken = (String) data.get("topic");
		String username = (String) client.getAttribute("username");

		if (this.bot == null) {
			this.bot = this.getBotInstance();
			if (this.bot == null)
				throw new IOException("unable to locate bot "
						+ this.serviceName);
		}

		this.clients.put(replyToken, client);
		this.bot.onRequest(params, replyToken, username);

		return true;
	}

	@Override
	public boolean syncEvent(ServerSession client, Message message)
			throws IOException {
		log.fine("LocalTransport::syncEvent");
		log.fine("message = " + message);

		Map data = message.getDataAsMap();

		String username = (String) client.getAttribute("username");

		if (this.bot == null) {
			this.bot = this.getBotInstance();
			if (this.bot == null)
				throw new IOException("unable to locate bot "
						+ this.serviceName);
		}

		this.bot.onSync(data, username);

		return true;
	}

	@Override
	public void reply(Bot bot, String replyToken, Map obj) {

		log.fine("LocalTransport::reply");
		log.fine("reply data = " + obj);
		ServerSession client = this.clients.get(replyToken);

		if (client == null) {
			log.fine("LocalTransport::error sending bot reply client not found");
			// TODO send error.
			return;
		}

		HashMap data = new HashMap();
		data.put("value", obj);
		data.put("topic", replyToken);

		// HashMap payload = new HashMap();
		// payload.put("data", data);

		log.fine("LocalTransport::reply");
		log.fine("payload = " + data);
		log.fine("replyToken = " + replyToken);

		client.deliver(this.server, "/service/bot/" + this.serviceName
				+ "/response", data, null);
		this.clients.remove(replyToken);
	}

	@Override
	public void publish(Bot bot, Map obj) {
		log.fine("LocalTransport::publish");

		HashMap data = new HashMap();
		data.put("value", obj);

		HashMap payload = new HashMap();
		payload.put("data", data);

		ServerChannel channel = this.getResponseChannel();
		for (ServerSession client : this.subscribers) {
			client.deliver(this.server, channel.getId(), data, null);
		}

		log.fine("LocalTransport::publish");
		log.fine(data.toString());
		log.fine("channel = " + channel);
		/*
		 * Set clients = channel.getSubscribers();
		 * for(ServerSession client: clients) {
		 * System.out.println(client.getAttribute("username")); }
		 */
		// channel.publish(this.server, data, null);
	}

	private Bot getBotInstance() throws IOException {

		String className = (String) this.botConfig.get("class");
		if (className == null)
			throw new IOException("Error parsing bot descriptor");

		try {
			Class clazz = Class.forName(className).asSubclass(
					Bot.class);
			bot = clazz.newInstance();
			bot.setProxy(this);
			bot.init();

		} catch (Exception e) {
			throw new IOException("Error attempting to initialize bot "
					+ className + e.toString());
		}

		return bot;
	}

	private ServerChannel getResponseChannel() {

		String channelName = "/bot/" + this.serviceName;
		ServerChannel serverChannel = this.bayeuxServer.getChannel(channelName);
		if (serverChannel != null)
			return serverChannel;

		ServerChannel.Initializer initializer = new ServerChannel.Initializer() {
			@Override
			public void configureChannel(ConfigurableServerChannel channel) {
				channel.setPersistent(true);
				channel.setLazy(false);
			}
		};

		this.bayeuxServer.createIfAbsent(channelName, initializer);
		return this.bayeuxServer.getChannel(channelName);
	}
}