CallBackServlet.java

/**
	TwitStreet - Twitter Stock Market Game
    Copyright (C) 2012  Engin Guller (bisanthe@gmail.com), Cagdas Ozek (cagdasozek@gmail.com)

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
**/

package com.twitstreet.servlet;

import java.io.IOException;
import java.util.Calendar;

import javax.inject.Inject;
import javax.servlet.ServletException;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.log4j.Logger;

import com.google.inject.Singleton;
import com.twitstreet.config.ConfigMgr;
import com.twitstreet.db.data.User;
import com.twitstreet.session.UserMgr;
import com.twitstreet.session.UserMgrImpl;

import twitter4j.Twitter;
import twitter4j.TwitterException;
import twitter4j.TwitterFactory;
import twitter4j.auth.AccessToken;
import twitter4j.auth.RequestToken;

@SuppressWarnings("serial")
@Singleton
public class CallBackServlet extends TwitStreetServlet {
	private static Logger logger = Logger.getLogger(CallBackServlet.class);
	public static final String COOKIE_ID = "id";
	public static final String COOKIE_OAUTHTOKEN = "oauthtoken";
	public static final String REQUEST_TOKEN = "requestToken";
	private static final String OAUTH_VERIFIER = "oauth_verifier";

	private static final int COOKIE_EXPIRE = 30 * 24 * 60 * 60;
	@Inject
	UserMgr userMgr;
	@Inject
	ConfigMgr configMgr;

	protected void doGet(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		response.setContentType("text/html;charset=utf-8");
		response.setHeader("Cache-Control","no-cache"); //HTTP 1.1
		response.setHeader("Pragma","no-cache"); //HTTP 1.0
		response.setDateHeader ("Expires", 0); //prevents caching at the proxy server
		
		Twitter twitter = new TwitterFactory().getInstance();
		twitter.setOAuthConsumer(configMgr.getConsumerKey(),
				configMgr.getConsumerSecret());

		RequestToken requestToken = (RequestToken) request.getSession().getAttribute(REQUEST_TOKEN);
		String verifier = request.getParameter(OAUTH_VERIFIER);
		try {
			AccessToken accessToken = twitter.getOAuthAccessToken(requestToken,
					verifier);
			long userId = accessToken.getUserId();
			String screenName = accessToken.getScreenName();
			String oauthToken = accessToken.getToken();
			String oauthTokenSecret = accessToken.getTokenSecret();
			User user = null;
			user = userMgr.getUserById(userId);
			//new user
			if (user == null) {
				twitter4j.User twUser = twitter.showUser(userId);
				user = new User();
				user.setId(userId);
				user.setUserName(screenName);
				user.setLastIp(request.getRemoteAddr());
				user.setOauthToken(oauthToken);
				user.setOauthTokenSecret(oauthTokenSecret);
				user.setCash(configMgr.getInitialMoney());
				user.setPictureUrl(twUser.getProfileImageURL().toExternalForm());

				
				user.setDescription(twUser.getDescription());
				user.setLongName(twUser.getName());
				user.setLocation(twUser.getLocation());
				user.setLanguage(twUser.getLang());
				user.setUrl(twUser.getURL() == null ? null : twUser.getURL().toExternalForm());
				
				userMgr.saveUser(user);
				String referenceId = (String) request.getSession().getAttribute(HomePageServlet.REFERENCE_ID);
				long referenceLong = -1;
				try {
					referenceLong = Long.parseLong(referenceId);
				}
				catch (NumberFormatException e) {
					logger.error("Parsing reference id failed: " + referenceId);
				}
				if(referenceLong > -1){
					userMgr.invite(referenceLong, user.getId());
				}
				request.getSession().removeAttribute(HomePageServlet.REFERENCE_ID);
				
			} else {
				//existing user logging in again
				user = new User();
				user.setId(userId);
				user.setUserName(screenName);
				user.setLastIp(request.getRemoteHost());
				user.setOauthToken(oauthToken);
				user.setOauthTokenSecret(oauthTokenSecret);
				user.setAutoPlayer(false);
				userMgr.updateUser(user);
			}
			request.getSession().setAttribute(User.USER_ID, user.getId());
			Cookie cookies[] = createCookie(userId, oauthToken);
			writeCookies(response, cookies);
		} catch (TwitterException e) {
			throw new ServletException(e);
		}
		response.sendRedirect(request.getContextPath() + "/");
	}

	public Cookie[] createCookie(long userId, String oauthToken) {
		Cookie ck1 = new Cookie(COOKIE_ID, String.valueOf(userId));
		Cookie ck2 = new Cookie(COOKIE_OAUTHTOKEN, oauthToken);
		ck1.setMaxAge(COOKIE_EXPIRE);
		ck2.setMaxAge(COOKIE_EXPIRE);
		return new Cookie[] { ck1, ck2 };
	}

	public void writeCookies(HttpServletResponse response, Cookie[] cookies) {
		for (Cookie cookie : cookies) {
			response.addCookie(cookie);
		}

	}
}