src/com/android/browser/WebsiteSettingsActivity.java

/*
 * Copyright (C) 2009 The Android Open Source Project
 *
 * 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.
 */

package com.android.browser;

import android.app.AlertDialog;
import android.app.ListActivity;
import android.content.Context;
import android.content.DialogInterface;
import android.database.Cursor;
import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.net.Uri;
import android.os.Bundle;
import android.provider.Browser;
import android.util.Log;
import android.view.KeyEvent;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.webkit.WebIconDatabase;
import android.webkit.WebStorage;
import android.widget.ArrayAdapter;
import android.widget.AdapterView;
import android.widget.AdapterView.OnItemClickListener;
import android.widget.ImageView;
import android.widget.TextView;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
import java.util.Vector;

/**
 * Manage the settings for an origin.
 * We use it to keep track of the HTML5 settings, i.e. database (webstorage).
 */
public class WebsiteSettingsActivity extends ListActivity {

    private String LOGTAG = "WebsiteSettingsActivity";
    private static String sMBStored = null;
    private SiteAdapter mAdapter = null;

    class Site {
        private String mOrigin;
        private String mTitle;
        private Bitmap mIcon;

        public Site(String origin) {
            mOrigin = origin;
            mTitle = null;
            mIcon = null;
        }

        public String getOrigin() {
            return mOrigin;
        }

        public void setTitle(String title) {
            mTitle = title;
        }

        public void setIcon(Bitmap icon) {
            mIcon = icon;
        }

        public Bitmap getIcon() {
            return mIcon;
        }

        public String getPrettyOrigin() {
            return mTitle == null ? null : hideHttp(mOrigin);
        }

        public String getPrettyTitle() {
            return mTitle == null ? hideHttp(mOrigin) : mTitle;
        }

        private String hideHttp(String str) {
            Uri uri = Uri.parse(str);
            return "http".equals(uri.getScheme()) ?  str.substring(7) : str;
        }
    }

    class SiteAdapter extends ArrayAdapter<Site>
            implements AdapterView.OnItemClickListener {
        private int mResource;
        private LayoutInflater mInflater;
        private Bitmap mDefaultIcon;
        private Site mCurrentSite;
        private final static int STORED_DATA = 0;

        public SiteAdapter(Context context, int rsc) {
            super(context, rsc);
            mResource = rsc;
            mInflater = (LayoutInflater)context.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
            mDefaultIcon = BitmapFactory.decodeResource(getResources(),
                    R.drawable.ic_launcher_shortcut_browser_bookmark);
            populateOrigins();
        }

        public void populateOrigins() {
            clear();

            // Get the list of origins we want to display
            Set origins = WebStorage.getInstance().getOrigins();
            Set sites = new HashSet<Site>();
            if (origins != null) {
                Iterator<String> iter = origins.iterator();
                while (iter.hasNext()) {
                    String origin = iter.next();
                    Site site = new Site(origin);
                    sites.add(site);
                }
            }

            // Create a map from host to origin. This is used to add metadata
            // (title, icon) for this origin from the bookmarks DB.
            HashMap hosts = new HashMap<String, Set<Site> >();
            Iterator<Site> sitesIter = sites.iterator();
            while (sitesIter.hasNext()) {
                Site site = sitesIter.next();
                String host = Uri.parse(site.getOrigin()).getHost();
                Set hostSites = null;
                if (hosts.containsKey(host)) {
                    hostSites = (Set) hosts.get(host);
                } else {
                    hostSites = new HashSet<Site>();
                    hosts.put(host, hostSites);
                }
                hostSites.add(site);
            }

            // Check the bookmark DB. If we have data for a host used by any of
            // our origins, use it to set their title and favicon
            Cursor c = getContext().getContentResolver().query(Browser.BOOKMARKS_URI,
                    new String[] { Browser.BookmarkColumns.URL, Browser.BookmarkColumns.TITLE,
                    Browser.BookmarkColumns.FAVICON }, "bookmark = 1", null, null);

            if ((c != null) && c.moveToFirst()) {
                int urlIndex = c.getColumnIndex(Browser.BookmarkColumns.URL);
                int titleIndex = c.getColumnIndex(Browser.BookmarkColumns.TITLE);
                int faviconIndex = c.getColumnIndex(Browser.BookmarkColumns.FAVICON);
                do {
                    String url = c.getString(urlIndex);
                    String host = Uri.parse(url).getHost();
                    if (hosts.containsKey(host)) {
                        String title = c.getString(titleIndex);
                        Bitmap bmp = null;
                        byte[] data = c.getBlob(faviconIndex);
                        if (data != null) {
                            bmp = BitmapFactory.decodeByteArray(data, 0, data.length);
                        }
                        Set matchingSites = (Set) hosts.get(host);
                        sitesIter = matchingSites.iterator();
                        while (sitesIter.hasNext()) {
                            Site site = sitesIter.next();
                            site.setTitle(title);
                            if (bmp != null) {
                                site.setIcon(bmp);
                            }
                        }
                    }
                } while (c.moveToNext());
            }

            // We can now simply populate our array with Site instances
            sitesIter = sites.iterator();
            while (sitesIter.hasNext()) {
                Site site = sitesIter.next();
                add(site);
            }

            if (getCount() == 0) {
                finish(); // we close the screen
            }
        }

        public int getCount() {
            if (mCurrentSite == null) {
                return super.getCount();
            }
            return 1; // db view
        }

        public String sizeValueToString(long value) {
            float mb = (float) value / (1024.0F * 1024.0F);
            int val = (int) (mb * 10);
            float ret = (float) (val / 10.0F);
            if (ret <= 0) {
                return "0";
            }
            return String.valueOf(ret);
        }

        /*
         * If we receive the back event and are displaying
         * site's settings, we want to go back to the main
         * list view. If not, we just do nothing (see
         * dispatchKeyEvent() below).
         */
        public boolean backKeyPressed() {
            if (mCurrentSite != null) {
                mCurrentSite = null;
                populateOrigins();
                notifyDataSetChanged();
                return true;
            }
            return false;
        }

        public View getView(int position, View convertView, ViewGroup parent) {
            View view;
            TextView title;
            TextView subtitle;
            ImageView icon;

            if (convertView == null) {
                view = mInflater.inflate(mResource, parent, false);
            } else {
                view = convertView;
            }

            title = (TextView) view.findViewById(R.id.title);
            subtitle = (TextView) view.findViewById(R.id.subtitle);
            icon = (ImageView) view.findViewById(R.id.icon);

            if (mCurrentSite == null) {
                Site site = getItem(position);
                title.setText(site.getPrettyTitle());
                subtitle.setText(site.getPrettyOrigin());
                icon.setVisibility(View.VISIBLE);
                Bitmap bmp = site.getIcon();
                if (bmp == null) {
                    bmp = mDefaultIcon;
                }
                icon.setImageBitmap(bmp);
                // We set the site as the view's tag,
                // so that we can get it in onItemClick()
                view.setTag(site);
            } else {
                icon.setVisibility(View.GONE);
                if (position == STORED_DATA) {
                    String origin = mCurrentSite.getOrigin();
                    long usageValue = WebStorage.getInstance().getUsageForOrigin(origin);
                    String usage = sizeValueToString(usageValue) + " " + sMBStored;

                    title.setText(R.string.webstorage_clear_data_title);
                    subtitle.setText(usage);
                }
            }

            return view;
        }

        public void onItemClick(AdapterView<?> parent,
                                View view,
                                int position,
                                long id) {
            if (mCurrentSite != null) {
                if (position == STORED_DATA) {
                    new AlertDialog.Builder(getContext())
                        .setTitle(R.string.webstorage_clear_data_dialog_title)
                        .setMessage(R.string.webstorage_clear_data_dialog_message)
                        .setPositiveButton(R.string.webstorage_clear_data_dialog_ok_button,
                                           new AlertDialog.OnClickListener() {
                            public void onClick(DialogInterface dlg, int which) {
                                WebStorage.getInstance().deleteOrigin(mCurrentSite.getOrigin());
                                mCurrentSite = null;
                                populateOrigins();
                                notifyDataSetChanged();
                            }})
                        .setNegativeButton(R.string.webstorage_clear_data_dialog_cancel_button, null)
                        .setIcon(android.R.drawable.ic_dialog_alert)
                        .show();
                }
            } else {
                mCurrentSite = (Site) view.getTag();
                notifyDataSetChanged();
            }
        }
    }

    /**
     * Intercepts the back key to immediately notify
     * NativeDialog that we are done.
     */
    public boolean dispatchKeyEvent(KeyEvent event) {
        if ((event.getKeyCode() == KeyEvent.KEYCODE_BACK)
            && (event.getAction() == KeyEvent.ACTION_DOWN)) {
            if ((mAdapter != null) && (mAdapter.backKeyPressed())){
                return true; // event consumed
            }
        }
        return super.dispatchKeyEvent(event);
    }

    @Override
    protected void onCreate(Bundle icicle) {
        super.onCreate(icicle);
        if (sMBStored == null) {
            sMBStored = getString(R.string.webstorage_origin_summary_mb_stored);
        }
        mAdapter = new SiteAdapter(this, R.layout.application);
        setListAdapter(mAdapter);
        getListView().setOnItemClickListener(mAdapter);
    }
}