package org.xmlsh.aws;

import java.io.IOException;
import java.util.List;
import javax.xml.stream.XMLStreamException;
import org.xmlsh.aws.util.AWSSDBCommand;
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 com.amazonaws.services.simpledb.model.GetAttributesRequest;
import com.amazonaws.services.simpledb.model.GetAttributesResult;
import net.sf.saxon.s9api.SaxonApiException;

public class sdbGetAttributes extends AWSSDBCommand {

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

    Options opts = getOptions("c=consistant");
    opts.parse(args);
    setSerializeOpts(this.getSerializeOpts(opts));

    args = opts.getRemainingArgs();

    boolean bConsistant = opts.hasOpt("consistant");

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

    }

    if(args.size() != 2) {
      usage(getName() + " domain item");

    }
    String domain = args.remove(0).toString();
    String item = args.remove(0).toString();

    int ret = -1;
    ret = getAttributes(domain, item, bConsistant);

    return ret;

  }

  private int getAttributes(String domainName, String itemName,
      boolean bConsistantRead)
      throws IOException, XMLStreamException, SaxonApiException, CoreException {

    OutputPort stdout = this.getStdout();
    mWriter = stdout.asXMLStreamWriter(getSerializeOpts());

    startDocument();
    startElement(getName());

    GetAttributesRequest getAttributesRequest = new GetAttributesRequest(
        domainName, itemName).withConsistentRead(bConsistantRead);

    traceCall("getAttributes");

    GetAttributesResult result = getAWSClient()
        .getAttributes(getAttributesRequest);

    if(result.getAttributes().size() > 0)
      writeItem(itemName, result);

    endElement();
    endDocument();

    closeWriter();
    stdout.writeSequenceTerminator(getSerializeOpts());
    return 0;

  }

  private void writeItem(String itemName, GetAttributesResult result)
      throws XMLStreamException {
    startElement("item");
    attribute("name", itemName);
    writeAttributes(result.getAttributes());
    endElement();
  }

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

}