Telegram 搜群机器人 springboot+谷歌搜索引擎实现

package com.hkphantom.earch.bot; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.hkphantom.earch.util.NsHttp; import com.vdurmont.emoji.EmojiParser; import org.apache.commons.lang3.StringUtils; import org.jetbrains.annotations.Nullable; import org.telegram.telegrambots.bots.DefaultBotOptions; import org.telegram.telegrambots.bots.TelegramLongPollingBot; import org.telegram.telegrambots.meta.api.methods.ParseMode; import org.telegram.telegrambots.meta.api.methods.send.SendMessage; import org.telegram.telegrambots.meta.api.objects.CallbackQuery; import org.telegram.telegrambots.meta.api.objects.Message; import org.telegram.telegrambots.meta.api.objects.Update; import org.telegram.telegrambots.meta.api.objects.replykeyboard.InlineKeyboardMarkup; import org.telegram.telegrambots.meta.api.objects.replykeyboard.buttons.InlineKeyboardButton; import org.telegram.telegrambots.meta.exceptions.TelegramApiException; import java.nio.charset.StandardCharsets; import java.text.DateFormat; import java.text.SimpleDateFormat; import java.util.*; /** * 电报搜群机器人 */ public class TelegramSearch extends TelegramLongPollingBot { private String token = ""; // 电报机器人token private String userName = ""; // 电报机器人 userName String proxyHost = "127.0.0.1"; // 本地代理 String proxyPort = "10809"; // 本地代理端口 String httpUrl = "https://www.googleapis.com/customsearch/v1"; // 谷歌搜索引擎地址 String cx = "b0bcb1b09813012d8"; // 搜索引擎编号 String h1 = "zh-CN"; String key = ""; // 搜索引擎API秘钥 Map<String, HashMap> map = new HashMap<String, HashMap>(); public TelegramSearch() { this(new DefaultBotOptions()); } public TelegramSearch(DefaultBotOptions options) { super(options); } /** * 处理搜索查询 * * @param update */ @Override public void onUpdateReceived(Update update) { if (update.hasMessage()) {// 处理普通文本消息 // 获取查询关键字 Message message = update.getMessage(); if (message.hasText()) { // 打印日志 this.log(update); String message_text = message.getText();//发送的消息内容 Long chat_id = message.getChatId(); //发送信息的用户id //执行查询结果 StringBuffer stringBuffer = search(chat_id, message_text); if (stringBuffer == null) { return; } SendMessage sendMessage = getSendMessage(stringBuffer.toString(), message.getChatId()); try { execute(sendMessage); } catch (TelegramApiException e) { e.printStackTrace(); } } } else if (update.hasCallbackQuery()) { // 处理内联消息 CallbackQuery callbackQuery = update.getCallbackQuery(); Message msg = callbackQuery.getMessage(); String data = callbackQuery.getData(); SendMessage sendMessage = new SendMessage(); if (data.equals("下一页")) { String s = ""; // 获取上次关键词和开始页码 HashMap hashMap = map.get(String.valueOf(msg.getChatId())); System.out.println("hashMap==========" + hashMap); if (hashMap == null) { s = EmojiParser.parseToUnicode("只能查询自己的关键词 "); } else { StringBuffer search = search(msg.getChatId(), ""); System.out.println(map); s = search.toString(); } sendMessage = getSendMessage(s, msg.getChatId()); } try { execute(sendMessage); } catch (TelegramApiException e) { e.printStackTrace(); } } else { return; } } @Nullable private StringBuffer search(Long chat_id, String q) { //进来查询先创建查询数据 //查询是否已经创建记录 // 如果包含就是下一页直接获取，如果没有记录就是第一次查询创建记录 if (!map.containsKey(chat_id) && StringUtils.isNotBlank(q)) { HashMap<String, String> dataMap = new HashMap<>(); dataMap.put("chatId", String.valueOf(chat_id)); dataMap.put("q", q); dataMap.put("start", String.valueOf(1)); System.out.println(dataMap); map.put(String.valueOf(chat_id), dataMap); } // 对http开启代理 System.setProperty("http.proxyHost", proxyHost); System.setProperty("http.proxyPort", proxyPort); // 对https也开启代理 System.setProperty("https.proxyHost", proxyHost); System.setProperty("https.proxyPort", proxyPort); //根据关键词进行查询 //获取关键词 HashMap hashMap = map.get(String.valueOf(chat_id)); String start = hashMap.get("start").toString(); NsHttp nsHttp = new NsHttp(); HashMap<String, Object> paramsMap = new HashMap<>(); paramsMap.put("key", key); paramsMap.put("cx", cx); paramsMap.put("h1", h1); paramsMap.put("q", hashMap.get("q")); paramsMap.put("start", start); paramsMap.put("num", 10); System.out.println("start=======" + hashMap.get("start")); String response = nsHttp.doGet(httpUrl, paramsMap); if (StringUtils.isBlank(response)) { return null; } JSONObject jsonObject = JSONObject.parseObject(response); System.out.println(jsonObject); JSONArray items = jsonObject.getJSONArray("items"); StringBuffer stringBuffer = new StringBuffer(); System.out.println(items); Integer j = Integer.valueOf(start); for (int i = 0; i < items.size(); i++) { JSONObject item = (JSONObject) items.get(i); String link = item.getString("link"); if (link.contains("www.")) { link = link.replace("www.", ""); } if (link.contains("/s")) { link = link.replace("/s", ""); } String title = item.getString("title"); if (title.contains("–")) { title = title.replace("–", ""); } if (title.contains("Telegram")) { title = title.replace("Telegram", ""); } title = title.replaceAll("", ""); String authorizeStr = EmojiParser.parseToUnicode( (j) + ". :bust_in_silhouette: <a href=\"" + link + "\">" + title + "</a>\n\n" ); stringBuffer.append(authorizeStr); j++; } //查询结束更新map的值 hashMap.put("start", Integer.valueOf(hashMap.get("start").toString()) + 10); map.put(String.valueOf(chat_id), hashMap); System.out.println(stringBuffer); return stringBuffer; } @Override public String getBotUsername() { return this.userName; } @Override public String getBotToken() { return this.token; } //发送普通消息 private SendMessage getSendMessage(String message_text, Long chat_id) { SendMessage message = new SendMessage();// Create a message object object message.setChatId(String.valueOf(chat_id)); message.setText(message_text); message.setParseMode(ParseMode.HTML); //�