com.sundbybergsit.servlets.OAuth2CallbackServlet.java Source code

Introduction

Here is the source code for com.sundbybergsit.servlets.OAuth2CallbackServlet.java

Source

package com.sundbybergsit.servlets;

/*
 * Copyright (c) 2012 Google Inc.
 *
 * 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.
 */

import com.google.api.client.auth.oauth2.AuthorizationCodeFlow;
import com.google.api.client.auth.oauth2.AuthorizationCodeResponseUrl;
import com.google.api.client.auth.oauth2.Credential;
import com.google.api.client.extensions.servlet.auth.oauth2.AbstractAuthorizationCodeCallbackServlet;
import com.sundbybergsit.fatman.jsf.FatmanLoginBean;
import com.sundbybergsit.fatman.ConfigurationLoader;
import com.sundbybergsit.google.Util;
import org.apache.commons.lang3.StringEscapeUtils;

import javax.inject.Inject;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.logging.Logger;

import static javax.servlet.http.HttpServletResponse.SC_INTERNAL_SERVER_ERROR;

/**
 * A servlet which handles the OAuth 2.0 authentication result and retrieves an access
 * token on successful authentication.
 *
 * @author Jennifer Murphy
 * @author Lee Denison
 */
@WebServlet(name = "oauth2callback", urlPatterns = "/oauth2callback")
public class OAuth2CallbackServlet extends AbstractAuthorizationCodeCallbackServlet {

    private static final long serialVersionUID = 1L;

    private static final Logger LOG = Logger.getLogger(OAuth2CallbackServlet.class.getName());

    @Override
    protected void onSuccess(HttpServletRequest request, HttpServletResponse response, Credential credential)
            throws ServletException, IOException {
        response.sendRedirect("/");
        FatmanLoginBean loginBean = (FatmanLoginBean) request.getSession().getAttribute("loginBean");
        if (loginBean == null) {
            loginBean = new FatmanLoginBean();
            request.getSession().setAttribute("loginBean", loginBean);
        }
        loginBean.setAccessToken(credential.getAccessToken());
        loginBean.setSessionId(request.getSession(true).getId());
        loginBean.setLoggedIn(true);
    }

    @Override
    protected void onError(HttpServletRequest req, HttpServletResponse resp,
            AuthorizationCodeResponseUrl errorResponse) throws ServletException, IOException {

        StringBuilder err = new StringBuilder("There was a problem during authentication: ");

        if (errorResponse != null) {
            err.append(errorResponse.getError());

            if (errorResponse.getErrorUri() != null) {
                err.append(" [").append(errorResponse.getErrorUri()).append("]");
            }

            if (errorResponse.getErrorDescription() != null) {
                err.append(": ").append(errorResponse.getErrorDescription());
            }
        }

        LOG.warning(err.toString());
        resp.sendError(SC_INTERNAL_SERVER_ERROR, StringEscapeUtils.escapeHtml4(Util.stripTags(err.toString())));
    }

    @Override
    protected String getRedirectUri(HttpServletRequest request) throws ServletException, IOException {
        return ConfigurationLoader.REDIRECT_URI;
    }

    @Override
    protected AuthorizationCodeFlow initializeFlow() throws IOException {
        return Util.getFlow();
    }

    @Override
    protected String getUserId(HttpServletRequest request) throws ServletException, IOException {
        return request.getSession(true).getId();
    }

}