天气.js

import plugin from '../../../lib/plugins/plugin.js'
import fetch from "node-fetch";
import { Config, Version } from '../components/index.js'
const city_list = {};
export class xiaofei_weather extends plugin {
	constructor() {
		super({
			/** 功能名称 */
			name: '小飞插件_天气',
			/** 功能描述 */
			dsc: '请求腾讯天气网站进行页面截图，目前支持以下命令：【#天气】',
			/** https://oicqjs.github.io/oicq/#events */
			event: 'message',
			/** 优先级，数字越小等级越高 */
			priority: 2000,
			rule: [
				{
					/** 命令正则匹配 */
					reg: '^#?(小飞)?(.*)天气$',
					/** 执行方法 */
					fnc: 'query_weather'
				}
			]
		});

		try {
			let setting = Config.getdefSet('setting', 'system') || {};
			this.priority = setting['weather'] == true ? 10 : 2000;
		} catch (err) { }

	}

	async query_weather() {
		if (/^#?小飞设置.*$/.test(this.e.msg)) return false;

		let msg = this.e.msg
			.replace('#', '')
			.replace('小飞', '')
			.replace('天气', '');
		return await weather(this.e, msg);
	}

}

async function weather(e, search) {
	if (search.replace(/ /g, '') == '' || search == '地区') {
		if (e.msg.includes('#')) e.reply("格式：#地区天气\r\n例如：#北京天气", true);
		return true;
	}
	var area_id = -1, reg = null, province = '', city = '', district = '';
	search = search.replace(/\s\s/g, ' ').replace(/\s\s/g, ' ');
	reg = /((.*)省)?((.*)市)?((.*)区)?/.exec(search);
	if (reg[2]) { province = reg[2]; search = search.replace(province + '省', ' '); }
	if (reg[4]) { city = reg[4]; search = search.replace('市', ' '); }
	if (reg[6]) { district = reg[6]; search = search.replace('区', ' '); }

	let res = null;
	let arr = search.trim().split(' ').reverse();
	arr.push(search.trim());

	for (let index in arr) {
		let value = arr[index];
		let url = `https://wis.qq.com/city/matching?source=xw&city=${encodeURI(value)}`;//地区名取area_id接口
		let response = await fetch(url); //获取area_id列表
		try {
			res = await response.json();
		} catch (err) { }
		if (res == null || res.status != 200 || !res.data?.internal || res.data?.internal.length < 1) {
			continue;
		}
		let internal = res.data.internal;
		let keys = Object.keys(internal).reverse();
		for (let key of keys) {
			for (let i = parseInt(index) + 1; i < arr.length; i++) {
				if (internal[key].includes(arr[i]) || arr[i].includes(internal[key])) {
					area_id = key;
					break;
				}
			}
			if (area_id != -1) break;
		}
		if (area_id != -1) break;
	}

	if (res == null || res.status != 200 || !res.data?.internal || res.data?.internal.length < 1) {
		if (e.msg.includes('#')) e.reply('没有查询到该地区的天气！', true);
		return true;
	}

	let internal = res.data.internal;
	let keys = Object.keys(internal).reverse();
	for (let key of keys) {
		let arr = internal[key].split(', ');

		if (province && !province.includes(arr[0])) {
			continue;
		}

		if (city && !city.includes(arr[1])) {
			continue;
		}

		if (district && !district.includes(arr[2])) {
			continue;
		}

		if (arr[0]) province = arr[0];
		if (arr[1]) city = arr[1];
		if (arr[2]) district = arr[2];
		area_id = key;
		break;
	}

	if (area_id == -1) {
		if (e.msg.includes('#')) e.reply('没有查询到该地区的天气！', true);
		return true;
	}
	//let setting = Config.getdefSet('setting', 'system') || {};

	var attentionCity = JSON.stringify([{
		province: province,
		city: city,
		district: district,
		isDefault: true
	}]);

	let img = null;
	try {
		const browser = await xiaofei_plugin.puppeteer.browserInit();
		const page = await browser.newPage();
		await page.setViewport({
			width: 1280,
			height: 1320
		});

		await page.setUserAgent('Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36 Edg/122.0.0.0');
		await page.goto('view-source:https://tianqi.qq.com/');
		await page.evaluate(`localStorage.setItem('attentionCity', '${attentionCity}')`);//设置默认地区信息
		await page.setRequestInterception(true);
		page.on('request', req => {
			let urls = [
				'trace.qq.com',
			];

			let url = req.url();
			if (urls.find(val => { return url.includes(val) })) {
				req.abort();
			} else {
				req.continue();
			}
		});
		await page.goto('https://tianqi.qq.com/');//请求天气页面

		await page.evaluate(() => {
			$('a').remove();
			$('#ct-footer').remove();//删除底部导航栏
		});

		await page.evaluate(`$('body').append('<p style="text-align: center;font-size: 15px;margin-top: -25px;">Created By Yunzai-Bot ${Version.yunzai} &amp; xiaofei-Plugin ${Version.ver}</p><br>');`);//增加版本号显示

		let body = await page.$('body');
		img = await body.screenshot({
			//fullPage: true,
			type: 'jpeg',
			omitBackground: false,
			quality: 100,
		});

		page.close().catch((err) => logger.error(err));

		xiaofei_plugin.puppeteer.renderNum++;
		xiaofei_plugin.puppeteer.restart();
	} catch (err) {
		logger.error(err);
	}

	if (!img) {
		if (e.msg.includes('#')) await e.reply('[小飞插件]天气截图失败！');
		return false;
	}
	if (img?.type != 'image') img = segment.image(img);
	if (img?.file) img.file = Buffer.from(img.file);
	await e.reply(img);

	return true;
}