Big Sky :: picojsonとnanowwwを使って「関係ないbot」書いた。

tokuhiromさんがnanowwwという、C++から簡単にHTTPが扱えるクライアントライブラリを書いてくれたので、botを書いてみた。

tokuhirom's nanowww at master - GitHub

C++ lightweight, fast, portable HTTP client library
http://github.com/tokuhirom/nanowww

picojsonは以前ご紹介した通り、kazuhoさんが書いたC++からSTLと親和性の高いJSONパーサです。
今回はtokuhiromさんが、C++で、もちろんSTLと親和性の高いHTTPクライアントライブラリを書いてくれました。
これを使えば、例えばhttp://example.com/fooというURLにPOSTで"name=hasegawa"を送信するとした場合


  nanowww::Client client;

  nanowww::Request req("POST", "http://example.com/foo", "name=hasegawa");

  nanowww::Response res;

  req.set_header("Content-Type", "application/x-www-form-urlencoded");

  if (client.send_request(req, &res))

      std::cout << res.content() << std::endl;

  else

      std::cout << client.errstr() << std::endl;

これだけのコードでHTTP通信出来てしまいます。
C++らしくていいですね。CURLよりはインタフェースが優れていると思います。ただ現状はPROXYに対応していなかったり、Windowsで使えなかったりしますので、用途を絞られますが、今後に期待したいと思います。
さて今日はこれを使って、IRCに常駐し、指定のチャネルに入り、指定の単語が現れると「それXXX関係ないだろ！jk」と発言する、「関係ないbot」を書きました。実際にIRCプロトコルを喋っている訳でなく、webchat.freenode.netのAjax通信を、nanowwwとpicojsonによる擬似Ajax通信を行って実現しています。
以下、少々長いですが全ソースコード。


#include "picojson.h"

#include "nanowww.h"



using namespace std;

using namespace picojson;

using namespace nanowww;



#ifdef _WIN32

static char* utf8_to_str_alloc(const char* utf8) {

    size_t in_len = strlen(utf8);

    wchar_t* wcsdata;

    char* mbsdata;

    size_t mbssize, wcssize;



    wcssize = MultiByteToWideChar(CP_UTF8, 0, utf8, in_len,  NULL, 0);

    wcsdata = (wchar_t*) malloc((wcssize + 1) * sizeof(wchar_t));

    wcssize = MultiByteToWideChar(CP_UTF8, 0, utf8, in_len, wcsdata, wcssize + 1);

    wcsdata[wcssize] = 0;



    mbssize = WideCharToMultiByte(GetACP(), 0, (LPCWSTR) wcsdata, -1, NULL, 0, NULL, NULL);

    mbsdata = (char*) malloc((mbssize + 1));

    mbssize = WideCharToMultiByte(GetACP(), 0, (LPCWSTR) wcsdata, -1, mbsdata, mbssize, NULL, NULL);

    mbsdata[mbssize] = 0;

    free(wcsdata);

    return mbsdata;

}

#endif



value get_json(string url, map<string, string>& postdata) {

  value v;

  char error[256];

  string data;

  map<string, string>::iterator it;



  for (it = postdata.begin(); it != postdata.end(); it++) {

      if (data.size()) data += "&";

      data += it->first + "=";

      data += it->second;

  }



  Client client;

  Request req("POST", url.c_str(), data.c_str());

  Response res;

  req.set_header("Content-Type", "application/x-www-form-urlencoded");

  if (client.send_request(req, &res)) {

    const char* ptr = res.content().c_str();

    string err = parse(v, ptr, ptr + strlen(ptr));

    if (!err.empty()) cerr << err << endl;

  }

  return v;

}



int

main(int argc, char* argv[]) {

  map<string, string> postdata;

  value v;

  bool loop;



  if (argc != 4) {

      std::cerr << "usage: kankeinaibot [nick] [channel] [word]" << std::endl;

      return -1;

  }

  string nick = argv[1];

  string channel = argv[2];

  string word = argv[3];



  postdata.clear();

  postdata["nick"] = nick;

  v = get_json("http://webchat.freenode.net/e/n", postdata);

  string sid = v.get<array>()[1].get<string>();



  loop = true;

  postdata.clear();

  postdata["s"] = sid;

  while (loop) {

    v = get_json("http://webchat.freenode.net/e/s", postdata);

    array a = v.get<array>();

    for (array::iterator it = a.begin(); it != a.end(); it++) {

      if (!it->is<array>()) continue;

      array line = it->get<array>();

      if (line[0].to_str() == "c") {

        if (line[1].to_str() == "443") {

            postdata["nick"] += "_";

            v = get_json("http://webchat.freenode.net/e/n", postdata);

            sid = v.get<array>()[1].get<string>();

            break;

        }

        if (line[1].to_str() == "376") {

            loop = false;

            break;

        }

      }

    }

  }



  loop = true;

  postdata.clear();

  postdata["c"] = "JOIN #";

  postdata["c"] += channel;

  postdata["s"] = sid;

  v = get_json("http://webchat.freenode.net/e/p", postdata);

  while (loop) {

    v = get_json("http://webchat.freenode.net/e/s", postdata);

    array a = v.get<array>();

    for (array::iterator n = a.begin(); n != a.end(); n++) {

      array line = n->get<array>();

      if (line[0].to_str() == "c") {

        array aa = line[3].get<array>();

        if (aa.size() > 1 && aa[0].to_str() != nick) {

          cout << aa[0].serialize() << endl;

          cout << aa[1].serialize() << endl;

          if (!strncmp(aa[1].to_str().c_str(), word.c_str(), word.size())) {

              map<string, string> saydata;

              saydata["s"] = sid;

              std::stringstream s;

              s << "PRIVMSG #" << channel << " :" <<

                  "それ" << word << "関係ないだろ！jk";

              saydata["c"] = s.str();

              cout << saydata["c"] << endl;

              v = get_json("http://webchat.freenode.net/e/p", saydata);

          }

        }

        string s = line[3].serialize();

#ifdef _WIN32

        char* p = utf8_to_str_alloc(s.c_str());

        cout << p;

        cout << " ";

        free((void*)p);

        cout << endl;

#else

        cout << s << endl;

#endif

      }

    }

  }

  

  return 0;

}