package org.xmlsh.aws;

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

public class snsPublish extends AWSSNSCommand {

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

    Options opts = getOptions("t=topic:,m=message:,s=subject:");
    parseOptions(opts, args);

    args = opts.getRemainingArgs();
    setSerializeOpts(this.getSerializeOpts(opts));

    String topic = opts.getOptStringRequired("topic");
    String message = opts.getOptStringRequired("message");
    String subject = opts.getOptString("subject", null);

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

    }

    int ret;

    ret = publish(topic, subject, message);

    return ret;

  }

  private int publish(String topic, String subject, String message)
      throws IOException, XMLStreamException, SaxonApiException, CoreException {

    PublishRequest request = new PublishRequest().withTopicArn(topic)
        .withMessage(message).withSubject(subject);
    traceCall("publish");

    PublishResult result = getAWSClient().publish(request);

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

    startDocument();
    startElement(getName());

    startElement("message");
    attribute("id", result.getMessageId());
    endElement();

    endElement();
    endDocument();
    closeWriter();
    stdout.writeSequenceTerminator(getSerializeOpts());

    return 0;

  }

}