package com.sforce.async;

import com.sforce.ws.ConnectionException;
import com.sforce.ws.parser.PullParserException;
import com.sforce.ws.transport.Transport;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

public class CsvBatchRequest {
  private final OutputStream csvStream;
  
  private final Transport transport;
  
  private boolean isNew;
  
  private final String UTF8 = "UTF-8";
  
  public CsvBatchRequest(Transport transport, OutputStream out) throws IOException {
    this.transport = transport;
    this.csvStream = out;
    this.isNew = true;
  }
  
  public void addHeader(String[] columns) throws AsyncApiException {
    if (this.isNew) {
      addRow(columns);
    } else {
      throw new AsyncApiException("Can't add header to populated CSV", AsyncExceptionCode.ClientInputError);
    } 
  }
  
  public void addRow(String[] columns) throws AsyncApiException {
    try {
      for (int i = 0; i < columns.length; i++) {
        if (i == 0) {
          addFirstColumn(columns[i]);
        } else {
          addColumn(columns[i]);
        } 
      } 
      this.isNew = false;
    } catch (IOException e) {
      throw new AsyncApiException("Failed to add row", AsyncExceptionCode.ClientInputError, e);
    } 
  }
  
  public BatchInfo completeRequest() throws AsyncApiException {
    try {
      this.csvStream.close();
      InputStream in = this.transport.getContent();
      if (this.transport.isSuccessful())
        return BatchRequest.loadBatchInfo(in); 
      BulkConnection.parseAndThrowException(in, ContentType.CSV);
    } catch (IOException e) {
      throw new AsyncApiException("Failed to complete request", AsyncExceptionCode.ClientInputError, e);
    } catch (PullParserException e) {
      throw new AsyncApiException("Failed to complete request", AsyncExceptionCode.ClientInputError, e);
    } catch (ConnectionException e) {
      throw new AsyncApiException("Failed to complete request", AsyncExceptionCode.ClientInputError, e);
    } 
    return null;
  }
  
  private void addFirstColumn(String value) throws IOException {
    this.csvStream.write(",".getBytes("UTF-8"));
    addColumn(value);
  }
  
  private void addColumn(String value) throws IOException {
    if (value != null) {
      this.csvStream.write("\"".getBytes("UTF-8"));
      this.csvStream.write(value.replace("\"", "\"\"").getBytes("UTF-8"));
      this.csvStream.write("\"".getBytes("UTF-8"));
    } 
  }
}