• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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.
PromptForAddress(BufferedReader stdin, PrintStream stdout)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.
main(String[] args)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 }
96