/*
 * Copyright 2013 Eediom Inc.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.araqne.logdb.client.http.impl;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.concurrent.TimeoutException;

import org.araqne.logdb.client.AbstractLogDbSession;
import org.araqne.logdb.client.Message;
import org.araqne.logdb.client.MessageException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * HTTP 롱 폴링을 이용하여 아라크네 메시지버스 RPC 통신을 구현합니다.
 * 
 * @author xeraph@eediom.com
 * @deprecated
 */
public class CometSession extends AbstractLogDbSession implements TrapListener {
	private final Logger logger = LoggerFactory.getLogger(CometSession.class);
	private String host;
	private int port;
	private String cookie;
	private TrapReceiver trapReceiver;

	public CometSession(String host) {
		this(host, 80);
	}

	public CometSession(String host, int port) {
		this.host = host;
		this.port = port;
	}

	@Override
	public void login(String loginName, String password, boolean force) throws IOException {
		super.login(loginName, password, force);

		trapReceiver = new TrapReceiver(host, port, cookie);
		trapReceiver.addListener(this);
		trapReceiver.start();
	}

	public Message rpc(Message req, int timeout) throws IOException {
		String json = MessageCodec.encode(req);

		HttpURLConnection con = null;
		InputStream is = null;

		try {
			con = (HttpURLConnection) new URL("http://" + host + ":" + port + "/msgbus/request").openConnection();
			con.setRequestMethod("POST");
			con.setDoOutput(true);
			if (cookie != null) {
				con.setRequestProperty("Cookie", cookie);
			}

			con.getOutputStream().write(json.getBytes("utf-8"));

			is = con.getInputStream();
			if (con.getHeaderField("Set-Cookie") != null) {
				cookie = con.getHeaderField("Set-Cookie").split(";")[0];
			}

			ByteArrayOutputStream bos = null;
			if (con.getContentLength() > 0)
				bos = new ByteArrayOutputStream(con.getContentLength());
			else
				bos = new ByteArrayOutputStream();

			byte[] b = new byte[8096];

			while (true) {
				int read = is.read(b);
				if (read < 0)
					break;

				bos.write(b, 0, read);
			}

			String text = new String(bos.toByteArray(), "utf-8");
			Message msg = MessageCodec.decode(text);

			if (msg.getErrorCode() != null)
				throw new MessageException(msg.getErrorCode(), msg.getErrorMessage(), msg.getParameters());

			return msg;
		} finally {
			if (is != null)
				is.close();

			if (con != null)
				con.disconnect();
		}
	}

	@Override
	public void onTrap(Message msg) {
		for (TrapListener listener : listeners) {
			try {
				listener.onTrap(msg);
			} catch (Throwable t) {
				logger.error("logdb client: trap listener should not throw any exception", t);
			}
		}
	}

	@Override
	public void onClose(Throwable t) {
		trapReceiver = null;

		for (TrapListener listener : listeners) {
			try {
				listener.onClose(t);
			} catch (Throwable t2) {
				logger.error("logdb client: trap listener should not throw any exception", t2);
			}
		}
	}

	public void close() throws IOException {
		if (isClosed())
			return;

		try {
			// may be hang here.
			if (isLogin)
				rpc("org.araqne.logdb.msgbus.ManagementPlugin.logout", 1);
		} catch (TimeoutException e) {
		}

		super.close();

		if (trapReceiver != null) {
			trapReceiver.close();
			trapReceiver = null;
		}
	}
}