94 lines
2.3 KiB
C++
94 lines
2.3 KiB
C++
// Copyright Andrew Resch 2008. Use, modification and distribution is
|
|
// subject to the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt
|
|
|
|
#include "boost_python.hpp"
|
|
#include "bytes.hpp"
|
|
#include <libtorrent/session.hpp>
|
|
#include <libtorrent/torrent.hpp>
|
|
#include <libtorrent/magnet_uri.hpp>
|
|
#include "gil.hpp"
|
|
#include "bytes.hpp"
|
|
|
|
using namespace boost::python;
|
|
using namespace lt;
|
|
|
|
extern void dict_to_add_torrent_params(dict params, add_torrent_params& p);
|
|
|
|
namespace {
|
|
|
|
#if TORRENT_ABI_VERSION == 1
|
|
torrent_handle _add_magnet_uri(lt::session& s, std::string uri, dict params)
|
|
{
|
|
add_torrent_params p;
|
|
|
|
dict_to_add_torrent_params(params, p);
|
|
|
|
allow_threading_guard guard;
|
|
|
|
p.url = uri;
|
|
|
|
#ifndef BOOST_NO_EXCEPTIONS
|
|
return s.add_torrent(p);
|
|
#else
|
|
error_code ec;
|
|
return s.add_torrent(p, ec);
|
|
#endif
|
|
}
|
|
#endif
|
|
|
|
dict parse_magnet_uri_dict(std::string const& uri)
|
|
{
|
|
error_code ec;
|
|
add_torrent_params p = parse_magnet_uri(uri, ec);
|
|
|
|
if (ec) throw system_error(ec);
|
|
|
|
dict ret;
|
|
|
|
if (p.ti) ret["ti"] = p.ti;
|
|
list tracker_list;
|
|
for (std::vector<std::string>::const_iterator i = p.trackers.begin()
|
|
, end(p.trackers.end()); i != end; ++i)
|
|
tracker_list.append(*i);
|
|
ret["trackers"] = tracker_list;
|
|
|
|
list nodes_list;
|
|
for (auto const& i : p.dht_nodes)
|
|
nodes_list.append(boost::python::make_tuple(i.first, i.second));
|
|
ret["dht_nodes"] = nodes_list;
|
|
ret["info_hash"] = bytes(p.info_hash.to_string());
|
|
ret["name"] = p.name;
|
|
ret["save_path"] = p.save_path;
|
|
ret["storage_mode"] = p.storage_mode;
|
|
#if TORRENT_ABI_VERSION == 1
|
|
ret["url"] = p.url;
|
|
ret["uuid"] = p.uuid;
|
|
#endif
|
|
ret["flags"] = p.flags;
|
|
return ret;
|
|
}
|
|
|
|
add_torrent_params parse_magnet_uri_wrap(std::string const& uri)
|
|
{
|
|
error_code ec;
|
|
add_torrent_params p = parse_magnet_uri(uri, ec);
|
|
if (ec) throw system_error(ec);
|
|
return p;
|
|
}
|
|
|
|
std::string (*make_magnet_uri0)(torrent_handle const&) = make_magnet_uri;
|
|
std::string (*make_magnet_uri1)(torrent_info const&) = make_magnet_uri;
|
|
}
|
|
|
|
void bind_magnet_uri()
|
|
{
|
|
#if TORRENT_ABI_VERSION == 1
|
|
def("add_magnet_uri", &_add_magnet_uri);
|
|
#endif
|
|
def("make_magnet_uri", make_magnet_uri0);
|
|
def("make_magnet_uri", make_magnet_uri1);
|
|
def("parse_magnet_uri", parse_magnet_uri_wrap);
|
|
def("parse_magnet_uri_dict", parse_magnet_uri_dict);
|
|
}
|