1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
/*
* Copyright (C) 2009 Google Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.google.i18n.phonenumbers.tools;
import com.google.i18n.phonenumbers.Phonemetadata.PhoneMetadata;
import com.google.i18n.phonenumbers.Phonemetadata.PhoneMetadataCollection;
import java.io.BufferedWriter;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.util.Formatter;
import java.util.List;
import java.util.Map;
/**
* Tool to convert phone number metadata from the XML format to protocol buffer format.
*
* @author Shaopeng Jia
*/
public class BuildMetadataProtoFromXml extends Command {
private static final String PACKAGE_NAME = "com/google/i18n/phonenumbers";
private static final String META_DATA_FILE_PREFIX =
"/com/google/i18n/phonenumbers/data/PhoneNumberMetadataProto";
private static final String TEST_META_DATA_FILE_PREFIX =
"/com/google/i18n/phonenumbers/data/PhoneNumberMetadataProtoForTesting";
private static final String TEST_COUNTRY_CODE_TO_REGION_CODE_MAP_CLASS_NAME =
"CountryCodeToRegionCodeMapForTesting";
private static final String COUNTRY_CODE_TO_REGION_CODE_MAP_CLASS_NAME =
"CountryCodeToRegionCodeMap";
private static final String HELP_MESSAGE =
"Usage:\n" +
"BuildMetadataProtoFromXml <inputFile> <outputDir> <forTesting> [<liteBuild>]\n" +
"\n" +
"where:\n" +
" inputFile The input file containing phone number metadata in XML format.\n" +
" outputDir The output directory to store phone number metadata in proto\n" +
" format (one file per region) and the country code to region code\n" +
" mapping file.\n" +
" forTesting Flag whether to generate metadata for testing purposes or not.\n" +
" liteBuild Whether to generate the lite-version of the metadata (default:\n" +
" false). When set to true certain metadata will be omitted.\n" +
" At this moment, example numbers information is omitted.\n" +
"\n" +
"Metadata will be stored in:\n" +
" <outputDir>" + META_DATA_FILE_PREFIX + "_*\n" +
"Mapping file will be stored in:\n" +
" <outputDir>/" + PACKAGE_NAME + "/" +
COUNTRY_CODE_TO_REGION_CODE_MAP_CLASS_NAME + ".java\n" +
"\n" +
"Example command line invocation:\n" +
"BuildMetadataProtoFromXml PhoneNumberMetadata.xml src false false\n";
@Override
public String getCommandName() {
return "BuildMetadataProtoFromXml";
}
@Override
public boolean start() {
String[] args = getArgs();
if (args.length != 4 && args.length != 5) {
System.err.println(HELP_MESSAGE);
return false;
}
String inputFile = args[1];
String outputDir = args[2];
boolean forTesting = args[3].equals("true");
boolean liteBuild = args.length > 4 && args[4].equals("true");
String filePrefix;
if (forTesting) {
filePrefix = outputDir + TEST_META_DATA_FILE_PREFIX;
} else {
filePrefix = outputDir + META_DATA_FILE_PREFIX;
}
try {
PhoneMetadataCollection metadataCollection =
BuildMetadataFromXml.buildPhoneMetadataCollection(inputFile, liteBuild);
for (PhoneMetadata metadata : metadataCollection.getMetadataList()) {
String regionCode = metadata.getId();
PhoneMetadataCollection outMetadataCollection = new PhoneMetadataCollection();
outMetadataCollection.addMetadata(metadata);
FileOutputStream outputForRegion = new FileOutputStream(filePrefix + "_" + regionCode);
ObjectOutputStream out = new ObjectOutputStream(outputForRegion);
outMetadataCollection.writeExternal(out);
out.close();
}
Map<Integer, List<String>> countryCodeToRegionCodeMap =
BuildMetadataFromXml.buildCountryCodeToRegionCodeMap(metadataCollection);
writeCountryCallingCodeMappingToJavaFile(countryCodeToRegionCodeMap, outputDir, forTesting);
} catch (Exception e) {
System.err.println(HELP_MESSAGE);
return false;
}
return true;
}
private static final String MAPPING_IMPORTS =
"import java.util.ArrayList;\n" +
"import java.util.HashMap;\n" +
"import java.util.List;\n" +
"import java.util.Map;\n";
private static final String MAPPING_COMMENT =
" // A mapping from a country code to the region codes which denote the\n" +
" // country/region represented by that country code. In the case of multiple\n" +
" // countries sharing a calling code, such as the NANPA countries, the one\n" +
" // indicated with \"isMainCountryForCode\" in the metadata should be first.\n";
private static final double MAPPING_LOAD_FACTOR = 0.75;
private static final String MAPPING_COMMENT_2 =
" // The capacity is set to %d as there are %d different country codes,\n" +
" // and this offers a load factor of roughly " + MAPPING_LOAD_FACTOR + ".\n";
private static void writeCountryCallingCodeMappingToJavaFile(
Map<Integer, List<String>> countryCodeToRegionCodeMap,
String outputDir, boolean forTesting) throws IOException {
String mappingClassName;
if (forTesting) {
mappingClassName = TEST_COUNTRY_CODE_TO_REGION_CODE_MAP_CLASS_NAME;
} else {
mappingClassName = COUNTRY_CODE_TO_REGION_CODE_MAP_CLASS_NAME;
}
String mappingFile =
outputDir + "/" + PACKAGE_NAME.replaceAll("\\.", "/") + "/" + mappingClassName + ".java";
int capacity = (int) (countryCodeToRegionCodeMap.size() / MAPPING_LOAD_FACTOR);
BufferedWriter writer = new BufferedWriter(new FileWriter(mappingFile));
writer.write(CopyrightNotice.TEXT);
if (PACKAGE_NAME.length() > 0) {
writer.write("package " + PACKAGE_NAME + ";\n\n");
}
writer.write(MAPPING_IMPORTS);
writer.write("\n");
writer.write("public class " + mappingClassName + " {\n");
writer.write(MAPPING_COMMENT);
writer.write(" static Map<Integer, List<String>> getCountryCodeToRegionCodeMap() {\n");
Formatter formatter = new Formatter(writer);
formatter.format(MAPPING_COMMENT_2, capacity, countryCodeToRegionCodeMap.size());
writer.write(" Map<Integer, List<String>> countryCodeToRegionCodeMap =\n");
writer.write(" new HashMap<Integer, List<String>>(" + capacity + ");\n");
writer.write("\n");
writer.write(" ArrayList<String> listWithRegionCode;\n");
writer.write("\n");
for (Map.Entry<Integer, List<String>> entry : countryCodeToRegionCodeMap.entrySet()) {
int countryCallingCode = entry.getKey();
List<String> regionCodes = entry.getValue();
writer.write(" listWithRegionCode = new ArrayList<String>(" + regionCodes.size() + ");\n");
for (String regionCode : regionCodes) {
writer.write(" listWithRegionCode.add(\"" + regionCode + "\");\n");
}
writer.write(" countryCodeToRegionCodeMap.put(" + countryCallingCode +
", listWithRegionCode);\n");
writer.write("\n");
}
writer.write(" return countryCodeToRegionCodeMap;\n");
writer.write(" }\n");
writer.write("}\n");
writer.flush();
writer.close();
}
}
|
