[api] Added bind() method to Socket object

Author: pajama-coder

src/socket.cpp

@@ -1192,6 +1192,26 @@ void SocketRaw::on_send(Data *data, const std::error_code &ec, std::size_t n) {
 // Socket
 //
 
+void Socket::bind(const std::string &ip_port) {
+  std::string ip;
+  int port;
+  if (utils::get_host_port(ip_port, ip, port)) {
+    bind(ip, port);
+  } else {
+    throw std::runtime_error("invalid [ip]:port format");
+  }
+}
+
+void Socket::bind(const std::string &ip, int port) {
+  if (m_fd) {
+    tcp::endpoint ep(asio::ip::make_address(ip), port);
+    auto addr = ep.data();
+    ::bind(m_fd, addr, sizeof(addr));
+  } else {
+    throw std::runtime_error("socket is gone");
+  }
+}
+
 auto Socket::get_raw_option(int level, int option, Data *data) -> int {
   if (!m_fd) throw std::runtime_error("socket is gone");
   char buf[1000];
@@ -1235,6 +1255,22 @@ namespace pjs {
 using namespace pipy;
 
 template<> void ClassDef<Socket>::init() {
+  method("bind", [](Context &ctx, Object *obj, Value &ret) {
+    Str *ip;
+    int port;
+    try {
+      if (ctx.argc() > 1) {
+        if (!ctx.arguments(2, &ip, &port)) return;
+        obj->as<Socket>()->bind(ip->str(), port);
+      } else {
+        if (!ctx.arguments(1, &ip)) return;
+        obj->as<Socket>()->bind(ip->str());
+      }
+    } catch (std::runtime_error &err) {
+      ctx.error(err);
+    }
+  });
+
   method("getRawOption", [](Context &ctx, Object *obj, Value &ret) {
     int level, option;
     pipy::Data *data;

src/socket.hpp

@@ -459,6 +459,8 @@ class SocketRaw :
 
 class Socket : public pjs::ObjectTemplate<Socket> {
 public:
+  void bind(const std::string &ip_port);
+  void bind(const std::string &ip, int port);
   auto get_raw_option(int level, int option, Data *data) -> int;
   auto set_raw_option(int level, int option, Data *data) -> int;
   auto io_control(int64_t op, const Data &input, Data *output = nullptr) -> int;