/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-21 14:47:57 UTC
  • Revision ID: tim@ed.am-20121221144757-5cb1lgsp7fdt7p2n
updated TODO

Show diffs side-by-side

added added

removed removed

133
133
                        boolean in_vcard = false;
134
134
                        while( ( line = reader.readLine() ) != null )
135
135
                        {
136
 
                                if( !in_vcard )
137
 
                                {
 
136
                                if( !in_vcard ) {
138
137
                                        // look for vcard beginning
139
 
                                        if( line.matches( "(?i)BEGIN[ \t]*:[ \t]*VCARD.*" ) ) {
 
138
                                        if( line.matches( "^BEGIN:VCARD" ) ) {
140
139
                                                in_vcard = true;
141
140
                                                _vcard_count++;
142
141
                                        }
143
 
                                        // check for vMsg files
144
 
                                        else if( line.matches( "(?i)BEGIN[ \t]*:[ \t]*VMSG.*" ) ) {
145
 
                                                showError( getText( R.string.error_vcf_vmsgfile )
146
 
                                                        + file.getName() );
147
 
                                        }
148
142
                                }
149
 
                                else if( line.matches( "(?i)END[ \t]*:[ \t]*VCARD.*" ) )
 
143
                                else if( line.matches( "^END:VCARD" ) )
150
144
                                        in_vcard = false;
151
145
                        }
152
146
 
181
175
                        // import
182
176
                        importVCardFileContent( content, file.getName() );
183
177
                }
184
 
                catch( OutOfMemoryError e ) {
185
 
                        showError( R.string.error_outofmemory );
186
 
                }
187
178
                catch( FileNotFoundException e ) {
188
179
                        showError( getText( R.string.error_filenotfound ) +
189
180
                                file.getName() );
202
193
                ContentLineIterator cli = new ContentLineIterator( content );
203
194
                while( cli.hasNext() )
204
195
                {
205
 
                        ContentLine content_line = cli.next();
 
196
                        ByteBuffer buffer = cli.next();
206
197
 
207
 
                        // get a US-ASCII version of the string, for processing
208
 
                        String line = content_line.getUsAsciiLine();
 
198
                        // get a US-ASCII version of the line for processing
 
199
                        String line;
 
200
                        try {
 
201
                                line = new String( buffer.array(), buffer.position(),
 
202
                                        buffer.limit() - buffer.position(), "US-ASCII" );
 
203
                        }
 
204
                        catch( UnsupportedEncodingException e ) {
 
205
                                // we know US-ASCII *is* supported, so appease the compiler...
 
206
                                line = "";
 
207
                        }
209
208
 
210
209
                        if( vcard == null ) {
211
210
                                // look for vcard beginning
212
 
                                if( line.matches( "(?i)BEGIN[ \t]*:[ \t]*VCARD.*" ) ) {
 
211
                                if( line.matches( "^BEGIN[ \t]*:[ \t]*VCARD" ) ) {
213
212
                                        setProgress( _progress++ );
214
213
                                        vcard = new Vcard();
215
214
                                        vcard_start_line = cli.getLineNumber();
217
216
                        }
218
217
                        else {
219
218
                                // look for vcard content or ending
220
 
                                if( line.matches( "(?i)END[ \t]*:[ \t]*VCARD.*" ) )
 
219
                                if( line.matches( "^END[ \t]*:[ \t]*VCARD" ) )
221
220
                                {
222
221
                                        // finalise the vcard/contact
223
222
                                        try {
260
259
                                {
261
260
                                        // try giving the line to the vcard
262
261
                                        try {
263
 
                                                vcard.parseLine( content_line );
 
262
                                                vcard.parseLine( buffer, line,
 
263
                                                        cli.doesNextLineLookFolded() );
264
264
                                        }
265
265
                                        catch( Vcard.ParseException e ) {
266
266
                                                skipContact();
289
289
                }
290
290
        }
291
291
 
292
 
        class ContentLine
293
 
        {
294
 
                private ByteBuffer _buffer;
295
 
                private boolean _folded_next;
296
 
                private String _line;
297
 
 
298
 
                public ContentLine( ByteBuffer buffer, boolean folded_next )
299
 
                {
300
 
                        _buffer = buffer;
301
 
                        _folded_next = folded_next;
302
 
                        _line = null;
303
 
                }
304
 
 
305
 
                public ByteBuffer getBuffer()
306
 
                {
307
 
                        return _buffer;
308
 
                }
309
 
 
310
 
                public boolean doesNextLineLookFolded()
311
 
                {
312
 
                        return _folded_next;
313
 
                }
314
 
 
315
 
                public String getUsAsciiLine()
316
 
                {
317
 
                        // generated line and cache it
318
 
                        if( _line == null ) {
319
 
                                try {
320
 
                                        _line = new String( _buffer.array(), _buffer.position(),
321
 
                                                _buffer.limit() - _buffer.position(), "US-ASCII" );
322
 
                                }
323
 
                                catch( UnsupportedEncodingException e ) {
324
 
                                        // we know US-ASCII *is* supported, so appease the
325
 
                                        // compiler...
326
 
                                }
327
 
                        }
328
 
 
329
 
                        // return cached line
330
 
                        return _line;
331
 
                }
332
 
        }
333
 
 
334
 
        class ContentLineIterator implements Iterator< ContentLine >
 
292
        class ContentLineIterator implements Iterator< ByteBuffer >
335
293
        {
336
294
                protected byte[] _content = null;
337
295
                protected int _pos = 0;
349
307
                }
350
308
 
351
309
                @Override
352
 
                public ContentLine next()
 
310
                public ByteBuffer next()
353
311
                {
354
312
                        int initial_pos = _pos;
355
313
 
362
320
                                                _pos > initial_pos )? _pos - 1 : _pos;
363
321
                                        _pos++;
364
322
                                        _line++;
365
 
                                        return new ContentLine(
366
 
                                                ByteBuffer.wrap( _content, initial_pos,
367
 
                                                        to - initial_pos ),
368
 
                                                doesNextLineLookFolded() );
 
323
                                        return ByteBuffer.wrap( _content, initial_pos,
 
324
                                                to - initial_pos );
369
325
                                }
370
326
 
371
327
                        // we didn't find one, but were there bytes left?
373
329
                                int to = _pos;
374
330
                                _pos++;
375
331
                                _line++;
376
 
                                return new ContentLine(
377
 
                                        ByteBuffer.wrap( _content, initial_pos,
378
 
                                                to - initial_pos ),
379
 
                                        doesNextLineLookFolded() );
 
332
                                return ByteBuffer.wrap( _content, initial_pos,
 
333
                                        to - initial_pos );
380
334
                        }
381
335
 
382
336
                        // no bytes left
394
348
                 * onto the end of this one?
395
349
                 * @return
396
350
                 */
397
 
                private boolean doesNextLineLookFolded()
 
351
                public boolean doesNextLineLookFolded()
398
352
                {
399
353
                        return _pos > 0 && _pos < _content.length &&
400
354
                                _content[ _pos - 1 ] == '\n' &&
419
373
                private final static int MULTILINE_FOLDED = 3;  // MIME-DIR folding
420
374
 
421
375
                private String _version = null;
422
 
                private Vector< ContentLine > _content_lines = null;
 
376
                private Vector< ByteBuffer > _buffers = null;
423
377
                private int _name_level = NAMELEVEL_NONE;
424
378
                private int _parser_multiline_state = MULTILINE_NONE;
425
379
                private String _parser_current_name_and_params = null;
468
422
                @SuppressWarnings("serial")
469
423
                protected class SkipImportException extends Exception { }
470
424
 
471
 
                private String extractCollonPartFromLine( ContentLine content_line,
472
 
                        boolean former )
 
425
                private String extractCollonPartFromLine( ByteBuffer buffer,
 
426
                        String line, boolean former )
473
427
                {
474
428
                        String ret = null;
475
429
 
 
430
                        // get a US-ASCII version of the line for processing, unless we were
 
431
                        // supplied with one
 
432
                        if( line == null ) {
 
433
                                try {
 
434
                                        line = new String( buffer.array(), buffer.position(),
 
435
                                                buffer.limit() - buffer.position(), "US-ASCII" );
 
436
                                }
 
437
                                catch( UnsupportedEncodingException e ) {
 
438
                                        // we know US-ASCII is supported, so appease the compiler...
 
439
                                        line = "";
 
440
                                }
 
441
                        }
 
442
 
476
443
                        // split line into name and value parts and check to make sure we
477
444
                        // only got 2 parts and that the first part is not zero in length
478
 
                        String[] parts = content_line.getUsAsciiLine().split( ":", 2 );
 
445
                        String[] parts = line.split( ":", 2 );
479
446
                        if( parts.length == 2 && parts[ 0 ].length() > 0 )
480
447
                                ret = parts[ former? 0 : 1 ];
481
448
 
482
449
                        return ret;
483
450
                }
484
451
 
485
 
                private String extractNameAndParamsFromLine( ContentLine content_line )
486
 
                {
487
 
                        return extractCollonPartFromLine( content_line, true ).trim();
488
 
                }
489
 
 
490
 
                private String extractValueFromLine( ContentLine content_line )
491
 
                {
492
 
                        return extractCollonPartFromLine( content_line, false );
493
 
                }
494
 
 
495
 
                public void parseLine( ContentLine content_line )
 
452
                private String extractNameAndParamsFromLine( ByteBuffer buffer,
 
453
                        String line )
 
454
                {
 
455
                        return extractCollonPartFromLine( buffer, line, true );
 
456
                }
 
457
 
 
458
                private String extractValueFromLine( ByteBuffer buffer, String line )
 
459
                {
 
460
                        return extractCollonPartFromLine( buffer, line, false );
 
461
                }
 
462
 
 
463
                public void parseLine( ByteBuffer buffer, String line,
 
464
                        boolean next_line_looks_folded )
496
465
                        throws ParseException, SkipImportException,
497
466
                        AbortImportException
498
467
                {
501
470
                        {
502
471
                                // tentatively get name and params from line
503
472
                                String name_and_params =
504
 
                                        extractNameAndParamsFromLine( content_line );
 
473
                                        extractNameAndParamsFromLine( buffer, line );
505
474
 
506
475
                                // is it a version line?
507
476
                                if( name_and_params != null &&
508
477
                                        name_and_params.equalsIgnoreCase( "VERSION" ) )
509
478
                                {
510
479
                                        // yes, get it!
511
 
                                        String value = extractValueFromLine( content_line ).trim();
 
480
                                        String value = extractValueFromLine( buffer, line );
512
481
                                        if( !value.equals( "2.1" ) && !value.equals( "3.0" ) )
513
482
                                                throw new ParseException( R.string.error_vcf_version );
514
483
                                        _version = value;
515
484
 
516
485
                                        // parse any buffers we've been accumulating while we waited
517
486
                                        // for a version
518
 
                                        if( _content_lines != null )
519
 
                                                for( int i = 0; i < _content_lines.size(); i++ )
520
 
                                                        parseLine( _content_lines.get( i ) );
521
 
                                        _content_lines = null;
 
487
                                        if( _buffers != null )
 
488
                                                for( int i = 0; i < _buffers.size(); i++ )
 
489
                                                        parseLine( _buffers.get( i ), null,
 
490
                                                                i + 1 < _buffers.size() &&
 
491
                                                                _buffers.get( i + 1 ).hasRemaining() &&
 
492
                                                                _buffers.get( i + 1 ).get(
 
493
                                                                        _buffers.get( i + 1 ).position() ) == ' ' );
 
494
                                        _buffers = null;
522
495
                                }
523
496
                                else
524
497
                                {
525
498
                                        // no, so stash this line till we get a version
526
 
                                        if( _content_lines == null )
527
 
                                                _content_lines = new Vector< ContentLine >();
528
 
                                        _content_lines.add( content_line );
 
499
                                        if( _buffers == null )
 
500
                                                _buffers = new Vector< ByteBuffer >();
 
501
                                        _buffers.add( buffer );
529
502
                                }
530
503
                        }
531
504
                        else
543
516
 
544
517
                                        // skip some initial line characters, depending on the type
545
518
                                        // of multi-line we're handling
546
 
                                        pos = content_line.getBuffer().position();
 
519
                                        pos = buffer.position();
547
520
                                        switch( _parser_multiline_state )
548
521
                                        {
549
522
                                        case MULTILINE_FOLDED:
550
523
                                                pos++;
551
524
                                                break;
552
525
                                        case MULTILINE_ENCODED:
553
 
                                                while( pos < content_line.getBuffer().limit() && (
554
 
                                                        content_line.getBuffer().get( pos ) == ' ' ||
555
 
                                                        content_line.getBuffer().get( pos ) == '\t' ) )
 
526
                                                while( pos < buffer.limit() && (
 
527
                                                        buffer.get( pos ) == ' ' ||
 
528
                                                        buffer.get( pos ) == '\t' ) )
556
529
                                                {
557
530
                                                        pos++;
558
531
                                                }
568
541
                                else
569
542
                                {
570
543
                                        // skip empty lines
571
 
                                        if( content_line.getUsAsciiLine().trim().length() == 0 )
572
 
                                                return;
 
544
                                        if( line.trim().length() == 0 ) return;
573
545
 
574
546
                                        // get name and params from line, and since we're not
575
547
                                        // parsing a subsequent line in a multi-line, this should
576
548
                                        // not fail, or it's an error
577
549
                                        name_and_params =
578
 
                                                extractNameAndParamsFromLine( content_line );
 
550
                                                extractNameAndParamsFromLine( buffer, line );
579
551
                                        if( name_and_params == null )
580
552
                                                throw new ParseException(
581
553
                                                        R.string.error_vcf_malformed );
582
554
 
583
555
                                        // calculate how many chars to skip from beginning of line
584
556
                                        // so we skip the property "name:" part
585
 
                                        pos = content_line.getBuffer().position() +
586
 
                                                name_and_params.length() + 1;
 
557
                                        pos = buffer.position() + name_and_params.length() + 1;
587
558
 
588
559
                                        // reset the saved multi-line state
589
560
                                        _parser_current_name_and_params = name_and_params;
592
563
 
593
564
                                // get value from buffer, as raw bytes
594
565
                                ByteBuffer value;
595
 
                                value = ByteBuffer.wrap( content_line.getBuffer().array(), pos,
596
 
                                        content_line.getBuffer().limit() - pos );
 
566
                                value = ByteBuffer.wrap( buffer.array(), pos,
 
567
                                        buffer.limit() - pos );
597
568
 
598
569
                                // get parameter parts
599
570
                                String[] name_param_parts = name_and_params.split( ";", -1 );
687
658
                                // if we know we're not in an encoding-based multi-line, check
688
659
                                // to see if we're in a folded multi-line
689
660
                                if( _parser_multiline_state == MULTILINE_NONE &&
690
 
                                        content_line.doesNextLineLookFolded() )
 
661
                                        next_line_looks_folded )
691
662
                                {
692
663
                                        _parser_multiline_state = MULTILINE_FOLDED;
693
664
                                }
849
820
                                        for( int b = 0; b < name_part_parts.length; b++ )
850
821
                                                if( name_part_parts[ b ].length() > 0 )
851
822
                                                {
852
 
                                                        if( value.length() > 0 ) value += " ";
 
823
                                                        if( value.length() == 0 ) value += " ";
853
824
                                                        value += name_part_parts[ b ];
854
825
                                                }
855
826
                                }
1033
1004
                        throws ParseException, ContactNotIdentifiableException
1034
1005
                {
1035
1006
                        // missing version (and data is present)
1036
 
                        if( _version == null && _content_lines != null )
 
1007
                        if( _version == null && _buffers != null )
1037
1008
                                throw new ParseException( R.string.error_vcf_malformed );
1038
1009
 
1039
1010
                        // finalise the parent class