/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/VcardImporter.java

  • Committer: edam
  • Date: 2012-12-19 17:51:35 UTC
  • Revision ID: tim@ed.am-20121219175135-1cpuafp76jg1ib1p
added preliminary (buggy) ContactsContract backend

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/am/ed/importcontacts/VcardImporter.java
1
1
/*
2
2
 * VCFImporter.java
3
3
 *
4
 
 * Copyright (C) 2009 to 2011 Tim Marston <edam@waxworlds.org>
 
4
 * Copyright (C) 2009 to 2011 Tim Marston <tim@ed.am>
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://www.waxworlds.org/edam/software/android/import-contacts
 
8
 * http://ed.am/dev/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 org.waxworlds.edam.importcontacts;
 
24
package am.ed.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;
41
42
import java.util.NoSuchElementException;
42
43
import java.util.Set;
43
44
import java.util.Vector;
44
45
import java.util.regex.Matcher;
45
46
import java.util.regex.Pattern;
46
47
 
 
48
import android.annotation.SuppressLint;
47
49
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" )
61
62
        @Override
62
63
        protected void onImport() throws AbortImportException
63
64
        {
82
83
                                // get files
83
84
                                class VCardFilter implements FilenameFilter {
84
85
                                        public boolean accept( File dir, String name ) {
85
 
                                                return name.toLowerCase().endsWith( ".vcf" );
 
86
                                                return name.toLowerCase( Locale.US ).endsWith( ".vcf" );
86
87
                                        }
87
88
                                }
88
89
                                files = file.listFiles( new VCardFilter() );
362
363
        private class Vcard extends ContactData
363
364
        {
364
365
                private final static int NAMELEVEL_NONE = 0;
365
 
                private final static int NAMELEVEL_FN = 1;
366
 
                private final static int NAMELEVEL_N = 2;
 
366
                private final static int NAMELEVEL_N = 1;
 
367
                private final static int NAMELEVEL_FN = 2;
367
368
 
368
369
                private final static int MULTILINE_NONE = 0;
369
370
                private final static int MULTILINE_ENCODED = 1; // v2.1 quoted-printable
568
569
 
569
570
                                // determine whether we care about this entry
570
571
                                final HashSet< String > interesting_fields =
571
 
                                        new HashSet< String >( Arrays.asList( new String[]
572
 
                                                { "N", "FN", "ORG", "TITLE", "TEL", "EMAIL", "ADR" }
 
572
                                        new HashSet< String >( Arrays.asList( new String[] { "N",
 
573
                                                "FN", "ORG", "TITLE", "TEL", "EMAIL", "ADR", "LABEL" }
573
574
                                ) );
574
575
                                boolean is_interesting_field =
575
576
                                        interesting_fields.contains( name_param_parts[ 0 ] );
576
577
 
577
578
                                // parse encoding parameter
578
579
                                String encoding = checkParam( name_param_parts, "ENCODING" );
579
 
                                if( encoding != null ) encoding = encoding.toUpperCase();
 
580
                                if( encoding != null )
 
581
                                        encoding = encoding.toUpperCase( Locale.US );
580
582
                                if( is_interesting_field && encoding != null &&
581
583
                                        !encoding.equals( "8BIT" ) &&
582
584
                                        !encoding.equals( "QUOTED-PRINTABLE" ) )
587
589
 
588
590
                                // parse charset parameter
589
591
                                String charset = checkParam( name_param_parts, "CHARSET" );
590
 
                                if( charset != null ) charset = charset.toUpperCase();
 
592
                                if( charset != null )
 
593
                                        charset = charset.toUpperCase( Locale.US );
591
594
                                if( charset != null &&
592
595
                                        !charset.equals( "US-ASCII" ) &&
593
596
                                        !charset.equals( "ASCII" ) &&
609
612
                                                _parser_multiline_state = MULTILINE_ENCODED;
610
613
                                }
611
614
 
612
 
                                // convert 8-bit ASCII charset to US-ASCII
613
 
                                if( charset == null || charset.equals( "ASCII" ) ) {
 
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
                                {
614
622
                                        value = transcodeAsciiToUtf8( value );
615
 
                                        charset = "UTF-8";
616
623
                                }
617
624
 
618
 
                                // process charset
 
625
                                // process charset (value is now in UTF-8)
619
626
                                String string_value;
620
627
                                try {
621
628
                                        string_value = new String( value.array(), value.position(),
622
 
                                                value.limit() - value.position(), charset );
 
629
                                                value.limit() - value.position(), "UTF-8" );
623
630
                                } catch( UnsupportedEncodingException e ) {
624
631
                                        throw new ParseException( R.string.error_vcf_charset );
625
632
                                }
672
679
                                        parseEMAIL( name_param_parts, complete_value );
673
680
                                else if( name_param_parts[ 0 ].equals( "ADR" ) )
674
681
                                        parseADR( name_param_parts, complete_value );
 
682
                                else if( name_param_parts[ 0 ].equals( "LABEL" ) )
 
683
                                        parseLABEL( name_param_parts, complete_value );
675
684
                        }
676
685
                }
677
686
 
690
699
                        return ( count & 1 ) == 1;
691
700
                }
692
701
 
693
 
                private String[] splitValueBySemicolon( String value )
 
702
                private String[] splitValueByCharacter( String value, char character )
694
703
                {
695
 
                        // split string in to parts by semicolon
 
704
                        // split string in to parts by specified character
696
705
                        ArrayList< String > parts = new ArrayList< String >(
697
 
                                Arrays.asList( value.split(  ";" ) ) );
 
706
                                Arrays.asList( value.split( "" + character ) ) );
698
707
 
699
708
                        // go through parts
700
709
                        for( int a = 0; a < parts.size(); a++ )
708
717
                                if( a < parts.size() - 1 &&
709
718
                                        doesStringEndInAnEscapeChar( str ) )
710
719
                                {
711
 
                                        // join the next part to this part and remove the next part
 
720
                                        // append the escaped character, join the next part to this
 
721
                                        // part and remove the next part
712
722
                                        parts.set( a, str.substring( 0, str.length() - 1 ) +
713
 
                                                ';' + parts.get( a + 1 ) );
 
723
                                                character + parts.get( a + 1 ) );
714
724
                                        parts.remove( a + 1 );
715
725
 
716
726
                                        // re-visit this part
727
737
                        return parts.toArray( ret );
728
738
                }
729
739
 
 
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
 
730
783
                private void parseN( String[] params, String value )
731
784
                {
732
785
                        // already got a better name?
733
786
                        if( _name_level >= NAMELEVEL_N ) return;
734
787
 
735
788
                        // get name parts
736
 
                        String[] name_parts = splitValueBySemicolon( value );
 
789
                        String[] name_parts = splitValueByCharacter( value, ';' );
737
790
 
738
791
                        // build name
739
792
                        value = "";
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 ];
 
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
                                }
744
808
 
745
809
                        // set name
746
 
                        setName( value );
 
810
                        setName( unescapeValue( value ) );
747
811
                        _name_level = NAMELEVEL_N;
748
812
                }
749
813
 
753
817
                        if( _name_level >= NAMELEVEL_FN ) return;
754
818
 
755
819
                        // set name
756
 
                        setName( value );
 
820
                        setName( unescapeValue( value ) );
757
821
                        _name_level = NAMELEVEL_FN;
758
822
                }
759
823
 
760
824
                private void parseORG( String[] params, String value )
761
825
                {
762
826
                        // get org parts
763
 
                        String[] org_parts = splitValueBySemicolon( value );
 
827
                        String[] org_parts = splitValueByCharacter( value, ';' );
764
828
                        if( org_parts == null || org_parts.length < 1 ) return;
765
829
 
766
830
                        // build organisation name
768
832
                                String.valueOf( org_parts[ 0 ] ) );
769
833
                        for( int a = 1; a < org_parts.length; a++ )
770
834
                                builder.append( ", " ).append( org_parts[ a ] );
771
 
                        String organisation = builder.toString();
 
835
                        String organisation = unescapeValue( builder.toString() );
772
836
 
773
837
                        // set organisation name (using a title we've previously found)
774
838
                        addOrganisation( organisation, _cached_title, true );
785
849
 
786
850
                private void parseTITLE( String[] params, String value )
787
851
                {
 
852
                        value = unescapeValue( value );
 
853
 
788
854
                        // if we previously had an organisation, look it up and append this
789
855
                        // title to it
790
856
                        if( _cached_organisation != null && hasOrganisations() ) {
816
882
                        int type;
817
883
                        if( types.contains( "FAX" ) )
818
884
                                if( types.contains( "HOME" ) )
819
 
                                        type = PhonesColumns.TYPE_FAX_HOME;
 
885
                                        type = TYPE_FAX_HOME;
820
886
                                else
821
 
                                        type = PhonesColumns.TYPE_FAX_WORK;
 
887
                                        type = TYPE_FAX_WORK;
822
888
                        else if( types.contains( "CELL" ) || types.contains( "VIDEO" ) )
823
 
                                type = PhonesColumns.TYPE_MOBILE;
 
889
                                type = TYPE_MOBILE;
824
890
                        else if( types.contains( "PAGER" ) )
825
 
                                type = PhonesColumns.TYPE_PAGER;
 
891
                                type = TYPE_PAGER;
826
892
                        else if( types.contains( "WORK" ) )
827
 
                                type = PhonesColumns.TYPE_WORK;
 
893
                                type = TYPE_WORK;
828
894
                        else
829
 
                                type = PhonesColumns.TYPE_HOME;
 
895
                                type = TYPE_HOME;
830
896
 
831
897
                        // add phone number
832
898
                        addNumber( value, type, is_preferred );
843
909
                        boolean is_preferred = types.contains( "PREF" );
844
910
                        int type;
845
911
                        if( types.contains( "WORK" ) )
846
 
                                type = Contacts.ContactMethods.TYPE_WORK;
 
912
                                type = TYPE_WORK;
847
913
                        else
848
 
                                type = Contacts.ContactMethods.TYPE_HOME;
 
914
                                type = TYPE_HOME;
849
915
 
850
 
                        addEmail( value, type, is_preferred );
 
916
                        addEmail( unescapeValue( value ), type, is_preferred );
851
917
                }
852
918
 
853
919
                private void parseADR( String[] params, String value )
854
920
                {
855
921
                        // get address parts
856
 
                        String[] adr_parts = splitValueBySemicolon( value );
 
922
                        String[] adr_parts = splitValueByCharacter( value, ';' );
857
923
 
858
924
                        // build address
859
925
                        value = "";
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 );
 
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 );
876
966
                }
877
967
 
878
968
                public void finaliseVcard()
886
976
                        finalise();
887
977
                }
888
978
 
 
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
                 */
889
986
                private String checkParam( String[] params, String name )
890
987
                {
 
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
 
891
1002
                        Pattern p = Pattern.compile(
892
1003
                                "^" + name + "[ \\t]*=[ \\t]*(\"?)(.*)\\1$" );
893
1004
                        for( int i = 0; i < params.length; i++ ) {
894
1005
                                Matcher m = p.matcher( params[ i ] );
895
1006
                                if( m.matches() )
896
 
                                        return m.group( 2 );
 
1007
                                        ret.add( m.group( 2 ) );
897
1008
                        }
898
 
                        return null;
 
1009
 
 
1010
                        return (String[]) ret.toArray( new String[ ret.size() ] );
899
1011
                }
900
1012
 
 
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
                 */
901
1021
                private Set< String > extractTypes( String[] params,
902
1022
                        List< String > valid_types )
903
1023
                {
904
1024
                        HashSet< String > types = new HashSet< String >();
905
1025
 
906
1026
                        // get 3.0-style TYPE= param
907
 
                        String type_param;
908
 
                        if( ( type_param = checkParam( params, "TYPE" ) ) != null ) {
909
 
                                String[] parts = type_param.split( "," );
 
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( "," );
910
1033
                                for( int i = 0; i < parts.length; i++ )
911
1034
                                        if( valid_types.contains( parts[ i ] ) )
912
1035
                                                types.add( parts[ i ] );