com.joshlong.esb.springintegration.modules.social.twitter.TwitterTweetSendingMessageHandler.java Source code

Introduction

Here is the source code for com.joshlong.esb.springintegration.modules.social.twitter.TwitterTweetSendingMessageHandler.java

Source

/*
 * Copyright 2010 the original author or authors
 *
 *     Licensed under the Apache License, Version 2.0 (the "License");
 *     you may not use this file except in compliance with the License.
 *     You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 *     Unless required by applicable law or agreed to in writing, software
 *     distributed under the License is distributed on an "AS IS" BASIS,
 *     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *     See the License for the specific language governing permissions and
 *     limitations under the License.
 */
package com.joshlong.esb.springintegration.modules.social.twitter;

import org.apache.commons.lang.StringUtils;
import org.apache.commons.lang.exception.ExceptionUtils;

import org.apache.log4j.Logger;

import org.springframework.beans.factory.InitializingBean;

import org.springframework.integration.core.Message;
import org.springframework.integration.core.MessageHeaders;
import org.springframework.integration.message.MessageDeliveryException;
import org.springframework.integration.message.MessageHandler;
import org.springframework.integration.message.MessageHandlingException;
import org.springframework.integration.message.MessageRejectedException;

import org.springframework.util.Assert;

import twitter4j.Twitter;
import twitter4j.TwitterException;

/**
 * @author <a href="mailto:josh@joshlong.com">Josh Long</a>
 */
public class TwitterTweetSendingMessageHandler implements InitializingBean, MessageHandler {
    private static Logger logger = Logger.getLogger(TwitterTweetSendingMessageHandler.class);
    private volatile String password;
    private volatile String username;
    private volatile Twitter twitter;
    private volatile TwitterMessageType type;

    public void afterPropertiesSet() throws Exception {
        Assert.state(null != type);

        if (twitter == null) {
            Assert.state(!StringUtils.isEmpty(username));
            Assert.state(!StringUtils.isEmpty(password));

            twitter = new Twitter();
            twitter.setUserId(username);
            twitter.setPassword(password);
        } else { // it isnt null, in which case it becomes canonical memory
            setPassword(twitter.getPassword());
            setUsername(twitter.getUserId());
        }
    }

    public String getPassword() {
        return password;
    }

    public Twitter getTwitter() {
        return twitter;
    }

    public TwitterMessageType getType() {
        return type;
    }

    public String getUsername() {
        return username;
    }

    public void handleMessage(Message<?> message)
            throws MessageRejectedException, MessageHandlingException, MessageDeliveryException {
        Object payload = message.getPayload();

        if (payload instanceof String) {
            String msg = (String) payload;
            tweet(message.getHeaders(), msg);
        } else if (payload instanceof Tweet) {
            Tweet twt = (Tweet) payload;
            tweet(message.getHeaders(), twt.getMessage());
        } else {
            throw new RuntimeException("Can't tweet! Payload was not a 'Tweet' or 'String'");
        }
    }

    public void setPassword(String password) {
        this.password = password;
    }

    public void setTwitter(Twitter twitter) {
        this.twitter = twitter;
    }

    public void setType(TwitterMessageType type) {
        this.type = type;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    public void tweet(MessageHeaders headers, String tweet) {
        try {
            String dmTarget = (String) headers.get(TwitterConstants.TWITTER_DM_TARGET_USER_ID);

            if (type.equals(TwitterMessageType.FRIENDS)) {
                twitter.updateStatus(tweet);
            } else if (type.equals(TwitterMessageType.DM) && !StringUtils.isEmpty(dmTarget)) {
                twitter.sendDirectMessage(dmTarget, tweet);
            } else {
                throw new RuntimeException("can't send direct message without proper"
                        + " message header for TwitterConstants.TWITTER_DM_TARGET_USER_ID!");
            }
        } catch (TwitterException e) {
            logger.debug(ExceptionUtils.getFullStackTrace(e));
        }
    }
}