list_people.cpp

// Copyright 2018 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include <fstream>
#include <google/protobuf/util/time_util.h>
#include <iostream>
#include <string>

#include "addressbook.pb.h"
#include "ListPeople.h"
#include <unistd.h>

using namespace std;
using google::protobuf::util::TimeUtil;

// Iterates through all people in the AddressBook and prints info about them.
void ListPeople(const tutorial::AddressBook& address_book) {
  cout << "Reading AddressBook Proto using C++\n";
  // Loop through all persons in address book
  for (int i = 0; i < address_book.people_size(); i++) {
    const tutorial::Person& person = address_book.people(i);
    cout << "Person ID: " << person.id() << endl;
    cout << "  Name: " << person.name() << endl;
    if (person.email() != "") {
      cout << "  E-mail address: " << person.email() << endl;
    }

    // Loop through all phone numbers
    for (int j = 0; j < person.phones_size(); j++) {
      const tutorial::Person::PhoneNumber& phone_number = person.phones(j);
      
      switch (phone_number.type()) {
        case tutorial::Person::MOBILE:
          cout << "  Mobile phone #: ";
          break;
        case tutorial::Person::HOME:
          cout << "  Home phone #: ";
          break;
        case tutorial::Person::WORK:
          cout << "  Work phone #: ";
          break;
        default:
          cout << "  Unknown phone #: ";
          break;
      	}
      	cout << phone_number.number() << endl;
      }

      if (person.has_last_updated()) {
      	cout << "  Updated: " << TimeUtil::ToString(person.last_updated()) << endl;
      }
  }
}

// Prompt to get new person record in address book.
void PromptForAddress(tutorial::Person* person) {
  cout <<"\n\n\nEnter new  person information"<<endl;
  cout << "Enter person ID number: ";
  int id;
  cin >> id;
  person->set_id(id);
  cin.ignore(256, '\n');

  cout << "Enter name: ";
  getline(cin, *person->mutable_name());

  cout << "Enter email address (blank for none): ";
  string email;
  getline(cin, email);
  if (!email.empty()) {
    person->set_email(email);
  }

   while (true) {
     cout << "Enter a phone number (or leave blank to finish): ";
     string number;
     getline(cin, number);
		
     if (number.empty()) {
       break;
     }

     tutorial::Person::PhoneNumber* phone_number = person->add_phones();
     phone_number->set_number(number);

     cout << "Is this a mobile, home, or work phone? ";
     string type;
     getline(cin, type);
     if (type == "mobile") {
       phone_number->set_type(tutorial::Person::MOBILE);
     } else if (type == "home") {
       phone_number->set_type(tutorial::Person::HOME);
     } else if (type == "work") {
       phone_number->set_type(tutorial::Person::WORK);
     } else {
       cout << "Unknown phone type.  Using default." << endl;
     }
  }  
  *person->mutable_last_updated() = TimeUtil::SecondsToTimestamp(time(NULL));
}

// initialize proto
void initProto(){
  //Verify that the version of the library that we linksed against is
  //compatible with the version of the headers we compiled agaist.
  GOOGLE_PROTOBUF_VERIFY_VERSION;
}

// This method reads proto from a file name passed as an argument.
JNIEXPORT void JNICALL Java_ListPeople_readProtoNative
  (JNIEnv *env, jclass thisClass,jstring jFilePath){
  cout << "C++ reading of Proto from file.\n";
  initProto();
  const char *fileStr = env->GetStringUTFChars(jFilePath, NULL);

  cout <<"File name provided by Java: "<<fileStr <<endl;

  tutorial::AddressBook address_book;
  {
     // Read the existing address book.
     fstream input(fileStr, ios::in | ios::binary);
		
     if (!address_book.ParseFromIstream(&input)) {
       cerr << "Failed to parse address book." << endl;
       exit(0);
     }
  }
  env->ReleaseStringUTFChars(jFilePath, fileStr);  // release resources
  // List address book
  ListPeople(address_book);
}

// This method reads byte array from java and converts it to Protobuf object.
JNIEXPORT void JNICALL Java_ListPeople_readProtoNativeByteArray
  (JNIEnv *env, jclass thisClass, jbyteArray bArray){
  cout << "C++ reading of Proto from ByteArray.\n";
  initProto();
  tutorial::AddressBook address_book;
	  
  jbyte *bufferElems = env->GetByteArrayElements(bArray, 0);
  int len = env->GetArrayLength(bArray);
	  
  try {
    address_book.ParseFromArray(reinterpret_cast<unsigned char*>(bufferElems),len);
    // handle message here
  } catch (...) {}
  env->ReleaseByteArrayElements(bArray, bufferElems, JNI_ABORT);
  ListPeople(address_book);
}

// This method first reads byte array from java and converts to Protobuf object.
// Then makes modifications to Protobuf and sends byte array to java.
JNIEXPORT jbyteArray JNICALL Java_ListPeople_readProtoNativeByteArrayAndModify
  (JNIEnv *env, jclass thisClass, jbyteArray bArray){
  cout << "C++ reading and modification of Proto from ByteArray.\n";
  initProto();
  tutorial::AddressBook address_book;
	  
  jbyte *bufferElems = env->GetByteArrayElements(bArray, 0);
  int len = env->GetArrayLength(bArray);
	  
  try {
    address_book.ParseFromArray(reinterpret_cast<unsigned char*>(bufferElems),len);
    ListPeople(address_book);
  } catch (...) {}
  env->ReleaseByteArrayElements(bArray, bufferElems, JNI_ABORT);

  cout<<"\n\nLet's add objects to Proto";
  PromptForAddress(address_book.add_people());
  
  // Finally convert Proto to bytes and return to java
  int size = address_book.ByteSize();
  char *ret = new char[size];
  address_book.SerializeToArray(ret,size);

  jbyteArray array = env->NewByteArray(size);
  env->SetByteArrayRegion( array, 0, size, (const jbyte*)ret );

  delete[] ret;
  return array;
}

// Main function:  Reads the entire address book from a file and prints all
// the information inside.
int main(int argc, char* argv[]) {
  // Verify that the version of the library that we linked against is
  // compatible with the version of the headers we compiled against.
  GOOGLE_PROTOBUF_VERIFY_VERSION;

  if (argc != 2) {
    cerr << "Usage:  " << argv[0] << " ADDRESS_BOOK_FILE" << endl;
    return -1;
  }

  tutorial::AddressBook address_book;
  {
     // Read the existing address book.
     fstream input(argv[1], ios::in | ios::binary);
     if (!input) {	
       cout << argv[1] << ": File not found.  Creating a new file." << endl;
     } else if (!address_book.ParseFromIstream(&input)) {
       cerr << "Failed to parse address book." << endl;
       return -1;
     }
  }

  // Add an address.
  PromptForAddress(address_book.add_people());
  {
     // Write the new address book back to disk.
     fstream output(argv[1], ios::out | ios::trunc | ios::binary);
     if (!address_book.SerializeToOstream(&output)) {
       cerr << "Failed to write address book." << endl;
       return -1;
     }
  }

  ListPeople(address_book);

  // Optional:  Delete all global objects allocated by libprotobuf.
  google::protobuf::ShutdownProtobufLibrary();

  return 0;
}