{

BOOL fError;

switch (message)

{

case WM_INITDIALOG:

if (IsDlgButtonChecked(hwndDlg, IDC_CHECK_XMPP) != BST_CHECKED) {

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_XMPP_CHANNEL), FALSE);

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_XMPP_NICK), FALSE);

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_XMPP_PWD), FALSE);

}

if (IsDlgButtonChecked(hwndDlg, IDC_CHECK_IRC) != BST_CHECKED) {

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_IRC_CHANNEL), FALSE);

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_IRC_NICK), FALSE);

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_IRC_PWD), FALSE);

}

return TRUE;

case WM_COMMAND:

switch (LOWORD(wParam))

{

case IDOK:

// 通知主函数写入数据并重启

PostMessage(NULL, WM_RESTART_AV_BOT, 0, 0);

return TRUE;

case IDCANCEL:

DestroyWindow(hwndDlg);

// 退出消息循环

PostMessage(NULL, WM_QUIT, 0, 0);

return TRUE;

case IDC_CHECK_XMPP:{

BOOL enable = FALSE;

if (IsDlgButtonChecked(hwndDlg, IDC_CHECK_XMPP) == BST_CHECKED) enable = TRUE;

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_XMPP_CHANNEL), enable);

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_XMPP_NICK), enable);

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_XMPP_PWD), enable);

}return TRUE;

case IDC_CHECK_IRC:{

BOOL enable = FALSE;

if (IsDlgButtonChecked(hwndDlg, IDC_CHECK_IRC) == BST_CHECKED) enable = TRUE;

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_IRC_CHANNEL), enable);

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_IRC_NICK), enable);

EnableWindow(GetDlgItem(hwndDlg, IDC_EDIT_IRC_PWD), enable);

}return TRUE;

}

}

return FALSE;

{

const char * data = boost::asio::buffer_cast<const char*>( imgbuf );

size_t imgsize = boost::asio::buffer_size( imgbuf );

fs::path imgpath = fs::path( logfile.log_path() ) / "vercode.jpeg";

std::ofstream img( imgpath.string().c_str(), std::ofstream::openmode(std::ofstream::binary | std::ofstream::out) );

img.write( data, imgsize );

qqneedvc = true;

// send to xmpp and irc

mybot.broadcast_message( "请查看qqlog目录下的vercode.jpeg 然后用\".qqbot vc XXX\"输入验证码:" );

std::cerr << console_out_str("请查看qqlog目录下的vercode.jpeg 然后输入验证码: ") << std::flush ;

do_vc_code = boost::bind(&avbot::feed_login_verify_code, &mybot, _1);

bool operator()(const std::string & str)

{

return str.substr(0, 3) == "qq:";

}

{

std::vector<std::string> gs;

boost::split(gs, chanelmapstring, boost::is_any_of(";"));

int m = 1;

BOOST_FOREACH(std::string pergroup, gs)

{

std::vector<std::string> groups;

boost::split(groups, pergroup, boost::is_any_of(","));

// 对于包含了 qq: 的那个组，组名就是 qq号码，而其他的嘛，组名使用 groupXXX

std::vector<std::string>::iterator qqgroup = std::find_if(groups.begin(), groups.end(), build_group_has_qq() );

if (qqgroup != groups.end())

{

BOOST_FOREACH(std::string c, groups)

{

mybot.add_to_channel((*qqgroup).substr(3), c);

}

}else{

m ++;

std::string group_name = std::string("group") + boost::lexical_cast<std::string>(m);

BOOST_FOREACH(std::string c, groups)

{

mybot.add_to_channel(group_name, c);

}

}

}

{

std::string linemessage;

// 首先是根据 nick 格式化

if( message.get<std::string>( "protocol" ) != "mail" )

{

linemessage += message.get<std::string>( "preamble", "" );

BOOST_FOREACH( const avbot::av_message_tree::value_type & v, message.get_child( "message" ) )

{

if( v.first == "text" )

{

std::string txt = v.second.data();

// 将 < > 给转义.

boost::replace_all( txt, "&", "&amp;" );

boost::replace_all( txt, "<", "&lt;" );

boost::replace_all( txt, ">", "&gt;" );

boost::replace_all( txt, " ", "&nbsp;" );

linemessage += txt;

}

else if( v.first == "url" )

{

linemessage += boost::str( boost::format( "<a href=\"%s\">%s</a>" ) % v.second.data() % v.second.data() );

}

else if( v.first == "cface" )

{

if( mybot.fetch_img ){

std::string cface = v.second.data();

linemessage += boost::str( boost::format( "<img src=\"../images/%s/%s\" />" ) % avbot::image_subdir_name(cface) % cface );

}

else

linemessage += boost::str( boost::format( "<img src=\"http://w.qq.com/cgi-bin/get_group_pic?pic=%s\" />" ) % v.second.data() );

}

else if( v.first == "img" )

{

linemessage += boost::str( boost::format( "<img src=\"%s\" />" ) % v.second.data() );

}

}

std::string channel_name = message.get<std::string>( "channel", "" );

if( channel_name.empty() )

{

// 全部记录?

// TODO

}

else

{

// 如果最好的办法就是遍历组里的所有QQ群，都记录一次.

avbot::av_chanel_map channelmap = mybot.get_channel_map( channel_name );

// 如果没有Q群，诶，只好，嘻嘻.

logfile.add_log( channel_name, linemessage );

BOOST_FOREACH( std::string roomname, channelmap )

{

if( roomname.substr( 0, 3 ) == "qq:" )

{

// 避免重复记录.

if( roomname.substr( 3 ) != channel_name )

{

logfile.add_log( roomname.substr( 3 ), linemessage );

}

}

}

}

}

else

{

// TODO , 暂时不记录邮件吧！

}

{

try{

std::string textmessage;

try{

textmessage = message.get<std::string>("newbee");

// 如果没有发生异常, 那就是新人入群消息了, 嘻嘻.

// 格式化为 .qqbot newbee XXX, 哼!

std::string nick = message.get<std::string>("who.name");

textmessage = boost::str(boost::format(".qqbot newbee %s") % nick);

message.put("message.text", textmessage);

message.put("op", 1);

boost::delayedcallsec(mybot.get_io_service(), 6, boost::bind(on_bot_command, message, boost::ref(mybot)));

return;

}catch (const boost::property_tree::ptree_bad_path &){

textmessage = message.get<std::string>("message.text");

}

on_bot_command(message, mybot);

}catch (...){}

{

// nothing...

return -1;

{

std::string qqnumber, qqpwd;

std::string ircnick, ircroom, ircpwd;

std::string xmppuser, xmppserver, xmpppwd, xmpproom, xmppnick;

std::string cfgfile;

std::string logdir;

std::string chanelmap;

std::string mailaddr, mailpasswd, pop3server, smtpserver;

fs::path config; // 配置文件的路径

unsigned rpcport;

boost::asio::io_service io_service;

avbot mybot(io_service);

progname = fs::basename( argv[0] );

setlocale( LC_ALL, "" );

po::variables_map vm;

po::options_description desc( "qqbot options" );

desc.add_options()

( "version,v", "output version" )

( "help,h", "produce help message" )

( "daemon,d", "go to background" )

#ifdef WIN32

( "gui,g", "pop up settings dialog" )

#endif

( "config,c", po::value<fs::path>( &config ), "use an alternative configuration file." )

( "qqnum,u", po::value<std::string>( &qqnumber ), "QQ number" )

( "qqpwd,p", po::value<std::string>( &qqpwd ), "QQ password" )

( "logdir", po::value<std::string>( &logdir ), "dir for logfile" )

( "ircnick", po::value<std::string>( &ircnick ), "irc nick" )

( "ircpwd", po::value<std::string>( &ircpwd ), "irc password" )

( "ircrooms", po::value<std::string>( &ircroom ), "irc room" )

( "xmppuser", po::value<std::string>( &xmppuser ), "id for XMPP, eg: ([email protected])" )

( "xmppserver", po::value<std::string>( &xmppserver ), "server to connect for XMPP, eg: (xmpp.l.google.com)" )

( "xmpppwd", po::value<std::string>( &xmpppwd ), "password for XMPP" )

( "xmpprooms", po::value<std::string>( &xmpproom ), "xmpp rooms" )

( "xmppnick", po::value<std::string>( &xmppnick ), "nick in xmpp rooms" )

( "map", po::value<std::string>( &chanelmap ), "map channels. eg: --map=qq:12345,irc:avplayer;qq:56789,irc:ubuntu-cn" )

( "mail", po::value<std::string>( &mailaddr ), "fetch mail from this address" )

( "mailpasswd", po::value<std::string>( &mailpasswd ), "password of mail" )

( "pop3server", po::value<std::string>( &pop3server ), "pop server of mail, default to pop.[domain]" )

( "smtpserver", po::value<std::string>( &smtpserver ), "smtp server of mail, default to smtp.[domain]" )

( "localimage", po::value<bool>( &(mybot.fetch_img) )->default_value(true), "fetch qq image to local disk and store it there")

( "rpcport", po::value<unsigned>(&rpcport)->default_value(6176), "run rpc server on port 6176")

( "preambleqq", po::value<std::string>( &preamble_qq_fmt )->default_value( console_out_str("qq(%a) :") ),

console_out_str("为QQ设置的发言前缀, 默认是 qq(%a):").c_str() )

( "preambleirc", po::value<std::string>( &preamble_irc_fmt )->default_value( console_out_str("%a 说 :") ),

console_out_str("为IRC设置的发言前缀, 默认是 %a 说: ").c_str() )

( "preamblexmpp", po::value<std::string>( &preamble_xmpp_fmt )->default_value( console_out_str("(%a) :") ),

console_out_str(

"为XMPP设置的发言前缀, 默认是 (%a): \n\n "

"前缀里的含义 \n"

"\t %a 为自动选择\n\t %q 为QQ号码\n\t %n 为昵称\n\t %c 为群名片 \n"

"\t %r为房间名(群号, XMPP房名, IRC频道名) \n"

"可以包含多个, 例如想记录QQ号码的可以使用 qq(%a, %q)说: \n"

"注意在shell下可能需要使用\\(来转义(\n配置文件无此问题 \n\n").c_str() )

;

po::store( po::parse_command_line( argc, argv, desc ), vm );

po::notify( vm );

if( vm.count( "help" ) ) {

std::cerr << desc << std::endl;

return 1;

}

#ifdef WIN32

// 从 windows 控制台输入的能有啥好编码，转到utf8吧.

preamble_qq_fmt = ansi_utf8(preamble_qq_fmt);

preamble_irc_fmt = ansi_utf8(preamble_irc_fmt);

preamble_xmpp_fmt = ansi_utf8(preamble_xmpp_fmt);

#endif

if( qqnumber.empty() ) {

// 命令行没指定选项，读取配置文件

try {

if (config.empty()) {

// 命令行没指定配置文件？使用默认的！

config = configfilepath();

}

std::cout << "loading config from: " << config.string() << std::endl;

po::store( po::parse_config_file<char>( config.string().c_str(), desc ), vm );

po::notify( vm );

} catch( char const* e ) {

std::cout << "no command line arg and config file not found neither." << std::endl;

std::cout << "try to add command line arg or put config file in /etc/qqbotrc or ~/.qqbotrc" << std::endl;

}

}

if( vm.count( "daemon" ) ) {

daemon( 0, 0 );

}

if( vm.count( "version" ) ) {

printf( "qqbot version %s (%s %s) \n", QQBOT_VERSION, __DATE__, __TIME__ );

exit( EXIT_SUCCESS );

}

if( !logdir.empty() ) {

if( !fs::exists( logdir ) )

fs::create_directory( logdir );

}

// 设置到中国的时区，否则 qq 消息时间不对啊.

#ifndef _WIN32

setenv("TZ", "Asia/Shanghai", 1);

# else

putenv( ( char* )"TZ=Asia/Shanghai" );

# endif

#ifdef WIN32

::InitCommonControls();

// windows下面弹出选项设置框框

if( qqnumber.empty() || qqpwd.empty() || vm.count( "gui" )>0)

{

HMODULE hIns = GetModuleHandle(NULL);

HWND hDlg = NULL;

hDlg = CreateDialog(hIns, MAKEINTRESOURCE(IDD_AVSETTINGS_DIALOG), NULL, (DLGPROC)DlgProc);

ShowWindow(hDlg, SW_SHOW);

// 启动windows消息循环

MSG msg;

while (true) {

if (GetMessage(&msg, NULL, 0, 0)) {

if (msg.message == WM_QUIT) exit(1);

if (msg.message == WM_RESTART_AV_BOT) {

// 看起来avbot用的是多字节,std::string not std::wstring

TCHAR temp[MAX_PATH];

bool use_xmpp = false;

bool use_irc = false;

// qq setting

GetDlgItemText(hDlg, IDC_EDIT_USER_NAME, temp, MAX_PATH);

qqnumber = temp;

GetDlgItemText(hDlg, IDC_EDIT_PWD, temp, MAX_PATH);

qqpwd = temp;

// irc setting

if (IsDlgButtonChecked(hDlg, IDC_CHECK_IRC) == BST_CHECKED)

{

GetDlgItemText(hDlg, IDC_EDIT_IRC_CHANNEL, temp, MAX_PATH);

ircroom = temp;

GetDlgItemText(hDlg, IDC_EDIT_IRC_NICK, temp, MAX_PATH);

ircnick = temp;

GetDlgItemText(hDlg, IDC_EDIT_IRC_PWD, temp, MAX_PATH);

ircpwd = temp;

use_irc = true;

}

// xmpp setting

if (IsDlgButtonChecked(hDlg, IDC_CHECK_XMPP) == BST_CHECKED)

{

GetDlgItemText(hDlg, IDC_EDIT_XMPP_CHANNEL, temp, MAX_PATH);

xmpproom = temp;

GetDlgItemText(hDlg, IDC_EDIT_XMPP_NICK, temp, MAX_PATH);

xmppnick = temp;

GetDlgItemText(hDlg, IDC_EDIT_XMPP_PWD, temp, MAX_PATH);

xmpppwd = temp;

use_xmpp = true;

}

// save data to config file

try {

fs::path config_file;

if (exist_config_file()) {

config_file = configfilepath();

}

else {

std::cerr << "config file not exist." << std::endl;

TCHAR file_path[MAX_PATH];

GetModuleFileName(NULL, file_path, MAX_PATH);

config_file = fs::path(file_path).parent_path();

config_file += "\\qqbotrc";

}

fs::ofstream file(config_file);

file << "# qq config" << std::endl;

file << "qqnum=" << qqnumber << std::endl;

file << "qqpwd=" << qqpwd << std::endl;

file << std::endl;

if(use_irc)

{

file << "# irc config" << std::endl;

file << "ircnick=" << ircnick << std::endl;

file << "ircpwd=" << ircpwd << std::endl;

file << "ircrooms=" << ircroom << std::endl;

file << std::endl;

}

if (use_xmpp)

{

file << "# xmpp config" << std::endl;

file << "xmppuser=" << xmppnick << std::endl;

file << "xmpppwd=" << xmpppwd << std::endl;

file << "xmpproom=" << xmpproom << std::endl;

file << std::endl;

}

}

catch (...) {

std::cerr << "error while handle config file" << std::endl;

}

TCHAR file_path[MAX_PATH];

GetModuleFileName(NULL, file_path, MAX_PATH);

// now, create process

STARTUPINFO si;

PROCESS_INFORMATION pi;

ZeroMemory(&si, sizeof(si));

si.cb = sizeof(si);

ZeroMemory(&pi, sizeof(pi));

CreateProcess(file_path, NULL, NULL, NULL,

FALSE, 0, NULL, NULL,

&si, &pi);

CloseHandle(pi.hProcess);

CloseHandle(pi.hThread);

// now,parent process exit.

exit(1);

}

TranslateMessage(&msg);

DispatchMessage(&msg);

}

}

}

#endif

{

#ifndef _WIN32

std::string env_oldpwd = fs::complete(fs::current_path()).string();

setenv("O_PWD", env_oldpwd.c_str(), 1);

# else

std::string env_oldpwd = std::string("O_PWD=") + fs::complete(fs::current_path()).string();

putenv( (char*)env_oldpwd.c_str() );

#endif

}

// 设置日志自动记录目录.

if( ! logdir.empty() ) {

logfile.log_path( logdir );

chdir( logdir.c_str() );

}

if( qqnumber.empty() || qqpwd.empty() ) {

std::cerr << console_out_str("请设置qq号码和密码") << std::endl;

exit( 1 );

}

mybot.preamble_irc_fmt = preamble_irc_fmt;

mybot.preamble_qq_fmt = preamble_qq_fmt;

mybot.preamble_xmpp_fmt = preamble_xmpp_fmt;

mybot.signal_new_channel.connect(boost::bind(new_channel_set_extension, boost::ref(io_service), boost::ref(mybot), _1));

mybot.set_qq_account( qqnumber, qqpwd, boost::bind( on_verify_code, _1, boost::ref( mybot ) ) );

if( !ircnick.empty() )

mybot.set_irc_account( ircnick, ircpwd );

if( !xmppuser.empty() )

mybot.set_xmpp_account( xmppuser, xmpppwd, xmppserver, xmppnick );

if( !mailaddr.empty() )

mybot.set_mail_account( mailaddr, mailpasswd, pop3server, smtpserver );

build_group( chanelmap, mybot );

// 记录到日志.

mybot.on_message.connect(boost::bind(avbot_log, _1, boost::ref(mybot)));

// 开启 bot 控制.

mybot.on_message.connect(boost::bind(my_on_bot_command, _1, boost::ref(mybot)));

std::vector<std::string> ircrooms;

boost::split( ircrooms, ircroom, boost::is_any_of( "," ) );

BOOST_FOREACH( std::string room , ircrooms )

{

mybot.irc_join_room(std::string( "#" ) + room);

}

std::vector<std::string> xmpprooms;

boost::split( xmpprooms, xmpproom, boost::is_any_of( "," ) );

BOOST_FOREACH( std::string room , xmpprooms )

{

mybot.xmpp_join_room(room);

}

boost::asio::io_service::work work( io_service );

if( !vm.count( "daemon" ) ) {

#ifdef BOOST_ASIO_HAS_POSIX_STREAM_DESCRIPTOR

boost::shared_ptr<boost::asio::posix::stream_descriptor> stdin( new boost::asio::posix::stream_descriptor( io_service, 0 ) );

boost::shared_ptr<boost::asio::streambuf> inputbuffer( new boost::asio::streambuf );

boost::asio::async_read_until( *stdin, *inputbuffer, '\n', boost::bind( inputread , _1, _2, stdin, inputbuffer, boost::ref( mybot ) ) );

#else

boost::thread( boost::bind( input_thread, boost::ref( io_service ), boost::ref( mybot ) ) );

#endif

}

if (rpcport > 0)

{

try

{

// 调用 acceptor_server 跑 avbot_rpc_server 。 在端口 6176 上跑哦!

boost::acceptor_server( io_service,

boost::asio::ip::tcp::endpoint( boost::asio::ip::tcp::v6(), rpcport ),

boost::bind( avbot_rpc_server, _1, boost::ref( mybot ) )

);

}

catch( ... )

{

try

{

boost::acceptor_server( io_service,

boost::asio::ip::tcp::endpoint( boost::asio::ip::tcp::v4(), rpcport ),

boost::bind( avbot_rpc_server, _1, boost::ref( mybot ) )

);

}catch(...)

{

std::cerr << "bind to port " << rpcport << " failed!" << std::endl;

std::cerr << "Did you happened to already run an avbot? " << std::endl;

std::cerr << "Now avbot will run without RPC support. " << std::endl;

}

}

}

avloop_run( io_service);

return 0;