com.symphony.bdk.bot.sdk.command.MultiResponseComposerImpl Maven / Gradle / Ivy

Go to download
package com.symphony.bdk.bot.sdk.command;

import lombok.Getter;

import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import com.symphony.bdk.bot.sdk.event.model.MessageAttachmentFile;
import com.symphony.bdk.bot.sdk.symphony.model.SymphonyMessage;

/**
 * Implementation of {@link MultiResponseComposer}
 *
 * @author Gabriel Berberian
 */
public class MultiResponseComposerImpl implements MultiResponseComposer, ComposerMessageDefinition,
    ComposerAttachmentOrStreamDefinition, ComposerStreamsDefinition {

  @Getter private boolean complete;

  @Getter private Map> composedResponse;
  private SymphonyMessage message;

  public MultiResponseComposerImpl() {
    this.complete = true;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerMessageDefinition compose() {
    this.composedResponse = new HashMap<>();
    this.complete = false;
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerAttachmentOrStreamDefinition withMessage(String message) {
    this.message = new SymphonyMessage(message);
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerAttachmentOrStreamDefinition withEnrichedMessage(String message, String entityName,
      Object entity, String version) {
    this.message = new SymphonyMessage();
    this.message.setEnrichedMessage(message, entityName, entity, version);
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerAttachmentOrStreamDefinition withTemplateMessage(String templateMessage,
      Object templateData) {
    this.message = new SymphonyMessage();
    this.message.setTemplateMessage(templateMessage, templateData);
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerAttachmentOrStreamDefinition withEnrichedTemplateMessage(String templateMessage,
      Object templateData, String entityName, Object entity, String version) {
    this.message = new SymphonyMessage();
    this.message.setEnrichedTemplateMessage(
        templateMessage, templateData, entityName, entity, version);
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerAttachmentOrStreamDefinition withTemplateFile(String templateFile,
      Object templateData) {
    this.message = new SymphonyMessage();
    this.message.setTemplateFile(templateFile, templateData);
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerAttachmentOrStreamDefinition withEnrichedTemplateFile(String templateFile,
      Object templateData, String entityName, Object entity, String version) {
    this.message = new SymphonyMessage();
    this.message.setEnrichedTemplateFile(templateFile, templateData, entityName, entity, version);
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerMessageDefinition toStreams(String... streamIds) {
    composedResponse.put(message, Arrays.stream(streamIds).collect(Collectors.toSet()));
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerMessageDefinition toStreams(Collection streamIds) {
    composedResponse.put(message, new HashSet<>(streamIds));
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerStreamsDefinition withAttachments(MessageAttachmentFile... attachments) {
    this.message.setAttachments(Arrays.asList(attachments));
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public ComposerStreamsDefinition withAttachments(Collection attachments) {
    this.message.setAttachments(attachments.stream().collect(Collectors.toList()));
    return this;
  }

  /**
   * {@inheritDoc}
   */
  @Override
  public void complete() {
    this.complete = true;
  }

  protected boolean hasContent() {
    return composedResponse != null && !composedResponse.isEmpty();
  }
}