自动向新人问候头文件

Author: lovey599

auto_question.hpp

@@ -0,0 +1,148 @@
+/**
+ * @file
+ * @author
+ * @date
+ * 
+ * usage:
+ * 	auto_question question;
+ * 	auto_question::value_qq_list list;
+ * 
+ * 	list.push_back("lovey599");
+ * 
+ * 	questioin.add_to_list(list);
+ * 	questioin.on_handle_message(group, qqclient);
+ */
+
+#ifndef __QQBOT_AUTO_QUESTION_H__
+#define __QQBOT_AUTO_QUESTION_H__
+
+#include <iostream>
+#include <fstream>
+#include <string>
+#include <vector>
+#include <map>
+
+#include <boost/noncopyable.hpp>
+#include <boost/assign/list_of.hpp>
+#include <boost/assign.hpp>
+#include <boost/bind.hpp>
+#include <boost/foreach.hpp>
+#include <boost/format.hpp>
+#include <boost/asio.hpp>
+#include <boost/date_time/posix_time/posix_time.hpp>
+
+#include "libwebqq/webqq.h"
+
+void qq_msg_send_callback(const boost::system::error_code& ec)
+{
+	// nothing to do
+}
+
+class auto_question : public boost::noncopyable
+{
+public:
+	typedef std::vector<std::string> value_qq_list;
+	
+	auto_question(std::string filename = "question.txt")
+		: _filename(filename)
+		, _is_processing(false)
+	{
+		try
+		{
+			load_question();
+		}
+		catch (std::exception& ex)
+		{
+			std::cerr << "load questioin error." << std::endl;
+		}
+	}
+	
+	void add_to_list(value_qq_list list)
+	{
+		if (_is_processing)
+		{
+			_is_processing = false;
+		}
+		
+		BOOST_FOREACH(std::string item, list)
+		{
+			_process_count.insert(std::pair<std::string, int>(item, _questioin_count));
+		}
+	}
+	
+	void on_handle_message(qqGroup& group, webqq& qqclient)
+	{
+		handle_question(group, qqclient);
+	}
+	
+protected:
+	void load_question()
+	{
+		std::fstream file(_filename.c_str(), std::ios_base::in);
+		char question[512 + 1], answer[512 + 1];
+		
+		while (file.good())
+		{
+			file.getline(question, 512);
+			file.getline(answer, 512);
+			
+			if (strlen(question) == 0)
+			{
+				continue;
+			}
+			
+			_question.insert(std::pair<std::string, std::string>(question, answer));
+		}
+		
+		file.close();
+		
+		_questioin_count = _question.size();
+	}
+	
+	void handle_question(qqGroup& group, webqq& qqclient)
+	{
+		boost::asio::io_service io;
+		std::map<std::string, int>::iterator iter;
+		std::string str_msg_body = build_message();
+		
+		for (iter = _process_count.begin(); iter != _process_count.end(); ++iter)
+		{
+			std::string str_msg = boost::str(boost::format("@%1% 你好,欢迎你加入本群.请先回答以下问题:\n%2%谢谢\n") % iter->first % str_msg_body);
+			
+			std::cout << str_msg << std::endl;
+			qqclient.send_group_message(group, str_msg, qq_msg_send_callback);
+			
+			io.reset();
+			boost::asio::deadline_timer timer(io, boost::posix_time::seconds(1));
+			timer.wait();
+		}
+	}
+	
+	std::string build_message()
+	{
+		value_question::iterator iter;
+		std::string str_msg = "";
+		int index = 1;
+		
+		for (iter = _question.begin(); iter != _question.end(); ++iter)
+		{
+			str_msg += boost::str(boost::format("问题%1%.%2%\n") % (index++) % iter->first);
+		}
+		
+		return str_msg;
+	}
+	
+private:
+	std::string _filename;
+	
+	typedef std::map<std::string, std::string> value_question;
+	value_question _question;
+	
+	int _questioin_count;
+	
+	bool _is_processing;
+	
+	std::map<std::string, int> _process_count;
+};
+
+#endif