bindings/python/src/extensions.cpp

   1 // Copyright Daniel Wallin, Arvid Norberg 2007. Use, modification and distribution is
   2 // subject to the Boost Software License, Version 1.0. (See accompanying
   3 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   4
   5 #include <libtorrent/extensions.hpp>
   6 #include <libtorrent/entry.hpp>
   7 #include <libtorrent/peer_request.hpp>
   8 #include <libtorrent/peer_connection.hpp>
   9 #include <libtorrent/extensions/ut_pex.hpp>
  10 #include <libtorrent/extensions/metadata_transfer.hpp>
  11 #include <boost/python.hpp>
  12 #include "gil.hpp"
  13
  14 using namespace boost::python;
  15 using namespace libtorrent;
  16
  17 namespace
  18 {
  19
  20   struct torrent_plugin_wrap : torrent_plugin, wrapper<torrent_plugin>
  21   {
  22       boost::shared_ptr<peer_plugin> new_connection(peer_connection* p)
  23       {
  24           lock_gil lock;
  25
  26           if (override f = this->get_override("new_connection"))
  27               return f(ptr(p));
  28           return torrent_plugin::new_connection(p);
  29       }
  30
  31       boost::shared_ptr<peer_plugin> default_new_connection(peer_connection* p)
  32       {
  33           return this->torrent_plugin::new_connection(p);
  34       }
  35
  36       void on_piece_pass(int index)
  37       {
  38           lock_gil lock;
  39
  40           if (override f = this->get_override("on_piece_pass"))
  41               f(index);
  42           else
  43             torrent_plugin::on_piece_pass(index);
  44       }
  45
  46       void default_on_piece_pass(int index)
  47       {
  48           this->torrent_plugin::on_piece_pass(index);
  49       }
  50
  51       void on_piece_failed(int index)
  52       {
  53           lock_gil lock;
  54
  55           if (override f = this->get_override("on_piece_failed"))
  56               f(index);
  57           else
  58               torrent_plugin::on_piece_failed(index);
  59       }
  60
  61       void default_on_piece_failed(int index)
  62       {
  63           return this->torrent_plugin::on_piece_failed(index);
  64       }
  65
  66       void tick()
  67       {
  68           lock_gil lock;
  69
  70           if (override f = this->get_override("tick"))
  71               f();
  72           else
  73               torrent_plugin::tick();
  74       }
  75
  76       void default_tick()
  77       {
  78           return this->torrent_plugin::tick();
  79       }
  80
  81       bool on_pause()
  82       {
  83           lock_gil lock;
  84
  85           if (override f = this->get_override("on_pause"))
  86               return f();
  87           return torrent_plugin::on_pause();
  88       }
  89
  90       bool default_on_pause()
  91       {
  92           return this->torrent_plugin::on_pause();
  93       }
  94
  95       bool on_resume()
  96       {
  97           lock_gil lock;
  98
  99           if (override f = this->get_override("on_resume"))
 100               return f();
 101           return torrent_plugin::on_resume();
 102       }
 103
 104       bool default_on_resume()
 105       {
 106           return this->torrent_plugin::on_resume();
 107       }
 108   };
 109
 110 } // namespace unnamed
 111
 112
 113 boost::shared_ptr<torrent_plugin> create_metadata_plugin_wrapper(torrent* t) {
 114     return create_metadata_plugin(t, NULL);
 115 }
 116
 117 boost::shared_ptr<torrent_plugin> create_ut_pex_plugin_wrapper(torrent* t) {
 118     return create_ut_pex_plugin(t, NULL);
 119 }
 120
 121 void bind_extensions()
 122 {
 123     class_<
 124         torrent_plugin_wrap, boost::shared_ptr<torrent_plugin_wrap>, boost::noncopyable
 125     >("torrent_plugin")
 126         .def(
 127             "new_connection"
 128           , &torrent_plugin::new_connection, &torrent_plugin_wrap::default_new_connection
 129         )
 130         .def(
 131             "on_piece_pass"
 132           , &torrent_plugin::on_piece_pass, &torrent_plugin_wrap::default_on_piece_pass
 133         )
 134         .def(
 135             "on_piece_failed"
 136           , &torrent_plugin::on_piece_failed, &torrent_plugin_wrap::default_on_piece_failed
 137         )
 138         .def(
 139             "tick"
 140           , &torrent_plugin::tick, &torrent_plugin_wrap::default_tick
 141         )
 142         .def(
 143             "on_pause"
 144           , &torrent_plugin::on_pause, &torrent_plugin_wrap::default_on_pause
 145         )
 146         .def(
 147             "on_resume"
 148           , &torrent_plugin::on_resume, &torrent_plugin_wrap::default_on_resume
 149         );
 150
 151     // TODO move to it's own file
 152     class_<peer_connection, boost::noncopyable>("peer_connection", no_init);
 153
 154     class_<torrent_plugin, boost::shared_ptr<torrent_plugin> >("torrent_plugin", no_init);
 155     def("create_ut_pex_plugin", create_ut_pex_plugin_wrapper);
 156     def("create_metadata_plugin", create_metadata_plugin_wrapper);
 157 }
 158
 159