stig

#include "stig/stig.hh"

#include "stig/lang.hh"

#include "ob/term.hh"
namespace aec = OB::Term::ANSI_Escape_Codes;

#include "ob/belle.hh"
namespace Belle = OB::Belle;

#include "lib/json.hh"
using Json = nlohmann::json;

#include <unistd.h>

#include <cstddef>
#include <cmath>
#include <ctime>

#include <string>
#include <iostream>
#include <iomanip>
#include <vector>
#include <utility>
#include <optional>

namespace OB::Stig
{

// prototypes
void on_error(std::string const& str);
void on_http_error(Belle::Client& app);
std::pair<std::string, std::string> fuzzy_time(long int const sec);
void ostream_fmt(OB::Term::ostream& os, bool const is_term, bool const is_color);
void search_print(Json const& js, std::pair<int, int> rate, std::size_t page,
  std::size_t per_page, std::string const& color);

void on_error(std::string const& str)
{
  throw std::runtime_error(
    str +
    "\nview the help output with '-h'"
  );
}

void on_http_error(Belle::Client& app)
{
  app.on_http_error([&](auto& ctx)
  {
    on_error(ctx.ec.message());
  });
}

template<typename T>
std::optional<T> json_value(Json const& js)
{
  if (js.is_null())
  {
    return {};
  }
  return js.get<T>();
}

std::pair<std::string, std::string> fuzzy_time(long int const sec)
{
  std::pair<std::string, std::string> res;

  long int constexpr t_second {1};
  long int constexpr t_minute {t_second * 60};
  long int constexpr t_hour   {t_minute * 60};
  long int constexpr t_day    {t_hour * 24};
  long int constexpr t_week   {t_day * 7};
  long int constexpr t_month  (t_day * 30.4);
  long int constexpr t_year   {t_month * 12};

  std::time_t const now {std::time(nullptr)};
  long int const dif {now - sec};

  auto const fuzzy_string = [&](long int const time_ref, std::string const time_str)
  {
    long int const fuzzy (dif / time_ref);

    res.first = std::to_string(fuzzy);
    res.second = time_str;
  };

  if (dif >= t_year)
  {
    fuzzy_string(t_year, "Y");
  }
  else if (dif >= t_month)
  {
    fuzzy_string(t_month, "M");
  }
  else if (dif >= t_week)
  {
    fuzzy_string(t_week, "W");
  }
  else if (dif >= t_day)
  {
    fuzzy_string(t_day, "D");
  }
  else if (dif >= t_hour)
  {
    fuzzy_string(t_hour, "h");
  }
  else if (dif >= t_minute)
  {
    fuzzy_string(t_minute, "m");
  }
  else if (dif >= t_second)
  {
    fuzzy_string(t_second, "s");
  }
  else
  {
    res.first = "0";
    res.second = "s";
  }

  return res;
}

void search(std::string const& host, std::string const& query,
  std::string const& sort, std::string const& order, std::size_t page,
  std::size_t per_page, std::string const& token, std::string const& color)
{
  Json js;
  std::pair<int, int> rate;

  Belle::Client app {host, 443, true};
  on_http_error(app);

  Belle::Request req;
  req.method(Belle::Method::get);

  if (host == "api.github.com")
  {
    req.target("/search/repositories");
  }
  else
  {
    req.target("/api/v3/search/repositories");
  }

  req.set(Belle::Header::accept, "application/vnd.github.mercy-preview+json");

  req.params().emplace("q", query);
  req.params().emplace("sort", sort);
  req.params().emplace("order", order);
  req.params().emplace("page", std::to_string(page));
  req.params().emplace("per_page", std::to_string(per_page));

  if (! token.empty())
  {
    req.params().emplace("access_token", token);
  }

  app.on_http(req.move(), [&js, &rate](auto& ctx)
  {
    if (ctx.res.result() != Belle::Status::ok)
    {
      std::stringstream ss; ss
      << "HTTP "
      << "'" << ctx.res.result_int() << "'"
      << " " << ctx.res.reason();
      on_error(ss.str());
      return;
    }

    rate.first = std::stoi(std::string(ctx.res.at("X-RateLimit-Remaining")));
    rate.second = std::stoi(std::string(ctx.res.at("X-RateLimit-Limit")));
    js = Json::parse(std::move(ctx.res.body()));
  });

  auto const total = app.queue().size();
  auto const completed = app.connect();

  if (completed != total)
  {
    on_error("an HTTP error occurred");
  }

  search_print(js, rate, page, per_page, color);
}

void ostream_fmt(OB::Term::ostream& os, bool const is_term, bool const is_color)
{
  os.indent(4);
  os.first_wrap(false);
  os.white_space(false);
  os.escape_codes(is_color);

  if (is_term)
  {
    os.width(OB::Term::width(STDOUT_FILENO));
  }
  else
  {
    os.line_wrap(false);
  }
}

void search_print(Json const& js, std::pair<int, int> rate, std::size_t page,
  std::size_t per_page, std::string const& color)
{
  // is stdout a tty
  bool is_term {OB::Term::is_term(STDOUT_FILENO)};

  // color auto
  bool is_color {is_term};
  if (color == "on")
  {
    // color on
    is_color = true;
  }
  else if (color == "off")
  {
    // color off
    is_color = false;
  }

  // output formatter
  OB::Term::ostream out {std::cout};
  ostream_fmt(out, is_term, is_color);

  for (auto const& e : js["items"])
  {
    auto const owner = e["owner"]["login"].get<std::string>();
    auto const repo = e["name"].get<std::string>();
    auto const stars = e["stargazers_count"].get<std::size_t>();
    auto const fork = e["fork"].get<bool>();
    auto const forks = e["forks_count"].get<std::size_t>();
    auto const issues = e["open_issues_count"].get<std::size_t>();
    auto const lang = json_value<std::string>(e["language"]).value_or("");
    auto desc = json_value<std::string>(e["description"]).value_or("");

    if (desc.size() > 256)
    {
      // limit repo description size to 256 chars
      desc.erase(253);
      desc.append("...");
    }

    std::string const fork_symbol {fork ? ">" : "<"};

    std::pair<std::string, std::string> updated;
    std::tm t = {};
    std::stringstream ss (e["pushed_at"].get<std::string>());
    if (ss >> std::get_time(&t, "%Y-%m-%dT%H:%M:%SZ"))
    {
      updated = fuzzy_time(std::mktime(&t));
    }

    out
    << aec::wrap(owner, {aec::fg_magenta, aec::bold}, is_color)
    << "/"
    << aec::wrap(repo, {aec::fg_white, aec::bold}, is_color)
    << " *"
    << aec::wrap(stars, {aec::fg_green, aec::bold}, is_color)
    << " " << fork_symbol
    << aec::wrap(forks, {aec::fg_blue, aec::bold}, is_color)
    << " !"
    << aec::wrap(issues, {aec::fg_cyan, aec::bold}, is_color)
    << " ["
    << aec::wrap(lang, {aec::fg_true(Lang::color(lang)), aec::bold}, is_color)
    << "] "
    << aec::wrap(updated.first, {aec::fg_yellow, aec::bold}, is_color)
    << aec::wrap(updated.second, {aec::fg_yellow, aec::bold}, is_color)
    << "\n";

    if (! desc.empty())
    {
      out
      << OB::Term::iomanip::push()
      << desc
      << OB::Term::iomanip::pop();
    }
  }

  auto const total_results = js["total_count"].get<std::size_t>();
  auto const total_pages = static_cast<std::size_t>(std::ceil(static_cast<double>(total_results) / per_page));

  if (total_pages == 0)
  {
    page = 0;
  }

  std::size_t begin_results {js["items"].size() ? 1ul : 0ul};
  std::size_t end_results {per_page > total_results ? total_results : per_page};

  if (begin_results && page > 1)
  {
    end_results = page * per_page;
    begin_results = end_results - per_page + 1;

    if (end_results > total_results)
    {
      end_results = total_results;
      begin_results = total_results - (total_results % per_page);
    }

    if (page == total_pages)
    {
      ++begin_results;
    }
  }
  else if (! begin_results)
  {
    end_results = 0;
  }

  out
  << "\n"
  << aec::wrap(begin_results, {aec::fg_magenta, aec::bold}, is_color)
  << "-"
  << aec::wrap(end_results, {aec::fg_magenta, aec::bold}, is_color)
  << "/"
  << aec::wrap(total_results, {aec::fg_white, aec::bold}, is_color)
  << " results | "
  << aec::wrap(page, {aec::fg_magenta, aec::bold}, is_color)
  << "/"
  << aec::wrap(total_pages, {aec::fg_white, aec::bold}, is_color)
  << " pages | "
  << aec::wrap(rate.first, {aec::fg_magenta, aec::bold}, is_color)
  << "/"
  << aec::wrap(rate.second, {aec::fg_white, aec::bold}, is_color)
  << " limit"
  << "\n";
}

void readme(std::string const& host, std::string const& repo, std::string const& ref)
{
  std::string res;

  Belle::Client app {host, 443, true};
  on_http_error(app);

  Belle::Request req;
  req.method(Belle::Method::get);

  if (host == "api.github.com")
  {
    req.target("/repos/" + repo + "/readme");
  }
  else
  {
    req.target("/api/v3/repos/" + repo + "/readme");
  }

  req.set(Belle::Header::accept, "application/vnd.github.VERSION.raw");

  if (! ref.empty())
  {
    req.params().emplace("ref", ref);
  }

  app.on_http(req.move(), [&res](auto& ctx)
  {
    if (ctx.res.result() != Belle::Status::ok)
    {
      std::stringstream ss; ss
      << "HTTP "
      << "'" << ctx.res.result_int() << "'"
      << " " << ctx.res.reason();
      on_error(ss.str());
      return;
    }

    res = std::move(ctx.res.body());
  });

  auto const total = app.queue().size();
  auto const completed = app.connect();

  if (completed != total)
  {
    on_error("an HTTP error occurred");
  }

  std::cout << res << std::flush;
}

} // namespace OB::Stig