package com.srv4pos.server.api.infrastructure;
   
   import com.srv4pos.server.api.infrastructure.http.HttpHelper;
   
   import java.io.IOException;
   import java.io.OutputStream;
   import java.io.OutputStreamWriter;
   import java.nio.charset.Charset;
   import java.util.List;
  
  /**
   * Implementation of {@link ConnectionWriter} which write data from {@link String}.
   * <p>User: Kirill, Date: 23.08.13 14:51</p>
   */
  public class StringConnectionWriter implements ConnectionWriter {
      private String body;
  
      public StringConnectionWriter(String body) {
          this.body = body;
      }
  
      public StringConnectionWriter(List<String> body) {
          this.body = flatString(body);
      }
  
      private String flatString(List<String> lines) {
          final String delimiter = "\n";
          final StringBuilder builder = new StringBuilder();
          for (String line : lines) {
              if (builder.length() != 0) {
                  builder.append(delimiter);
              }
              builder.append(line);
          }
          return builder.toString();
      }
  
      public void write(OutputStream outputStream) throws IOException {
          OutputStreamWriter output = new OutputStreamWriter(outputStream, HttpHelper.ENCODING_FOR_BODY);
          output.write(body);
          output.close();
      }
  
      public Integer getContentLength() {
          return body.getBytes(Charset.forName("UTF-8")).length;
      }
  }