com.github.ajasmin.telususageandroidwidget.TelusReportFetcher.java Source code

Introduction

Here is the source code for com.github.ajasmin.telususageandroidwidget.TelusReportFetcher.java

Source

/*
 * Copyright (C) 2011 by Alexandre Jasmin <alexandre.jasmin@gmail.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */

package com.github.ajasmin.telususageandroidwidget;

import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
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.impl.client.DefaultHttpClient;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.params.CoreProtocolPNames;
import org.apache.http.util.EntityUtils;

import android.content.Context;
import android.util.Log;

import com.github.ajasmin.telususageandroidwidget.TelusWidgetPreferences.PreferencesData;

public class TelusReportFetcher {
    private static final long CACHE_LIFETIME = 1 /*hour*/ * 60 * 60 * 1000;

    @SuppressWarnings("serial")
    public static class InvalidCredentialsException extends Exception {
    }

    @SuppressWarnings("serial")
    public static class NetworkErrorException extends Exception {
        public NetworkErrorException(String string, Throwable e) {
            super(string, e);
        }
    }

    public static void retriveUsageSummaryData(int appWidgetId)
            throws InvalidCredentialsException, NetworkErrorException {
        Context context = MyApp.getContext();
        String fileName = "" + appWidgetId;

        TelusWidgetPreferences.PreferencesData prefs = TelusWidgetPreferences.getPreferences(appWidgetId);

        if (!context.getFileStreamPath(fileName).exists()
                || System.currentTimeMillis() - prefs.lastUpdateTime > CACHE_LIFETIME) {
            fetchFromTelusSite(prefs);
        }
    }

    private static void fetchFromTelusSite(final PreferencesData prefs)
            throws NetworkErrorException, InvalidCredentialsException {
        final DefaultHttpClient httpclient = new DefaultHttpClient();
        httpclient.getParams().setParameter(CoreProtocolPNames.USER_AGENT, "ajasmin-widget");
        logIn(httpclient, prefs);
        fetchUsageSummaryPage(httpclient, prefs);

        prefs.markAsUpdatedNow();

        // Log out to avoid session limit
        // on background thread to avoid extra delay
        new Thread(new Runnable() {
            @Override
            public void run() {
                try {
                    fetchLogOutPage(httpclient);
                    Log.i("TelusWebScraper", "Logged out " + prefs.email);
                } catch (IOException e) {
                    Log.e("TelusWebScraper", "Couldn't fetch logout page for " + prefs.email, e);
                }
            }
        }).run();
    }

    private static void logIn(DefaultHttpClient httpclient, PreferencesData prefs)
            throws NetworkErrorException, InvalidCredentialsException {
        try {
            final String url = "https://mobile.telus.com/login.htm";
            HttpPost httpPost = new HttpPost(url);

            List<NameValuePair> formparams = new ArrayList<NameValuePair>();
            formparams.add(new BasicNameValuePair("username", prefs.email));
            formparams.add(new BasicNameValuePair("password", prefs.password));
            formparams.add(new BasicNameValuePair("_rememberMe", "on"));
            formparams.add(new BasicNameValuePair("forwardAction", "/index.htm?lang=en"));

            UrlEncodedFormEntity entity = new UrlEncodedFormEntity(formparams, "UTF-8");
            httpPost.setEntity(entity);

            HttpResponse response = httpclient.execute(httpPost);
            HttpEntity responseEntity = response.getEntity();
            String str = EntityUtils.toString(responseEntity);
            if (str.contains("The email or password you entered is invalid.")) {
                throw new InvalidCredentialsException();
            }
        } catch (IOException e) {
            throw new NetworkErrorException("Error logging in", e);
        }
    }

    private static void fetchUsageSummaryPage(DefaultHttpClient httpclient, PreferencesData prefs)
            throws NetworkErrorException {
        try {
            final String url = "https://mobile.telus.com/index.htm";
            HttpPost httpPost = new HttpPost(url);

            List<NameValuePair> formparams = new ArrayList<NameValuePair>();
            if (prefs.subscriber != null) {
                formparams.add(new BasicNameValuePair("subscriber", prefs.subscriber));
            }

            UrlEncodedFormEntity entity = new UrlEncodedFormEntity(formparams, "UTF-8");
            httpPost.setEntity(entity);

            HttpResponse response = httpclient.execute(httpPost);
            HttpEntity responseEntity = response.getEntity();

            // Save the page to disk
            String fileName = Integer.toString(prefs.appWidgetId);
            FileOutputStream fileOutput = MyApp.getContext().openFileOutput(fileName, Context.MODE_PRIVATE);
            responseEntity.writeTo(fileOutput);
            fileOutput.close();
        } catch (IOException e) {
            throw new NetworkErrorException("Error fetching data from Telus website", e);
        }
    }

    private static void fetchLogOutPage(DefaultHttpClient httpclient) throws IOException {
        final String url = "https://mobile.telus.com/logout.htm";
        HttpGet httpGet = new HttpGet(url);

        HttpResponse response = httpclient.execute(httpGet);
        HttpEntity responseEntity = response.getEntity();
        // fetch the contents
        responseEntity.getContent().close();
    }
}