/*
    *  Licensed to the Apache Software Foundation (ASF) under one
    *  or more contributor license agreements.  See the NOTICE file
    *  distributed with this work for additional information
    *  regarding copyright ownership.  The ASF licenses this file
    *  to you 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 org.apache.mina.example.chat.client;
  
  import java.net.SocketAddress;
  
  import javax.net.ssl.SSLContext;
  
  import org.apache.mina.core.filterchain.IoFilter;
  import org.apache.mina.core.future.ConnectFuture;
  import org.apache.mina.core.service.IoHandler;
  import org.apache.mina.core.session.IoSession;
  import org.apache.mina.example.echoserver.ssl.BogusSslContextFactory;
  import org.apache.mina.filter.ssl.SslFilter;
  import org.apache.mina.filter.codec.ProtocolCodecFilter;
  import org.apache.mina.filter.codec.textline.TextLineCodecFactory;
  import org.apache.mina.filter.compression.CompressionFilter;
  import org.apache.mina.filter.logging.LoggingFilter;
  import org.apache.mina.filter.logging.MdcInjectionFilter;
  import org.apache.mina.transport.socket.nio.NioSocketConnector;
  
  /**
   * A simple chat client for a given user.
   *
   * @author <a href="http://mina.apache.org">Apache MINA Project</a>
   */
  public class ChatClientSupport {
      private final IoHandler handler;
  
      private final String name;
  
      private IoSession session;
  
      public ChatClientSupport(String name, IoHandler handler) {
          if (name == null) {
              throw new IllegalArgumentException("Name can not be null");
          }
          this.name = name;
          this.handler = handler;
      }
  
      public boolean connect(NioSocketConnector connector, SocketAddress address,
              boolean useSsl) {
          if (session != null && session.isConnected()) {
              throw new IllegalStateException(
                      "Already connected. Disconnect first.");
          }
  
          try {
              IoFilter LOGGING_FILTER = new LoggingFilter();
  
              IoFilter CODEC_FILTER = new ProtocolCodecFilter(
                      new TextLineCodecFactory());
              
              connector.getFilterChain().addLast("mdc", new MdcInjectionFilter());
              
              // Add the compression filter
              connector.getFilterChain().addLast( "Compression", new CompressionFilter() );
              connector.getFilterChain().addLast("codec", CODEC_FILTER);
              connector.getFilterChain().addLast("logger", LOGGING_FILTER);
  
              if (useSsl) {
                  SSLContext sslContext = BogusSslContextFactory
                          .getInstance(false);
                  SslFilterslFilter.html#SslFilter">SslFilter sslFilter = new SslFilter(sslContext);
                  sslFilter.setUseClientMode(true);
                  connector.getFilterChain().addFirst("sslFilter", sslFilter);
              }
  
              connector.setHandler(handler);
              ConnectFuture future1 = connector.connect(address);
              future1.awaitUninterruptibly();
              if (!future1.isConnected()) {
                  return false;
              }
              session = future1.getSession();
              login();
  
              return true;
          } catch (Exception e) {
              return false;
          }
     }
 
     public void login() {
         session.write("LOGIN " + name);
     }
 
     public void broadcast(String message) {
         try {
             session.write("BROADCAST " + message);
         } catch ( Exception e ) {
             e.printStackTrace();
         }
     }
 
     public void quit() {
         if (session != null) {
             if (session.isConnected()) {
                 session.write("QUIT");
                 // Wait until the chat ends.
                 session.getCloseFuture().awaitUninterruptibly();
             }
             session.closeNow();
         }
     }
 
 }