OLD | NEW |
(Empty) | |
| 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. |
| 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. |
| 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 } |
OLD | NEW |