package org.kuali.common.httplib.api.model;
   
   import static com.google.common.io.ByteSource.wrap;
   import static com.google.common.io.ByteStreams.copy;
   import static org.kuali.common.httplib.impl.HttpUtils.fromOptionalHeader;
   import static org.kuali.common.jute.base.Optionals.fromAbsentToNegative;
   import static org.kuali.common.jute.reflect.Reflection.checkNoNulls;
   
   import java.io.ByteArrayOutputStream;
  import java.io.IOException;
  import java.io.InputStream;
  import java.io.OutputStream;
  
  import org.apache.http.Header;
  import org.apache.http.HttpEntity;
  
  import com.fasterxml.jackson.annotation.JsonIgnore;
  import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
  import com.google.common.io.ByteSource;
  
  @JsonDeserialize(builder = ByteSourceEntity.Builder.class)
  public final class ByteSourceEntity implements HttpEntity {
  
      private final ContentMetadata metadata;
      private final ByteSource byteSource;
      private final boolean truncated;
  
      private ByteSourceEntity(Builder builder) {
          this.truncated = builder.truncated;
          this.metadata = builder.metadata;
          this.byteSource = builder.byteSource;
      }
  
      public static ByteSourceEntity copyOf(HttpEntity entity) throws IOException {
          if (entity instanceof ByteSourceEntity) {
              return (ByteSourceEntity) entity;
          } else {
              Builder builder = builder(entity);
              ByteArrayOutputStream baos = new ByteArrayOutputStream();
              entity.writeTo(baos);
              builder.withByteSource(wrap(baos.toByteArray()));
              return builder.build();
          }
      }
  
      public static ByteSourceEntity build(HttpEntity entity, ByteSource byteSource, boolean truncated) {
          return builder(entity).withByteSource(byteSource).withTruncated(truncated).build();
      }
  
      public static Builder builder(HttpEntity entity) {
          Builder builder = builder();
          builder.withMetadata(ContentMetadata.build(entity));
          return builder;
      }
  
      public static Builder builder() {
          return new Builder();
      }
  
      public static class Builder implements org.apache.commons.lang3.builder.Builder<ByteSourceEntity> {
  
          private boolean truncated = false;
          private ContentMetadata metadata;
          private ByteSource byteSource;
  
          public Builder withMetadata(ContentMetadata metadata) {
              this.metadata = metadata;
              return this;
          }
  
          public Builder withByteSource(ByteSource byteSource) {
              this.byteSource = byteSource;
              return this;
          }
  
          public Builder withTruncated(boolean truncated) {
              this.truncated = truncated;
              return this;
          }
  
          @Override
          public ByteSourceEntity build() {
              return checkNoNulls(new ByteSourceEntity(this));
          }
  
      }
  
      public boolean isTruncated() {
          return truncated;
      }
  
      public ContentMetadata getMetadata() {
          return metadata;
      }
  
      @Override
      @JsonIgnore
      public boolean isChunked() {
          return false;
     }
 
     @Override
     @JsonIgnore
     public boolean isStreaming() {
         return false;
     }
 
     @Override
     @JsonIgnore
     public boolean isRepeatable() {
         return true;
     }
 
     @Override
     @JsonIgnore
     public long getContentLength() {
         return fromAbsentToNegative(metadata.getContentLength());
     }
 
     @Override
     @JsonIgnore
     public Header getContentType() {
         return fromOptionalHeader(metadata.getContentType());
     }
 
     @Override
     @JsonIgnore
     public Header getContentEncoding() {
         return fromOptionalHeader(metadata.getContentEncoding());
     }
 
     @Override
     @JsonIgnore
     public InputStream getContent() throws IOException {
         return byteSource.openStream();
     }
 
     public ByteSource getByteSource() {
         return byteSource;
     }
 
     @Override
     public void writeTo(OutputStream out) throws IOException {
         copy(getContent(), out);
     }
 
     /**
      * Do not call this method. It doesn't do anything
      *
      * @deprecated
      * @see getContent()
      * @see writeTo(OutputStream out)
      */
     @Override
     @Deprecated
     public void consumeContent() {}
 
 }