/*
    * 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.netty.channel.socket;
  
  import io.netty.buffer.ByteBufAllocator;
  import io.netty.channel.ChannelException;
  import io.netty.channel.ChannelOption;
  import io.netty.channel.DefaultChannelConfig;
  import io.netty.channel.MessageSizeEstimator;
  import io.netty.channel.RecvByteBufAllocator;
  import io.netty.channel.WriteBufferWaterMark;
  import io.netty.util.internal.ObjectUtil;
  import io.netty.util.internal.PlatformDependent;
  
  import java.net.Socket;
  import java.net.SocketException;
  import java.util.Map;
  
  import static io.netty.channel.ChannelOption.*;
  
  /**
   * The default {@link SocketChannelConfig} implementation.
   */
  public class DefaultSocketChannelConfig extends DefaultChannelConfig
                                          implements SocketChannelConfig {
  
      protected final Socket javaSocket;
      private volatile boolean allowHalfClosure;
  
      /**
       * Creates a new instance.
       */
      public DefaultSocketChannelConfig(SocketChannel channel, Socket javaSocket) {
          super(channel);
          this.javaSocket = ObjectUtil.checkNotNull(javaSocket, "javaSocket");
  
          // Enable TCP_NODELAY by default if possible.
          if (PlatformDependent.canEnableTcpNoDelayByDefault()) {
              try {
                  setTcpNoDelay(true);
              } catch (Exception e) {
                  // Ignore.
              }
          }
      }
  
      @Override
      public Map<ChannelOption<?>, Object> getOptions() {
          return getOptions(
                  super.getOptions(),
                  SO_RCVBUF, SO_SNDBUF, TCP_NODELAY, SO_KEEPALIVE, SO_REUSEADDR, SO_LINGER, IP_TOS,
                  ALLOW_HALF_CLOSURE);
      }
  
      @SuppressWarnings("unchecked")
      @Override
      public <T> T getOption(ChannelOption<T> option) {
          if (option == SO_RCVBUF) {
              return (T) Integer.valueOf(getReceiveBufferSize());
          }
          if (option == SO_SNDBUF) {
              return (T) Integer.valueOf(getSendBufferSize());
          }
          if (option == TCP_NODELAY) {
              return (T) Boolean.valueOf(isTcpNoDelay());
          }
          if (option == SO_KEEPALIVE) {
              return (T) Boolean.valueOf(isKeepAlive());
          }
          if (option == SO_REUSEADDR) {
              return (T) Boolean.valueOf(isReuseAddress());
          }
          if (option == SO_LINGER) {
              return (T) Integer.valueOf(getSoLinger());
          }
          if (option == IP_TOS) {
              return (T) Integer.valueOf(getTrafficClass());
          }
          if (option == ALLOW_HALF_CLOSURE) {
              return (T) Boolean.valueOf(isAllowHalfClosure());
          }
  
          return super.getOption(option);
      }
  
      @Override
     public <T> boolean setOption(ChannelOption<T> option, T value) {
         validate(option, value);
 
         if (option == SO_RCVBUF) {
             setReceiveBufferSize((Integer) value);
         } else if (option == SO_SNDBUF) {
             setSendBufferSize((Integer) value);
         } else if (option == TCP_NODELAY) {
             setTcpNoDelay((Boolean) value);
         } else if (option == SO_KEEPALIVE) {
             setKeepAlive((Boolean) value);
         } else if (option == SO_REUSEADDR) {
             setReuseAddress((Boolean) value);
         } else if (option == SO_LINGER) {
             setSoLinger((Integer) value);
         } else if (option == IP_TOS) {
             setTrafficClass((Integer) value);
         } else if (option == ALLOW_HALF_CLOSURE) {
             setAllowHalfClosure((Boolean) value);
         } else {
             return super.setOption(option, value);
         }
 
         return true;
     }
 
     @Override
     public int getReceiveBufferSize() {
         try {
             return javaSocket.getReceiveBufferSize();
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public int getSendBufferSize() {
         try {
             return javaSocket.getSendBufferSize();
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public int getSoLinger() {
         try {
             return javaSocket.getSoLinger();
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public int getTrafficClass() {
         try {
             return javaSocket.getTrafficClass();
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public boolean isKeepAlive() {
         try {
             return javaSocket.getKeepAlive();
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public boolean isReuseAddress() {
         try {
             return javaSocket.getReuseAddress();
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public boolean isTcpNoDelay() {
         try {
             return javaSocket.getTcpNoDelay();
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public SocketChannelConfig setKeepAlive(boolean keepAlive) {
         try {
             javaSocket.setKeepAlive(keepAlive);
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
         return this;
     }
 
     @Override
     public SocketChannelConfig setPerformancePreferences(
             int connectionTime, int latency, int bandwidth) {
         javaSocket.setPerformancePreferences(connectionTime, latency, bandwidth);
         return this;
     }
 
     @Override
     public SocketChannelConfig setReceiveBufferSize(int receiveBufferSize) {
         try {
             javaSocket.setReceiveBufferSize(receiveBufferSize);
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
         return this;
     }
 
     @Override
     public SocketChannelConfig setReuseAddress(boolean reuseAddress) {
         try {
             javaSocket.setReuseAddress(reuseAddress);
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
         return this;
     }
 
     @Override
     public SocketChannelConfig setSendBufferSize(int sendBufferSize) {
         try {
             javaSocket.setSendBufferSize(sendBufferSize);
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
         return this;
     }
 
     @Override
     public SocketChannelConfig setSoLinger(int soLinger) {
         try {
             if (soLinger < 0) {
                 javaSocket.setSoLinger(false, 0);
             } else {
                 javaSocket.setSoLinger(true, soLinger);
             }
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
         return this;
     }
 
     @Override
     public SocketChannelConfig setTcpNoDelay(boolean tcpNoDelay) {
         try {
             javaSocket.setTcpNoDelay(tcpNoDelay);
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
         return this;
     }
 
     @Override
     public SocketChannelConfig setTrafficClass(int trafficClass) {
         try {
             javaSocket.setTrafficClass(trafficClass);
         } catch (SocketException e) {
             throw new ChannelException(e);
         }
         return this;
     }
 
     @Override
     public boolean isAllowHalfClosure() {
         return allowHalfClosure;
     }
 
     @Override
     public SocketChannelConfig setAllowHalfClosure(boolean allowHalfClosure) {
         this.allowHalfClosure = allowHalfClosure;
         return this;
     }
 
     @Override
     public SocketChannelConfig setConnectTimeoutMillis(int connectTimeoutMillis) {
         super.setConnectTimeoutMillis(connectTimeoutMillis);
         return this;
     }
 
     @Override
     @Deprecated
     public SocketChannelConfig setMaxMessagesPerRead(int maxMessagesPerRead) {
         super.setMaxMessagesPerRead(maxMessagesPerRead);
         return this;
     }
 
     @Override
     public SocketChannelConfig setWriteSpinCount(int writeSpinCount) {
         super.setWriteSpinCount(writeSpinCount);
         return this;
     }
 
     @Override
     public SocketChannelConfig setAllocator(ByteBufAllocator allocator) {
         super.setAllocator(allocator);
         return this;
     }
 
     @Override
     public SocketChannelConfig setRecvByteBufAllocator(RecvByteBufAllocator allocator) {
         super.setRecvByteBufAllocator(allocator);
         return this;
     }
 
     @Override
     public SocketChannelConfig setAutoRead(boolean autoRead) {
          super.setAutoRead(autoRead);
         return this;
     }
 
     @Override
     public SocketChannelConfig setAutoClose(boolean autoClose) {
         super.setAutoClose(autoClose);
         return this;
     }
 
     @Override
     public SocketChannelConfig setWriteBufferHighWaterMark(int writeBufferHighWaterMark) {
         super.setWriteBufferHighWaterMark(writeBufferHighWaterMark);
         return this;
     }
 
     @Override
     public SocketChannelConfig setWriteBufferLowWaterMark(int writeBufferLowWaterMark) {
         super.setWriteBufferLowWaterMark(writeBufferLowWaterMark);
         return this;
     }
 
     @Override
     public SocketChannelConfig setWriteBufferWaterMark(WriteBufferWaterMark writeBufferWaterMark) {
         super.setWriteBufferWaterMark(writeBufferWaterMark);
         return this;
     }
 
     @Override
     public SocketChannelConfig setMessageSizeEstimator(MessageSizeEstimator estimator) {
         super.setMessageSizeEstimator(estimator);
         return this;
     }
 }