lora / 不知道 /回收站 /1 /webPreview.js
lintonxue00's picture
Upload webPreview.js
4598406
raw
history blame
4.35 kB
import plugin from '../../../lib/plugins/plugin.js'
import { segment } from 'oicq'
import lodash from "lodash";
import { createRequire } from 'module'
const require = createRequire(import.meta.url)
export class webPreview extends plugin {
constructor() {
super({
/** 功能名称 */
name: '截图预览网页内容',
/** 功能描述 */
dsc: '群里发送网页地址,截图预览网页内容',
/** https://oicqjs.github.io/oicq/#events */
event: 'message',
/** 优先级,数字越小等级越高 */
priority: 1006,
rule: [
{
/** 命令正则匹配 */
reg: '^(?:(http|https):\/\/)?((?:[\\w-]+\\.)+[a-z0-9]+)(?::\\d+)?((?:\/[^\/?#]*)+)?(\\?[^#]+)?(#.+)?$'
,
/** 执行方法 */
fnc: 'webPreview'
},
{
/** 命令正则匹配 */
reg: '^#*必应(.*)$',
/** 执行方法 */
fnc: 'bingWeb'
},
]
})
}
/**
*
* @param e oicq传递的事件参数e
*/
async webPreview(e) {
const puppeteer = require('puppeteer');
const browser = await puppeteer.launch({
headless: true,
args: [
'--disable-gpu',
'--disable-dev-shm-usage',
'--disable-setuid-sandbox',
'--no-first-run',
'--no-sandbox',
'--no-zygote',
'--single-process'
]
});
const page = await browser.newPage();
try {
await page.goto(e.msg, { timeout: 30000, waitUntil: 'networkidle0' });
} catch (err) {
console.error(`Failed to navigate to ${e.msg}: ${err.message}`);
await browser.close();
return;
}
await page.setViewport({ width: 1920, height: 1080 });
let screenshot;
try {
screenshot = await page.screenshot({ fullPage: true });
} catch (err) {
console.error(`Failed to take screenshot: ${err.message}`);
await browser.close();
return;
}
await this.reply(segment.image(screenshot));
await browser.close();
}
/**
*
* @param e oicq传递的事件参数e
*/
async bingWeb(e) {
let webkeywd = e.msg.replace(/#|必应/gm, '');
webkeywd = webkeywd.replace(/,| |,/g, ",");
let wdKey = webkeywd.split(",");
wdKey = lodash.compact(wdKey);
let keyWd = '';
let searchKey = '';
if(wdKey.length > 1){
searchKey = wdKey[0]
keyWd = wdKey[1];
}
console.log("keyWd",keyWd);
let weburl = `https://www.bing.com/search?q=${searchKey?searchKey:webkeywd}`;
const puppeteer = require('puppeteer');
const browser = await puppeteer.launch({
headless: true,
args: [
'--disable-gpu',
'--disable-dev-shm-usage',
'--disable-setuid-sandbox',
'--no-first-run',
'--no-sandbox',
'--no-zygote',
'--single-process'
]
});
const page = await browser.newPage();
await page.goto(weburl);
await page.setViewport({
width: 1920,
height: 1080
});
if(keyWd != '列表'){
let link = await page.evaluate(async keyWd => {
return [...document.querySelectorAll('.result a')].filter(item => {
return item.innerText && item.innerText.includes(keyWd)
})[0].toString();
},keyWd);
if(link){
// console.log("找到");
link = link.toString();
}else{
// console.log("没找到");
link = weburl;
}
console.log(link);
await page.goto(link);
}
await this.reply(segment.image(await page.screenshot({
fullPage: true
})))
await browser.close();
}
}