/*
    * SPDX-License-Identifier: MIT
    * See LICENSE file for details.
    *
    * Copyright 2010-2026 The Waffle Project Contributors: https://github.com/Waffle/waffle/graphs/contributors
    */
   package waffle.apache.catalina;
   
   import java.util.ArrayList;
  import java.util.Collection;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.apache.catalina.connector.Response;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  /**
   * Simple HTTP Response.
   */
  public class SimpleHttpResponse extends Response {
  
      /**
       * Instantiates a new simple http response.
       *
       * @param coyoteResponse
       *            the coyote response
       */
      public SimpleHttpResponse(final org.apache.coyote.Response coyoteResponse) {
          super(coyoteResponse);
      }
  
      /** The Constant LOGGER. */
      private static final Logger LOGGER = LoggerFactory.getLogger(SimpleHttpResponse.class);
  
      /** The status. */
      private int status = 500;
  
      /** The headers. */
      private final Map<String, List<String>> headers = new HashMap<>();
  
      @Override
      public void addHeader(final String headerName, final String headerValue) {
          List<String> current = this.headers.get(headerName);
          if (current == null) {
              current = new ArrayList<>();
          }
          current.add(headerValue);
          this.headers.put(headerName, current);
      }
  
      @Override
      public void flushBuffer() {
          SimpleHttpResponse.LOGGER.info("{} {}", Integer.valueOf(this.status), this.getStatusString());
          for (final String header : this.headers.keySet()) {
              for (final String headerValue : this.headers.get(header)) {
                  SimpleHttpResponse.LOGGER.info("{}: {}", header, headerValue);
              }
          }
      }
  
      @Override
      public String getHeader(final String headerName) {
          final List<String> headerValues = this.headers.get(headerName);
          return headerValues == null ? null : String.join(", ", headerValues);
      }
  
      @Override
      public Collection<String> getHeaderNames() {
          return this.headers.keySet();
      }
  
      /**
       * Gets the header values.
       *
       * @param headerName
       *            the header name
       *
       * @return the header values
       */
      public String[] getHeaderValues(final String headerName) {
          final List<String> headerValues = this.headers.get(headerName);
          return headerValues == null ? null : headerValues.toArray(new String[0]);
      }
  
      @Override
      public int getStatus() {
          return this.status;
      }
  
      /**
       * Gets the status string.
       *
       * @return the status string
       */
      public String getStatusString() {
          return this.status == 401 ? "Unauthorized" : "Unknown";
      }
 
     @Override
     public void sendError(final int rc) {
         this.status = rc;
     }
 
     @Override
     public void sendError(final int rc, final String message) {
         this.status = rc;
     }
 
     @Override
     public void setHeader(final String headerName, final String headerValue) {
         List<String> current = this.headers.get(headerName);
         if (current == null) {
             current = new ArrayList<>();
         } else {
             current.clear();
         }
         current.add(headerValue);
         this.headers.put(headerName, current);
     }
 
     @Override
     public void setStatus(final int value) {
         this.status = value;
     }
 
 }