// Copyright (C) 2010, 2011, 2012, 2013 GlavSoft LLC.
// All rights reserved.
//
//-------------------------------------------------------------------------
// This file is part of the TightVNC software.  Please visit our Web site:
//
//                       http://www.tightvnc.com/
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
//-------------------------------------------------------------------------
//

package com.glavsoft.rfb.protocol.state;

import com.glavsoft.exceptions.AuthenticationFailedException;
import com.glavsoft.exceptions.FatalException;
import com.glavsoft.exceptions.TransportException;
import com.glavsoft.exceptions.UnsupportedProtocolVersionException;
import com.glavsoft.exceptions.UnsupportedSecurityTypeException;
import com.glavsoft.rfb.encoding.ServerInitMessage;
import com.glavsoft.rfb.protocol.ProtocolContext;
import com.glavsoft.rfb.protocol.ProtocolSettings;
import jp.ac.u_ryukyu.treevnc.TreeRFBProto;

/**
 * ClientInit
 *
 * Once the client and server are sure that they're happy to talk to one
 * another, the client sends an initialisation message.  At present this
 * message onl@!,@!,y consists of a boolean indicating whether the server should try
 * to share the desktop by leaving other clients connected, or give exclusive
 * access to this client by disconnecting all other clients.
 *
 * 1 - U8 - shared-���ag
 *
 * Shared-���ag is non-zero (true) if the server should try to share the desktop by leaving
 * other clients connected, zero (false) if it should give exclusive access to this client by
 * disconnecting all other clients.
 *
 * ServerInit
 *
 * After receiving the ClientInit message, the server sends a ServerInit message. This
 * tells the client the width and height of the server���s framebuffer, its pixel format and the
 * name associated with the desktop.
 */
public class InitState extends ProtocolState {

    private int singleWidth;
    private int singleHeight;
    private byte[] originalInitData;

    public InitState(ProtocolContext context) {
        super(context);
    }

    @Override
    public boolean next() throws UnsupportedProtocolVersionException, TransportException,
            UnsupportedSecurityTypeException, AuthenticationFailedException, FatalException {
        clientAndServerInit();
        return false;
    }

    protected void clientAndServerInit() throws TransportException {
        ServerInitMessage serverInitMessage = getServerInitMessage();

        if (super.rfb.filterSingleDisplay) {
            singleWidth = super.rfb.getSingleWidth();
            singleHeight = super.rfb.getSingleHeight();
            originalInitData = super.rfb.createOriginalInitData(singleWidth, singleHeight, serverInitMessage.getName());
            ProtocolSettings settings = context.getSettings();
            settings.enableAllEncodingCaps();
            completeContextData(serverInitMessage, originalInitData);
        } else {

            ProtocolSettings settings = context.getSettings();
            settings.enableAllEncodingCaps();
            completeContextData(serverInitMessage);
        }
    }

    public void completeContextData(ServerInitMessage serverInitMessage, byte[] originalInitData) {
        context.setPixelFormat(serverInitMessage.getPixelFormat());
        context.setFbWidth(serverInitMessage.getFrameBufferWidth());
        context.setFbHeight(serverInitMessage.getFrameBufferHeight());
        context.setRemoteDesktopName(serverInitMessage.getName());
        context.setInitData(originalInitData);
    }


    public void completeContextData(ServerInitMessage serverInitMessage) {
        context.setPixelFormat(serverInitMessage.getPixelFormat());
        context.setFbWidth(serverInitMessage.getFrameBufferWidth());
        context.setFbHeight(serverInitMessage.getFrameBufferHeight());
        context.setRemoteDesktopName(serverInitMessage.getName());
        context.setInitData(serverInitMessage.getInitData());
    }

    protected ServerInitMessage getServerInitMessage() throws TransportException {
        writer.write(context.getSettings().getSharedFlag());
        ServerInitMessage serverInitMessage = new ServerInitMessage(reader);
        return serverInitMessage;
    }


}