package alice.datasegment;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.concurrent.BlockingQueue;
import java.util.zip.Deflater;
import java.util.zip.DeflaterOutputStream;

import org.apache.log4j.Logger;
import org.msgpack.MessagePack;

import alice.codesegment.CodeSegment;
import alice.codesegment.SingletonMessage;
import alice.daemon.CommandMessage;
import alice.daemon.Connection;

/**
 * DSMで使われる各コマンドのセット（ReceiveDataからのDSの読み込み）
 */
public class Command {
    public CommandType type;//PEEK, PUTなどのコマンドタイプ
    public String key;
    public Receiver receiver;
    public ReceiveData rData;
    public int index;//使ってない。アクセス用のindex。負の遺産。
    public int seq;//DSの待ち合わせを行っているCSを表すunique number。リモート用。対応コマンドを表す。
    public Connection connection; // for remote
    public BlockingQueue<Command> replyQueue;//PEEK/TAKE必要な返り値？
    public CodeSegment cs;
    public String reverseKey;//どこからput/updateされたか
    private boolean quickFlag = false;//SEDAを挟まずに処理を行うかどうか。trueだとリモート
    private boolean compressFlag = false;//trueだったら圧縮する必要がある

    private Logger logger = Logger.getLogger("MessagePack");

    /**
     * for PEEK/TAKE
     */
    public Command(CommandType cmdType, Receiver receiver, String key, ReceiveData rData, int index, int seq, BlockingQueue<Command> replyQueue, CodeSegment cs, String reverseKey) {
        this.type = cmdType;
        this.receiver = receiver;
        this.key = key;
        this.rData = rData;
        this.index = index;
        this.seq = seq;
        this.replyQueue = replyQueue;
        this.cs = cs;
        this.reverseKey = reverseKey;
    }

    /**
     * for PUT/UPDATE/REPLY/PING/RESPONSE
     */
    public Command(CommandType cmdType, Receiver receiver, String key, ReceiveData rData, int index, int seq, CodeSegment cs, String reverseKey, Connection connection) {
        this.type = cmdType;
        this.receiver = receiver;
        this.key = key;
        this.rData = rData;
        this.index = index;
        this.seq = seq;
        this.connection = connection;
        this.cs = cs;
        this.reverseKey = reverseKey;
    }

    /**
     * String型でコマンドを取得するメソッド。たぶんログ表示用。
     * @return
     */
    public String getCommandString() {
        String csName = "null";
        if (cs != null) {
            csName = cs.toString();
        }
        return this.type + "\t" + key + "\t" + rData + "\tindex=" + index + "\tcs=" + csName;
    }

    /**
     * @return serialized ByteBuffer
     */
    public ByteBuffer convert() {//メッセージパックでbyteArrayに変換
        ByteBuffer buf = null;
        MessagePack msg = SingletonMessage.getInstance();
        try {
            byte[] header = null;//DSのメタデータ用byteArray
            byte[] data = null;//DS本体用byteArray
            byte[] dataSize = null;//DSのサイズ
            boolean serialized = false;//DSがシリアライズ状態かのフラグ
            boolean compressed = false;//DSが圧縮されているかのフラグ
            switch (type) {
            /*
             * UPDATE, PUT, REPLY need send DataSegment to RemoteDataSegment
             * case UPDATE and PUT
             * compress and serialize flag are selected by user, so if true, need convert.
             * case REPLY
             * these flags represent DataSegment status.
             * for example, serializeFlag is true. DataSegment had already converted, so no need convert.
             */
            case UPDATE:
            case PUT:
            case REPLY://ReceiveDataからREPLYするDSを取得
                if (rData.compressed()) {//圧縮されている場合：各フラグの状態とDS本体を取得
                    // have already converted
                    data = (byte[]) rData.getObj();
                    compressed = rData.compressed();
                    serialized = rData.serialized();//シリアライズされているか＝Remoteから送られてきたか
                } else {//圧縮されていない場合：状態にあわせてDS本体を取得し各フラグを立てる
                    if (!rData.serialized() && !rData.isByteArray()) {
                        data = msg.write(rData.getObj());//シリアライズ
                        serialized = true;
                    } else { // rData is RAW ByteArray or already serialized
                        data = (byte[]) rData.getObj();
                        serialized = rData.serialized();
                    }
                    if (compressFlag) {//圧縮する ここはあとで消す
                        data = rData.zip(data);
                        compressed = true;
                    }
                }
                CommandMessage cm = new CommandMessage(type.id, index, seq, key, false, serialized, compressed);
                if (rData.setTime) {//AliceVNCの計測用（消してもいい）
                    cm.setTime = true;
                    cm.time = rData.time;
                    cm.depth = rData.depth + 1;
                }

                //MessagePackでDSを作成（ヘッダー・データ本体のサイズ・データ本体）
                header = msg.write(cm);
                dataSize = msg.write(data.length);
                buf = ByteBuffer.allocate(header.length+dataSize.length+data.length);
                buf.put(header);
                buf.put(dataSize);
                buf.put(data);

                //System.out.print("messagePack = " + Integer.toHexString(buf.getInt() & 0xff));
                logger.debug("messagePack = " + Integer.toHexString(buf.getInt() & 0xff));

                break;
            default://PEEK, TAKE, RemoveならCommandMessageとそのサイズのみセット
                header = msg.write(new CommandMessage(type.id, index, seq, key, quickFlag, false, compressFlag));
                buf = ByteBuffer.allocate(header.length);
                buf.put(header);
                //System.out.print("messagePack = " + Integer.toHexString(buf.getInt() & 0xff));
                break;
            }

            buf.flip();
        } catch (IOException e) {
            e.printStackTrace();
        }//ここに圧縮機能を入れる
        return buf;
    }

    /**
     * If this flag is true, command isn't send queue.
     * command is executed right now.
     *
     * @param flag
     */

    public void setQuickFlag(boolean flag){//SEDA処理の有無フラグのsetter
        quickFlag = flag;
    }

    public boolean getQuickFlag(){//SEDA処理の有無フラグのgetter
        return quickFlag;
    }

    /**
     * Before sending Remote DataSegment, DataSegment type is ByteArray.
     * If this flag true, ByteArray is compressed with ZRLEE(ZRIB) algorithm
     *
     * @param flag
     */

    public void setCompressFlag(boolean flag){//圧縮フラグのsetter
        compressFlag = flag;
    }

    public boolean getCompressFlag(){//圧縮フラグのgetter
        return compressFlag;
    }

}