/android/import-contacts

To get this branch, use:
bzr branch http://bzr.ed.am/android/import-contacts

« back to all changes in this revision

Viewing changes to src/am/ed/importcontacts/ContactsCache.java

  • Committer: edam
  • Date: 2012-04-24 09:56:17 UTC
  • Revision ID: tim@ed.am-20120424095617-lsfkkpkrlqk9xthn
updated all URLs, email addresses and package names to ed.am

Show diffs side-by-side

added added

removed removed

src/am/ed/importcontacts/ContactsCache.java
1
1
/*
2
2
 * ContactsCache.java
3
3
 *
4
 
 * Copyright (C) 2011 to 2012 Tim Marston <tim@ed.am>
 
4
 * Copyright (C) 2011 Tim Marston <tim@ed.am>
5
5
 *
6
6
 * This file is part of the Import Contacts program (hereafter referred
7
 
 * to as "this program").  For more information, see
 
7
 * to as "this program"). For more information, see
8
8
 * http://ed.am/dev/android/import-contacts
9
9
 *
10
10
 * This program is free software: you can redistribute it and/or modify
25
25
 
26
26
import java.util.HashMap;
27
27
import java.util.HashSet;
28
 
import java.util.Locale;
 
28
 
 
29
import am.ed.importcontacts.Importer.AbortImportException;
 
30
 
 
31
import android.app.Activity;
 
32
import android.database.Cursor;
 
33
import android.provider.Contacts;
 
34
 
29
35
 
30
36
public class ContactsCache
31
37
{
32
38
        /**
33
 
         * A thing that can be used to identify (or lookup) a contact within the
34
 
         * contacts cache.  It is not a reference to a cache entry and may not
35
 
         * identify an existing contact in the cache.
 
39
         * Information that can be used to identify a contact within the cache
36
40
         */
37
 
        public static class CacheIdentifier
 
41
        static public class CacheIdentifier
38
42
        {
39
 
                public enum Type { NAME, ORGANISATION, PRIMARY_NUMBER, PRIMARY_EMAIL }
 
43
                public enum Type {
 
44
                        NONE, NAME, ORGANISATION, PRIMARY_NUMBER, PRIMARY_EMAIL }
40
45
 
41
46
                private Type _type;
42
47
                private String _detail;
43
48
 
44
 
                /**
45
 
                 * Obtain a cache identifier.  This routine is designed to be as robust
46
 
                 * as possible (in terms of bad or null detail values), and to return
47
 
                 * null when a cache identifier can not be created.
48
 
                 *
49
 
                 * @param type the detail type
50
 
                 * @param detail the detail
51
 
                 * @return the cache identifier, or null
52
 
                 */
53
 
                public static CacheIdentifier factory( Type type, String detail )
54
 
                {
55
 
                        switch( type )
56
 
                        {
57
 
                        case NAME: detail = normaliseName( detail ); break;
58
 
                        case ORGANISATION: detail = normaliseOrganisation( detail ); break;
59
 
                        case PRIMARY_NUMBER: detail = normalisePhoneNumber( detail ); break;
60
 
                        case PRIMARY_EMAIL: detail = normaliseEmailAddress( detail ); break;
61
 
                        default: return null;
62
 
                        }
63
 
                        if( detail == null ) return null;
64
 
                        return new CacheIdentifier( type, detail );
65
 
                }
66
 
 
67
 
                /**
68
 
                 * Obtain a cache identifier from contact data.  This routine is
69
 
                 * designed to be as robust as possible and may return null when a cache
70
 
                 * identifier can not be created.
71
 
                 *
72
 
                 * @param contact the contact data
73
 
                 * @return the cache identifier, or null
74
 
                 */
75
 
                public static CacheIdentifier factory( Importer.ContactData contact )
76
 
                {
77
 
                        CacheIdentifier identifier = null;
78
 
 
79
 
                        if( contact.hasName() )
80
 
                                identifier = factory( CacheIdentifier.Type.NAME,
81
 
                                        contact.getName() );
82
 
                        if( identifier != null ) return identifier;
83
 
 
84
 
                        if( contact.hasPrimaryOrganisation() )
85
 
                                identifier = factory( CacheIdentifier.Type.ORGANISATION,
86
 
                                        contact.getPrimaryOrganisation() );
87
 
                        if( identifier != null ) return identifier;
88
 
 
89
 
                        if( contact.hasPrimaryNumber() )
90
 
                                identifier = factory( CacheIdentifier.Type.PRIMARY_NUMBER,
91
 
                                        contact.getPrimaryNumber() );
92
 
                        if( identifier != null ) return identifier;
93
 
 
94
 
                        if( contact.hasPrimaryEmail() )
95
 
                                identifier = factory( CacheIdentifier.Type.PRIMARY_EMAIL,
96
 
                                        contact.getPrimaryEmail() );
97
 
                        if( identifier != null ) return identifier;
98
 
 
99
 
                        return null;
 
49
                protected CacheIdentifier()
 
50
                {
 
51
                        _type = Type.NONE;
100
52
                }
101
53
 
102
54
                protected CacheIdentifier( Type type, String detail )
117
69
        }
118
70
 
119
71
        // mappings of contact names, organisations and primary numbers to ids
120
 
        private HashMap< String, Long > _contactsByName
121
 
                = new HashMap< String, Long >();
122
 
        private HashMap< String, Long > _contactsByOrg
123
 
                = new HashMap< String, Long >();
124
 
        private HashMap< String, Long > _contactsByNumber
125
 
                = new HashMap< String, Long >();
126
 
        private HashMap< String, Long > _contactsByEmail
127
 
                = new HashMap< String, Long >();
 
72
        private HashMap< String, Long > _contactsByName;
 
73
        private HashMap< String, Long > _contactsByOrg;
 
74
        private HashMap< String, Long > _contactsByNumber;
 
75
        private HashMap< String, Long > _contactsByEmail;
128
76
 
129
77
        // mapping of contact ids to sets of associated data
130
 
        private HashMap< Long, HashSet< String > > _contactNumbers
131
 
                = new HashMap< Long, HashSet< String > >();
132
 
        private HashMap< Long, HashSet< String > > _contactEmails
133
 
                = new HashMap< Long, HashSet< String > >();
134
 
        private HashMap< Long, HashSet< String > > _contactAddresses
135
 
                = new HashMap< Long, HashSet< String > >();
136
 
        private HashMap< Long, HashSet< String > > _contactOrganisations
137
 
                = new HashMap< Long, HashSet< String > >();
138
 
        private HashMap< Long, HashSet< String > > _contactNotes
139
 
                = new HashMap< Long, HashSet< String > >();
 
78
        private HashMap< Long, HashSet< String > > _contactNumbers;
 
79
        private HashMap< Long, HashSet< String > > _contactEmails;
 
80
        private HashMap< Long, HashSet< String > > _contactAddresses;
 
81
        private HashMap< Long, HashSet< String > > _contactOrganisations;
 
82
 
 
83
        public static CacheIdentifier createIdentifier(
 
84
                Importer.ContactData contact )
 
85
        {
 
86
                if( contact.hasName() ) {
 
87
                        String name = normaliseName( contact.getName() );
 
88
                        if( name != null )
 
89
                                return new CacheIdentifier(
 
90
                                        CacheIdentifier.Type.NAME, name );
 
91
                }
 
92
 
 
93
                if( contact.hasPrimaryOrganisation() ) {
 
94
                        String organisation = normaliseOrganisation(
 
95
                                contact.getPrimaryOrganisation() );
 
96
                        if( organisation != null )
 
97
                                return new CacheIdentifier(
 
98
                                        CacheIdentifier.Type.ORGANISATION, organisation );
 
99
                }
 
100
 
 
101
                if( contact.hasPrimaryNumber() ) {
 
102
                        String number = normalisePhoneNumber( contact.getPrimaryNumber() );
 
103
                        if( number != null )
 
104
                        return new CacheIdentifier(
 
105
                                CacheIdentifier.Type.PRIMARY_NUMBER, number );
 
106
                }
 
107
 
 
108
                if( contact.hasPrimaryEmail() ) {
 
109
                        String email = normaliseEmailAddress( contact.getPrimaryEmail() );
 
110
                        if( email != null )
 
111
                        return new CacheIdentifier(
 
112
                                CacheIdentifier.Type.PRIMARY_EMAIL, email );
 
113
                }
 
114
 
 
115
                return null;
 
116
        }
140
117
 
141
118
        public boolean canLookup( CacheIdentifier identifier )
142
119
        {
143
120
                return lookup( identifier ) != null;
144
121
        }
145
122
 
146
 
        /**
147
 
         * Retrieve the contact id of a contact identified by the specified cache
148
 
         * identifier, if it exists.
149
 
         *
150
 
         * @param identifier the cache identifier
151
 
         * @return a contact id, or null
152
 
         */
153
123
        public Long lookup( CacheIdentifier identifier )
154
124
        {
155
125
                switch( identifier.getType() )
166
136
                return null;
167
137
        }
168
138
 
169
 
        /**
170
 
         * Remove any cache entry that is identified by the cache identifier.
171
 
         *
172
 
         * @param identifier the cache identifier
173
 
         * @return the contact id of the contact that was removed, or null
174
 
         */
175
139
        public Long removeLookup( CacheIdentifier identifier )
176
140
        {
177
141
                switch( identifier.getType() )
188
152
                return null;
189
153
        }
190
154
 
191
 
        /**
192
 
         * Add a lookup from a contact identifier to a contact id to the cache.
193
 
         *
194
 
         * @param identifier the cache identifier
195
 
         * @param id teh contact id
196
 
         */
197
155
        public void addLookup( CacheIdentifier identifier, Long id )
198
156
        {
199
157
                switch( identifier.getType() )
213
171
                }
214
172
        }
215
173
 
216
 
        /**
217
 
         * Remove any data that is associated with an contact id.
218
 
         *
219
 
         * @param id
220
 
         */
221
174
        public void removeAssociatedData( Long id )
222
175
        {
223
176
                _contactNumbers.remove( id );
224
177
                _contactEmails.remove( id );
225
178
                _contactAddresses.remove( id );
226
179
                _contactOrganisations.remove( id );
227
 
                _contactNotes.remove( id );
228
180
        }
229
181
 
230
182
        public boolean hasAssociatedNumber( Long id, String number )
246
198
                        set = new HashSet< String >();
247
199
                        _contactNumbers.put( id, set );
248
200
                }
249
 
                set.add( number );
 
201
                set.add( normalisePhoneNumber( number ) );
250
202
        }
251
203
 
252
204
        public boolean hasAssociatedEmail( Long id, String email )
255
207
                if( email == null ) return false;
256
208
 
257
209
                HashSet< String > set = _contactEmails.get( id );
258
 
                return set != null && set.contains( email );
 
210
                return set != null && set.contains( normaliseEmailAddress( email ) );
259
211
        }
260
212
 
261
213
        public void addAssociatedEmail( Long id, String email )
268
220
                        set = new HashSet< String >();
269
221
                        _contactEmails.put( id, set );
270
222
                }
271
 
                set.add( email );
 
223
                set.add( normaliseEmailAddress( email ) );
272
224
        }
273
225
 
274
226
        public boolean hasAssociatedAddress( Long id, String address )
277
229
                if( address == null ) return false;
278
230
 
279
231
                HashSet< String > set = _contactAddresses.get( id );
280
 
                return set != null && set.contains( address );
 
232
                return set != null && set.contains( normaliseAddress( address ) );
281
233
        }
282
234
 
283
235
        public void addAssociatedAddress( Long id, String address )
290
242
                        set = new HashSet< String >();
291
243
                        _contactAddresses.put( id, set );
292
244
                }
293
 
                set.add( address );
 
245
                set.add( normaliseAddress( address ) );
294
246
        }
295
247
 
296
248
        public boolean hasAssociatedOrganisation( Long id, String organisation )
299
251
                if( organisation == null ) return false;
300
252
 
301
253
                HashSet< String > set = _contactOrganisations.get( id );
302
 
                return set != null && set.contains( organisation );
 
254
                return set != null && set.contains(
 
255
                        normaliseOrganisation( organisation ) );
303
256
        }
304
257
 
305
258
        public void addAssociatedOrganisation( Long id, String organisation )
312
265
                        set = new HashSet< String >();
313
266
                        _contactOrganisations.put( id, set );
314
267
                }
315
 
                set.add( organisation );
316
 
        }
317
 
 
318
 
        public boolean hasAssociatedNote( Long id, String note )
319
 
        {
320
 
                note = normaliseNote( note );
321
 
                if( note == null ) return false;
322
 
 
323
 
                HashSet< String > set = _contactNotes.get( id );
324
 
                return set != null && set.contains( note );
325
 
        }
326
 
 
327
 
        public void addAssociatedNote( Long id, String note )
328
 
        {
329
 
                note = normaliseNote( note );
330
 
                if( note == null ) return;
331
 
 
332
 
                HashSet< String > set = _contactNotes.get( id );
333
 
                if( set == null ) {
334
 
                        set = new HashSet< String >();
335
 
                        _contactNotes.put( id, set );
336
 
                }
337
 
                set.add( note );
338
 
        }
339
 
 
340
 
        static public String normaliseName( String name )
 
268
                set.add( normaliseOrganisation( organisation ) );
 
269
        }
 
270
 
 
271
        public void buildCache( Activity activity )
 
272
                throws AbortImportException
 
273
        {
 
274
                Cursor cur;
 
275
 
 
276
                // init id lookups
 
277
                _contactsByName = new HashMap< String, Long >();
 
278
                _contactsByOrg = new HashMap< String, Long >();
 
279
                _contactsByNumber = new HashMap< String, Long >();
 
280
                _contactsByEmail = new HashMap< String, Long >();
 
281
 
 
282
                // init associated data cache
 
283
                _contactNumbers = new HashMap< Long, HashSet< String > >();
 
284
                _contactEmails = new HashMap< Long, HashSet< String > >();
 
285
                _contactAddresses = new HashMap< Long, HashSet< String > >();
 
286
                _contactOrganisations = new HashMap< Long, HashSet< String > >();
 
287
 
 
288
                // set of contact ids that we have not yet added
 
289
                HashSet< Long > unadded = new HashSet< Long >();
 
290
 
 
291
                // get all contacts
 
292
                cur = activity.managedQuery( Contacts.People.CONTENT_URI,
 
293
                        new String[] {
 
294
                                Contacts.People._ID,
 
295
                                Contacts.People.NAME,
 
296
                        }, null, null, null );
 
297
                while( cur.moveToNext() ) {
 
298
                        Long id = cur.getLong(
 
299
                                cur.getColumnIndex( Contacts.People._ID ) );
 
300
                        String name = normaliseName( cur.getString(
 
301
                                cur.getColumnIndex( Contacts.People.NAME ) ) );
 
302
                        if( name != null )
 
303
                        {
 
304
                                // if we can, add a lookup for the contact id by name
 
305
                                if( name.length() > 0 ) {
 
306
                                        addLookup( new CacheIdentifier(
 
307
                                                CacheIdentifier.Type.NAME, name ), id );
 
308
                                        continue;
 
309
                                }
 
310
                        }
 
311
 
 
312
                        // record that a lookup for this contact's id still needs to be
 
313
                        // added by some other means
 
314
                        unadded.add( id );
 
315
                }
 
316
 
 
317
                // get contact organisations, primary ones first
 
318
                cur = activity.managedQuery( Contacts.Organizations.CONTENT_URI,
 
319
                        new String[] {
 
320
                                Contacts.Phones.PERSON_ID,
 
321
                                Contacts.Organizations.COMPANY,
 
322
                        }, null, null, Contacts.Organizations.ISPRIMARY + " DESC" );
 
323
                while( cur.moveToNext() ) {
 
324
                        Long id = cur.getLong( cur.getColumnIndex(
 
325
                                Contacts.Organizations.PERSON_ID ) );
 
326
                        String organisation = normaliseOrganisation( cur.getString(
 
327
                                cur.getColumnIndex( Contacts.Organizations.COMPANY ) ) );
 
328
                        if( organisation != null )
 
329
                        {
 
330
                                // if this is an organisation name for a contact for whom we
 
331
                                // have not added a lookup, add a lookup for the contact id
 
332
                                // by organisation
 
333
                                if( unadded.contains( id ) ) {
 
334
                                        addLookup( new CacheIdentifier(
 
335
                                                CacheIdentifier.Type.ORGANISATION, organisation ), id );
 
336
                                        unadded.remove( id );
 
337
                                }
 
338
 
 
339
                                // add associated data
 
340
                                addAssociatedOrganisation( id, organisation );
 
341
                        }
 
342
                }
 
343
 
 
344
                // get all phone numbers, primary ones first
 
345
                cur = activity.managedQuery( Contacts.Phones.CONTENT_URI,
 
346
                        new String[] {
 
347
                                Contacts.Phones.PERSON_ID,
 
348
                                Contacts.Phones.NUMBER,
 
349
                        }, null, null, Contacts.Phones.ISPRIMARY + " DESC" );
 
350
                while( cur.moveToNext() ) {
 
351
                        Long id = cur.getLong(
 
352
                                cur.getColumnIndex( Contacts.Phones.PERSON_ID ) );
 
353
                        String number = normalisePhoneNumber( cur.getString(
 
354
                                cur.getColumnIndex( Contacts.Phones.NUMBER ) ) );
 
355
                        if( number != null )
 
356
                        {
 
357
                                // if this is a number for a contact for whom we have not
 
358
                                // added a lookup, add a lookup for the contact id by phone
 
359
                                // number
 
360
                                if( unadded.contains( id ) ) {
 
361
                                        addLookup( new CacheIdentifier(
 
362
                                                CacheIdentifier.Type.PRIMARY_NUMBER, number ), id );
 
363
                                        unadded.remove( id );
 
364
                                }
 
365
 
 
366
                                // add associated data
 
367
                                addAssociatedNumber( id, number );
 
368
                        }
 
369
                }
 
370
 
 
371
                // now get all email addresses, primary ones first, and postal addresses
 
372
                cur = activity.managedQuery( Contacts.ContactMethods.CONTENT_URI,
 
373
                        new String[] {
 
374
                                Contacts.ContactMethods.PERSON_ID,
 
375
                                Contacts.ContactMethods.DATA,
 
376
                                Contacts.ContactMethods.KIND,
 
377
                        }, Contacts.ContactMethods.KIND + " IN( ?, ? )", new String[] {
 
378
                                "" + Contacts.KIND_EMAIL,
 
379
                                "" + Contacts.KIND_POSTAL,
 
380
                        }, Contacts.ContactMethods.ISPRIMARY + " DESC" );
 
381
                while( cur.moveToNext() ) {
 
382
                        Long id = cur.getLong(
 
383
                                cur.getColumnIndex( Contacts.ContactMethods.PERSON_ID ) );
 
384
                        int kind = cur.getInt(
 
385
                                cur.getColumnIndex( Contacts.ContactMethods.KIND ) );
 
386
                        if( kind == Contacts.KIND_EMAIL )
 
387
                        {
 
388
                                String email = normaliseEmailAddress( cur.getString(
 
389
                                        cur.getColumnIndex( Contacts.ContactMethods.DATA ) ) );
 
390
                                if( email != null )
 
391
                                {
 
392
                                        // if this is an email address for a contact for whom we
 
393
                                        // have not added a lookup, add a lookup for the contact
 
394
                                        // id by email address
 
395
                                        if( unadded.contains( id ) ) {
 
396
                                                addLookup( new CacheIdentifier(
 
397
                                                        CacheIdentifier.Type.PRIMARY_EMAIL, email ), id );
 
398
                                                unadded.remove( id );
 
399
                                        }
 
400
 
 
401
                                        // add associated data
 
402
                                        addAssociatedEmail( id, email );
 
403
                                }
 
404
                        }
 
405
                        else if( kind == Contacts.KIND_POSTAL )
 
406
                        {
 
407
                                String address = normaliseAddress( cur.getString(
 
408
                                        cur.getColumnIndex( Contacts.ContactMethods.DATA ) ) );
 
409
                                if( address != null )
 
410
                                {
 
411
                                        // add associated data
 
412
                                        addAssociatedAddress( id, address );
 
413
                                }
 
414
                        }
 
415
                }
 
416
        }
 
417
 
 
418
        static private String normaliseName( String name )
341
419
        {
342
420
                if( name == null ) return null;
343
421
                name = name.trim();
344
422
                return name.length() > 0? name : null;
345
423
        }
346
424
 
347
 
        static public String normalisePhoneNumber( String number )
 
425
        static private String normalisePhoneNumber( String number )
348
426
        {
349
427
                if( number == null ) return null;
350
428
                number = number.trim().replaceAll( "[-\\(\\) ]", "" );
351
429
                return number.length() > 0? number : null;
352
430
        }
353
431
 
354
 
        static public String normaliseEmailAddress( String email )
 
432
        static private String normaliseEmailAddress( String email )
355
433
        {
356
434
                if( email == null ) return null;
357
 
                email = email.trim().toLowerCase( Locale.US );
 
435
                email = email.trim().toLowerCase();
358
436
                return email.length() > 0? email : null;
359
437
        }
360
438
 
361
 
        static public String normaliseOrganisation( String organisation )
 
439
        static private String normaliseOrganisation( String organisation )
362
440
        {
363
441
                if( organisation == null ) return null;
364
442
                organisation = organisation.trim();
365
443
                return organisation.length() > 0? organisation : null;
366
444
        }
367
445
 
368
 
        static public String normaliseAddress( String address )
 
446
        static private String normaliseAddress( String address )
369
447
        {
370
448
                if( address == null ) return null;
371
449
                address = address.trim();
372
450
                return address.length() > 0? address : null;
373
451
        }
374
 
 
375
 
        static public String normaliseNote( String note )
376
 
        {
377
 
                if( note == null ) return null;
378
 
                note = note.trim();
379
 
                return note.length() > 0? note : null;
380
 
        }
381
452
}