package alice.daemon;

import java.io.IOException;
import java.net.Socket;
import java.nio.ByteBuffer;
import java.nio.channels.ClosedChannelException;
import java.util.concurrent.LinkedBlockingQueue;

import alice.codesegment.SingletonMessage;
import alice.datasegment.Command;
import alice.datasegment.DataSegment;

public class Connection {

    public Socket socket;
    public LinkedBlockingQueue<Command> sendQueue = new LinkedBlockingQueue<Command>();

    public Connection(Socket socket) {
        this.socket = socket;
    }

    public Connection() {}

    public void sendCommand(Command cmd) {
        try {
            sendQueue.put(cmd);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    public String getInfoString() {
        return socket.getInetAddress().getHostName()
                + ":" + socket.getPort();
    }

    public synchronized void write(Command cmd) {	
        CommandMessage cmdMsg = cmd.convert();
        ByteBuffer buffer;
        try {
            buffer = ByteBuffer.wrap(SingletonMessage.getInstance().write(cmdMsg));
            while (buffer.hasRemaining()) {
                socket.getChannel().write(buffer);
            }
        } catch (ClosedChannelException e) {
            // when put dataSegment to remote if connection close this dataSemgent put.
            putConnectionInfo();

        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void close(){
        try {
            socket.shutdownOutput();
            socket.shutdownInput();
            socket.close();
        } catch (ClosedChannelException e) {
            putConnectionInfo();
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

    public void putConnectionInfo() {
        ConnectionInfo c = new ConnectionInfo(socket.getInetAddress().toString() ,socket.getPort());
        DataSegment.getLocal().put("disconnect", c);

    }
}