lay/toREST/include/torrents.hpp

#ifndef _TR_TORRENTS_HPP_
#define _TR_TORRENTS_HPP_
#include <http.hpp>
#include <libtorrent/magnet_uri.hpp>
#include <util.hpp>

namespace tr {
namespace session {
namespace torrents {
template <class torrent_session, class request, class response>
void get(torrent_session &session, response resp, request req) {
  auto http_response = http::response();
  if (session.is_valid()) {
    auto torrents = session.get_torrents();
    auto response_object = nlohmann::json::object();
    auto torrents_json = nlohmann::json::array();
    for (auto &torrent : torrents) {
      if (torrent.is_valid()) {
        const auto hash = torrent.info_hash().to_string();
        const auto status = torrent.status(
            torrent.query_name |
            torrent.query_save_path);
        torrents_json.push_back({{"hash", hash},
                                 {"paused", status.paused},
                                 {"seeding", status.is_seeding},
                                 {"state", status.state},
                                 {"priority", status.priority},
                                 {"name", status.name}});
      }
    }
    response_object["torrents"] = torrents_json;
    http_response.set_status(http::ok);
    http_response.set_body(response_object);
  } else {
    auto response_code = http::code(http::service_unavailable);
    http_response.set_body({{"code", response_code.first}, {"status", response_code.second}});
    http_response.set_status(response_code.first);
  }
  *resp << http_response;
}
template <class settings, class torrent_session, class request, class response>
void post(settings opts, torrent_session &session, response resp, request req) {
  auto http_response = http::response();
  const auto respond = [&](http::status status) {
    const auto response_code = http::code(status);
    http_response.set_body({{"code", response_code.first}, {"status", response_code.second}});
    http_response.set_status(response_code.first);
    *resp << http_response;
  };

  if (!session.is_valid()) {
    return respond(http::service_unavailable);
  }

  auto request_object = util::json::parse(req->content);

  if (request_object.is_null()) {
    return respond(http::bad_request);
  }

  if (!request_object.is_object()) {
    return respond(http::bad_request);
  }

  if (request_object.find("magnet_uri") == request_object.end()) {
    return respond(http::bad_request);
  }

  std::string uri = request_object.at("magnet_uri");
  boost::system::error_code ec;
  libtorrent::add_torrent_params params;
  libtorrent::parse_magnet_uri(uri, params, ec);
  auto magnet = util::uri::parse(uri);
  if (ec) {
    return respond(http::bad_request);
  }

  params.save_path = opts.default_download_dir.string();


  http_response.add_header({"Location", "/session/torrents/hash"});

  return respond(http::created);

  // if(ec){
  //   auto response_code = http::code(http::bad_request);
  //   http_response.set_body({{"code", response_code.first}, {"status", response_code.second + ec.message()}});
  //   http_response.set_status(response_code.first);
  // }

  // if(request_object.find("save_path") != request_object.end())
  //   params.save_path=request_object.at("save_path");
  // else
  //   params.save_path=opts.default_download_dir.string();
  // session.async_add_torrent(params);
  // http_response.add_header({"Location",params.info_hash.to_string()});
}
}
}
}
#endif