package org.xmlsh.aws;

import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import javax.xml.stream.XMLStreamException;
import org.xmlsh.aws.util.AWSGlacierCommand;
import org.xmlsh.core.CoreException;
import org.xmlsh.core.Options;
import org.xmlsh.core.UnexpectedException;
import org.xmlsh.core.XValue;
import org.xmlsh.core.io.OutputPort;
import org.xmlsh.util.Util;
import com.amazonaws.services.glacier.model.DescribeJobRequest;
import com.amazonaws.services.glacier.model.DescribeJobResult;
import com.amazonaws.services.glacier.model.GetJobOutputRequest;
import com.amazonaws.services.glacier.model.GetJobOutputResult;
import com.amazonaws.services.glacier.model.InitiateJobRequest;
import com.amazonaws.services.glacier.model.InitiateJobResult;
import com.amazonaws.services.glacier.model.JobParameters;
import com.amazonaws.services.glacier.model.ListVaultsRequest;
import net.sf.saxon.s9api.SaxonApiException;

public class glacierInventoryVault extends AWSGlacierCommand {

  /**
   * @param args
   * @throws IOException
   */
  @Override
  public int run(List<XValue> args) throws Exception {

    Options opts = getOptions();
    parseOptions(opts, args);
    setSerializeOpts(this.getSerializeOpts(opts));

    args = opts.getRemainingArgs();
    if(args.size() != 1)
      usage();

    String vault = args.get(0).toString();

    try {
      getGlacierClient(opts);
    } catch (UnexpectedException e) {
      usage(e.getLocalizedMessage());
      return 1;

    }

    int ret = -1;
    ret = inventory(vault);

    return ret;

  }

  private int inventory(String vault) throws IOException, XMLStreamException,
      SaxonApiException, CoreException, InterruptedException {

    OutputPort stdout = this.getStdout();

    ListVaultsRequest request = new ListVaultsRequest();

    InitiateJobRequest initiateJobRequest = new InitiateJobRequest(vault,
        new JobParameters().withType("inventory-retrieval"));

    traceCall("initiateJob");

    InitiateJobResult result = getAWSClient().initiateJob(initiateJobRequest);

    DescribeJobRequest describeJobRequest = new DescribeJobRequest(vault,
        result.getJobId());

    String status = null;
    DescribeJobResult describeResult = null;
    do {
      traceCall("describeJob");

      describeResult = getAWSClient().describeJob(describeJobRequest);

      status = describeResult.getStatusCode();

      mShell.printOut(status);
      if(!status.equals("InProgress"))
        break;

      Thread.sleep(10 * 1000);

    } while(true);

    if(status.equals("Succeeded")) {

      GetJobOutputRequest getJobOutputRequest = new GetJobOutputRequest(vault,
          result.getJobId(), null);

      traceCall("getJobOutput");

      GetJobOutputResult jobOutputResult = getAWSClient()
          .getJobOutput(getJobOutputRequest);
      InputStream jobOutput = jobOutputResult.getBody();
      Util.copyStream(jobOutput, stdout.asOutputStream(getSerializeOpts()));
      jobOutput.close();

    }

    return 0;

  }

  @Override
  public void usage() {
    super.usage();
  }

}