/android/import-contacts

/*
 * ContactsCache.java
 *
 * Copyright (C) 2009 Tim Marston <edam@waxworlds.org>
 *
 * This file is part of the Import Contacts program (hereafter referred
 * to as "this program"). For more information, see
 * http://www.waxworlds.org/edam/software/android/import-contacts
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.waxworlds.edam.importcontacts;

import java.util.HashMap;
import java.util.HashSet;

import org.waxworlds.edam.importcontacts.Importer.AbortImportException;

import android.app.Activity;
import android.database.Cursor;
import android.provider.Contacts;


public class ContactsCache
{
	private HashMap< String, Long > _contacts;

	private HashMap< Long, HashSet< String > > _contactNumbers;
	private HashMap< Long, HashSet< String > > _contactEmails;
	private HashMap< Long, HashSet< String > > _contactAddresses;

	public boolean exists( String name )
	{
		return _contacts.containsKey( name );
	}

	public Long getId( String name )
	{
		return _contacts.get( name );
	}

	public void remove( String name )
	{
		_contacts.remove( name );
	}

	public void put( Long id, String name )
	{
		_contacts.put( name, id );
	}

	public boolean hasNumber( Long id, String number )
	{
		HashSet< String > cache = _contactNumbers.get( id );
		return cache != null && cache.contains( number );
	}

	public void addNumber( Long id, String number )
	{
		HashSet< String > cache = _contactNumbers.get( id );
		if( cache == null ) {
			cache = new HashSet< String >();
			_contactNumbers.put( id, cache );
		}
		cache.add( number );
	}

	public boolean hasEmail( Long id, String email )
	{
		HashSet< String > cache = _contactEmails.get( id );
		return cache != null && cache.contains( email );
	}

	public void addEmail( Long id, String email )
	{
		HashSet< String > cache = _contactEmails.get( id );
		if( cache == null ) {
			cache = new HashSet< String >();
			_contactEmails.put( id, cache );
		}
		cache.add( email );
	}

	public boolean hasAddress( Long id, String address )
	{
		HashSet< String > cache = _contactAddresses.get( id );
		return cache != null && cache.contains( address );
	}

	public void addAddress( Long id, String address )
	{
		HashSet< String > cache = _contactAddresses.get( id );
		if( cache == null ) {
			cache = new HashSet< String >();
			_contactAddresses.put( id, cache );
		}
		cache.add( address );
	}

	public void buildCache( Activity activity )
		throws AbortImportException
	{
		String[] cols;
		Cursor cur;

		// init contacts caches
		_contacts = new HashMap< String, Long >();
		_contactNumbers = new HashMap< Long, HashSet< String > >();
		_contactEmails = new HashMap< Long, HashSet< String > >();
		_contactAddresses = new HashMap< Long, HashSet< String > >();

		// query and store map of contact names to ids
		cols = new String[] { Contacts.People._ID, Contacts.People.NAME };
		cur = activity.managedQuery( Contacts.People.CONTENT_URI,
			cols, null, null, null);
		if( cur.moveToFirst() ) {
			int idCol = cur.getColumnIndex( Contacts.People._ID );
			int nameCol = cur.getColumnIndex( Contacts.People.NAME );
			do {
				_contacts.put( cur.getString( nameCol ), cur.getLong( idCol ) );
			} while( cur.moveToNext() );
		}

		// query and store map of contact ids to sets of phone numbers
		cols = new String[] { Contacts.Phones.PERSON_ID,
				Contacts.Phones.NUMBER };
		cur = activity.managedQuery( Contacts.Phones.CONTENT_URI,
			cols, null, null, null);
		if( cur.moveToFirst() ) {
			int personIdCol = cur.getColumnIndex( Contacts.Phones.PERSON_ID );
			int numberCol = cur.getColumnIndex( Contacts.Phones.NUMBER );
			do {
				Long id = cur.getLong( personIdCol );
				String number = Importer.sanitisePhoneNumber(
						cur.getString( numberCol ) );
				if( number != null ) {
					HashSet< String > numbers = _contactNumbers.get( id );
					if( numbers == null ) {
						numbers = new HashSet< String >();
						_contactNumbers.put( id, numbers );
					}
					numbers.add( number );
				}
			} while( cur.moveToNext() );
		}

		// query and store map of contact ids to sets of email addresses
		cols = new String[] { Contacts.ContactMethods.PERSON_ID,
				Contacts.ContactMethods.DATA };
		cur = activity.managedQuery( Contacts.ContactMethods.CONTENT_URI,
				cols, Contacts.ContactMethods.KIND + " = ?",
				new String[] { "" + Contacts.KIND_EMAIL }, null );
		if( cur.moveToFirst() ) {
			int personIdCol = cur.getColumnIndex(
				Contacts.ContactMethods.PERSON_ID );
			int addressCol = cur.getColumnIndex(
				Contacts.ContactMethods.DATA );
			do {
				Long id = cur.getLong( personIdCol );
				String address = Importer.sanitiseEmailAddress(
					cur.getString( addressCol ) );
				if( address != null ) {
					HashSet< String > addresses = _contactEmails.get( id );
					if( addresses == null ) {
						addresses = new HashSet< String >();
						_contactEmails.put( id, addresses );
					}
					addresses.add( address );
				}
			} while( cur.moveToNext() );
		}

		// query and store map of contact ids to sets of postal addresses
		cols = new String[] { Contacts.ContactMethods.PERSON_ID,
			Contacts.ContactMethods.DATA };
		cur = activity.managedQuery( Contacts.ContactMethods.CONTENT_URI,
			cols, Contacts.ContactMethods.KIND + " = ?",
			new String[] { "" + Contacts.KIND_POSTAL }, null );
		if( cur.moveToFirst() ) {
			int personIdCol = cur.getColumnIndex(
				Contacts.ContactMethods.PERSON_ID );
			int addressCol = cur.getColumnIndex(
				Contacts.ContactMethods.DATA );
			do {
				Long id = cur.getLong( personIdCol );
				String address = cur.getString( addressCol );
				if( address != null ) {
					HashSet< String > addresses = _contactAddresses.get( id );
					if( addresses == null ) {
						addresses = new HashSet< String >();
						_contactAddresses.put( id, addresses );
					}
					addresses.add( address );
				}
			} while( cur.moveToNext() );
		}
	}
}