re-organize the source code

Author: jvmvik

oobt/Makefile

@@ -0,0 +1,5 @@
+all: build run
+
+build:
+
+run:

oobt/main.cpp

@@ -0,0 +1,59 @@
+#include <cstdint>
+#include <iostream>
+#include <memory>
+#include <string>
+#include <curl/curl.h>
+#include <serpapi.hpp>
+
+#include "rapidjson/document.h"
+#include "rapidjson/writer.h"
+#include "rapidjson/stringbuffer.h"
+#include "rapidjson/prettywriter.h"
+#include "rapidjson/pointer.h"
+
+using namespace rapidjson;
+using namespace std;
+
+void info(string msg) {
+    cout << "\nINFO: " << msg << endl;
+}
+void info(double msg) {
+    cout << "\nINFO: " << msg << endl;
+}
+
+void info(const Document& document) {
+    StringBuffer buffer;
+    PrettyWriter<StringBuffer> writer(buffer);
+    document.Accept(writer);
+    cout << "\nINFO: " << buffer.GetString();
+}
+
+// RapidJSON parser documentation is available: https://rapidjson.org
+int main()
+{
+    // initialize a client
+    const char* env_p = std::getenv("API_KEY");
+    std::string apiKey(env_p);
+    std::map<string, string> default_parameter;
+    default_parameter["api_key"] = apiKey;
+    default_parameter["engine"] = "google";
+    
+    // using namespace serpapi;
+    serpapi::Client client(default_parameter);
+
+    // execute search 
+    std::map<string, string> parameter;
+    parameter["q"] = "coffee";
+    parameter["location"] = "Austin,TX";
+    //  using namespace rapidjson;
+    rapidjson::Document d = client.search(parameter);
+    info("document loaded");
+    assert(d.HasMember("search_metadata"));
+    assert(d["search_metadata"]["status"] == "Success");
+    assert(d["search_metadata"]["id"].IsString());
+
+    string id = d["search_metadata"]["id"].GetString();
+    client.searchArchive(id);
+    assert(d["search_metadata"]["status"] == "Success");
+    info(d);
+}

src/callback.cpp

@@ -0,0 +1,16 @@
+#include <cstdint>
+#include <iostream>
+#include <memory>
+#include <string>
+
+using namespace std;
+
+namespace serpapi
+{
+ size_t callback(const char* in, size_t size, size_t num, string* out)
+ {
+  const size_t totalBytes(size*num);
+  out->append(in, totalBytes);
+  return totalBytes;
+ }
+}

src/callback.hpp

@@ -0,0 +1,11 @@
+#include <cstdint>
+#include <iostream>
+#include <memory>
+#include <string>
+
+using namespace std;
+
+namespace serpapi
+{
+ size_t callback(const char* in, size_t size, size_t num, string* out);
+}

src/serpapi.cpp

@@ -0,0 +1,115 @@
+/***
+ * Run search engine powered by SerpApi
+ */
+
+#include <serpapi.hpp>
+
+namespace serpapi 
+{
+
+const static string JSON_FORMAT = "json";
+const static string HTML_FORMAT = "html";
+const static string HOST = "https://serpapi.com";
+
+using namespace rapidjson;
+using namespace std;
+
+Client::Client(map<string, string> parameter)
+{
+   this->parameter = parameter;
+}
+
+/***
+* Get HTML search results
+*/
+string Client::html(map<string, string> parameter)
+{
+   return Client::get("/search", HTML_FORMAT, parameter);
+}
+
+Document Client::search(map<string, string> parameter) 
+{
+   return Client::json("/search", parameter);
+}
+
+Document Client::searchArchive(string searchID) 
+{
+   return Client::json("/searches/" + searchID + ".json", map<string, string>());
+}
+
+Document Client::account(map<string, string> parameter) 
+{
+   return Client::json("/account.json", parameter);
+}
+
+Document Client::location(map<string, string> parameter) 
+{
+   return Client::json("/locations.json", parameter);
+}
+
+Document Client::json(string uri, map<string, string> parameter)
+{
+   string buf = get(uri, JSON_FORMAT, parameter);
+   const char* json = buf.c_str();
+   Document d;
+   d.Parse(json);
+   return d;
+}
+
+string Client::url(string output, map<string, string> parameter)
+{
+   map<string, string> param(parameter);
+   string s = "";
+   map<string,string>::iterator it;
+   for (it = param.begin(); it != param.end(); ++it)
+   {
+      if (s != "")
+      {
+         s += "&";
+      }
+      // encode each value in case of special character
+      s += it->first + "=" + it->second;
+   }
+
+   // append output format
+   s += "&output=" + output;
+
+   // append source language
+   s += "&source=serpapi-cpp:1.0.0";
+
+   return s;
+}
+
+string Client::get(string uri, string output, map<string, string> parameter)
+{
+   curl_global_init(CURL_GLOBAL_DEFAULT);
+   curl = curl_easy_init();
+   const string url = HOST + uri + "?" + this->url(output, parameter);
+   cout << "url: " + url << endl;
+   curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
+   curl_easy_setopt(curl, CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4);
+   curl_easy_setopt(curl, CURLOPT_TIMEOUT, timeout);
+   curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1L);
+   curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 0L);
+   curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 0L);
+
+   long httpCode(0);
+   unique_ptr<string> httpData(new string());
+   curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, callback);
+   curl_easy_setopt(curl, CURLOPT_WRITEDATA, httpData.get());
+
+   // execute search
+   curl_easy_perform(curl);
+   curl_easy_getinfo(curl, CURLINFO_RESPONSE_CODE, &httpCode);
+   curl_easy_cleanup(curl);
+
+   if(httpCode != 200) {
+      // TODO what is the best solution in C++
+      return nullptr;
+   }
+   curl_global_cleanup();
+   return *httpData.get();
+}
+
+}
+

src/serpapi.hpp

@@ -0,0 +1,47 @@
+/***
+ * Run search engine powered by SerpApi
+ */
+#ifndef SERPAPI
+#define SERPAPI
+
+#include <iostream>
+#include <map>
+#include <string>
+#include <curl/curl.h>
+#include <callback.hpp>
+#include "rapidjson/document.h"
+
+namespace serpapi 
+{
+using namespace rapidjson;
+using namespace std;
+
+class Client {
+
+    map<string, string> parameter;
+    CURL *curl;
+    int timeout = 60;
+    
+    public:
+
+    Client(map<string, string> parameter);
+
+    Document search(map<string, string> parameter);
+
+    string html(map<string, string> parameter);
+
+    Document searchArchive(string searchID);
+
+    Document account(map<string, string> parameter);
+
+    Document location(map<string, string> parameter);
+
+    Document json(string uri, map<string, string> parameter);
+
+    string url(string output, map<string, string> parameter);
+
+    string get(string uri, string output, map<string, string> parameter);
+};
+}
+
+#endif