package qz.communication;

import org.apache.commons.lang3.ArrayUtils;
import org.codehaus.jettison.json.JSONException;
import org.codehaus.jettison.json.JSONObject;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import qz.utils.DeviceUtilities;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.Socket;
import java.nio.charset.Charset;
import java.util.ArrayList;

public class SocketIO {

    private static final Logger log = LogManager.getLogger(SocketIO.class);

    private String host;
    private int port;
    private Charset encoding;

    private Socket socket;
    private DataOutputStream dataOut;
    private DataInputStream dataIn;

    public SocketIO(String host, int port, Charset encoding) {
        this.host = host;
        this.port = port;
        this.encoding = encoding;
    }

    public boolean open() throws IOException {
        socket = new Socket(host, port);
        dataOut = new DataOutputStream(socket.getOutputStream());
        dataIn = new DataInputStream(socket.getInputStream());

        return socket.isConnected();
    }

    public boolean isOpen() {
        return socket.isConnected();
    }

    public void sendData(JSONObject params) throws JSONException, IOException {
        log.debug("Sending data over [{}:{}]", host, port);
        dataOut.write(DeviceUtilities.getDataBytes(params, encoding));
        dataOut.flush();
    }

    public String processSocketResponse() throws IOException {
        byte[] response = new byte[1024];
        ArrayList<Byte> fullResponse = new ArrayList<>();
        do {
            int size = dataIn.read(response);
            for(int i = 0; i < size; i++) {
                fullResponse.add(response[i]);
            }
        }
        while(dataIn.available() > 0);

        return new String(ArrayUtils.toPrimitive(fullResponse.toArray(new Byte[0])), encoding);
    }

    public void close() throws IOException {
        dataOut.close();
        socket.close();
    }

    public String getHost() {
        return host;
    }

    public int getPort() {
        return port;
    }

}