• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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_phones();
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_people());
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