package com.srv4pos.server.api.infrastructure.http;
   
   import com.srv4pos.server.api.infrastructure.ConnectionWriter;
   import com.srv4pos.server.api.infrastructure.HttpMethod;
   
   import java.io.IOException;
   import java.io.InputStream;
   import java.net.HttpURLConnection;
   import java.net.URL;
  import java.util.Locale;
  
  /**
   * Provides HTTP/HTTPS network layer for server api.
   * <p/>
   * <p>User: Kirill, Date: 13.09.13 13:35</p>
   */
  public class HttpUrlTrasport implements Transport {
  
      private HttpURLConnection connection;
      private boolean isConnectionClose;
  
      public HttpUrlTrasport() {
      }
  
      public HttpUrlTrasport(boolean isConnectionClose) {
          this.isConnectionClose = isConnectionClose;
      }
  
      public Response invoke(HttpMethod httpMethod, String uri, String authentication, String eTag, String mimeType,
                             ConnectionWriter body, Locale acceptLanguage) {
          try {
              URL url = new URL(uri);
  
              if (connection != null) {
                  closeConnection();
              }
  
              connection = (HttpURLConnection) url.openConnection();
              connection.setRequestMethod(httpMethod.name());
              if (authentication != null) {
                  connection.setRequestProperty(HttpHelper.AUTHORIZATION, authentication);
              }
              if (eTag != null) {
                  if (httpMethod.equals(httpMethod.GET)) {
                      connection.setRequestProperty("If-None-Match", eTag);
                  } else {
                      connection.setRequestProperty("If-Match", eTag);
                  }
              }
              if (acceptLanguage != null) {
                  connection.setRequestProperty("Accept-Language", HttpHelper.formatAcceptLanguageHeader(acceptLanguage));
              }
              boolean doOutput = HttpMethod.POST.equals(httpMethod) || HttpMethod.PUT.equals(httpMethod);
              connection.addRequestProperty(HttpHelper.ACCEPT, mimeType);
              connection.addRequestProperty(HttpHelper.CONTENT_TYPE, mimeType);
              if (isConnectionClose) {
                  connection.addRequestProperty(HttpHelper.CONNECTION, "close");
              }
              connection.setDoOutput(doOutput);
              connection.setUseCaches(false);
              connection.setInstanceFollowRedirects(false); // setting needed to switch of redirect if status code == 401
              Integer contentLength = body.getContentLength();
              if (doOutput && contentLength != null) {
                  connection.setFixedLengthStreamingMode(contentLength);
              }
  
              connection.connect();
              if (doOutput) {
                  body.write(connection.getOutputStream());
              }
              int responseCode = connection.getResponseCode();
              InputStream inputStream = HttpHelper.isStatusCodeInSuccessfulClass(responseCode) ? connection.getInputStream() : connection.getErrorStream();
              return new Response(
                      inputStream,
                      responseCode,
                      connection.getContentEncoding(),
                      connection.getContentType());
          } catch (IOException e) {
              throw new RuntimeException(e);
          }
      }
  
      public void closeConnection() {
          if (connection != null) {
              connection.disconnect();
              connection = null;
          }
      }
  }