/*
 * Copyright (c) 2001-2004 Ant-Contrib project.  All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.sf.antcontrib.antserver.client;

import java.io.*;
import java.net.Socket;
import java.net.SocketException;

import org.apache.tools.ant.Project;

import net.sf.antcontrib.antserver.Command;
import net.sf.antcontrib.antserver.Response;
import net.sf.antcontrib.antserver.Util;
import net.sf.antcontrib.antserver.commands.DisconnectCommand;


/****************************************************************************
 * Place class description here.
 *
 * @author <a href='mailto:mattinger@yahoo.com'>Matthew Inger</a>
 * @author		<additional author>
 *
 * @since
 *
 ****************************************************************************/


public class Client
{
    private String machine;
    private int port;
    private Project project;


    public Client(Project project, String machine, int port)
    {
        super();
        this.machine = machine;
        this.port = port;
        this.project = project;
    }


    private Socket socket;
    private OutputStream os;
    private InputStream is;
    private ObjectOutputStream oos;
    private ObjectInputStream ois;
    private boolean connected;


    public void connect()
            throws IOException
    {
        project.log("Opening connection to " + machine + ":" + port,
                Project.MSG_DEBUG);

        try
        {
            socket = new Socket(machine, port);
            socket.setKeepAlive(true);
            project.log("Got connection to " + machine + ":" + port,
                    Project.MSG_DEBUG);

            os = socket.getOutputStream();
            is = socket.getInputStream();

            oos = new ObjectOutputStream(os);
            ois = new ObjectInputStream(is);

            connected = true;
            try
            {
                // Read the initial response object so that the
                // object stream is initialized
                ois.readObject();
            }
            catch (ClassNotFoundException e)
            {
                ; // gulp
            }
        }
        finally
        {
            // If we were unable to connect, close everything
            if (!connected)
            {

                try
                {
                    if (os != null)
                        os.close();
                    os = null;
                    oos = null;
                }
                catch (IOException e)
                {

                }

                try
                {
                    if (is != null)
                        is.close();
                    is = null;
                    ois = null;
                }
                catch (IOException e)
                {

                }

                try
                {
                    if (socket != null)
                        socket.close();
                    socket = null;
                }
                catch (IOException e)
                {

                }
            }
        }


    }

    public void shutdown()
    {
        try
        {
            if (os != null)
                os.close();
        }
        catch (IOException e)
        {
            ; // gulp

        }
        os = null;
        oos = null;

        try
        {
            if (is != null)
                is.close();
        }
        catch (IOException e)
        {
            ; // gulp

        }
        is = null;
        ois = null;

        try
        {
            socket.close();
        }
        catch (IOException e)
        {
            ; // gulp
        }
        socket = null;

        connected = false;
    }


    public void disconnect()
            throws IOException
    {
        if (!connected)
            return;

        try {
            oos.writeObject(DisconnectCommand.DISCONNECT_COMMAND);
            try
            {
                // Read disconnect response
                ois.readObject();
            }
            catch (ClassNotFoundException e)
            {
                ; // gulp
            }

            shutdown();
        }
        catch (SocketException e) {
            ; // connection was closed
        }
        catch (EOFException e) {
            ; // connection was closed
        }
    }


    public Response sendCommand(Command command)
        throws IOException
    {
        project.log("Sending command: " + command,
                Project.MSG_DEBUG);
        oos.writeObject(command);

        if (command.getContentLength() > 0)
        {
            Util.transferBytes(command.getContentStream(),
                    command.getContentLength(),
                    os,
                    true);
        }

        Response response = null;

        try
        {
            // Read the response object
            response = (Response) ois.readObject();
            project.log("Received Response: " + response,
                    Project.MSG_DEBUG);
            if (response.getContentLength() != 0)
            {
                command.respond(project,
                        response.getContentLength(),
                        is);
            }
        }
        catch (ClassNotFoundException e)
        {
            ; // gulp
        }

        return response;
    }

}