/*
    * Copyright 2016 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.kqueue;
  
  import io.netty.buffer.ByteBufAllocator;
  import io.netty.channel.ChannelException;
  import io.netty.channel.ChannelOption;
  import io.netty.channel.MessageSizeEstimator;
  import io.netty.channel.RecvByteBufAllocator;
  import io.netty.channel.WriteBufferWaterMark;
  import io.netty.channel.socket.SocketChannelConfig;
  import io.netty.util.internal.PlatformDependent;
  import io.netty.util.internal.UnstableApi;
  
  import java.io.IOException;
  import java.util.Map;
  
  import static io.netty.channel.ChannelOption.ALLOW_HALF_CLOSURE;
  import static io.netty.channel.ChannelOption.IP_TOS;
  import static io.netty.channel.ChannelOption.SO_KEEPALIVE;
  import static io.netty.channel.ChannelOption.SO_LINGER;
  import static io.netty.channel.ChannelOption.SO_RCVBUF;
  import static io.netty.channel.ChannelOption.SO_REUSEADDR;
  import static io.netty.channel.ChannelOption.SO_SNDBUF;
  import static io.netty.channel.ChannelOption.TCP_NODELAY;
  import static io.netty.channel.kqueue.KQueueChannelOption.SO_SNDLOWAT;
  import static io.netty.channel.kqueue.KQueueChannelOption.TCP_NOPUSH;
  
  @UnstableApi
  public final class KQueueSocketChannelConfig extends KQueueChannelConfig implements SocketChannelConfig {
      private volatile boolean allowHalfClosure;
      private volatile boolean tcpFastopen;
  
      KQueueSocketChannelConfig(KQueueSocketChannel channel) {
          super(channel);
          if (PlatformDependent.canEnableTcpNoDelayByDefault()) {
              setTcpNoDelay(true);
          }
          calculateMaxBytesPerGatheringWrite();
      }
  
      @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, SO_SNDLOWAT, TCP_NOPUSH);
      }
  
      @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());
          }
          if (option == SO_SNDLOWAT) {
              return (T) Integer.valueOf(getSndLowAt());
          }
          if (option == TCP_NOPUSH) {
              return (T) Boolean.valueOf(isTcpNoPush());
          }
          if (option == ChannelOption.TCP_FASTOPEN_CONNECT) {
              return (T) Boolean.valueOf(isTcpFastOpenConnect());
          }
          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 if (option == SO_SNDLOWAT) {
             setSndLowAt((Integer) value);
         } else if (option == TCP_NOPUSH) {
             setTcpNoPush((Boolean) value);
         } else if (option == ChannelOption.TCP_FASTOPEN_CONNECT) {
             setTcpFastOpenConnect((Boolean) value);
         } else {
             return super.setOption(option, value);
         }
 
         return true;
     }
 
     @Override
     public int getReceiveBufferSize() {
         try {
             return ((KQueueSocketChannel) channel).socket.getReceiveBufferSize();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public int getSendBufferSize() {
         try {
             return ((KQueueSocketChannel) channel).socket.getSendBufferSize();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public int getSoLinger() {
         try {
             return ((KQueueSocketChannel) channel).socket.getSoLinger();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public int getTrafficClass() {
         try {
             return ((KQueueSocketChannel) channel).socket.getTrafficClass();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public boolean isKeepAlive() {
         try {
             return ((KQueueSocketChannel) channel).socket.isKeepAlive();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public boolean isReuseAddress() {
         try {
             return ((KQueueSocketChannel) channel).socket.isReuseAddress();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public boolean isTcpNoDelay() {
         try {
             return ((KQueueSocketChannel) channel).socket.isTcpNoDelay();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     public int getSndLowAt() {
         try {
             return ((KQueueSocketChannel) channel).socket.getSndLowAt();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     public void setSndLowAt(int sndLowAt)  {
         try {
             ((KQueueSocketChannel) channel).socket.setSndLowAt(sndLowAt);
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     public boolean isTcpNoPush() {
         try {
             return ((KQueueSocketChannel) channel).socket.isTcpNoPush();
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     public void setTcpNoPush(boolean tcpNoPush)  {
         try {
             ((KQueueSocketChannel) channel).socket.setTcpNoPush(tcpNoPush);
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public KQueueSocketChannelConfig setKeepAlive(boolean keepAlive) {
         try {
             ((KQueueSocketChannel) channel).socket.setKeepAlive(keepAlive);
             return this;
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public KQueueSocketChannelConfig setReceiveBufferSize(int receiveBufferSize) {
         try {
             ((KQueueSocketChannel) channel).socket.setReceiveBufferSize(receiveBufferSize);
             return this;
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public KQueueSocketChannelConfig setReuseAddress(boolean reuseAddress) {
         try {
             ((KQueueSocketChannel) channel).socket.setReuseAddress(reuseAddress);
             return this;
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public KQueueSocketChannelConfig setSendBufferSize(int sendBufferSize) {
         try {
             ((KQueueSocketChannel) channel).socket.setSendBufferSize(sendBufferSize);
             calculateMaxBytesPerGatheringWrite();
             return this;
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public KQueueSocketChannelConfig setSoLinger(int soLinger) {
         try {
             ((KQueueSocketChannel) channel).socket.setSoLinger(soLinger);
             return this;
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public KQueueSocketChannelConfig setTcpNoDelay(boolean tcpNoDelay) {
         try {
             ((KQueueSocketChannel) channel).socket.setTcpNoDelay(tcpNoDelay);
             return this;
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public KQueueSocketChannelConfig setTrafficClass(int trafficClass) {
         try {
             ((KQueueSocketChannel) channel).socket.setTrafficClass(trafficClass);
             return this;
         } catch (IOException e) {
             throw new ChannelException(e);
         }
     }
 
     @Override
     public boolean isAllowHalfClosure() {
         return allowHalfClosure;
     }
 
     /**
      * Enables client TCP fast open, if available.
      */
     public KQueueSocketChannelConfig setTcpFastOpenConnect(boolean fastOpenConnect) {
         tcpFastopen = fastOpenConnect;
         return this;
     }
 
     /**
      * Returns {@code true} if TCP fast open is enabled, {@code false} otherwise.
      */
     public boolean isTcpFastOpenConnect() {
         return tcpFastopen;
     }
 
     @Override
     public KQueueSocketChannelConfig setRcvAllocTransportProvidesGuess(boolean transportProvidesGuess) {
         super.setRcvAllocTransportProvidesGuess(transportProvidesGuess);
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setPerformancePreferences(
             int connectionTime, int latency, int bandwidth) {
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setAllowHalfClosure(boolean allowHalfClosure) {
         this.allowHalfClosure = allowHalfClosure;
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setConnectTimeoutMillis(int connectTimeoutMillis) {
         super.setConnectTimeoutMillis(connectTimeoutMillis);
         return this;
     }
 
     @Override
     @Deprecated
     public KQueueSocketChannelConfig setMaxMessagesPerRead(int maxMessagesPerRead) {
         super.setMaxMessagesPerRead(maxMessagesPerRead);
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setWriteSpinCount(int writeSpinCount) {
         super.setWriteSpinCount(writeSpinCount);
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setAllocator(ByteBufAllocator allocator) {
         super.setAllocator(allocator);
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setRecvByteBufAllocator(RecvByteBufAllocator allocator) {
         super.setRecvByteBufAllocator(allocator);
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setAutoRead(boolean autoRead) {
         super.setAutoRead(autoRead);
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setAutoClose(boolean autoClose) {
         super.setAutoClose(autoClose);
         return this;
     }
 
     @Override
     @Deprecated
     public KQueueSocketChannelConfig setWriteBufferHighWaterMark(int writeBufferHighWaterMark) {
         super.setWriteBufferHighWaterMark(writeBufferHighWaterMark);
         return this;
     }
 
     @Override
     @Deprecated
     public KQueueSocketChannelConfig setWriteBufferLowWaterMark(int writeBufferLowWaterMark) {
         super.setWriteBufferLowWaterMark(writeBufferLowWaterMark);
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setWriteBufferWaterMark(WriteBufferWaterMark writeBufferWaterMark) {
         super.setWriteBufferWaterMark(writeBufferWaterMark);
         return this;
     }
 
     @Override
     public KQueueSocketChannelConfig setMessageSizeEstimator(MessageSizeEstimator estimator) {
         super.setMessageSizeEstimator(estimator);
         return this;
     }
 
     private void calculateMaxBytesPerGatheringWrite() {
         // Multiply by 2 to give some extra space in case the OS can process write data faster than we can provide.
         int newSendBufferSize = getSendBufferSize() << 1;
         if (newSendBufferSize > 0) {
             setMaxBytesPerGatheringWrite(newSendBufferSize);
         }
     }
 }