/android/export-contacts

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

« back to all changes in this revision

Viewing changes to src/am/ed/exportcontacts/VcardExporter.java

  • Committer: edam
  • Date: 2012-12-19 15:51:48 UTC
  • Revision ID: tim@ed.am-20121219155148-9ut7lwtpi21aj54m
changed "homepage" to "webpage"

Show diffs side-by-side

added added

removed removed

src/am/ed/exportcontacts/VcardExporter.java
 
1
/*
 
2
 * Exporter.java
 
3
 *
 
4
 * Copyright (C) 2011 Tim Marston <tim@ed.am>
 
5
 *
 
6
 * This file is part of the Export Contacts program (hereafter referred
 
7
 * to as "this program"). For more information, see
 
8
 * http://ed.am/dev/android/export-contacts
 
9
 *
 
10
 * This program is free software: you can redistribute it and/or modify
 
11
 * it under the terms of the GNU General Public License as published by
 
12
 * the Free Software Foundation, either version 3 of the License, or
 
13
 * (at your option) any later version.
 
14
 *
 
15
 * This program is distributed in the hope that it will be useful,
 
16
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
17
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
18
 * GNU General Public License for more details.
 
19
 *
 
20
 * You should have received a copy of the GNU General Public License
 
21
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
22
 */
 
23
 
 
24
package am.ed.exportcontacts;
 
25
 
 
26
import java.io.File;
 
27
import java.io.FileNotFoundException;
 
28
import java.io.FileOutputStream;
 
29
import java.io.IOException;
 
30
import java.util.AbstractCollection;
 
31
import java.util.ArrayList;
 
32
import java.util.Iterator;
 
33
 
 
34
import android.content.SharedPreferences;
 
35
import android.provider.Contacts;
 
36
 
 
37
public class VcardExporter extends Exporter
 
38
{
 
39
        protected FileOutputStream _ostream = null;
 
40
 
 
41
        public VcardExporter( Doit doit )
 
42
        {
 
43
                super( doit );
 
44
        }
 
45
 
 
46
        @Override
 
47
        protected void preExport() throws AbortExportException
 
48
        {
 
49
                SharedPreferences prefs = getSharedPreferences();
 
50
 
 
51
                // create output filename
 
52
                String filename = prefs.getString( "filename", "android-contacts.vcf" );
 
53
                File file = new File( "/sdcard" + prefs.getString( "location", "/" ) +
 
54
                        filename );
 
55
 
 
56
                // check if the output file already exists
 
57
                if( file.exists() && file.length() > 0 )
 
58
                        if( !showContinue( R.string.error_vcf_exists ) )
 
59
                                finish( ACTION_ABORT );
 
60
 
 
61
                // open file
 
62
                try {
 
63
                        _ostream = new FileOutputStream( file );
 
64
                }
 
65
                catch( FileNotFoundException e ) {
 
66
                        showError( R.string.error_filenotfound );
 
67
                }
 
68
        }
 
69
 
 
70
        /**
 
71
         * Do line folding at 75 chars
 
72
         * @param raw string
 
73
         * @return folded string
 
74
         */
 
75
        private String fold( String line )
 
76
        {
 
77
                StringBuilder ret = new StringBuilder( line.length() );
 
78
 
 
79
                // keep pulling off the first line's worth of chars, while the string is
 
80
                // still longer than a line should be
 
81
                while( line.length() > 75 )
 
82
                {
 
83
                        // length of the line we'll be pulling off
 
84
                        int len = 75;
 
85
 
 
86
                        // if splitting at this length would break apart a codepoint, use
 
87
                        // one less char
 
88
                        if( Character.isHighSurrogate( line.charAt( len - 1 ) ) )
 
89
                                len--;
 
90
 
 
91
                        // count how many backslashes would be at the end of the line we're
 
92
                        // pulling off
 
93
                        int count = 0;
 
94
                        for( int a = len - 1; a >= 0; a-- )
 
95
                                if( line.charAt( a ) == '\\' )
 
96
                                        count++;
 
97
                                else
 
98
                                        break;
 
99
 
 
100
                        // if there would be an odd number of slashes at the end of the line
 
101
                        // then pull off one fewer characters so that we don't break apart
 
102
                        // escape sequences
 
103
                        if( count % 2 == 1 )
 
104
                                len--;
 
105
 
 
106
                        // pull off the line and add it to the output, folded
 
107
                        ret.append( line.substring( 0, len ) + "\n " );
 
108
                        line = line.substring( len );
 
109
                }
 
110
 
 
111
                // add any remaining data
 
112
                ret.append( line );
 
113
 
 
114
                return ret.toString();
 
115
        }
 
116
 
 
117
        /**
 
118
         * Do unsafe character escaping
 
119
         * @param raw string
 
120
         * @return escaped string
 
121
         */
 
122
        private String escape( String str )
 
123
        {
 
124
                StringBuilder ret = new StringBuilder( str.length() );
 
125
                for( int a = 0; a < str.length(); a++ )
 
126
                {
 
127
                        int c = str.codePointAt( a );
 
128
                        switch( c )
 
129
                        {
 
130
                        case '\n':
 
131
                                // append escaped newline
 
132
                                ret.append( "\\n" );
 
133
                                break;
 
134
                        case ',':
 
135
                        case ';':
 
136
                        case '\\':
 
137
                                // append return character
 
138
                                ret.append( '\\' );
 
139
                                // fall through
 
140
                        default:
 
141
                                // append character
 
142
                                ret.append( Character.toChars( c ) );
 
143
                        }
 
144
                }
 
145
 
 
146
                return ret.toString();
 
147
        }
 
148
 
 
149
        /**
 
150
         * join
 
151
         */
 
152
        @SuppressWarnings( "rawtypes" )
 
153
        public static String join( AbstractCollection s, String delimiter)
 
154
        {
 
155
                StringBuffer buffer = new StringBuffer();
 
156
                Iterator iter = s.iterator();
 
157
                if( iter.hasNext() ) {
 
158
                        buffer.append( iter.next() );
 
159
                        while( iter.hasNext() ) {
 
160
                                buffer.append( delimiter );
 
161
                                buffer.append( iter.next() );
 
162
                        }
 
163
                }
 
164
                return buffer.toString();
 
165
        }
 
166
 
 
167
 
 
168
        @Override
 
169
        protected boolean exportContact( ContactData contact )
 
170
                throws AbortExportException
 
171
        {
 
172
                StringBuilder out = new StringBuilder();
 
173
 
 
174
                // skip if the contact has no identifiable features
 
175
                if( contact.getPrimaryIdentifier() == null )
 
176
                        return false;
 
177
 
 
178
                // append header
 
179
                out.append( "BEGIN:VCARD\n" );
 
180
                out.append( "VERSION:3.0\n" );
 
181
 
 
182
                // append formatted name
 
183
                String name = contact.getName();
 
184
                if( name == null ) name = "";
 
185
                out.append( fold( "FN:" + escape( name ) ) + "\n" );
 
186
 
 
187
                // append name
 
188
                String[] bits = name.split( " +" );
 
189
                StringBuilder tmp = new StringBuilder();
 
190
                for( int a = 1; a < bits.length - 1; a++ ) {
 
191
                        if( a > 1 ) tmp.append( " " );
 
192
                        tmp.append( escape( bits[ a ] ) );
 
193
                }
 
194
                String value = escape( bits[ bits.length - 1 ] ) + ";" +
 
195
                        ( bits.length > 1? escape( bits[ 0 ] ) : "" ) + ";" +
 
196
                        tmp.toString() + ";;";
 
197
                out.append( fold( "N:" + value ) + "\n" );
 
198
 
 
199
                // append organisations and titles
 
200
                ArrayList< Exporter.ContactData.OrganisationDetail > organisations =
 
201
                        contact.getOrganisations();
 
202
                if( organisations != null ) {
 
203
                        for( int a = 0; a < organisations.size(); a++ ) {
 
204
                                if( organisations.get( a ).getOrganisation() != null )
 
205
                                        out.append( fold( "ORG:" + escape(
 
206
                                                organisations.get( a ).getOrganisation() ) ) + "\n" );
 
207
                                if( organisations.get( a ).getTitle() != null )
 
208
                                        out.append( fold( "TITLE:" + escape(
 
209
                                                organisations.get( a ).getTitle() ) ) + "\n" );
 
210
                        }
 
211
                }
 
212
 
 
213
                // append phone numbers
 
214
                ArrayList< Exporter.ContactData.NumberDetail > numbers =
 
215
                        contact.getNumbers();
 
216
                if( numbers != null ) {
 
217
                        for( int a = 0; a < numbers.size(); a++ ) {
 
218
                                ArrayList< String > types = new ArrayList< String >();
 
219
                                switch( numbers.get( a ).getType() ) {
 
220
                                case Contacts.Phones.TYPE_HOME:
 
221
                                        types.add( "VOICE" ); types.add( "HOME" ); break;
 
222
                                case Contacts.Phones.TYPE_WORK:
 
223
                                        types.add( "VOICE" ); types.add( "WORK" ); break;
 
224
                                case Contacts.Phones.TYPE_FAX_HOME:
 
225
                                        types.add( "FAX" ); types.add( "HOME" ); break;
 
226
                                case Contacts.Phones.TYPE_FAX_WORK:
 
227
                                        types.add( "FAX" ); types.add( "WORK" ); break;
 
228
                                case Contacts.Phones.TYPE_PAGER:
 
229
                                        types.add( "PAGER" ); break;
 
230
                                case Contacts.Phones.TYPE_MOBILE:
 
231
                                        types.add( "VOICE" ); types.add( "CELL" ); break;
 
232
                                }
 
233
                                if( a == 0 ) types.add( "PREF" );
 
234
                                out.append( fold( "TEL" +
 
235
                                        ( types.size() > 0? ";TYPE=" + join( types, "," ) : "" ) +
 
236
                                        ":" + escape( numbers.get( a ).getNumber() ) ) + "\n" );
 
237
                        }
 
238
                }
 
239
 
 
240
                // append email addresses
 
241
                ArrayList< Exporter.ContactData.EmailDetail > emails =
 
242
                        contact.getEmails();
 
243
                if( emails != null ) {
 
244
                        for( int a = 0; a < emails.size(); a++ ) {
 
245
                                ArrayList< String > types = new ArrayList< String >();
 
246
                                types.add( "INTERNET" );
 
247
                                switch( emails.get( a ).getType() ) {
 
248
                                case Contacts.ContactMethods.TYPE_HOME:
 
249
                                        types.add( "HOME" ); break;
 
250
                                case Contacts.ContactMethods.TYPE_WORK:
 
251
                                        types.add( "WORK" ); break;
 
252
                                }
 
253
                                out.append( fold( "EMAIL" +
 
254
                                        ( types.size() > 0? ";TYPE=" + join( types, "," ) : "" ) +
 
255
                                        ":" + escape( emails.get( a ).getEmail() ) ) + "\n" );
 
256
                        }
 
257
                }
 
258
 
 
259
                // append addresses
 
260
                ArrayList< Exporter.ContactData.AddressDetail > addresses =
 
261
                        contact.getAddresses();
 
262
                if( addresses != null ) {
 
263
                        for( int a = 0; a < addresses.size(); a++ ) {
 
264
                                ArrayList< String > types = new ArrayList< String >();
 
265
                                types.add( "POSTAL" );
 
266
                                switch( addresses.get( a ).getType() ) {
 
267
                                case Contacts.ContactMethods.TYPE_HOME:
 
268
                                        types.add( "HOME" ); break;
 
269
                                case Contacts.ContactMethods.TYPE_WORK:
 
270
                                        types.add( "WORK" ); break;
 
271
                                }
 
272
                                out.append( fold( "LABEL" +
 
273
                                        ( types.size() > 0? ";TYPE=" + join( types, "," ) : "" ) +
 
274
                                        ":" + escape( addresses.get( a ).getAddress() ) ) + "\n" );
 
275
                        }
 
276
                }
 
277
 
 
278
                // append footer
 
279
                out.append( "END:VCARD\n" );
 
280
 
 
281
                // write to file
 
282
                try {
 
283
                        _ostream.write( out.toString().getBytes() );
 
284
                        _ostream.flush();
 
285
                }
 
286
                catch( IOException e ) {
 
287
                        showError( R.string.error_ioerror );
 
288
                }
 
289
 
 
290
                return true;
 
291
        }
 
292
 
 
293
}