/*
    * ====================================================================
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF 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
   *
   *   http://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.
   * ====================================================================
   *
   * This software consists of voluntary contributions made by many
   * individuals on behalf of the Apache Software Foundation.  For more
   * information on the Apache Software Foundation, please see
   * <http://www.apache.org/>.
   *
   */
  
  package org.apache.hc.client5.http.async.methods;
  
  import java.util.Iterator;
  
  import org.apache.hc.core5.http.ContentType;
  import org.apache.hc.core5.http.Header;
  import org.apache.hc.core5.http.HttpResponse;
  import org.apache.hc.core5.http.message.BasicHttpResponse;
  import org.apache.hc.core5.util.Args;
  
  /**
   * HTTP response that can enclose a body represented as a simple text string or an array of bytes.
   * <p>
   * IMPORTANT: {@link SimpleHttpResponse}s are intended for simple scenarios where entities inclosed
   * in responses are known to be small. It is generally recommended to use streaming
   * {@link org.apache.hc.core5.http.nio.AsyncResponseConsumer}s, for instance, such as based on
   * {@link AbstractCharResponseConsumer} or {@link AbstractBinResponseConsumer}.
   *
   * @since 5.0
   *
   * @see SimpleBody
   * @see AbstractCharResponseConsumer
   * @see AbstractBinResponseConsumer
   */
  public final class SimpleHttpResponse extends BasicHttpResponse {
  
      private static final long serialVersionUID = 1L;
      private SimpleBody body;
  
      public SimpleHttpResponse(final int code) {
          super(code);
      }
  
      public SimpleHttpResponse(final int code, final String reasonPhrase) {
          super(code, reasonPhrase);
      }
  
      public static SimpleHttpResponse copy(final HttpResponse original) {
          Args.notNull(original, "HTTP response");
          final SimpleHttpResponse copy = new SimpleHttpResponse(original.getCode());
          copy.setVersion(original.getVersion());
          for (final Iterator<Header> it = original.headerIterator(); it.hasNext(); ) {
              copy.addHeader(it.next());
          }
          return copy;
      }
  
      public static SimpleHttpResponse create(final int code) {
          return new SimpleHttpResponse(code);
      }
  
      public static SimpleHttpResponse create(final int code, final String content, final ContentType contentType) {
          final SimpleHttpResponse response = new SimpleHttpResponse(code);
          if (content != null) {
              response.setBody(content, contentType);
          }
          return response;
      }
  
      public static SimpleHttpResponse create(final int code, final String content) {
          return create(code, content, ContentType.TEXT_PLAIN);
      }
  
      public static SimpleHttpResponse create(final int code, final byte[] content, final ContentType contentType) {
          final SimpleHttpResponse response = new SimpleHttpResponse(code);
          if (content != null) {
              response.setBody(content, contentType);
          }
          return response;
      }
  
      public static SimpleHttpResponse create(final int code, final byte[] content) {
         return create(code, content, ContentType.TEXT_PLAIN);
     }
 
     public void setBody(final SimpleBody body) {
         this.body = body;
     }
 
     public void setBody(final byte[] bodyBytes, final ContentType contentType) {
         this.body = SimpleBody.create(bodyBytes, contentType);
     }
 
     public void setBody(final String bodyText, final ContentType contentType) {
         this.body = SimpleBody.create(bodyText, contentType);
     }
 
     public SimpleBody getBody() {
         return body;
     }
 
     public ContentType getContentType() {
         return body != null ? body.getContentType() : null;
     }
 
     public String getBodyText() {
         return body != null ? body.getBodyText() : null;
     }
 
     public byte[] getBodyBytes() {
         return body != null ? body.getBodyBytes() : null;
     }
 
 }