package jp.ac.u_ryukyu.alicevnc;

import com.glavsoft.drawing.Renderer;
import com.glavsoft.rfb.IRepaintController;
import com.glavsoft.rfb.encoding.PixelFormat;
import com.glavsoft.rfb.encoding.ServerInitMessage;
import com.glavsoft.rfb.protocol.Protocol;
import com.glavsoft.rfb.protocol.ProtocolSettings;
import com.glavsoft.viewer.Viewer;
import com.glavsoft.viewer.cli.Parser;
import com.glavsoft.viewer.swing.ParametersHandler;
import com.glavsoft.viewer.swing.SwingViewerWindow;
import com.glavsoft.viewer.swing.SwingViewerWindowFactory;

import alice.codesegment.CodeSegment;
import alice.datasegment.CommandType;
import alice.datasegment.Receiver;

public class ReceiveServerInitMessage extends CodeSegment {

    private Receiver info = ids.create(CommandType.PEEK);


    public ReceiveServerInitMessage(){
        info.setKey("ServerInitMessage");
    }

    public ReceiveServerInitMessage(String key){
        info.setKey(key, "MyServerInitMessage");
    }

    public void run() {
        PixelFormat pixelFormat = null;
        int width = 0;
        int height = 0;
        String name = "defalut";
        
        Parser parser = new Parser();
        ParametersHandler.completeParserOptions(parser);
        if (parser.isSet(ParametersHandler.ARG_HELP)) {
            System.exit(0);
        }
        Viewer viewer = new Viewer(parser);
        SwingViewerWindowFactory viewerWindowFactory = new SwingViewerWindowFactory(true, false, viewer);
        ProtocolSettings settings = ProtocolSettings.getDefaultSettings();
        Protocol context = new Protocol(null, null, null, settings);
        //context.createMessageQueue(); // need clear Queue
        if ("local".equals(info.from)){
            ServerInitMessage serverInitMessage = info.asClass(ServerInitMessage.class);
            pixelFormat = serverInitMessage.getPixelFormat();
            width = serverInitMessage.getFrameBufferWidth();
            height = serverInitMessage.getFrameBufferHeight();
            name = serverInitMessage.getName();
            
            MyServerInitMessage myServerInitMessage = new MyServerInitMessage(serverInitMessage);
            myServerInitMessage.width = width;
            myServerInitMessage.height = height;
            myServerInitMessage.name = name;
            
            ods.put("MyServerInitMessage", myServerInitMessage);
        } else {
            MyServerInitMessage serverInitMessage = info.asClass(MyServerInitMessage.class);
            pixelFormat = serverInitMessage.getPixelFormat();
            width = serverInitMessage.width;
            height = serverInitMessage.height;
            name = serverInitMessage.name;
        }

        context.setPixelFormat(pixelFormat);
        context.setFbWidth(width);
        context.setFbHeight(height);
        context.setRemoteDesktopName(name);
        
        SwingViewerWindow viewerWindow = viewerWindowFactory.createViewerWindow(context, settings, viewer.getUiSettings(), null, null);
        IRepaintController repaintController = viewerWindow.getSurface();
        Renderer renderer = repaintController.createRenderer(null, width, height, context.getPixelFormat());
        RenderAndControllerHolder holder = new RenderAndControllerHolder(renderer, repaintController);
        ods.put("Holder", holder);

        new ReceiveUpdateRectangle();
    }

}