Loosen BDAY parsing for vCard results
[zxing.git] / core / src / com / google / zxing / client / result / VCardResultParser.java
1 /*
2  * Copyright 2008 ZXing authors
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package com.google.zxing.client.result;
18
19 import com.google.zxing.Result;
20
21 import java.util.Vector;
22
23 /**
24  * Parses contact information formatted according to the VCard (2.1) format. This is not a complete
25  * implementation but should parse information as commonly encoded in 2D barcodes.
26  *
27  * @author Sean Owen
28  */
29 final class VCardResultParser extends ResultParser {
30
31   private VCardResultParser() {
32   }
33
34   public static AddressBookParsedResult parse(Result result) {
35     // Although we should insist on the raw text ending with "END:VCARD", there's no reason
36     // to throw out everything else we parsed just because this was omitted. In fact, Eclair
37     // is doing just that, and we can't parse its contacts without this leniency.
38     String rawText = result.getText();
39     if (rawText == null || !rawText.startsWith("BEGIN:VCARD")) {
40       return null;
41     }
42     String[] names = matchVCardPrefixedField("FN", rawText, true);
43     if (names == null) {
44       // If no display names found, look for regular name fields and format them
45       names = matchVCardPrefixedField("N", rawText, true);
46       formatNames(names);
47     }
48     String[] phoneNumbers = matchVCardPrefixedField("TEL", rawText, true);
49     String[] emails = matchVCardPrefixedField("EMAIL", rawText, true);
50     String note = matchSingleVCardPrefixedField("NOTE", rawText, false);
51     String address = matchSingleVCardPrefixedField("ADR", rawText, true);
52     address = formatAddress(address);
53     String org = matchSingleVCardPrefixedField("ORG", rawText, true);
54     String birthday = matchSingleVCardPrefixedField("BDAY", rawText, true);
55     if (!isLikeVCardDate(birthday)) {
56       return null;
57     }
58     String title = matchSingleVCardPrefixedField("TITLE", rawText, true);
59     String url = matchSingleVCardPrefixedField("URL", rawText, true);
60     return new AddressBookParsedResult(names, null, phoneNumbers, emails, note, address, org,
61         birthday, title, url);
62   }
63
64   private static String[] matchVCardPrefixedField(String prefix, String rawText, boolean trim) {
65     Vector matches = null;
66     int i = 0;
67     int max = rawText.length();
68     while (i < max) {
69       i = rawText.indexOf(prefix, i);
70       if (i < 0) {
71         break;
72       }
73       if (i > 0 && rawText.charAt(i - 1) != '\n') {
74         // then this didn't start a new token, we matched in the middle of something
75         i++;
76         continue;
77       }
78       i += prefix.length(); // Skip past this prefix we found to start
79       if (rawText.charAt(i) != ':' && rawText.charAt(i) != ';') {
80         continue;
81       }
82       while (rawText.charAt(i) != ':') { // Skip until a colon
83         i++;
84       }
85       i++; // skip colon
86       int start = i; // Found the start of a match here
87       i = rawText.indexOf((int) '\n', i); // Really, ends in \r\n
88       if (i < 0) {
89         // No terminating end character? uh, done. Set i such that loop terminates and break
90         i = max;
91       } else if (i > start) {
92         // found a match
93         if (matches == null) {
94           matches = new Vector(3); // lazy init
95         }
96         String element = rawText.substring(start, i);
97         if (trim) {
98           element = element.trim();
99         }
100         matches.addElement(element);
101         i++;
102       } else {
103         i++;
104       }
105     }
106     if (matches == null || matches.isEmpty()) {
107       return null;
108     }
109     return toStringArray(matches);
110   }
111
112   static String matchSingleVCardPrefixedField(String prefix, String rawText, boolean trim) {
113     String[] values = matchVCardPrefixedField(prefix, rawText, trim);
114     return values == null ? null : values[0];
115   }
116
117   private static boolean isLikeVCardDate(String value) {
118     // Not really sure this is true but matches practice
119     // Mach YYYYMMDD
120     if (isStringOfDigits(value, 8)) {
121       return true;
122     }
123     // or YYYY-MM-DD
124     return
125         value.length() == 10 &&
126         value.charAt(4) == '-' &&
127         value.charAt(7) == '-' &&
128         isSubstringOfDigits(value, 0, 4) &&
129         isSubstringOfDigits(value, 5, 2) &&
130         isSubstringOfDigits(value, 8, 2);
131   }
132
133   private static String formatAddress(String address) {
134     if (address == null) {
135       return null;
136     }
137     int length = address.length();
138     StringBuffer newAddress = new StringBuffer(length);
139     for (int j = 0; j < length; j++) {
140       char c = address.charAt(j);
141       if (c == ';') {
142         newAddress.append(' ');
143       } else {
144         newAddress.append(c);
145       }
146     }
147     return newAddress.toString().trim();
148   }
149
150   /**
151    * Formats name fields of the form "Public;John;Q.;Reverend;III" into a form like
152    * "Reverend John Q. Public III".
153    *
154    * @param names name values to format, in place
155    */
156   private static void formatNames(String[] names) {
157     if (names != null) {
158       for (int i = 0; i < names.length; i++) {
159         String name = names[i];
160         String[] components = new String[5];
161         int start = 0;
162         int end;
163         int componentIndex = 0;
164         while ((end = name.indexOf(';', start)) > 0) {
165           components[componentIndex] = name.substring(start, end);
166           componentIndex++;
167           start = end + 1;
168         }
169         components[componentIndex] = name.substring(start);
170         StringBuffer newName = new StringBuffer(100);
171         maybeAppendComponent(components, 3, newName);
172         maybeAppendComponent(components, 1, newName);
173         maybeAppendComponent(components, 2, newName);
174         maybeAppendComponent(components, 0, newName);
175         maybeAppendComponent(components, 4, newName);
176         names[i] = newName.toString().trim();
177       }
178     }
179   }
180
181   private static void maybeAppendComponent(String[] components, int i, StringBuffer newName) {
182     if (components[i] != null) {
183       newName.append(' ');
184       newName.append(components[i]);
185     }
186   }
187
188 }