/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/org/waxworlds/edam/importcontacts/VcardImporter.java

  • Committer: edam
  • Date: 2011-05-30 19:20:17 UTC
  • Revision ID: edam@waxworlds.org-20110530192017-5c09k4kgpov02gja
- added checks for Doit.this == null when handling dialog buttons (I managed to abort an import as a duplicate contacts dialog was shown, but can't reproduce it now)
- added line no.s to vcard parsing errors
- update progress bar after a contact is imported, not before
- fixed bug introduced in last commit where a contacts were imported after finaliseVcard()ing failed
- don't show unknown encoding errors for vcard fields that we don't care about (which ignores base64 encoded photos, for example)

Show diffs side-by-side

added added

removed removed

src/org/waxworlds/edam/importcontacts/VcardImporter.java
1
1
/*
2
2
 * VCFImporter.java
3
3
 *
4
 
 * Copyright (C) 2009 to 2011 Tim Marston <tim@ed.am>
 
4
 * Copyright (C) 2009 to 2011 Tim Marston <edam@waxworlds.org>
5
5
 *
6
6
 * This file is part of the Import Contacts program (hereafter referred
7
7
 * to as "this program"). For more information, see
8
 
 * http://ed.am/dev/android/import-contacts
 
8
 * http://www.waxworlds.org/edam/software/android/import-contacts
9
9
 *
10
10
 * This program is free software: you can redistribute it and/or modify
11
11
 * it under the terms of the GNU General Public License as published by
21
21
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
22
22
 */
23
23
 
24
 
package am.ed.importcontacts;
 
24
package org.waxworlds.edam.importcontacts;
25
25
 
26
26
import java.io.BufferedReader;
27
27
import java.io.File;
38
38
import java.util.HashSet;
39
39
import java.util.Iterator;
40
40
import java.util.List;
41
 
import java.util.Locale;
42
41
import java.util.NoSuchElementException;
43
42
import java.util.Set;
44
43
import java.util.Vector;
45
44
import java.util.regex.Matcher;
46
45
import java.util.regex.Pattern;
47
46
 
48
 
import android.annotation.SuppressLint;
49
47
import android.content.SharedPreferences;
 
48
import android.provider.Contacts;
 
49
import android.provider.Contacts.PhonesColumns;
50
50
 
51
51
public class VcardImporter extends Importer
52
52
{
58
58
                super( doit );
59
59
        }
60
60
 
61
 
        @SuppressLint( "SdCardPath" )
62
61
        @Override
63
62
        protected void onImport() throws AbortImportException
64
63
        {
83
82
                                // get files
84
83
                                class VCardFilter implements FilenameFilter {
85
84
                                        public boolean accept( File dir, String name ) {
86
 
                                                return name.toLowerCase( Locale.US ).endsWith( ".vcf" );
 
85
                                                return name.toLowerCase().endsWith( ".vcf" );
87
86
                                        }
88
87
                                }
89
88
                                files = file.listFiles( new VCardFilter() );
363
362
        private class Vcard extends ContactData
364
363
        {
365
364
                private final static int NAMELEVEL_NONE = 0;
366
 
                private final static int NAMELEVEL_N = 1;
367
 
                private final static int NAMELEVEL_FN = 2;
 
365
                private final static int NAMELEVEL_FN = 1;
 
366
                private final static int NAMELEVEL_N = 2;
368
367
 
369
368
                private final static int MULTILINE_NONE = 0;
370
369
                private final static int MULTILINE_ENCODED = 1; // v2.1 quoted-printable
569
568
 
570
569
                                // determine whether we care about this entry
571
570
                                final HashSet< String > interesting_fields =
572
 
                                        new HashSet< String >( Arrays.asList( new String[] { "N",
573
 
                                                "FN", "ORG", "TITLE", "TEL", "EMAIL", "ADR", "LABEL" }
 
571
                                        new HashSet< String >( Arrays.asList( new String[]
 
572
                                                { "N", "FN", "ORG", "TITLE", "TEL", "EMAIL", "ADR" }
574
573
                                ) );
575
574
                                boolean is_interesting_field =
576
575
                                        interesting_fields.contains( name_param_parts[ 0 ] );
577
576
 
578
577
                                // parse encoding parameter
579
578
                                String encoding = checkParam( name_param_parts, "ENCODING" );
580
 
                                if( encoding != null )
581
 
                                        encoding = encoding.toUpperCase( Locale.US );
 
579
                                if( encoding != null ) encoding = encoding.toUpperCase();
582
580
                                if( is_interesting_field && encoding != null &&
583
581
                                        !encoding.equals( "8BIT" ) &&
584
582
                                        !encoding.equals( "QUOTED-PRINTABLE" ) )
589
587
 
590
588
                                // parse charset parameter
591
589
                                String charset = checkParam( name_param_parts, "CHARSET" );
592
 
                                if( charset != null )
593
 
                                        charset = charset.toUpperCase( Locale.US );
 
590
                                if( charset != null ) charset = charset.toUpperCase();
594
591
                                if( charset != null &&
595
592
                                        !charset.equals( "US-ASCII" ) &&
596
593
                                        !charset.equals( "ASCII" ) &&
612
609
                                                _parser_multiline_state = MULTILINE_ENCODED;
613
610
                                }
614
611
 
615
 
                                // convert 8-bit US-ASCII charset to UTF-8 (where no charset is
616
 
                                // specified for a v2.1 vcard entry, we assume it's US-ASCII)
617
 
                                if( ( charset == null && _version.equals( "2.1" ) ) ||
618
 
                                        ( charset != null && (
619
 
                                                charset.equals( "ASCII" ) ||
620
 
                                                charset.equals( "US-ASCII" ) ) ) )
621
 
                                {
 
612
                                // convert 8-bit ASCII charset to US-ASCII
 
613
                                if( charset == null || charset.equals( "ASCII" ) ) {
622
614
                                        value = transcodeAsciiToUtf8( value );
 
615
                                        charset = "UTF-8";
623
616
                                }
624
617
 
625
 
                                // process charset (value is now in UTF-8)
 
618
                                // process charset
626
619
                                String string_value;
627
620
                                try {
628
621
                                        string_value = new String( value.array(), value.position(),
629
 
                                                value.limit() - value.position(), "UTF-8" );
 
622
                                                value.limit() - value.position(), charset );
630
623
                                } catch( UnsupportedEncodingException e ) {
631
624
                                        throw new ParseException( R.string.error_vcf_charset );
632
625
                                }
679
672
                                        parseEMAIL( name_param_parts, complete_value );
680
673
                                else if( name_param_parts[ 0 ].equals( "ADR" ) )
681
674
                                        parseADR( name_param_parts, complete_value );
682
 
                                else if( name_param_parts[ 0 ].equals( "LABEL" ) )
683
 
                                        parseLABEL( name_param_parts, complete_value );
684
675
                        }
685
676
                }
686
677
 
699
690
                        return ( count & 1 ) == 1;
700
691
                }
701
692
 
702
 
                private String[] splitValueByCharacter( String value, char character )
 
693
                private String[] splitValueBySemicolon( String value )
703
694
                {
704
 
                        // split string in to parts by specified character
 
695
                        // split string in to parts by semicolon
705
696
                        ArrayList< String > parts = new ArrayList< String >(
706
 
                                Arrays.asList( value.split( "" + character ) ) );
 
697
                                Arrays.asList( value.split(  ";" ) ) );
707
698
 
708
699
                        // go through parts
709
700
                        for( int a = 0; a < parts.size(); a++ )
717
708
                                if( a < parts.size() - 1 &&
718
709
                                        doesStringEndInAnEscapeChar( str ) )
719
710
                                {
720
 
                                        // append the escaped character, join the next part to this
721
 
                                        // part and remove the next part
 
711
                                        // join the next part to this part and remove the next part
722
712
                                        parts.set( a, str.substring( 0, str.length() - 1 ) +
723
 
                                                character + parts.get( a + 1 ) );
 
713
                                                ';' + parts.get( a + 1 ) );
724
714
                                        parts.remove( a + 1 );
725
715
 
726
716
                                        // re-visit this part
737
727
                        return parts.toArray( ret );
738
728
                }
739
729
 
740
 
                private String unescapeValue( String value )
741
 
                {
742
 
                        StringBuilder ret = new StringBuilder( value.length() );
743
 
                        boolean in_escape = false;
744
 
                        for( int a = 0; a < value.length(); a++ )
745
 
                        {
746
 
                                int c = value.codePointAt( a );
747
 
 
748
 
                                // process a normal character
749
 
                                if( !in_escape ) {
750
 
                                        if( c == '\\' )
751
 
                                                in_escape = true;
752
 
                                        else
753
 
                                                ret.append( Character.toChars( c ) );
754
 
                                        continue;
755
 
                                }
756
 
 
757
 
                                // process an escape sequence
758
 
                                in_escape = false;
759
 
                                switch( c )
760
 
                                {
761
 
                                case 'N':
762
 
                                case 'n':
763
 
                                        // add newline
764
 
                                        ret.append( '\n' );
765
 
                                        break;
766
 
                                case '\\':
767
 
                                case ',':
768
 
                                case ';':
769
 
                                        // add escaped character
770
 
                                        ret.append( Character.toChars( c ) );
771
 
                                        break;
772
 
                                default:
773
 
                                        // unknown escape sequence, so add it unescaped
774
 
                                        ret.append( "\\" );
775
 
                                        ret.append( Character.toChars( c ) );
776
 
                                        break;
777
 
                                }
778
 
                        }
779
 
 
780
 
                        return ret.toString();
781
 
                }
782
 
 
783
730
                private void parseN( String[] params, String value )
784
731
                {
785
732
                        // already got a better name?
786
733
                        if( _name_level >= NAMELEVEL_N ) return;
787
734
 
788
735
                        // get name parts
789
 
                        String[] name_parts = splitValueByCharacter( value, ';' );
 
736
                        String[] name_parts = splitValueBySemicolon( value );
790
737
 
791
738
                        // build name
792
739
                        value = "";
793
 
                        final int[] part_order = { 3, 1, 2, 0, 4 };
794
 
                        for( int a = 0; a < part_order.length; a++ )
795
 
                                if( name_parts.length > part_order[ a ] &&
796
 
                                        name_parts[ part_order[ a ] ].length() > 0 )
797
 
                                {
798
 
                                        // split this part in to it's comma-separated bits
799
 
                                        String[] name_part_parts = splitValueByCharacter(
800
 
                                                name_parts[ part_order[ a ] ], ',' );
801
 
                                        for( int b = 0; b < name_part_parts.length; b++ )
802
 
                                                if( name_part_parts[ b ].length() > 0 )
803
 
                                                {
804
 
                                                        if( value.length() == 0 ) value += " ";
805
 
                                                        value += name_part_parts[ b ];
806
 
                                                }
807
 
                                }
 
740
                        if( name_parts.length > 1 && name_parts[ 1 ].length() > 0 )
 
741
                                value += name_parts[ 1 ];
 
742
                        if( name_parts.length > 0 && name_parts[ 0 ].length() > 0 )
 
743
                                value += ( value.length() == 0? "" : " " ) + name_parts[ 0 ];
808
744
 
809
745
                        // set name
810
 
                        setName( unescapeValue( value ) );
 
746
                        setName( value );
811
747
                        _name_level = NAMELEVEL_N;
812
748
                }
813
749
 
817
753
                        if( _name_level >= NAMELEVEL_FN ) return;
818
754
 
819
755
                        // set name
820
 
                        setName( unescapeValue( value ) );
 
756
                        setName( value );
821
757
                        _name_level = NAMELEVEL_FN;
822
758
                }
823
759
 
824
760
                private void parseORG( String[] params, String value )
825
761
                {
826
762
                        // get org parts
827
 
                        String[] org_parts = splitValueByCharacter( value, ';' );
 
763
                        String[] org_parts = splitValueBySemicolon( value );
828
764
                        if( org_parts == null || org_parts.length < 1 ) return;
829
765
 
830
766
                        // build organisation name
832
768
                                String.valueOf( org_parts[ 0 ] ) );
833
769
                        for( int a = 1; a < org_parts.length; a++ )
834
770
                                builder.append( ", " ).append( org_parts[ a ] );
835
 
                        String organisation = unescapeValue( builder.toString() );
 
771
                        String organisation = builder.toString();
836
772
 
837
773
                        // set organisation name (using a title we've previously found)
838
774
                        addOrganisation( organisation, _cached_title, true );
849
785
 
850
786
                private void parseTITLE( String[] params, String value )
851
787
                {
852
 
                        value = unescapeValue( value );
853
 
 
854
788
                        // if we previously had an organisation, look it up and append this
855
789
                        // title to it
856
790
                        if( _cached_organisation != null && hasOrganisations() ) {
882
816
                        int type;
883
817
                        if( types.contains( "FAX" ) )
884
818
                                if( types.contains( "HOME" ) )
885
 
                                        type = TYPE_FAX_HOME;
 
819
                                        type = PhonesColumns.TYPE_FAX_HOME;
886
820
                                else
887
 
                                        type = TYPE_FAX_WORK;
 
821
                                        type = PhonesColumns.TYPE_FAX_WORK;
888
822
                        else if( types.contains( "CELL" ) || types.contains( "VIDEO" ) )
889
 
                                type = TYPE_MOBILE;
 
823
                                type = PhonesColumns.TYPE_MOBILE;
890
824
                        else if( types.contains( "PAGER" ) )
891
 
                                type = TYPE_PAGER;
 
825
                                type = PhonesColumns.TYPE_PAGER;
892
826
                        else if( types.contains( "WORK" ) )
893
 
                                type = TYPE_WORK;
 
827
                                type = PhonesColumns.TYPE_WORK;
894
828
                        else
895
 
                                type = TYPE_HOME;
 
829
                                type = PhonesColumns.TYPE_HOME;
896
830
 
897
831
                        // add phone number
898
832
                        addNumber( value, type, is_preferred );
909
843
                        boolean is_preferred = types.contains( "PREF" );
910
844
                        int type;
911
845
                        if( types.contains( "WORK" ) )
912
 
                                type = TYPE_WORK;
 
846
                                type = Contacts.ContactMethods.TYPE_WORK;
913
847
                        else
914
 
                                type = TYPE_HOME;
 
848
                                type = Contacts.ContactMethods.TYPE_HOME;
915
849
 
916
 
                        addEmail( unescapeValue( value ), type, is_preferred );
 
850
                        addEmail( value, type, is_preferred );
917
851
                }
918
852
 
919
853
                private void parseADR( String[] params, String value )
920
854
                {
921
855
                        // get address parts
922
 
                        String[] adr_parts = splitValueByCharacter( value, ';' );
 
856
                        String[] adr_parts = splitValueBySemicolon( value );
923
857
 
924
858
                        // build address
925
859
                        value = "";
926
 
                        for( int a = 0; a < adr_parts.length; a++ )
927
 
                                if( adr_parts[ a ].length() > 0 )
928
 
                                {
929
 
                                        // split this part in to it's comma-separated bits
930
 
                                        String[] adr_part_parts =
931
 
                                                splitValueByCharacter( adr_parts[ a ], ',' );
932
 
                                        for( int b = 0; b < adr_part_parts.length; b++ )
933
 
                                                if( adr_part_parts[ b ].length() > 0 )
934
 
                                                {
935
 
                                                        if( value.length() > 0 ) value += "\n";
936
 
                                                        value += adr_part_parts[ b ];
937
 
                                                }
938
 
                                }
939
 
 
940
 
                        Set< String > types = extractTypes( params, Arrays.asList(
941
 
                                "PREF", "WORK", "HOME" ) );
942
 
 
943
 
                        // add address
944
 
                        int type;
945
 
                        if( types.contains( "WORK" ) )
946
 
                                type = TYPE_WORK;
947
 
                        else
948
 
                                type = TYPE_HOME;
949
 
 
950
 
                        addAddress( unescapeValue( value ), type );
951
 
                }
952
 
 
953
 
                private void parseLABEL( String[] params, String value )
954
 
                {
955
 
                        Set< String > types = extractTypes( params, Arrays.asList(
956
 
                                "PREF", "WORK", "HOME" ) );
957
 
 
958
 
                        // add address
959
 
                        int type;
960
 
                        if( types.contains( "WORK" ) )
961
 
                                type = TYPE_WORK;
962
 
                        else
963
 
                                type = TYPE_HOME;
964
 
 
965
 
                        addAddress( unescapeValue( value ), type );
 
860
                        for( int a = 0; a < adr_parts.length; a++ ) {
 
861
                                if( value.length() > 0 ) value += "\n";
 
862
                                value += adr_parts[ a ].trim();
 
863
                        }
 
864
 
 
865
                        Set< String > types = extractTypes( params, Arrays.asList(
 
866
                                "PREF", "WORK", "HOME", "INTERNET" ) );
 
867
 
 
868
                        // add address
 
869
                        int type;
 
870
                        if( types.contains( "WORK" ) )
 
871
                                type = Contacts.ContactMethods.TYPE_WORK;
 
872
                        else
 
873
                                type = Contacts.ContactMethods.TYPE_HOME;
 
874
 
 
875
                        addAddress( value, type );
966
876
                }
967
877
 
968
878
                public void finaliseVcard()
976
886
                        finalise();
977
887
                }
978
888
 
979
 
                /**
980
 
                 * Amongst the params, find the value of the first, only, of any with
981
 
                 * the specified name
982
 
                 * @param params
983
 
                 * @param name
984
 
                 * @return a value, or null
985
 
                 */
986
889
                private String checkParam( String[] params, String name )
987
890
                {
988
 
                        String[] res = checkParams( params, name );
989
 
                        return res.length > 0? res[ 0 ] : null;
990
 
                }
991
 
 
992
 
                /**
993
 
                 * Amongst the params, find the values of any with the specified name
994
 
                 * @param params
995
 
                 * @param name
996
 
                 * @return an array of values, or null
997
 
                 */
998
 
                private String[] checkParams( String[] params, String name )
999
 
                {
1000
 
                        HashSet< String > ret = new HashSet< String >();
1001
 
 
1002
891
                        Pattern p = Pattern.compile(
1003
892
                                "^" + name + "[ \\t]*=[ \\t]*(\"?)(.*)\\1$" );
1004
893
                        for( int i = 0; i < params.length; i++ ) {
1005
894
                                Matcher m = p.matcher( params[ i ] );
1006
895
                                if( m.matches() )
1007
 
                                        ret.add( m.group( 2 ) );
 
896
                                        return m.group( 2 );
1008
897
                        }
1009
 
 
1010
 
                        return (String[]) ret.toArray( new String[ ret.size() ] );
 
898
                        return null;
1011
899
                }
1012
900
 
1013
 
                /**
1014
 
                 * Amongst the params, return any type values present. For v2.1 vCards,
1015
 
                 * those types are just parameters. For v3.0, they are prefixed with
1016
 
                 * "TYPE=". There may also be multiple type parameters.
1017
 
                 * @param params
1018
 
                 * @param a list of type values to look for
1019
 
                 * @return a set of present type values
1020
 
                 */
1021
901
                private Set< String > extractTypes( String[] params,
1022
902
                        List< String > valid_types )
1023
903
                {
1024
904
                        HashSet< String > types = new HashSet< String >();
1025
905
 
1026
906
                        // get 3.0-style TYPE= param
1027
 
                        String type_params[] = checkParams( params, "TYPE" );
1028
 
                        for( int a = 0; a < type_params.length; a++ )
1029
 
                        {
1030
 
                                // check for a comma-separated list of types (why? this isn't in
1031
 
                                // the specs!)
1032
 
                                String[] parts = type_params[ a ].split( "," );
 
907
                        String type_param;
 
908
                        if( ( type_param = checkParam( params, "TYPE" ) ) != null ) {
 
909
                                String[] parts = type_param.split( "," );
1033
910
                                for( int i = 0; i < parts.length; i++ )
1034
911
                                        if( valid_types.contains( parts[ i ] ) )
1035
912
                                                types.add( parts[ i ] );