com.wegas.core.security.facebook.FacebookRealm.java Source code

Introduction

Here is the source code for com.wegas.core.security.facebook.FacebookRealm.java

Source

/*
 * Wegas
 * http://wegas.albasim.ch
 *
 * Copyright (c) 2013, 2014, 2015 School of Business and Engineering Vaud, Comem
 * Licensed under the MIT License
 */
package com.wegas.core.security.facebook;

import com.wegas.core.Helper;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.AuthenticationInfo;
import org.apache.shiro.authc.AuthenticationToken;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.subject.PrincipalCollection;

/**
 *
 * @author Francois-Xavier Aeberhard <fx@red-agent.com>
 */
public class FacebookRealm extends AuthorizingRealm {

    private static final String APP_SECRET = Helper.getWegasProperty("facebook.appSecret");
    private static final String APP_ID = Helper.getWegasProperty("facebook.appId");
    private static final String REDIRECT_URL = Helper.getWegasProperty("facebook.redirectUrl");

    @Override
    public boolean supports(AuthenticationToken token) {
        if (token instanceof FacebookToken) {
            return true;
        }
        return false;
    }

    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {
        //return new FacebookAuthorizationInfo();
        return new SimpleAuthorizationInfo();
    }

    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
        FacebookToken facebookToken = (FacebookToken) token;

        // do all the facebook gubbins
        if (facebookToken.getCode() != null && facebookToken.getCode().trim().length() > 0) {
            URL authUrl;
            try {
                authUrl = new URL("https://graph.facebook.com/oauth/access_token?" + "client_id=" + APP_ID
                        + "&redirect_uri=" + REDIRECT_URL + "&client_secret=" + APP_SECRET + "&code="
                        + facebookToken.getCode());

                String authResponse = readURL(authUrl);
                System.out.println(authResponse);
                String accessToken = getPropsMap(authResponse).get("access_token");
                URL url = new URL("https://graph.facebook.com/me?access_token=" + accessToken);
                String fbResponse = readURL(url);
                FacebookUserDetails fud = new FacebookUserDetails(fbResponse);
                return new FacebookAuthenticationInfo(fud, this.getName());
            } catch (MalformedURLException e1) {
                e1.printStackTrace();
                throw new AuthenticationException(e1);
            } catch (IOException ioe) {
                ioe.printStackTrace();
                throw new AuthenticationException(ioe);
            } catch (Throwable e) {
                e.printStackTrace();
            }
        }
        return null;
    }

    // ------------------------------------------------------------
    // STUFF here should be in a more generic place TODO
    // ------------------------------------------------------------
    private String readURL(URL url) throws IOException {
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        InputStream is = url.openStream();
        int r;
        while ((r = is.read()) != -1) {
            baos.write(r);
        }
        return new String(baos.toByteArray(), StandardCharsets.UTF_8);
    }

    private Map<String, String> getPropsMap(String someString) {
        String[] pairs = someString.split("&");
        Map<String, String> props = new HashMap<>();
        for (String propPair : pairs) {
            String[] pair = propPair.split("=");
            props.put(pair[0], pair[1]);
        }
        return props;
    }
}