net.gpedro.integrations.slack.SlackMessage Maven / Gradle / Ivy

Go to download
package net.gpedro.integrations.slack;

import com.google.gson.JsonArray;
import com.google.gson.JsonObject;

import java.util.ArrayList;
import java.util.List;

public class SlackMessage {

    private static final String CHANNEL = "channel";
    private static final String USERNAME = "username";
    private static final String HTTP = "http";
    private static final String ICON_URL = "icon_url";
    private static final String ICON_EMOJI = "icon_emoji";
    private static final String UNFURL_MEDIA = "unfurl_media";
    private static final String UNFURL_LINKS = "unfurl_links";
    private static final String TEXT = "text";
    private static final String ATTACHMENTS = "attachments";

    private List attach = new ArrayList();
    private String channel = null;
    private String icon = null;
    private JsonObject slackMessage = new JsonObject();

    private String text = null;
    private String username = null;

    private boolean unfurlMedia = false;
    private boolean unfurlLinks = false;

    public SlackMessage() {
    }

    public SlackMessage(String text) {
        this(null, null, text);
    }

    public SlackMessage(String username, String text) {
        this(null, username, text);
    }

    public SlackMessage(String channel, String username, String text) {
        if (channel != null) {
            this.channel = channel;
        }

        if (username != null) {
            this.username = username;
        }

        this.text = text;
    }

    public SlackMessage addAttachments(SlackAttachment attach) {
        this.attach.add(attach);

        return this;
    }

    /**
     * Convert SlackMessage to JSON
     * 
     * @return JsonObject
     */
    public JsonObject prepare() {
        if (channel != null) {
            slackMessage.addProperty(CHANNEL, channel);
        }

        if (username != null) {
            slackMessage.addProperty(USERNAME, username);
        }

        if (icon != null) {
            if (icon.contains(HTTP)) {
                slackMessage.addProperty(ICON_URL, icon);
            } else {
                slackMessage.addProperty(ICON_EMOJI, icon);
            }
        }

        slackMessage.addProperty(UNFURL_MEDIA, unfurlMedia);
        slackMessage.addProperty(UNFURL_LINKS, unfurlLinks);

        if (text == null) {
            throw new IllegalArgumentException(
                    "Missing Text field @ SlackMessage");
        } else {
            slackMessage.addProperty(TEXT, text);
        }

        if (!attach.isEmpty()) {
            slackMessage.add(ATTACHMENTS, this.prepareAttach());
        }

        return slackMessage;
    }

    private JsonArray prepareAttach() {
        final JsonArray attachs = new JsonArray();
        for (SlackAttachment attach : this.attach) {
            attachs.add(attach.toJson());
        }

        return attachs;
    }

    public SlackMessage removeAttachment(int index) {
        this.attach.remove(index);

        return this;
    }

    public SlackMessage setAttachments(List attach) {
        this.attach = attach;

        return this;
    }

    public SlackMessage setChannel(String channel) {
        if (channel != null) {
            this.channel = channel;
        }

        return this;
    }

    /**
     * See more icons in http://www.emoji-cheat-sheet.com/
     * 
     * @param icon
     *            Avatar
     * @return SlackMessage
     */
    public SlackMessage setIcon(String icon) {
        if (icon != null) {
            this.icon = icon;
        }

        return this;
    }

    public SlackMessage setText(String message) {
        if (message != null) {
            this.text = message;
        }

        return this;
    }

    public SlackMessage setUsername(String username) {
        if (username != null) {
            this.username = username;
        }

        return this;
    }

    public SlackMessage setUnfurlMedia(boolean unfurlMedia) {
        this.unfurlMedia = unfurlMedia;

        return this;
    }

    public SlackMessage setUnfurlLinks(boolean unfurlLinks) {
        this.unfurlLinks = unfurlLinks;

        return this;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        final SlackMessage that = (SlackMessage) o;

        if (unfurlMedia != that.unfurlMedia) return false;
        if (unfurlLinks != that.unfurlLinks) return false;
        if (attach != null ? !attach.equals(that.attach) : that.attach != null) return false;
        if (channel != null ? !channel.equals(that.channel) : that.channel != null) return false;
        if (icon != null ? !icon.equals(that.icon) : that.icon != null) return false;
        if (text != null ? !text.equals(that.text) : that.text != null) return false;
        return !(username != null ? !username.equals(that.username) : that.username != null);

    }

    @Override
    public int hashCode() {
        int result = attach != null ? attach.hashCode() : 0;
        result = 31 * result + (channel != null ? channel.hashCode() : 0);
        result = 31 * result + (icon != null ? icon.hashCode() : 0);
        result = 31 * result + (text != null ? text.hashCode() : 0);
        result = 31 * result + (username != null ? username.hashCode() : 0);
        result = 31 * result + (unfurlMedia ? 1 : 0);
        result = 31 * result + (unfurlLinks ? 1 : 0);
        return result;
    }

    @Override
    public String toString() {
        return "SlackMessage{" +
                "attach=" + attach +
                ", channel='" + channel + '\'' +
                ", icon='" + icon + '\'' +
                ", slackMessage=" + slackMessage +
                ", text='" + text + '\'' +
                ", username='" + username + '\'' +
                ", unfurlMedia=" + unfurlMedia +
                ", unfurlLinks=" + unfurlLinks +
                '}';
    }
}