/*
    * Copyright 2012 The Netty Project
    *
    * The Netty Project 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:
    *
    *   https://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 io.netty5.example.securechat;
  
  import io.netty5.bootstrap.Bootstrap;
  import io.netty5.channel.Channel;
  import io.netty5.channel.EventLoopGroup;
  import io.netty5.channel.MultithreadEventLoopGroup;
  import io.netty5.channel.nio.NioHandler;
  import io.netty5.channel.socket.nio.NioSocketChannel;
  import io.netty5.example.telnet.TelnetClient;
  import io.netty5.handler.ssl.SslContext;
  import io.netty5.handler.ssl.SslContextBuilder;
  import io.netty5.handler.ssl.util.InsecureTrustManagerFactory;
  import io.netty5.util.concurrent.Future;
  
  import java.io.BufferedReader;
  import java.io.InputStreamReader;
  
  /**
   * Simple SSL chat client modified from {@link TelnetClient}.
   */
  public final class SecureChatClient {
  
      static final String HOST = System.getProperty("host", "127.0.0.1");
      static final int PORT = Integer.parseInt(System.getProperty("port", "8992"));
  
      public static void main(String[] args) throws Exception {
          // Configure SSL.
          final SslContext sslCtx = SslContextBuilder.forClient()
              .trustManager(InsecureTrustManagerFactory.INSTANCE).build();
  
          EventLoopGroup group = new MultithreadEventLoopGroup(NioHandler.newFactory());
          try {
              Bootstrap b = new Bootstrap();
              b.group(group)
               .channel(NioSocketChannel.class)
               .handler(new SecureChatClientInitializer(sslCtx));
  
              // Start the connection attempt.
              Channel ch = b.connect(HOST, PORT).asStage().get();
  
              // Read commands from the stdin.
              Future<Void> lastWriteFuture = null;
              BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
              for (;;) {
                  String line = in.readLine();
                  if (line == null) {
                      break;
                  }
  
                  // Sends the received line to the server.
                  lastWriteFuture = ch.writeAndFlush(line + "\r\n");
  
                  // If user typed the 'bye' command, wait until the server closes
                  // the connection.
                  if ("bye".equalsIgnoreCase(line)) {
                      ch.closeFuture().asStage().sync();
                      break;
                  }
              }
  
              // Wait until all messages are flushed before closing the channel.
              if (lastWriteFuture != null) {
                  lastWriteFuture.asStage().sync();
              }
          } finally {
              // The connection is closed automatically on shutdown.
              group.shutdownGracefully();
          }
      }
  }