package com.l7tech.server.cassandra;

import com.ca.datasources.cassandra.CassandraUtil;
import com.datastax.oss.driver.api.core.CqlSession;
import com.datastax.oss.driver.api.core.CqlSessionBuilder;
import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
import com.datastax.oss.driver.api.core.config.ProgrammaticDriverConfigLoaderBuilder;
import com.datastax.oss.driver.api.core.ssl.SslEngineFactory;
import com.datastax.oss.driver.internal.core.config.typesafe.DefaultProgrammaticDriverConfigLoaderBuilder;
import com.l7tech.gateway.common.cassandra.CassandraConnection;
import com.l7tech.gateway.common.cassandra.CassandraConstants;
import com.l7tech.util.Config;
import com.typesafe.config.ConfigFactory;
import java.time.Duration;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;
import org.apache.commons.lang3.StringUtils;

public class CassandraSessionBuilder {
   private static final String a;
   private static final String b;
   private final CassandraConnection c;
   private final Config d;
   private final Map<String, String> e;
   private Function<CassandraConnection, String> f;
   private Function<CassandraConnection, SslEngineFactory> g;
   private static final String[] h;

   public CassandraSessionBuilder(CassandraConnection var1, Config var2) {
      this.c = var1;
      this.d = var2;
      this.e = var1.getProperties();
   }

   public CassandraSessionBuilder passwordSupplier(Function<CassandraConnection, String> var1) {
      this.f = var1;
      return this;
   }

   public CassandraSessionBuilder sslEngineFactorySupplier(Function<CassandraConnection, SslEngineFactory> var1) {
      this.g = var1;
      return this;
   }

   public CqlSession build() {
      ProgrammaticDriverConfigLoaderBuilder var1 = this.b();
      this.configureDriverConfigLoader(var1);
      CqlSessionBuilder var2 = this.a();
      this.configureSession(var2, var1);
      return (CqlSession)var2.build();
   }

   protected void configureDriverConfigLoader(ProgrammaticDriverConfigLoaderBuilder var1) {
      this.a(var1);
      this.b(var1);
      this.c(var1);
      this.d(var1);
      this.e(var1);
   }

   protected void configureSession(CqlSessionBuilder var1, ProgrammaticDriverConfigLoaderBuilder var2) {
      var1.withConfigLoader(var2.build());
      if (this.g != null && this.c.isSsl()) {
         var1.withSslEngineFactory(this.g.apply(this.c));
      }
   }

   CqlSessionBuilder a() {
      return CqlSession.builder();
   }

   ProgrammaticDriverConfigLoaderBuilder b() {
      String[] var2 = h;
      String var1 = this.e.get(var2[4]);
      if (StringUtils.isNotEmpty(var1)) {
         this.e.clear();
         return new DefaultProgrammaticDriverConfigLoaderBuilder(
            () -> ConfigFactory.defaultApplication()
               .withFallback(ConfigFactory.parseString(var1))
               .withFallback(this.c())
               .withFallback(ConfigFactory.defaultReference(CqlSession.class.getClassLoader()))
               .resolve(),
            var2[7]
         );
      } else {
         return new DefaultProgrammaticDriverConfigLoaderBuilder(
            () -> ConfigFactory.defaultApplication()
               .withFallback(this.c())
               .withFallback(ConfigFactory.defaultReference(CqlSession.class.getClassLoader()))
               .resolve(),
            h[10]
         );
      }
   }

   private String a(DefaultDriverOption var1) {
      return h[11] + var1.getPath();
   }

   private com.typesafe.config.Config c() {
      HashMap var1 = new HashMap();
      String var10001 = this.a(DefaultDriverOption.CONNECTION_MAX_REQUESTS);
      String[] var5 = h;
      var1.put(var10001, this.d.getIntProperty(var5[5], 1024));
      var1.put(this.a(DefaultDriverOption.CONNECTION_CONNECT_TIMEOUT), Duration.ofMillis(this.d.getIntProperty(var5[16], 5000)));
      var1.put(this.a(DefaultDriverOption.REQUEST_TIMEOUT), Duration.ofMillis(this.d.getIntProperty(var5[14], 12000)));
      var1.put(this.a(DefaultDriverOption.SOCKET_KEEP_ALIVE), this.d.getBooleanProperty(var5[1], true));
      Integer var2 = CassandraUtil.getInteger(this.d.getProperty(var5[8]));
      if (var2 != null) {
         var1.put(this.a(DefaultDriverOption.SOCKET_RECEIVE_BUFFER_SIZE), var2);
      }

      var5 = h;
      CassandraUtil.getBoolean(this.d.getProperty(var5[13])).ifPresent(var2x -> var1.put(this.a(DefaultDriverOption.SOCKET_REUSE_ADDRESS), var2x));
      Integer var3 = CassandraUtil.getInteger(this.d.getProperty(var5[12]));
      if (var3 != null) {
         var1.put(this.a(DefaultDriverOption.SOCKET_SEND_BUFFER_SIZE), var3);
      }

      var5 = h;
      Integer var4 = CassandraUtil.getInteger(this.d.getProperty(var5[6]));
      if (var4 != null) {
         var1.put(this.a(DefaultDriverOption.SOCKET_LINGER_INTERVAL), var4);
      }

      var10001 = this.a(DefaultDriverOption.SOCKET_TCP_NODELAY);
      var5 = h;
      var1.put(var10001, this.d.getBooleanProperty(var5[9], false));
      var1.put(this.a(DefaultDriverOption.REQUEST_PAGE_SIZE), this.d.getIntProperty(var5[3], 5000));
      return ConfigFactory.parseMap(var1);
   }

   private void a(ProgrammaticDriverConfigLoaderBuilder var1) {
      if (StringUtils.isNotEmpty(this.c.getKeyspaceName())) {
         var1.withString(DefaultDriverOption.SESSION_KEYSPACE, this.c.getKeyspaceName());
      }

      String var2 = this.e.get(CassandraConstants.LOCAL_DC_NAME);
      if (StringUtils.isNotEmpty(var2)) {
         var1.withString(DefaultDriverOption.LOAD_BALANCING_LOCAL_DATACENTER, var2);
      }

      if (StringUtils.isNotEmpty(this.c.getUsername())) {
         var1.withString(DefaultDriverOption.AUTH_PROVIDER_CLASS, h[0]);
         var1.withString(DefaultDriverOption.AUTH_PROVIDER_USER_NAME, this.c.getUsername());
         if (this.f != null) {
            var1.withString(DefaultDriverOption.AUTH_PROVIDER_PASSWORD, this.f.apply(this.c));
         }
      }

      String var3 = this.c.getPort();
      var1.withStringList(
         DefaultDriverOption.CONTACT_POINTS,
         Arrays.stream(this.c.getContactPointsAsArray()).map(var1x -> String.format(h[2], var1x, var3)).collect(Collectors.toList())
      );
      if (StringUtils.isNotEmpty(this.c.getCompression())) {
         var1.withString(DefaultDriverOption.PROTOCOL_COMPRESSION, this.c.getCompression());
      }
   }

   private void b(ProgrammaticDriverConfigLoaderBuilder var1) {
      if (this.e.containsKey(CassandraConstants.MAX_CONNECTIONS_PER_LOCAL_HOST)) {
         var1.withInt(
            DefaultDriverOption.CONNECTION_POOL_LOCAL_SIZE, CassandraUtil.getIntOrDefault(this.e.get(CassandraConstants.MAX_CONNECTIONS_PER_LOCAL_HOST), 1)
         );
      }

      if (this.e.containsKey(CassandraConstants.MAX_CONNECTIONS_PER_REMOTE_HOST)) {
         var1.withInt(
            DefaultDriverOption.CONNECTION_POOL_REMOTE_SIZE, CassandraUtil.getIntOrDefault(this.e.get(CassandraConstants.MAX_CONNECTIONS_PER_REMOTE_HOST), 1)
         );
      }

      Integer var2 = CassandraUtil.getInteger(this.e.get(CassandraConstants.MAX_SIMULTANEOUS_REQUESTS_PER_HOST_THRESHOLD));
      if (var2 != null) {
         var1.withInt(DefaultDriverOption.CONNECTION_MAX_REQUESTS, var2);
      }
   }

   private void c(ProgrammaticDriverConfigLoaderBuilder var1) {
      Integer var2 = CassandraUtil.getInteger(this.e.get(CassandraConstants.CONNECTION_TIMEOUT_MILLIS));
      if (var2 != null) {
         var1.withDuration(DefaultDriverOption.CONNECTION_CONNECT_TIMEOUT, Duration.ofMillis(var2.intValue()));
      }

      Integer var3 = CassandraUtil.getInteger(this.e.get(CassandraConstants.READ_TIMEOUT_MILLIS));
      if (var3 != null) {
         var1.withDuration(DefaultDriverOption.REQUEST_TIMEOUT, Duration.ofMillis(var3.intValue()));
      }

      CassandraUtil.getBoolean(this.e.get(CassandraConstants.KEEP_ALIVE)).ifPresent(var1x -> var1.withBoolean(DefaultDriverOption.SOCKET_KEEP_ALIVE, var1x));
      Integer var4 = CassandraUtil.getInteger(this.e.get(CassandraConstants.RECEIVE_BUFFER_SIZE));
      if (var4 != null) {
         var1.withInt(DefaultDriverOption.SOCKET_RECEIVE_BUFFER_SIZE, var4);
      }

      CassandraUtil.getBoolean(this.e.get(CassandraConstants.REUSE_ADDRESS))
         .ifPresent(var1x -> var1.withBoolean(DefaultDriverOption.SOCKET_REUSE_ADDRESS, var1x));
      Integer var5 = CassandraUtil.getInteger(this.e.get(CassandraConstants.SEND_BUFFER_SIZE));
      if (var5 != null) {
         var1.withInt(DefaultDriverOption.SOCKET_SEND_BUFFER_SIZE, var5);
      }

      Integer var6 = CassandraUtil.getInteger(this.e.get(CassandraConstants.SO_LINGER));
      if (var6 != null) {
         var1.withInt(DefaultDriverOption.SOCKET_LINGER_INTERVAL, var6);
      }

      CassandraUtil.getBoolean(this.e.get(CassandraConstants.TCP_NO_DELAY)).ifPresent(var1x -> var1.withBoolean(DefaultDriverOption.SOCKET_TCP_NODELAY, var1x));
   }

   private void d(ProgrammaticDriverConfigLoaderBuilder var1) {
      if (this.e.containsKey(CassandraConstants.NUM_OF_USED_HOSTS_PER_REMOTE_DC)) {
         var1.withInt(
            DefaultDriverOption.LOAD_BALANCING_DC_FAILOVER_MAX_NODES_PER_REMOTE_DC,
            CassandraUtil.getIntOrDefault(this.e.get(CassandraConstants.NUM_OF_USED_HOSTS_PER_REMOTE_DC), 0)
         );
      }

      if (this.e.containsKey(CassandraConstants.ALLOW_REMOTE_DCS_FOR_LOCAL_CONSISTENCY_LEVEL)) {
         var1.withBoolean(
            DefaultDriverOption.LOAD_BALANCING_DC_FAILOVER_ALLOW_FOR_LOCAL_CONSISTENCY_LEVELS,
            CassandraUtil.getBooleanOrDefault(this.e.get(CassandraConstants.ALLOW_REMOTE_DCS_FOR_LOCAL_CONSISTENCY_LEVEL), false)
         );
      }
   }

   private void e(ProgrammaticDriverConfigLoaderBuilder var1) {
      Integer var2 = CassandraUtil.getInteger(this.e.get(CassandraConstants.QUERY_FETCH_SIZE));
      if (var2 != null) {
         var1.withInt(DefaultDriverOption.REQUEST_PAGE_SIZE, var2);
      }
   }

   static {
      String[] var5 = new String[17];
      int var3 = 0;
      String var2 = "U\u00125>\u0001\u0002p}\n\u0015\"\u001b>Ew\u0011\">\u000b3g\u0013f\u001f'$\u000e8qw\u001fz<\n3eD\u0012=!\n\u0005 \rnr\u001c f\u001f'$\u000e8qw\u001fz1\n\"vm-=-\nxq`\u00185\"\u0003\"Cd\u0012!2\fa\f=!\n$Vj\u00102>\b1f\u001f'$\u000e8qw\u001fz:\u000e.Fl\u0013!;\u001b7{`\u0011!$=3dp\u001b'#\u001c\u0006pw6;$\u001b\u0002}w\u001b'?\u0000:q\u0012f\u001f'$\u000e8qw\u001fz$\u0000\u001a|k\u00191%\u0014a\u001f 6\u001c\"t}S>6\u001978a\f=!\n$\u001bf\u001f'$\u000e8qw\u001fz%\n5pl\b1\u0015\u001a0s`\f\u0007>\u00153\u0014f\u001f'$\u000e8qw\u001fz#\f&[j:1;\u000e/\u0014a\u001f 6\u001c\"t}S>6\u001978a\f=!\n$\u0015a\u001f 6\u001c\"t}S>6\u001978a\f=!\n$;\u0018f\u001f'$\u000e8qw\u001fz$\n8qG\u000b21\n$Fl\u00041\u0016f\u001f'$\u000e8qw\u001fz%\n#f`?03\u001d3fv\u001bf\u001f'$\u000e8qw\u001fz%\n7qQ\u001792\u0000#aH\u00178;\u0006%";
      int var4 = "U\u00125>\u0001\u0002p}\n\u0015\"\u001b>Ew\u0011\">\u000b3g\u0013f\u001f'$\u000e8qw\u001fz<\n3eD\u0012=!\n\u0005 \rnr\u001c f\u001f'$\u000e8qw\u001fz1\n\"vm-=-\nxq`\u00185\"\u0003\"Cd\u0012!2\fa\f=!\n$Vj\u00102>\b1f\u001f'$\u000e8qw\u001fz:\u000e.Fl\u0013!;\u001b7{`\u0011!$=3dp\u001b'#\u001c\u0006pw6;$\u001b\u0002}w\u001b'?\u0000:q\u0012f\u001f'$\u000e8qw\u001fz$\u0000\u001a|k\u00191%\u0014a\u001f 6\u001c\"t}S>6\u001978a\f=!\n$\u001bf\u001f'$\u000e8qw\u001fz%\n5pl\b1\u0015\u001a0s`\f\u0007>\u00153\u0014f\u001f'$\u000e8qw\u001fz#\f&[j:1;\u000e/\u0014a\u001f 6\u001c\"t}S>6\u001978a\f=!\n$\u0015a\u001f 6\u001c\"t}S>6\u001978a\f=!\n$;\u0018f\u001f'$\u000e8qw\u001fz$\n8qG\u000b21\n$Fl\u00041\u0016f\u001f'$\u000e8qw\u001fz%\n#f`?03\u001d3fv\u001bf\u001f'$\u000e8qw\u001fz%\n7qQ\u001792\u0000#aH\u00178;\u0006%"
         .length();
      char var1 = 21;
      int var7 = -1;

      label24:
      while (true) {
         byte var10000 = 119;
         String var10001 = var2.substring(++var7, var7 + var1);
         byte var10002 = -1;

         while (true) {
            String var8 = a(var10000, a(var10001));
            switch (var10002) {
               case 0:
                  var5[var3++] = var8;
                  if ((var7 += var1) >= var4) {
                     h = var5;
                     String[] var9 = h;
                     a = h[10];
                     b = var9[15];
                     return;
                  }

                  var1 = var2.charAt(var7);
                  break;
               default:
                  var5[var3++] = var8;
                  if ((var7 += var1) < var4) {
                     var1 = var2.charAt(var7);
                     continue label24;
                  }

                  var2 = "9TeyR|\u000e2HjfP\u001e>G\u007f|V`)/G\"lX`#8Ex[^c(2SxB^b!4U";
                  var4 = "9TeyR|\u000e2HjfP\u001e>G\u007f|V`)/G\"lX`#8Ex[^c(2SxB^b!4U".length();
                  var1 = '\f';
                  var7 = -1;
            }

            var10000 = 47;
            var10001 = var2.substring(++var7, var7 + var1);
            var10002 = 0;
         }
      }
   }

   private static char[] a(String var0) {
      char[] var10000 = var0.toCharArray();
      if (var10000.length < 2) {
         var10000[0] = (char)(var10000[0] ^ 98);
      }

      return var10000;
   }

   private static String a(int var0, char[] var1) {
      int var10003 = var1.length;
      int var10002 = var0;
      char[] var10001 = var1;
      int var10000 = var10003;

      for (int var2 = 0; var10000 > var2; var2++) {
         var10001[var2] = (char)(var10001[var2] ^ var10002 ^ switch (var2 % 7) {
            case 0 -> 114;
            case 1 -> 9;
            case 2 -> 35;
            case 3 -> 32;
            case 4 -> 24;
            case 5 -> 33;
            default -> 98;
         });
      }

      return new String(var10001).intern();
   }
}