package com.l7tech.server.otel;

import com.l7tech.gateway.common.service.PublishedService;
import com.l7tech.gateway.common.stepdebug.DebugContextVariableData;
import com.l7tech.message.HttpRequestKnob;
import com.l7tech.objectmodel.Goid;
import com.l7tech.policy.assertion.Assertion;
import com.l7tech.policy.assertion.AssertionStatus;
import com.l7tech.policy.assertion.Include;
import com.l7tech.policy.assertion.SetVariableAssertion;
import com.l7tech.policy.assertion.Assertion.Comment;
import com.l7tech.server.audit.Auditor;
import com.l7tech.server.message.PolicyEnforcementContext;
import com.l7tech.server.otel.util.OtelBaggageUtil;
import com.l7tech.server.otel.util.OtelContextUtil;
import com.l7tech.server.otel.util.OtelTraceConfigUtil;
import com.l7tech.server.stepdebug.DebugPecData;
import com.l7tech.util.mdc.JulMdc;
import io.opentelemetry.api.GlobalOpenTelemetry;
import io.opentelemetry.api.baggage.Baggage;
import io.opentelemetry.api.common.Attributes;
import io.opentelemetry.api.common.AttributesBuilder;
import io.opentelemetry.api.internal.ImmutableSpanContext;
import io.opentelemetry.api.trace.Span;
import io.opentelemetry.api.trace.SpanKind;
import io.opentelemetry.api.trace.Tracer;
import io.opentelemetry.context.Context;
import io.opentelemetry.context.propagation.TextMapPropagator;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.StringJoiner;
import java.util.Map.Entry;
import java.util.logging.Level;
import java.util.logging.Logger;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import org.apache.commons.lang3.StringUtils;

public class OtelTraceContextImpl extends OtelMetricContextImpl {
   public static final String OTEL_TRACE_ID;
   public static final String OTEL_SPAN_ID;
   public static final String OTEL_TRACE_ATTRIBUTES_CV;
   public static final String OTEL_SPAN_ATTRIBUTES_CV;
   public static final String OTEL_BAGGAGE_TRACE_ATTRIBUTES_CV;
   public static final String DEFAULT_ATTR_DELIMITER = ",";
   public static final String DEFAULT_KV_DELIMITER = "=";
   private final Tracer l;
   private final String m;
   private final DebugPecData n;
   private final OtelTraceSamplingConfig o;
   private Collection<Integer> p;
   private Span q;
   private final String r;
   private final String s;
   private Map<String, String> t = new HashMap<>();
   private Map<String, String> u = new HashMap<>();
   private Map<String, String> v = new HashMap<>();
   private final Auditor w;
   private static final Logger x;
   private static final TextMapPropagator y;
   private final Set<String> z;
   private final Set<String> A = new HashSet<>();
   private boolean B;
   private static final String[] E;

   public OtelTraceContextImpl(String var1, String var2, String var3, OtelTraceSamplingConfig var4, String var5, String var6, boolean var7, Auditor var8) {
      super(var1, var2, var8);
      this.l = GlobalOpenTelemetry.tracerBuilder(E[6]).setInstrumentationVersion(OTEL_SCOPE_VERSION).build();
      this.m = var3;
      this.o = var4;
      this.r = var5;
      this.z = var6 != null ? Arrays.stream(var6.split(",")).map(String::trim).collect(Collectors.toUnmodifiableSet()) : Collections.emptySet();
      this.s = var2 != null ? var2 : "";
      this.B = var7;
      if (var7) {
         this.n = new DebugPecData(var8);
      } else {
         this.n = null;
      }

      this.w = var8;
   }

   @Override
   public Span getCurrentSpan() {
      return this.q;
   }

   @Override
   public Span startServiceSpan(PolicyEnforcementContext var1) {
      this.p = var1.getAssertionNumber();
      Context var2 = y.extract(Context.current(), var1.getRequest().getHeadersKnob(), k);
      Span var3 = this.l.spanBuilder(this.m).setParent(var2).setSpanKind(SpanKind.SERVER).startSpan();
      this.a(var3, var1);
      String[] var4 = E;
      JulMdc.put(var4[31], var3.getSpanContext().getTraceId() + "-" + var3.getSpanContext().getSpanId());
      var1.setVariable(var4[12], var3.getSpanContext().getTraceId());
      var1.setVariable(var4[0], var3.getSpanContext().getSpanId());
      this.h = Baggage.fromContext(var2);
      this.g.putAll(this.h.asMap());
      OtelBaggageUtil.setBaggageContextVariable(var1, this.g);
      this.q = var3;
      return var3;
   }

   @Override
   public void endServiceSpan(Span var1, AssertionStatus var2, PolicyEnforcementContext var3) {
      if (var1.getSpanContext() != ImmutableSpanContext.INVALID) {
         Goid var4 = var3.getService() == null ? PublishedService.DEFAULT_GOID : var3.getService().getGoid();
         String var5 = var3.getService() == null ? "" : var3.getService().getName();
         String var6 = var3.getService() == null ? "" : var3.getService().getRoutingUri();
         AttributesBuilder var7 = Attributes.builder();
         String[] var9 = E;
         var7.put(this.s + var9[5], var5);
         var7.put(this.s + var9[28], var6);
         var7.put(this.s + var9[23], var9[30]);
         var1.setAttribute(this.s + var9[24], var4.toString());
         if (var3.getRequest().isHttpRequest()) {
            HttpRequestKnob var8 = var3.getRequest().getHttpRequestKnob();
            var7.put(this.s + var9[9], var8.getRequestUrl());
            var7.put(this.s + var9[3], var8.getMethod().name());
         }

         var1.setAllAttributes(var7.build());
         if (this.B) {
            this.n.setPolicyResult(var2, this.p);
            this.a(var1, var3);
         }

         this.b(var1, var3);
         if (!this.u.isEmpty()) {
            this.a(var1, var3, this.u);
         }

         this.a(var1, var3, this.A);
         String var10 = OtelContextUtil.getResolvedContextValue(var3, this.o.getTraceNamePattern(), this.w);
         if (var10 != null && !var10.isBlank()) {
            var1.updateName(var10);
         }

         x.log(Level.FINEST, E[21] + var1.toString());
         var1.end();
      }
   }

   @Override
   public Span startAssertionSpan(Assertion var1, PolicyEnforcementContext var2) {
      this.p = var2.getAssertionNumber();
      this.c(var1);
      this.a(var1, var2);
      this.a(var1);
      if (this.b(var1)) {
         return Span.getInvalid();
      }

      String var3 = this.p.stream().map(String::valueOf).collect(Collectors.joining("."));
      StringJoiner var4 = new StringJoiner(" ");
      var4.add(var3);
      Comment var5 = var1.getAssertionComment();
      String var6 = this.getAssertionShortName(var1);
      String var7 = var1 instanceof Include var8 ? var8.getPolicyName() : null;
      AttributesBuilder var12 = Attributes.builder();
      String[] var11 = E;
      var12.put(this.s + var11[18], var3);
      if (var5 != null) {
         String var9 = var5.getAssertionComment(Comment.LEFT_COMMENT);
         String var10 = var5.getAssertionComment(Comment.RIGHT_COMMENT);
         if (var9 != null) {
            var12.put(this.s + var11[14], var9);
            var4.add(var9);
         }

         var11 = E;
         var4.add(String.format(var11[13], var6));
         if (var10 != null) {
            var12.put(this.s + var11[34], var10);
            var4.add(var10);
         }
      } else {
         var4.add(var6);
      }

      var11 = E;
      var12.put(this.s + var11[11], this.getAssertionFeatureSetName(var1));
      var12.put(this.s + var11[17], var6);
      if (var7 != null) {
         var4.add(String.format(var11[27], var7));
         var12.put(this.s + var11[22], var7);
      }

      Span var13 = this.l.spanBuilder(var4.toString()).setSpanKind(SpanKind.INTERNAL).setAllAttributes(var12.build()).startSpan();
      var11 = E;
      JulMdc.put(var11[2], var13.getSpanContext().getTraceId() + "-" + var13.getSpanContext().getSpanId());
      var2.setVariable(var11[32], var13.getSpanContext().getTraceId());
      var2.setVariable(var11[33], var13.getSpanContext().getSpanId());
      this.q = var13;
      return var13;
   }

   private void a(Assertion var1) {
      if (var1 instanceof SetVariableAssertion) {
         String var2 = ((SetVariableAssertion)var1).getVariableToSet();
         String var3 = ((SetVariableAssertion)var1).expression();
         if (!var3.isBlank()) {
            String[] var6 = E;
            if (var6[10].equalsIgnoreCase(var2)) {
               String var4 = Pattern.quote(",");
               String[] var5 = var3.split(var4);
               x.fine(var6[15] + Arrays.toString(var5));
               this.A.addAll(Arrays.stream(var5).map(String::trim).collect(Collectors.toSet()));
            }
         }
      }
   }

   @Override
   public void endAssertionSpan(Span var1, Assertion var2, PolicyEnforcementContext var3) {
      if (var1.getSpanContext() != ImmutableSpanContext.INVALID) {
         this.a(var1, var3);
         if (!this.v.isEmpty()) {
            this.a(var1, var3, this.v);
         }

         x.log(Level.FINEST, E[4] + var1.toString());
         var1.end();
      }
   }

   private boolean b(Assertion var1) {
      boolean var2 = false;
      List var3 = this.o.getIncludeAssertions();
      List var4 = this.o.getExcludeAssertions();
      String var5 = this.getAssertionShortName(var1);
      if (!var3.isEmpty()) {
         var2 = !var3.stream().anyMatch(var1x -> var1x.matcher(var5).matches());
      }

      if (!var2) {
         var2 = var4.stream().anyMatch(var1x -> var1x.matcher(var5).matches());
      }

      return var2;
   }

   private void a(Span var1, PolicyEnforcementContext var2) {
      if (this.B) {
         this.n.update(var2, Collections.emptySet());

         for (DebugContextVariableData var5 : this.n.getContextVariables()) {
            StringBuilder var6 = new StringBuilder();
            if (OtelTraceConfigUtil.shouldCtxVarBeIncluded(this.o.getIncludeCtxVars(), this.o.getExcludeCtxVars(), var5.getName())) {
               var6.append(var5).append(System.lineSeparator());
               if (!var5.getChildren().isEmpty()) {
                  this.a(var6, 1, var5.getName(), var5.getChildren());
               }
            }

            if (!var6.isEmpty()) {
               var1.addEvent(var6.toString());
            }
         }
      }
   }

   private void a(StringBuilder var1, int var2, String var3, Set<DebugContextVariableData> var4) {
      for (DebugContextVariableData var6 : var4) {
         String var7 = String.format(E[20], var3, var6.getName());
         if (OtelTraceConfigUtil.shouldCtxVarBeIncluded(this.o.getIncludeCtxVars(), this.o.getExcludeCtxVars(), var7)) {
            var1.append(StringUtils.repeat("\t", var2)).append(var6).append(System.lineSeparator());
            if (!var6.getChildren().isEmpty()) {
               this.a(var1, var2 + 1, var7, var6.getChildren());
            }
         }
      }
   }

   private void b(Span var1, PolicyEnforcementContext var2) {
      if (this.r != null && !this.r.isBlank()) {
         OtelTraceConfigUtil.setCustomAttributesMap(this.r, this.t);
         if (!this.t.isEmpty()) {
            this.a(var1, var2, this.t);
         }
      }
   }

   private void c(Assertion var1) {
      if (var1 instanceof SetVariableAssertion) {
         String var2 = ((SetVariableAssertion)var1).getVariableToSet();
         String var3 = ((SetVariableAssertion)var1).expression();
         String[] var4 = E;
         if (var4[26].equalsIgnoreCase(var2)) {
            if (!this.u.isEmpty()) {
               this.u.clear();
            }

            if (!var3.isBlank()) {
               OtelTraceConfigUtil.setCustomAttributesMap(var3, this.u);
            }
         } else if (E[8].equalsIgnoreCase(var2)) {
            if (!this.v.isEmpty()) {
               this.v.clear();
            }

            if (!var3.isBlank()) {
               OtelTraceConfigUtil.setCustomAttributesMap(var3, this.v);
            }
         }
      }
   }

   private void a(Span var1, PolicyEnforcementContext var2, Map<String, String> var3) {
      if (!var3.isEmpty()) {
         for (Entry var5 : var3.entrySet()) {
            String var6 = OtelContextUtil.getResolvedContextValue(var2, (String)var5.getValue(), this.w);
            var1.setAttribute(this.s + (String)var5.getKey(), var6);
         }
      }
   }

   private void a(Span var1, PolicyEnforcementContext var2, Set<String> var3) {
      Map var4 = this.createBaggage().asMap();
      HashSet var5 = new HashSet();
      if (var3.isEmpty()) {
         String[] var6 = E;
         x.fine(var6[29]);
         this.z.forEach(var3x -> var5.add(OtelContextUtil.getResolvedContextValue(var2, var3x, this.w)));
      } else {
         String[] var7 = E;
         x.fine(var7[16]);
         var3.forEach(var3x -> var5.add(OtelContextUtil.getResolvedContextValue(var2, var3x, this.w)));
      }

      x.fine(E[25] + var5);
      OtelBaggageUtil.setBaggageSpanAttributes(var1, var4, var5, this.s);
   }

   static {
      String[] var5 = new String[35];
      int var3 = 0;
      String var2 = "c\u0015#wS>\u0012m\u000f\u000f\u007f\u0014c\u0015#wS>\u0012m\u000fhz\t9\u0010e\u00033o\u0018>\u0007C5\u0003W\"\u0004&\u0006a\u00042s\u0012)\u001bC\u0015#w],\u0011\u007f\u00044o\u0014\"\f,\u00126z\u0013m\u0006i\u00033|Gm\u000b\u007f\u00044m\u0014.\u0007B\u0000+~\u0012o\u000e+5\u0011,\u001bi\u0013q5\u001a,\u0016i\u0016'b\u0015c\u0015#wS9\u0010m\u0002#5\u001c9\u0016~\b$n\t(\u0011\u0014c\u0015#wS>\u0012m\u000fhz\t9\u0010e\u00033o\u0018>\n~\u00047n\u0018>\u0016Y\u0013*\u001dc\u0015#wS/\u0003k\u0006'|\u0018c\u0016~\u0000%~S,\u0016x\u0013/y\b9\u0007\u007f\rm\u00125~\u000f9\u000bc\u000f\u0012b\r(\fc\u0015#wS9\u0010m\u0002#R\u0019\n,Ni;X>B#Nf\u000b`\u0004 o>\"\u000fa\u0004(o\u001d_\u00042;?,\u0005k\u0000!~]\u001e\u0012m\u000ffZ\t9\u0010e\u00033o\u0018>X,#Y\u0012/u\u001am\u000ec\u0002'w]/\u0003k\u0006'|\u0018m\u0011|\u0000(;\u001c9\u0016~\b$n\t(\u0011\t\u007f\t)i\t\u0003\u0003a\u0004\n`\b(~38\u000fn\u00044\u001dc\u0015#wS/\u0003k\u0006'|\u0018c\u0016~\u0000%~S,\u0016x\u0013/y\b9\u0007\u007f\u0005)\u0012h>\u000e\u0019C\u0015#w]>\u0007~\u0017/x\u0018m\u0011|\u0000(;\u0019(\u0000y\u0006|;\n|\u000e*r\u001e4,m\f#\rm\u00125~\u000f9\u000bc\u000f\u0012b\r(\u0004k\u000e/\u007f\u0019N\u0000!|\u001c*\u0007,26z\u0013m#x\u00154r\u001f8\u0016i\u0012|;\u0015c\u0015#wS9\u0010m\u0002#5\u001c9\u0016~\b$n\t(\u0011\u00046Ach\n\u007f\u00044m\u0014.\u0007Y\u0013/<@\u000e%z\u0011m m\u0006!z\u001a(Be\u0012f~\u0010=\u0016uMfn\u000e$\fkA!w\u0012/\u0003`A$z\u001a*\u0003k\u0004fh\r,\f,\u00002o\u000f$\u0000y\u0015#h\u0007_\u00044m\u0014.\u0007\u0007C5\u0003W\"\u0004&\fc\u0015#wS9\u0010m\u0002#R\u0019";
      int var4 = "c\u0015#wS>\u0012m\u000f\u000f\u007f\u0014c\u0015#wS>\u0012m\u000fhz\t9\u0010e\u00033o\u0018>\u0007C5\u0003W\"\u0004&\u0006a\u00042s\u0012)\u001bC\u0015#w],\u0011\u007f\u00044o\u0014\"\f,\u00126z\u0013m\u0006i\u00033|Gm\u000b\u007f\u00044m\u0014.\u0007B\u0000+~\u0012o\u000e+5\u0011,\u001bi\u0013q5\u001a,\u0016i\u0016'b\u0015c\u0015#wS9\u0010m\u0002#5\u001c9\u0016~\b$n\t(\u0011\u0014c\u0015#wS>\u0012m\u000fhz\t9\u0010e\u00033o\u0018>\n~\u00047n\u0018>\u0016Y\u0013*\u001dc\u0015#wS/\u0003k\u0006'|\u0018c\u0016~\u0000%~S,\u0016x\u0013/y\b9\u0007\u007f\rm\u00125~\u000f9\u000bc\u000f\u0012b\r(\fc\u0015#wS9\u0010m\u0002#R\u0019\n,Ni;X>B#Nf\u000b`\u0004 o>\"\u000fa\u0004(o\u001d_\u00042;?,\u0005k\u0000!~]\u001e\u0012m\u000ffZ\t9\u0010e\u00033o\u0018>X,#Y\u0012/u\u001am\u000ec\u0002'w]/\u0003k\u0006'|\u0018m\u0011|\u0000(;\u001c9\u0016~\b$n\t(\u0011\t\u007f\t)i\t\u0003\u0003a\u0004\n`\b(~38\u000fn\u00044\u001dc\u0015#wS/\u0003k\u0006'|\u0018c\u0016~\u0000%~S,\u0016x\u0013/y\b9\u0007\u007f\u0005)\u0012h>\u000e\u0019C\u0015#w]>\u0007~\u0017/x\u0018m\u0011|\u0000(;\u0019(\u0000y\u0006|;\n|\u000e*r\u001e4,m\f#\rm\u00125~\u000f9\u000bc\u000f\u0012b\r(\u0004k\u000e/\u007f\u0019N\u0000!|\u001c*\u0007,26z\u0013m#x\u00154r\u001f8\u0016i\u0012|;\u0015c\u0015#wS9\u0010m\u0002#5\u001c9\u0016~\b$n\t(\u0011\u00046Ach\n\u007f\u00044m\u0014.\u0007Y\u0013/<@\u000e%z\u0011m m\u0006!z\u001a(Be\u0012f~\u0010=\u0016uMfn\u000e$\fkA!w\u0012/\u0003`A$z\u001a*\u0003k\u0004fh\r,\f,\u00002o\u000f$\u0000y\u0015#h\u0007_\u00044m\u0014.\u0007\u0007C5\u0003W\"\u0004&\fc\u0015#wS9\u0010m\u0002#R\u0019"
         .length();
      char var1 = 11;
      int var7 = -1;

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

         while (true) {
            String var8 = b(var10000, b(var10001));
            switch (var10002) {
               case 0:
                  var5[var3++] = var8;
                  if ((var7 += var1) >= var4) {
                     E = var5;
                     String[] var9 = E;
                     OTEL_TRACE_ID = E[32];
                     OTEL_TRACE_ATTRIBUTES_CV = var9[7];
                     OTEL_BAGGAGE_TRACE_ATTRIBUTES_CV = var9[19];
                     OTEL_SPAN_ATTRIBUTES_CV = var9[1];
                     OTEL_SPAN_ID = var9[33];
                     x = Logger.getLogger(OtelTraceContextImpl.class.getName());
                     y = OtelContextFactory.getTextMapPropagator();
                     return;
                  }

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

                  var2 = "(^h<\u0018uY&DD4\f5Cj8BEF*Gh>B";
                  var4 = "(^h<\u0018uY&DD4\f5Cj8BEF*Gh>B".length();
                  var1 = 11;
                  var7 = -1;
            }

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

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

      return var10000;
   }

   private static String b(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 -> 77;
            case 1 -> 32;
            case 2 -> 7;
            case 3 -> 90;
            case 4 -> 60;
            case 5 -> 12;
            default -> 35;
         });
      }

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