gov.nasa.arc.geocam.memo.service.SiteAuthCookieImplementation.java Source code

Introduction

Here is the source code for gov.nasa.arc.geocam.memo.service.SiteAuthCookieImplementation.java

Source

// __BEGIN_LICENSE__
// Copyright (C) 2008-2010 United States Government as represented by
// the Administrator of the National Aeronautics and Space Administration.
// All Rights Reserved.
// __END_LICENSE__

package gov.nasa.arc.geocam.memo.service;

import gov.nasa.arc.geocam.memo.R;
import gov.nasa.arc.geocam.memo.exception.AuthenticationFailedException;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;

import org.apache.http.HttpResponse;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.params.HttpClientParams;
import org.apache.http.cookie.Cookie;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.params.HttpParams;
import org.apache.http.protocol.HTTP;

import roboguice.inject.InjectResource;
import android.util.Log;

/**
 * The Class SiteAuthCookieImplementation.
 */
public class SiteAuthCookieImplementation implements SiteAuthInterface {

    /** The server root url. */
    @InjectResource(R.string.url_server_root)
    String serverRootUrl;

    /** The app path. */
    @InjectResource(R.string.url_relative_app)
    String appPath;

    /** The http client. */
    private DefaultHttpClient httpClient;

    /** The session id cookie. */
    private Cookie sessionIdCookie;

    /** The username. */
    private String username;

    /** The password. */
    private String password;

    /* (non-Javadoc)
     * @see gov.nasa.arc.geocam.memo.service.SiteAuthInterface#setRoot(java.lang.String)
     */
    @Override
    public void setRoot(String siteRoot) {
        serverRootUrl = siteRoot;
    }

    /* (non-Javadoc)
     * @see gov.nasa.arc.geocam.memo.service.SiteAuthInterface#setAuth(java.lang.String, java.lang.String)
     */
    @Override
    public void setAuth(String username, String password) {
        this.username = username;
        this.password = password;
    }

    /* (non-Javadoc)
     * @see gov.nasa.arc.geocam.memo.service.SiteAuthInterface#post(java.lang.String, java.util.Map)
     */
    @Override
    public int post(String relativePath, Map<String, String> params)
            throws AuthenticationFailedException, IOException, ClientProtocolException {
        ensureAuthenticated();

        httpClient = new DefaultHttpClient();
        HttpParams httpParams = httpClient.getParams();
        HttpClientParams.setRedirecting(httpParams, false);
        httpParams.setParameter("http.protocol.handle-redirects", false);

        HttpPost post = new HttpPost(this.serverRootUrl + "/" + appPath + "/" + relativePath);
        post.setParams(httpParams);

        if (params != null) {
            List<BasicNameValuePair> nameValuePairs = new ArrayList<BasicNameValuePair>();
            for (String key : params.keySet()) {
                nameValuePairs.add(new BasicNameValuePair(key, params.get(key)));
            }

            post.setEntity(new UrlEncodedFormEntity(nameValuePairs, HTTP.ASCII));
        }

        httpClient.getCookieStore().addCookie(sessionIdCookie);
        //post.setHeader("Cookie", sessionIdCookie.toString());

        HttpResponse r = httpClient.execute(post);
        // TODO: check for redirect to login and call login if is the case

        return r.getStatusLine().getStatusCode();
    }

    /* (non-Javadoc)
     * @see gov.nasa.arc.geocam.memo.service.SiteAuthInterface#get(java.lang.String, java.util.Map)
     */
    @Override
    public String get(String relativePath, Map<String, String> params)
            throws AuthenticationFailedException, IOException, ClientProtocolException {
        ensureAuthenticated();
        httpClient = new DefaultHttpClient();

        HttpGet get = new HttpGet(this.serverRootUrl + "/" + appPath + "/" + relativePath);

        // TODO: add param parsing and query string construction as necessary

        httpClient.getCookieStore().addCookie(sessionIdCookie);
        //get.setHeader("Cookie", sessionIdCookie.toString());

        HttpResponse r = httpClient.execute(get);
        // TODO: check for redirect to login and call login if is the case

        InputStream content = r.getEntity().getContent();

        BufferedReader br = new BufferedReader(new InputStreamReader(content));
        StringBuilder sb = new StringBuilder();

        int c = 0;
        while ((c = br.read()) != -1) {
            sb.append((char) c);
        }

        br.close();
        return sb.toString();
    }

    /**
     * Ensure authenticated.
     *
     * @throws AuthenticationFailedException the authentication failed exception
     * @throws ClientProtocolException the client protocol exception
     * @throws IOException Signals that an I/O exception has occurred.
     */
    private void ensureAuthenticated() throws AuthenticationFailedException, ClientProtocolException, IOException {
        if (username == null || password == null) {
            throw new AuthenticationFailedException("Username and/or password not set.");
        } else {
            Date now = new Date();
            if (sessionIdCookie == null || sessionIdCookie.isExpired(now)) {
                // we're not logged in (at least we think. Let's log in)
                login();
            }
        }
    }

    /**
     * Login.
     *
     * @throws ClientProtocolException the client protocol exception
     * @throws IOException Signals that an I/O exception has occurred.
     * @throws AuthenticationFailedException the authentication failed exception
     */
    private void login() throws ClientProtocolException, IOException, AuthenticationFailedException {
        httpClient = new DefaultHttpClient();
        HttpParams params = httpClient.getParams();
        HttpClientParams.setRedirecting(params, false);

        Log.i("Talk", "Username:" + username);

        HttpPost p = new HttpPost(serverRootUrl + "/accounts/login/");
        p.setParams(params);

        List<BasicNameValuePair> nameValuePairs = new ArrayList<BasicNameValuePair>();
        nameValuePairs.add(new BasicNameValuePair("username", username));
        nameValuePairs.add(new BasicNameValuePair("password", password));

        p.setEntity(new UrlEncodedFormEntity(nameValuePairs, HTTP.ASCII));

        HttpResponse r = httpClient.execute(p);
        if (302 == r.getStatusLine().getStatusCode()) {
            for (Cookie c : httpClient.getCookieStore().getCookies()) {
                if (c.getName().contains("sessionid")) {
                    sessionIdCookie = c;
                    return;
                }
            }
            throw new AuthenticationFailedException("Session cookie was missing from server login response.");
        } else {
            throw new AuthenticationFailedException(
                    "Got unexpected response code from server: " + r.getStatusLine().getStatusCode());
        }
    }
}