/*
 * Daytona
 * Daytona AI platform API Docs
 *
 * The version of the OpenAPI document: 1.0
 * Contact: support@daytona.com
 *
 * NOTE: This class is auto generated by OpenAPI Generator (https://openapi-generator.tech).
 * https://openapi-generator.tech
 * Do not edit the class manually.
 */


package io.daytona.api.client.model;

import java.util.Objects;
import com.google.gson.TypeAdapter;
import com.google.gson.annotations.JsonAdapter;
import com.google.gson.annotations.SerializedName;
import com.google.gson.stream.JsonReader;
import com.google.gson.stream.JsonWriter;
import java.io.IOException;
import java.math.BigDecimal;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonArray;
import com.google.gson.JsonDeserializationContext;
import com.google.gson.JsonDeserializer;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParseException;
import com.google.gson.TypeAdapterFactory;
import com.google.gson.reflect.TypeToken;
import com.google.gson.TypeAdapter;
import com.google.gson.stream.JsonReader;
import com.google.gson.stream.JsonWriter;
import java.io.IOException;

import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import io.daytona.api.client.JSON;

/**
 * TraceSpan
 */
@javax.annotation.Generated(value = "org.openapitools.codegen.languages.JavaClientCodegen", comments = "Generator version: 7.21.0")
public class TraceSpan {
  public static final String SERIALIZED_NAME_TRACE_ID = "traceId";
  @SerializedName(SERIALIZED_NAME_TRACE_ID)
  @javax.annotation.Nonnull
  private String traceId;

  public static final String SERIALIZED_NAME_SPAN_ID = "spanId";
  @SerializedName(SERIALIZED_NAME_SPAN_ID)
  @javax.annotation.Nonnull
  private String spanId;

  public static final String SERIALIZED_NAME_PARENT_SPAN_ID = "parentSpanId";
  @SerializedName(SERIALIZED_NAME_PARENT_SPAN_ID)
  @javax.annotation.Nullable
  private String parentSpanId;

  public static final String SERIALIZED_NAME_SPAN_NAME = "spanName";
  @SerializedName(SERIALIZED_NAME_SPAN_NAME)
  @javax.annotation.Nonnull
  private String spanName;

  public static final String SERIALIZED_NAME_TIMESTAMP = "timestamp";
  @SerializedName(SERIALIZED_NAME_TIMESTAMP)
  @javax.annotation.Nonnull
  private String timestamp;

  public static final String SERIALIZED_NAME_DURATION_NS = "durationNs";
  @SerializedName(SERIALIZED_NAME_DURATION_NS)
  @javax.annotation.Nonnull
  private BigDecimal durationNs;

  public static final String SERIALIZED_NAME_SPAN_ATTRIBUTES = "spanAttributes";
  @SerializedName(SERIALIZED_NAME_SPAN_ATTRIBUTES)
  @javax.annotation.Nonnull
  private Map<String, String> spanAttributes = new HashMap<>();

  public static final String SERIALIZED_NAME_STATUS_CODE = "statusCode";
  @SerializedName(SERIALIZED_NAME_STATUS_CODE)
  @javax.annotation.Nullable
  private String statusCode;

  public static final String SERIALIZED_NAME_STATUS_MESSAGE = "statusMessage";
  @SerializedName(SERIALIZED_NAME_STATUS_MESSAGE)
  @javax.annotation.Nullable
  private String statusMessage;

  public TraceSpan() {
  }

  public TraceSpan traceId(@javax.annotation.Nonnull String traceId) {
    this.traceId = traceId;
    return this;
  }

  /**
   * Trace identifier
   * @return traceId
   */
  @javax.annotation.Nonnull
  public String getTraceId() {
    return traceId;
  }

  public void setTraceId(@javax.annotation.Nonnull String traceId) {
    this.traceId = traceId;
  }


  public TraceSpan spanId(@javax.annotation.Nonnull String spanId) {
    this.spanId = spanId;
    return this;
  }

  /**
   * Span identifier
   * @return spanId
   */
  @javax.annotation.Nonnull
  public String getSpanId() {
    return spanId;
  }

  public void setSpanId(@javax.annotation.Nonnull String spanId) {
    this.spanId = spanId;
  }


  public TraceSpan parentSpanId(@javax.annotation.Nullable String parentSpanId) {
    this.parentSpanId = parentSpanId;
    return this;
  }

  /**
   * Parent span identifier
   * @return parentSpanId
   */
  @javax.annotation.Nullable
  public String getParentSpanId() {
    return parentSpanId;
  }

  public void setParentSpanId(@javax.annotation.Nullable String parentSpanId) {
    this.parentSpanId = parentSpanId;
  }


  public TraceSpan spanName(@javax.annotation.Nonnull String spanName) {
    this.spanName = spanName;
    return this;
  }

  /**
   * Span name
   * @return spanName
   */
  @javax.annotation.Nonnull
  public String getSpanName() {
    return spanName;
  }

  public void setSpanName(@javax.annotation.Nonnull String spanName) {
    this.spanName = spanName;
  }


  public TraceSpan timestamp(@javax.annotation.Nonnull String timestamp) {
    this.timestamp = timestamp;
    return this;
  }

  /**
   * Span start timestamp
   * @return timestamp
   */
  @javax.annotation.Nonnull
  public String getTimestamp() {
    return timestamp;
  }

  public void setTimestamp(@javax.annotation.Nonnull String timestamp) {
    this.timestamp = timestamp;
  }


  public TraceSpan durationNs(@javax.annotation.Nonnull BigDecimal durationNs) {
    this.durationNs = durationNs;
    return this;
  }

  /**
   * Span duration in nanoseconds
   * @return durationNs
   */
  @javax.annotation.Nonnull
  public BigDecimal getDurationNs() {
    return durationNs;
  }

  public void setDurationNs(@javax.annotation.Nonnull BigDecimal durationNs) {
    this.durationNs = durationNs;
  }


  public TraceSpan spanAttributes(@javax.annotation.Nonnull Map<String, String> spanAttributes) {
    this.spanAttributes = spanAttributes;
    return this;
  }

  public TraceSpan putSpanAttributesItem(String key, String spanAttributesItem) {
    if (this.spanAttributes == null) {
      this.spanAttributes = new HashMap<>();
    }
    this.spanAttributes.put(key, spanAttributesItem);
    return this;
  }

  /**
   * Span attributes
   * @return spanAttributes
   */
  @javax.annotation.Nonnull
  public Map<String, String> getSpanAttributes() {
    return spanAttributes;
  }

  public void setSpanAttributes(@javax.annotation.Nonnull Map<String, String> spanAttributes) {
    this.spanAttributes = spanAttributes;
  }


  public TraceSpan statusCode(@javax.annotation.Nullable String statusCode) {
    this.statusCode = statusCode;
    return this;
  }

  /**
   * Status code of the span
   * @return statusCode
   */
  @javax.annotation.Nullable
  public String getStatusCode() {
    return statusCode;
  }

  public void setStatusCode(@javax.annotation.Nullable String statusCode) {
    this.statusCode = statusCode;
  }


  public TraceSpan statusMessage(@javax.annotation.Nullable String statusMessage) {
    this.statusMessage = statusMessage;
    return this;
  }

  /**
   * Status message
   * @return statusMessage
   */
  @javax.annotation.Nullable
  public String getStatusMessage() {
    return statusMessage;
  }

  public void setStatusMessage(@javax.annotation.Nullable String statusMessage) {
    this.statusMessage = statusMessage;
  }

  /**
   * A container for additional, undeclared properties.
   * This is a holder for any undeclared properties as specified with
   * the 'additionalProperties' keyword in the OAS document.
   */
  private Map<String, Object> additionalProperties;

  /**
   * Set the additional (undeclared) property with the specified name and value.
   * If the property does not already exist, create it otherwise replace it.
   *
   * @param key name of the property
   * @param value value of the property
   * @return the TraceSpan instance itself
   */
  public TraceSpan putAdditionalProperty(String key, Object value) {
    if (this.additionalProperties == null) {
        this.additionalProperties = new HashMap<String, Object>();
    }
    this.additionalProperties.put(key, value);
    return this;
  }

  /**
   * Return the additional (undeclared) property.
   *
   * @return a map of objects
   */
  public Map<String, Object> getAdditionalProperties() {
    return additionalProperties;
  }

  /**
   * Return the additional (undeclared) property with the specified name.
   *
   * @param key name of the property
   * @return an object
   */
  public Object getAdditionalProperty(String key) {
    if (this.additionalProperties == null) {
        return null;
    }
    return this.additionalProperties.get(key);
  }


  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }
    TraceSpan traceSpan = (TraceSpan) o;
    return Objects.equals(this.traceId, traceSpan.traceId) &&
        Objects.equals(this.spanId, traceSpan.spanId) &&
        Objects.equals(this.parentSpanId, traceSpan.parentSpanId) &&
        Objects.equals(this.spanName, traceSpan.spanName) &&
        Objects.equals(this.timestamp, traceSpan.timestamp) &&
        Objects.equals(this.durationNs, traceSpan.durationNs) &&
        Objects.equals(this.spanAttributes, traceSpan.spanAttributes) &&
        Objects.equals(this.statusCode, traceSpan.statusCode) &&
        Objects.equals(this.statusMessage, traceSpan.statusMessage)&&
        Objects.equals(this.additionalProperties, traceSpan.additionalProperties);
  }

  @Override
  public int hashCode() {
    return Objects.hash(traceId, spanId, parentSpanId, spanName, timestamp, durationNs, spanAttributes, statusCode, statusMessage, additionalProperties);
  }

  @Override
  public String toString() {
    StringBuilder sb = new StringBuilder();
    sb.append("class TraceSpan {\n");
    sb.append("    traceId: ").append(toIndentedString(traceId)).append("\n");
    sb.append("    spanId: ").append(toIndentedString(spanId)).append("\n");
    sb.append("    parentSpanId: ").append(toIndentedString(parentSpanId)).append("\n");
    sb.append("    spanName: ").append(toIndentedString(spanName)).append("\n");
    sb.append("    timestamp: ").append(toIndentedString(timestamp)).append("\n");
    sb.append("    durationNs: ").append(toIndentedString(durationNs)).append("\n");
    sb.append("    spanAttributes: ").append(toIndentedString(spanAttributes)).append("\n");
    sb.append("    statusCode: ").append(toIndentedString(statusCode)).append("\n");
    sb.append("    statusMessage: ").append(toIndentedString(statusMessage)).append("\n");
    sb.append("    additionalProperties: ").append(toIndentedString(additionalProperties)).append("\n");
    sb.append("}");
    return sb.toString();
  }

  /**
   * Convert the given object to string with each line indented by 4 spaces
   * (except the first line).
   */
  private String toIndentedString(Object o) {
    return o == null ? "null" : o.toString().replace("\n", "\n    ");
  }


  public static HashSet<String> openapiFields;
  public static HashSet<String> openapiRequiredFields;

  static {
    // a set of all properties/fields (JSON key names)
    openapiFields = new HashSet<String>(Arrays.asList("traceId", "spanId", "parentSpanId", "spanName", "timestamp", "durationNs", "spanAttributes", "statusCode", "statusMessage"));

    // a set of required properties/fields (JSON key names)
    openapiRequiredFields = new HashSet<String>(Arrays.asList("traceId", "spanId", "spanName", "timestamp", "durationNs", "spanAttributes"));
  }

  /**
   * Validates the JSON Element and throws an exception if issues found
   *
   * @param jsonElement JSON Element
   * @throws IOException if the JSON Element is invalid with respect to TraceSpan
   */
  public static void validateJsonElement(JsonElement jsonElement) throws IOException {
      if (jsonElement == null) {
        if (!TraceSpan.openapiRequiredFields.isEmpty()) { // has required fields but JSON element is null
          throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "The required field(s) %s in TraceSpan is not found in the empty JSON string", TraceSpan.openapiRequiredFields.toString()));
        }
      }

      // check to make sure all required properties/fields are present in the JSON string
      for (String requiredField : TraceSpan.openapiRequiredFields) {
        if (jsonElement.getAsJsonObject().get(requiredField) == null) {
          throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "The required field `%s` is not found in the JSON string: %s", requiredField, jsonElement.toString()));
        }
      }
        JsonObject jsonObj = jsonElement.getAsJsonObject();
      if (!jsonObj.get("traceId").isJsonPrimitive()) {
        throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "Expected the field `traceId` to be a primitive type in the JSON string but got `%s`", jsonObj.get("traceId").toString()));
      }
      if (!jsonObj.get("spanId").isJsonPrimitive()) {
        throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "Expected the field `spanId` to be a primitive type in the JSON string but got `%s`", jsonObj.get("spanId").toString()));
      }
      if ((jsonObj.get("parentSpanId") != null && !jsonObj.get("parentSpanId").isJsonNull()) && !jsonObj.get("parentSpanId").isJsonPrimitive()) {
        throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "Expected the field `parentSpanId` to be a primitive type in the JSON string but got `%s`", jsonObj.get("parentSpanId").toString()));
      }
      if (!jsonObj.get("spanName").isJsonPrimitive()) {
        throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "Expected the field `spanName` to be a primitive type in the JSON string but got `%s`", jsonObj.get("spanName").toString()));
      }
      if (!jsonObj.get("timestamp").isJsonPrimitive()) {
        throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "Expected the field `timestamp` to be a primitive type in the JSON string but got `%s`", jsonObj.get("timestamp").toString()));
      }
      if ((jsonObj.get("statusCode") != null && !jsonObj.get("statusCode").isJsonNull()) && !jsonObj.get("statusCode").isJsonPrimitive()) {
        throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "Expected the field `statusCode` to be a primitive type in the JSON string but got `%s`", jsonObj.get("statusCode").toString()));
      }
      if ((jsonObj.get("statusMessage") != null && !jsonObj.get("statusMessage").isJsonNull()) && !jsonObj.get("statusMessage").isJsonPrimitive()) {
        throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "Expected the field `statusMessage` to be a primitive type in the JSON string but got `%s`", jsonObj.get("statusMessage").toString()));
      }
  }

  public static class CustomTypeAdapterFactory implements TypeAdapterFactory {
    @SuppressWarnings("unchecked")
    @Override
    public <T> TypeAdapter<T> create(Gson gson, TypeToken<T> type) {
       if (!TraceSpan.class.isAssignableFrom(type.getRawType())) {
         return null; // this class only serializes 'TraceSpan' and its subtypes
       }
       final TypeAdapter<JsonElement> elementAdapter = gson.getAdapter(JsonElement.class);
       final TypeAdapter<TraceSpan> thisAdapter
                        = gson.getDelegateAdapter(this, TypeToken.get(TraceSpan.class));

       return (TypeAdapter<T>) new TypeAdapter<TraceSpan>() {
           @Override
           public void write(JsonWriter out, TraceSpan value) throws IOException {
             JsonObject obj = thisAdapter.toJsonTree(value).getAsJsonObject();
             obj.remove("additionalProperties");
             // serialize additional properties
             if (value.getAdditionalProperties() != null) {
               for (Map.Entry<String, Object> entry : value.getAdditionalProperties().entrySet()) {
                 if (entry.getValue() instanceof String)
                   obj.addProperty(entry.getKey(), (String) entry.getValue());
                 else if (entry.getValue() instanceof Number)
                   obj.addProperty(entry.getKey(), (Number) entry.getValue());
                 else if (entry.getValue() instanceof Boolean)
                   obj.addProperty(entry.getKey(), (Boolean) entry.getValue());
                 else if (entry.getValue() instanceof Character)
                   obj.addProperty(entry.getKey(), (Character) entry.getValue());
                 else {
                   JsonElement jsonElement = gson.toJsonTree(entry.getValue());
                   if (jsonElement.isJsonArray()) {
                     obj.add(entry.getKey(), jsonElement.getAsJsonArray());
                   } else {
                     obj.add(entry.getKey(), jsonElement.getAsJsonObject());
                   }
                 }
               }
             }
             elementAdapter.write(out, obj);
           }

           @Override
           public TraceSpan read(JsonReader in) throws IOException {
             JsonElement jsonElement = elementAdapter.read(in);
             validateJsonElement(jsonElement);
             JsonObject jsonObj = jsonElement.getAsJsonObject();
             // store additional fields in the deserialized instance
             TraceSpan instance = thisAdapter.fromJsonTree(jsonObj);
             for (Map.Entry<String, JsonElement> entry : jsonObj.entrySet()) {
               if (!openapiFields.contains(entry.getKey())) {
                 if (entry.getValue().isJsonPrimitive()) { // primitive type
                   if (entry.getValue().getAsJsonPrimitive().isString())
                     instance.putAdditionalProperty(entry.getKey(), entry.getValue().getAsString());
                   else if (entry.getValue().getAsJsonPrimitive().isNumber())
                     instance.putAdditionalProperty(entry.getKey(), entry.getValue().getAsNumber());
                   else if (entry.getValue().getAsJsonPrimitive().isBoolean())
                     instance.putAdditionalProperty(entry.getKey(), entry.getValue().getAsBoolean());
                   else
                     throw new IllegalArgumentException(String.format(java.util.Locale.ROOT, "The field `%s` has unknown primitive type. Value: %s", entry.getKey(), entry.getValue().toString()));
                 } else if (entry.getValue().isJsonArray()) {
                     instance.putAdditionalProperty(entry.getKey(), gson.fromJson(entry.getValue(), List.class));
                 } else { // JSON object
                     instance.putAdditionalProperty(entry.getKey(), gson.fromJson(entry.getValue(), HashMap.class));
                 }
               }
             }
             return instance;
           }

       }.nullSafe();
    }
  }

  /**
   * Create an instance of TraceSpan given an JSON string
   *
   * @param jsonString JSON string
   * @return An instance of TraceSpan
   * @throws IOException if the JSON string is invalid with respect to TraceSpan
   */
  public static TraceSpan fromJson(String jsonString) throws IOException {
    return JSON.getGson().fromJson(jsonString, TraceSpan.class);
  }

  /**
   * Convert an instance of TraceSpan to an JSON string
   *
   * @return JSON string
   */
  public String toJson() {
    return JSON.getGson().toJson(this);
  }
}

