package org.jboss.resteasy.test.client.jetty;

import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.time.Duration;
import java.util.Random;
import java.util.concurrent.CompletionStage;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

import javax.servlet.ServletInputStream;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.ProcessingException;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.ClientRequestContext;
import javax.ws.rs.client.ClientResponseContext;
import javax.ws.rs.client.ClientResponseFilter;
import javax.ws.rs.client.CompletionStageRxInvoker;
import javax.ws.rs.client.Entity;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.StreamingOutput;

import org.apache.http.entity.ContentType;
import org.eclipse.jetty.client.HttpClient;
import org.eclipse.jetty.server.Request;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.ServerConnector;
import org.eclipse.jetty.server.handler.AbstractHandler;
import org.jboss.resteasy.client.jaxrs.ResteasyClientBuilder;
import org.jboss.resteasy.client.jaxrs.engines.jetty.JettyClientEngine;
import org.junit.After;
import org.junit.Test;

public class JettyClientEngineTest {
   Server server = new Server(0);
   Client client;

   @After
   public void stop() throws Exception {
      if (client != null) {
         client.close();
      }
      server.stop();
   }

   private Client client() throws Exception {
      if (!server.isStarted()) {
         server.start();
      }
      if (client == null) {
         final HttpClient hc = new HttpClient();
         client = ((ResteasyClientBuilder)ClientBuilder.newBuilder()).httpEngine(new JettyClientEngine(hc)).build();
      }
      return client;
   }

   @Test
   public void testSimple() throws Exception {
      server.setHandler(new AbstractHandler() {
         @Override
         public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException {
            baseRequest.setHandled(true);
            if (baseRequest.getHeader("User-Agent").contains("Apache")) {
               response.setStatus(503);
            } else if (!"abracadabra".equals(baseRequest.getHeader("Password"))) {
               response.setStatus(403);
            } else {
               response.setStatus(200);
               response.getWriter().println("Success");
            }
         }
      });

      final Response response = client().target(baseUri()).request()
         .header("Password", "abracadabra")
         .get();

      assertEquals(200, response.getStatus());
      assertEquals("Success" + System.lineSeparator(), response.readEntity(String.class));
   }

   @Test
   public void testSimpleResponseRx() throws Exception {
      server.setHandler(new AbstractHandler() {
         @Override
         public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException {
            baseRequest.setHandled(true);
            if (baseRequest.getHeader("User-Agent").contains("Apache")) {
               response.setStatus(503);
            } else if (!"abracadabra".equals(baseRequest.getHeader("Password"))) {
               response.setStatus(403);
            } else {
               response.setStatus(200);
               response.setContentType(ContentType.TEXT_PLAIN.getMimeType());
               response.getWriter().println("Success");
            }
         }
      });

      final CompletionStage<Response> cs = client().target(baseUri()).request()
         .header("Password", "abracadabra").rx(CompletionStageRxInvoker.class)
         .get();

      Response response = cs.toCompletableFuture().get();
      assertEquals(200, response.getStatus());
      assertEquals("Success" + System.lineSeparator(), response.readEntity(String.class));
   }

   @Test
   public void testSimpleStringRx() throws Exception {
      server.setHandler(new AbstractHandler() {
         @Override
         public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException {
            baseRequest.setHandled(true);
            if (baseRequest.getHeader("User-Agent").contains("Apache")) {
               response.setStatus(503);
            } else if (!"abracadabra".equals(baseRequest.getHeader("Password"))) {
               response.setStatus(403);
            } else {
               response.setStatus(200);
               response.setContentType(ContentType.TEXT_PLAIN.getMimeType());
               response.getWriter().println("Success");
            }
         }
      });

      final CompletionStage<String> cs = client().target(baseUri()).request()
         .header("Password", "abracadabra").rx(CompletionStageRxInvoker.class)
         .get(String.class);

      String response = cs.toCompletableFuture().get();
      assertEquals("Success" + System.lineSeparator(), response);
   }

   @Test
   public void testBigly() throws Exception {
      server.setHandler(new EchoHandler());
      final byte[] valuableData = randomAlpha().getBytes(StandardCharsets.UTF_8);
      final Response response = client().target(baseUri()).request()
            .post(Entity.entity(valuableData, MediaType.APPLICATION_OCTET_STREAM_TYPE));

      assertEquals(200, response.getStatus());
      assertArrayEquals(valuableData, response.readEntity(byte[].class));
   }

   @Test
   public void testFutureResponse() throws Exception {
      server.setHandler(new EchoHandler());
      final String valuableData = randomAlpha();
      final Future<Response> response = client().target(baseUri()).request()
            .buildPost(Entity.entity(valuableData, MediaType.APPLICATION_OCTET_STREAM_TYPE))
            .submit();

      final Response resp = response.get(10, TimeUnit.SECONDS);
      assertEquals(200, resp.getStatus());
      assertEquals(valuableData, resp.readEntity(String.class));
   }

   @Test
   public void testFutureString() throws Exception {
      server.setHandler(new EchoHandler());
      final String valuableData = randomAlpha();
      final Future<String> response = client().target(baseUri()).request()
            .buildPost(Entity.entity(valuableData, MediaType.APPLICATION_OCTET_STREAM_TYPE))
            .submit(String.class);

      final String result = response.get(10, TimeUnit.SECONDS);
      assertEquals(valuableData, result);
   }

   private String randomAlpha() {
      final StringBuilder builder = new StringBuilder();
      final Random r = new Random();
      for (int i = 0; i < 20 * 1024 * 1024; i++) {
         builder.append((char) ('a' + (char) r.nextInt('z' - 'a')));
         if (i % 100 == 0) builder.append('\n');
      }
      return builder.toString();
   }

   @Test
   public void testTimeout() throws Exception {
      server.setHandler(new AbstractHandler() {
         @Override
         public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException {
            try {
               Thread.sleep(1000);
            } catch (InterruptedException e) {
               Thread.currentThread().interrupt();
               throw new AssertionError(e);
            }
            baseRequest.setHandled(true);
         }
      });

      try {
         client().target(baseUri()).request()
            .property(JettyClientEngine.REQUEST_TIMEOUT_MS, Duration.ofMillis(500))
            .get();
         fail();
      } catch (ProcessingException e) {
         assertTrue(e.getCause() instanceof TimeoutException);
      }
   }

   @Test
   public void testDeferContent() throws Exception {
      server.setHandler(new EchoHandler());
      final byte[] valuableData = randomAlpha().getBytes(StandardCharsets.UTF_8);
      final Response response = client().target(baseUri()).request()
            .post(Entity.entity(new StreamingOutput() {
               @Override
               public void write(OutputStream output) throws IOException, WebApplicationException {
                  try {
                     Thread.sleep(100);
                  } catch (InterruptedException e) {
                     Thread.currentThread().interrupt();
                     throw new AssertionError(e);
                  }
                  output.write(valuableData);
               }
            }, MediaType.APPLICATION_OCTET_STREAM_TYPE));

      assertEquals(200, response.getStatus());
      assertArrayEquals(valuableData, response.readEntity(byte[].class));
   }

   @Test
   public void testFilterBufferReplay() throws Exception {
      final String greeting = "Success";
      final byte[] expected = (greeting + System.lineSeparator()).getBytes(StandardCharsets.UTF_8);
      server.setHandler(new AbstractHandler() {
         @Override
         public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException {
            baseRequest.setHandled(true);
            response.setStatus(200);
            response.setContentType(MediaType.TEXT_PLAIN);
            response.getWriter().println(greeting);
         }
      });

      final byte[] content = new byte[expected.length];
      final ClientResponseFilter capturer = new ClientResponseFilter() {
         @Override
         public void filter(ClientRequestContext requestContext, ClientResponseContext responseContext) throws IOException {
            responseContext.getEntityStream().read(content);
         }
      };

      try (InputStream response = client().register(capturer).target(baseUri()).request()
         .get(InputStream.class)) {
         // ignored, we are checking filter
      }

      assertArrayEquals(expected, content);
   }

   public URI baseUri() {
      return URI.create("http://localhost:" + ((ServerConnector) server.getConnectors()[0]).getLocalPort());
   }

   static class EchoHandler extends AbstractHandler {
      @Override
      public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException {
         baseRequest.setHandled(true);
         String type = request.getContentType();
         if (type == null) {
            type = MediaType.TEXT_PLAIN;
         }

         response.setContentType(type);
         response.setStatus(200);
         int read;
         final byte[] data = new byte[1024];
         final ServletInputStream in = request.getInputStream();
         final ServletOutputStream out = response.getOutputStream();
         while ((read = in.read(data)) != -1) {
            out.write(data, 0, read);
         }
      }
   }
}
