OLD | NEW |
| (Empty) |
1 // See README.txt for information and build instructions. | |
2 | |
3 import com.example.tutorial.AddressBookProtos.AddressBook; | |
4 import com.example.tutorial.AddressBookProtos.Person; | |
5 import java.io.BufferedReader; | |
6 import java.io.FileInputStream; | |
7 import java.io.FileNotFoundException; | |
8 import java.io.FileOutputStream; | |
9 import java.io.InputStreamReader; | |
10 import java.io.IOException; | |
11 import java.io.PrintStream; | |
12 | |
13 class AddPerson { | |
14 // This function fills in a Person message based on user input. | |
15 static Person PromptForAddress(BufferedReader stdin, | |
16 PrintStream stdout) throws IOException { | |
17 Person.Builder person = Person.newBuilder(); | |
18 | |
19 stdout.print("Enter person ID: "); | |
20 person.setId(Integer.valueOf(stdin.readLine())); | |
21 | |
22 stdout.print("Enter name: "); | |
23 person.setName(stdin.readLine()); | |
24 | |
25 stdout.print("Enter email address (blank for none): "); | |
26 String email = stdin.readLine(); | |
27 if (email.length() > 0) { | |
28 person.setEmail(email); | |
29 } | |
30 | |
31 while (true) { | |
32 stdout.print("Enter a phone number (or leave blank to finish): "); | |
33 String number = stdin.readLine(); | |
34 if (number.length() == 0) { | |
35 break; | |
36 } | |
37 | |
38 Person.PhoneNumber.Builder phoneNumber = | |
39 Person.PhoneNumber.newBuilder().setNumber(number); | |
40 | |
41 stdout.print("Is this a mobile, home, or work phone? "); | |
42 String type = stdin.readLine(); | |
43 if (type.equals("mobile")) { | |
44 phoneNumber.setType(Person.PhoneType.MOBILE); | |
45 } else if (type.equals("home")) { | |
46 phoneNumber.setType(Person.PhoneType.HOME); | |
47 } else if (type.equals("work")) { | |
48 phoneNumber.setType(Person.PhoneType.WORK); | |
49 } else { | |
50 stdout.println("Unknown phone type. Using default."); | |
51 } | |
52 | |
53 person.addPhones(phoneNumber); | |
54 } | |
55 | |
56 return person.build(); | |
57 } | |
58 | |
59 // Main function: Reads the entire address book from a file, | |
60 // adds one person based on user input, then writes it back out to the same | |
61 // file. | |
62 public static void main(String[] args) throws Exception { | |
63 if (args.length != 1) { | |
64 System.err.println("Usage: AddPerson ADDRESS_BOOK_FILE"); | |
65 System.exit(-1); | |
66 } | |
67 | |
68 AddressBook.Builder addressBook = AddressBook.newBuilder(); | |
69 | |
70 // Read the existing address book. | |
71 try { | |
72 FileInputStream input = new FileInputStream(args[0]); | |
73 try { | |
74 addressBook.mergeFrom(input); | |
75 } finally { | |
76 try { input.close(); } catch (Throwable ignore) {} | |
77 } | |
78 } catch (FileNotFoundException e) { | |
79 System.out.println(args[0] + ": File not found. Creating a new file."); | |
80 } | |
81 | |
82 // Add an address. | |
83 addressBook.addPeople( | |
84 PromptForAddress(new BufferedReader(new InputStreamReader(System.in)), | |
85 System.out)); | |
86 | |
87 // Write the new address book back to disk. | |
88 FileOutputStream output = new FileOutputStream(args[0]); | |
89 try { | |
90 addressBook.build().writeTo(output); | |
91 } finally { | |
92 output.close(); | |
93 } | |
94 } | |
95 } | |
OLD | NEW |