/*
    * 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.mock.http;
   
   import jakarta.servlet.ServletOutputStream;
  import jakarta.servlet.WriteListener;
  import jakarta.servlet.http.HttpServletResponse;
  import jakarta.servlet.http.HttpServletResponseWrapper;
  
  import java.io.ByteArrayOutputStream;
  import java.io.IOException;
  import java.io.OutputStreamWriter;
  import java.io.PrintWriter;
  import java.nio.charset.StandardCharsets;
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.mockito.Mockito;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  
  /**
   * The Class SimpleHttpResponse.
   */
  public class SimpleHttpResponse extends HttpServletResponseWrapper {
  
      /** 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<>();
  
      /** The bytes. */
      private final ByteArrayOutputStream bytes = new ByteArrayOutputStream();
  
      /** The out. */
      private final ServletOutputStream out = new ServletOutputStream() {
          @Override
          public void write(final int b) {
              SimpleHttpResponse.this.bytes.write(b);
          }
  
          @Override
          public boolean isReady() {
              return false;
          }
  
          @Override
          public void setWriteListener(final WriteListener writeListener) {
              // Not used
          }
      };
  
      /** The writer. */
      private final PrintWriter writer = new PrintWriter(new OutputStreamWriter(this.bytes, StandardCharsets.UTF_8),
              true);
  
      /**
       * Instantiates a new simple http response.
       */
      public SimpleHttpResponse() {
          super(Mockito.mock(HttpServletResponse.class));
      }
  
      /**
       * Gets the status.
       *
       * @return the status
       */
      @Override
      public int getStatus() {
          return this.status;
      }
  
      @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 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;
     }
 
     /**
      * Gets the status string.
      *
      * @return the status string
      */
     public String getStatusString() {
         if (this.status == 401) {
             return "Unauthorized";
         }
         return "Unknown";
     }
 
     @Override
     public void flushBuffer() {
         SimpleHttpResponse.LOGGER.info("{}: {}", Integer.valueOf(this.status), this.getStatusString());
         for (final Map.Entry<String, List<String>> header : this.headers.entrySet()) {
             for (final String headerValue : header.getValue()) {
                 SimpleHttpResponse.LOGGER.info("{}: {}", header, headerValue);
             }
         }
     }
 
     /**
      * Use this for testing the number of headers.
      *
      * @return int header name size.
      */
     public int getHeaderNamesSize() {
         return this.headers.size();
     }
 
     /**
      * 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]);
     }
 
     /**
      * Gets the header.
      *
      * @param headerName
      *            the header name
      *
      * @return the header
      */
     @Override
     public String getHeader(final String headerName) {
         final List<String> headerValues = this.headers.get(headerName);
         return headerValues == null ? null : String.join(", ", headerValues);
     }
 
     @Override
     public void sendError(final int rc, final String message) {
         this.status = rc;
     }
 
     @Override
     public void sendError(final int rc) {
         this.status = rc;
     }
 
     @Override
     public PrintWriter getWriter() {
         return this.writer;
     }
 
     @Override
     public ServletOutputStream getOutputStream() throws IOException {
         return this.out;
     }
 
     /**
      * Gets the output text.
      *
      * @return the output text
      */
     public String getOutputText() {
         this.writer.flush();
         return this.bytes.toString(StandardCharsets.UTF_8);
     }
 }