1 // See README.txt for information and build instructions.
2
3 #include <iostream>
4 #include <fstream>
5 #include <string>
6 #include "addressbook.pb.h"
7 using namespace std;
8
9 // This function fills in a Person message based on user input.
PromptForAddress(tutorial::Person * person)10 void PromptForAddress(tutorial::Person* person) {
11 cout << "Enter person ID number: ";
12 int id;
13 cin >> id;
14 person->set_id(id);
15 cin.ignore(256, '\n');
16
17 cout << "Enter name: ";
18 getline(cin, *person->mutable_name());
19
20 cout << "Enter email address (blank for none): ";
21 string email;
22 getline(cin, email);
23 if (!email.empty()) {
24 person->set_email(email);
25 }
26
27 while (true) {
28 cout << "Enter a phone number (or leave blank to finish): ";
29 string number;
30 getline(cin, number);
31 if (number.empty()) {
32 break;
33 }
34
35 tutorial::Person::PhoneNumber* phone_number = person->add_phone();
36 phone_number->set_number(number);
37
38 cout << "Is this a mobile, home, or work phone? ";
39 string type;
40 getline(cin, type);
41 if (type == "mobile") {
42 phone_number->set_type(tutorial::Person::MOBILE);
43 } else if (type == "home") {
44 phone_number->set_type(tutorial::Person::HOME);
45 } else if (type == "work") {
46 phone_number->set_type(tutorial::Person::WORK);
47 } else {
48 cout << "Unknown phone type. Using default." << endl;
49 }
50 }
51 }
52
53 // Main function: Reads the entire address book from a file,
54 // adds one person based on user input, then writes it back out to the same
55 // file.
main(int argc,char * argv[])56 int main(int argc, char* argv[]) {
57 // Verify that the version of the library that we linked against is
58 // compatible with the version of the headers we compiled against.
59 GOOGLE_PROTOBUF_VERIFY_VERSION;
60
61 if (argc != 2) {
62 cerr << "Usage: " << argv[0] << " ADDRESS_BOOK_FILE" << endl;
63 return -1;
64 }
65
66 tutorial::AddressBook address_book;
67
68 {
69 // Read the existing address book.
70 fstream input(argv[1], ios::in | ios::binary);
71 if (!input) {
72 cout << argv[1] << ": File not found. Creating a new file." << endl;
73 } else if (!address_book.ParseFromIstream(&input)) {
74 cerr << "Failed to parse address book." << endl;
75 return -1;
76 }
77 }
78
79 // Add an address.
80 PromptForAddress(address_book.add_person());
81
82 {
83 // Write the new address book back to disk.
84 fstream output(argv[1], ios::out | ios::trunc | ios::binary);
85 if (!address_book.SerializeToOstream(&output)) {
86 cerr << "Failed to write address book." << endl;
87 return -1;
88 }
89 }
90
91 // Optional: Delete all global objects allocated by libprotobuf.
92 google::protobuf::ShutdownProtobufLibrary();
93
94 return 0;
95 }
96