com.twitstreet.servlet.CallBackServlet.java Source code

Introduction

Here is the source code for com.twitstreet.servlet.CallBackServlet.java

Source

/**
   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);
        }

    }
}