1// See README.txt for information and build instructions.
2
3#include <iostream>
4#include <fstream>
5#include <string>
6#include "addressbook.pb.h"
7using namespace std;
8
9// This function fills in a Person message based on user input.
10void 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.
56int 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