#include <iostream>
#include <list>

#include "json/json.h"

#include "webcc/http_client_session.h"
#include "webcc/logger.h"

#include "examples/common/book.h"
#include "examples/common/book_json.h"

#if (defined(WIN32) || defined(_WIN64))
#if defined(_DEBUG) && defined(WEBCC_ENABLE_VLD)
#pragma message ("< include vld.h >")
#include "vld/vld.h"
#pragma comment(lib, "vld")
#endif
#endif

// -----------------------------------------------------------------------------

class BookClientBase {
public:
  BookClientBase(const std::string& url, int timeout_seconds) : url_(url) {
    //session_.SetTimeout(timeout_seconds);

    //session_.set_content_type("application/json");
    session_.set_charset("utf-8");
  }

  virtual ~BookClientBase() = default;

protected:
  // Helper function to make a request.
  //webcc::HttpRequestPtr MakeRequest(const std::string& method,
  //                                  const std::string& url,
  //                                  std::string&& content = "") {
  //  auto request = webcc::HttpRequest::New(method, url, host_, port_);
  //  request->AcceptAppJson();
  //  if (!content.empty()) {
  //    request->SetContentInAppJsonUtf8(JsonToString(content), true);
  //  }
  //  request->Prepare();
  //  return request;
  //}

  // Check HTTP response status.
  bool CheckStatus(webcc::HttpResponsePtr response, int expected_status) {
    int status = response->status();
    if (status != expected_status) {
      LOG_ERRO("HTTP status error (actual: %d, expected: %d).",
               status, expected_status);
      return false;
    }
    return true;
  }

protected:
  std::string url_;

  webcc::HttpClientSession session_;
};

// -----------------------------------------------------------------------------

class BookListClient : public BookClientBase {
public:
  BookListClient(const std::string& url, int timeout_seconds)
      : BookClientBase(url, timeout_seconds) {
  }

  bool ListBooks(std::list<Book>* books) {
    try {
      auto r = session_.Get(url_ + "/books");
      
      if (!CheckStatus(r, webcc::http::Status::kOK)) {
        // Response HTTP status error.
        return false;
      }

      Json::Value rsp_json = StringToJson(r->content());

      if (!rsp_json.isArray()) {
        return false;  // Should be a JSON array of books.
      }

      for (Json::ArrayIndex i = 0; i < rsp_json.size(); ++i) {
        books->push_back(JsonToBook(rsp_json[i]));
      }

      return true;

    } catch (const webcc::Exception& e) {
      std::cerr << e.what() << std::endl;
      return false;
    }
  }

  bool CreateBook(const std::string& title, double price, std::string* id) {
    Json::Value req_json;
    req_json["title"] = title;
    req_json["price"] = price;

    try {
      auto r = session_.Post(url_ + "/books", JsonToString(req_json), true);

      if (!CheckStatus(r, webcc::http::Status::kCreated)) {
        return false;
      }

      Json::Value rsp_json = StringToJson(r->content());
      *id = rsp_json["id"].asString();

      return !id->empty();

    } catch (const webcc::Exception& e) {
      std::cerr << e.what() << std::endl;
      return false;
    }
  }
};

// -----------------------------------------------------------------------------

class BookDetailClient : public BookClientBase {
public:
  BookDetailClient(const std::string& url, int timeout_seconds)
      : BookClientBase(url, timeout_seconds) {
  }

  bool GetBook(const std::string& id, Book* book) {
    try {
      auto r = session_.Get(url_ + "/books" + id);

      if (!CheckStatus(r, webcc::http::Status::kOK)) {
        return false;
      }

      return JsonStringToBook(r->content(), book);

    } catch (const webcc::Exception& e) {
      std::cerr << e.what() << std::endl;
      return false;
    }
  }

  bool UpdateBook(const std::string& id, const std::string& title,
                  double price) {
    Json::Value json;
    json["title"] = title;
    json["price"] = price;

    try {
      auto r = session_.Put(url_ + "/books" + id, JsonToString(json), true);

      if (!CheckStatus(r, webcc::http::Status::kOK)) {
        return false;
      }

      return true;

    } catch (const webcc::Exception& e) {
      std::cerr << e.what() << std::endl;
      return false;
    }
  }

  bool DeleteBook(const std::string& id) {
    try {
      auto r = session_.Delete(url_ + "/books/" + id);

      if (!CheckStatus(r, webcc::http::Status::kOK)) {
        return false;
      }

      return true;

    } catch (const webcc::Exception& e) {
      std::cerr << e.what() << std::endl;
      return false;
    }
  }
};

// -----------------------------------------------------------------------------

void PrintSeparator() {
  std::cout << std::string(80, '-') << std::endl;
}

void PrintBook(const Book& book) {
  std::cout << "Book: " << book << std::endl;
}

void PrintBookList(const std::list<Book>& books) {
  std::cout << "Book list: " << books.size() << std::endl;
  for (const Book& book : books) {
    std::cout << "  Book: " << book << std::endl;
  }
}

// -----------------------------------------------------------------------------

void Help(const char* argv0) {
  std::cout << "Usage: " << argv0 << " <url> [timeout]" << std::endl;
  std::cout << "  E.g.," << std::endl;
  std::cout << "    " << argv0 << "http://localhost:8080" << std::endl;
  std::cout << "    " << argv0 << "http://localhost:8080 2" << std::endl;
}

int main(int argc, char* argv[]) {
  if (argc < 2) {
    Help(argv[0]);
    return 1;
  }

  WEBCC_LOG_INIT("", webcc::LOG_CONSOLE_FILE_OVERWRITE);

  std::string url = argv[1];

  int timeout_seconds = -1;
  if (argc > 2) {
    timeout_seconds = std::atoi(argv[2]);
  }

  BookListClient list_client(url, timeout_seconds);
  BookDetailClient detail_client(url, timeout_seconds);

  PrintSeparator();

  std::list<Book> books;
  if (list_client.ListBooks(&books)) {
    PrintBookList(books);
  }

  //PrintSeparator();

  //std::string id;
  //if (list_client.CreateBook("1984", 12.3, &id)) {
  //  std::cout << "Book ID: " << id << std::endl;
  //} else {
  //  id = "1";
  //  std::cout << "Book ID: " << id << " (faked)"<< std::endl;
  //}

  //PrintSeparator();

  //books.clear();
  //if (list_client.ListBooks(&books)) {
  //  PrintBookList(books);
  //}

  //PrintSeparator();

  //Book book;
  //if (detail_client.GetBook(id, &book)) {
  //  PrintBook(book);
  //}

  //PrintSeparator();

  //detail_client.UpdateBook(id, "1Q84", 32.1);

  //PrintSeparator();

  //if (detail_client.GetBook(id, &book)) {
  //  PrintBook(book);
  //}

  //PrintSeparator();

  //detail_client.DeleteBook(id);

  //PrintSeparator();

  //books.clear();
  //if (list_client.ListBooks(&books)) {
  //  PrintBookList(books);
  //}

  return 0;
}