nas/danmaku
1
0
Fork 0
mirror of https://github.com/lyz05/danmaku.git synced 2026-02-03 02:04:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复/sub的bug，增加自助完成注册的功能

Browse Source
This commit is contained in:
lyz05
2022-12-18 12:16:45 +08:00
parent 4ed7133a66
commit a9f92a1fd6
1 changed files with 153 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

243
tgbot/bot/airportsub.js
View File

@@ -3,16 +3,28 @@ const whacko = require("whacko");
const yaml = require("js-yaml");
const TelegramBot = require("node-telegram-bot-api");
const axios = require("axios");
const OSS = require("ali-oss");
const oss = require("../../utils/oss");
const goindex = require("./goindex");
// Environment variables
const OSS_OPTIONS = {
region: "oss-cn-hongkong",
accessKeyId: process.env.OSS_ACCESS_KEY,
accessKeySecret: process.env.OSS_ACCESS_KEY_SECRET,
bucket: "hkosslog",
};
async function finduserbychatid(chatid) {
const database = await oss.get("SUB/database.yaml");
const data = yaml.load(database);
const users = data.user;
// eslint-disable-next-line no-restricted-syntax
for (const user in users) {
if (users[user].chatID == chatid) {
return user;
}
}
return null;
}
async function setchatidbyuser(user, chatid) {
const database = await oss.get("SUB/database.yaml");
const data = yaml.load(database);
data.user[user].chatID = chatid;
oss.put("SUB/database.yaml", yaml.dump(data));
}
// function sleep(ms) {
// return new Promise((resolve) => setTimeout(resolve, ms));
@@ -22,16 +34,21 @@ module.exports = (TOKEN) => {
const game = {};
let setu = {};
const bot = new TelegramBot(TOKEN, { polling: true });
const client = new OSS(OSS_OPTIONS);
function sendSetu(chatId, i) {
const href = setu[i];
const prev = { text: "上一张", callback_data: i - 1 };
const next = { text: "下一张", callback_data: i + 1 };
const prev = {
text: "上一张",
callback_data: i - 1
};
const next = {
text: "下一张",
callback_data: i + 1
};
let replyMarkup = { inline_keyboard: [[prev, next]] };
if (i === 0) {
replyMarkup = { inline_keyboard: [[next]] };
} else if (i+1 === setu.length) {
} else if (i + 1 === setu.length) {
replyMarkup = { inline_keyboard: [[prev]] };
}
bot.sendMessage(chatId, href, { reply_markup: replyMarkup });
@@ -48,10 +65,11 @@ module.exports = (TOKEN) => {
bot.on("text", (msg) => {
if (msg.text.indexOf("/") === -1) {
bot.sendMessage(msg.chat.id, `you said: ${msg.text}`);
axios.get(`https://api.qingyunke.com/api.php?key=free&appid=0&msg=${encodeURI(msg.text)}`).then((res) => {
console.log(res.data);
bot.sendMessage(msg.chat.id, res.data.content);
});
axios.get(`https://api.qingyunke.com/api.php?key=free&appid=0&msg=${encodeURI(msg.text)}`)
.then((res) => {
console.log(res.data);
bot.sendMessage(msg.chat.id, res.data.content);
});
}
});
@@ -69,34 +87,41 @@ module.exports = (TOKEN) => {
// 发送用户头像
bot.onText(/\/sendpic/, (msg) => {
bot.getUserProfilePhotos(msg.chat.id).then((photos) => {
const photo = photos.photos[0][0];
bot.sendPhoto(msg.chat.id, photo.file_id, {
caption: "This is a picture of You!",
bot.getUserProfilePhotos(msg.chat.id)
.then((photos) => {
const photo = photos.photos[0][0];
bot.sendPhoto(msg.chat.id, photo.file_id, {
caption: "This is a picture of You!",
});
});
});
// bot.sendPhoto(msg.chat.id, "https://blog.home999.cc/images/avatar.jpg");
});
bot.onText(/\/register/, (msg) => {
bot.sendMessage(msg.chat.id, `Chat id: ${msg.chat.id}\n请把该id告诉管理员用于注册。`);
bot.onText(/\/register/, async (msg) => {
const user = await finduserbychatid(msg.chat.id);
if (user == null) {
const user = msg.text.replace("/register ", "");
if (msg.text === "/register") {
bot.sendMessage(msg.chat.id, `您的ChatId为: ${msg.chat.id}\n若要进行注册请跟上您的user信息，如： /register example`);
} else {
setchatidbyuser(user, msg.chat.id);
bot.sendMessage(msg.chat.id, "注册完成！");
}
} else {
bot.sendMessage(msg.chat.id, "您已经注册过了，请勿重复注册。");
}
});
bot.onText(/\/sub/, async (msg) => {
const database = await (await client.get("SUB/database.yaml")).content.toString();
const data = yaml.load(database);
const users = data.user;
// eslint-disable-next-line no-restricted-syntax
for (const user in users) {
if (users[user].chatID === msg.chat.id) {
bot.sendMessage(msg.chat.id, "您已经注册过了，请勿重复注册。");
bot.sendMessage(msg.chat.id, `你好，${user}。`);
const url = `https://fc.home999.cc/sub?user=${user}`;
bot.sendMessage(msg.chat.id, `您的订阅链接为：${url}`);
return;
}
const user = await finduserbychatid(msg.chat.id);
const url = `https://fc.home999.cc/sub?user=${user}`;
if (user == null) {
bot.sendMessage(msg.chat.id, "您未注册！请输入 /register 进行注册");
} else {
bot.sendMessage(msg.chat.id, `你好，${user}。`);
bot.sendMessage(msg.chat.id, `您的订阅链接为：${url}`);
}
bot.sendMessage(msg.chat.id, "您已经成功注册，请等待管理员审核");
});
// 猜数游戏
@@ -113,7 +138,10 @@ module.exports = (TOKEN) => {
await bot.sendMessage(chatID, "请输入你的猜测：(例：/game 50)");
return;
}
const { num, limit } = game[chatID];
const {
num,
limit
} = game[chatID];
if (limit <= 0) {
bot.sendMessage(chatID, `游戏结束！未猜出正确答案，正确答案为：${num}`);
game[chatID] = undefined;
@@ -132,16 +160,41 @@ module.exports = (TOKEN) => {
bot.onText(/\/help/, (msg) => {
const helpMsg = [
{ command: "start", description: "欢迎界面" },
{ command: "game", description: "猜数游戏" },
{ command: "sub", description: "订阅链接" },
{ command: "register", description: "注册" },
{ command: "sendpic", description: "发送你的头像" },
{ command: "setu", description: "随机色图，可加编号" },
{ command: "goindex", description: "查询GoIndex上的文件" },
{ command: "help", description: "帮助" },
{
command: "start",
description: "欢迎界面"
},
{
command: "game",
description: "猜数游戏"
},
{
command: "sub",
description: "订阅链接"
},
{
command: "register",
description: "注册"
},
{
command: "sendpic",
description: "发送你的头像"
},
{
command: "setu",
description: "随机色图，可加编号"
},
{
command: "goindex",
description: "查询GoIndex上的文件"
},
{
command: "help",
description: "帮助"
},
];
const helpMsgText = helpMsg.map((item) => `/${item.command} - ${item.description}`).join("\n");
const helpMsgText = helpMsg.map((item) => `/${item.command} - ${item.description}`)
.join("\n");
bot.sendMessage(msg.chat.id, helpMsgText, { parse_mode: "HTML" });
bot.setMyCommands(helpMsg);
});
@@ -151,7 +204,9 @@ module.exports = (TOKEN) => {
bot.sendMessage(msg.chat.id, "色图模式");
const res = await axios.get("https://asiantolick.com/ajax/buscar_posts.php", { params: { index } });
const $ = whacko.load(res.data);
setu = Object.values($(".miniatura")).map((item) => $(item).attr("href"));
setu = Object.values($(".miniatura"))
.map((item) => $(item)
.attr("href"));
sendSetu(msg.chat.id, 0);
});
@@ -165,51 +220,59 @@ module.exports = (TOKEN) => {
bot.onText(/\/goindex/, (msg) => {
const q = msg.text.replace("/goindex ", "");
bot.sendMessage(msg.chat.id, `正在搜寻“${q}”...`);
goindex.query(q).then((res) => {
// 筛选符合条件的文件
const videos = res.filter((e) => e.mimeType === "video/mp4").filter((e) => e.size < 50 * 1024 * 1024);
let images = res.filter((e) => e.mimeType === "image/jpeg");
const audios = res.filter((e) => e.mimeType === "audio/mp3").filter((e) => e.size < 50 * 1024 * 1024);
const folders = res.filter((e) => e.mimeType === "application/vnd.google-apps.folder");
goindex.query(q)
.then((res) => {
// 筛选符合条件的文件
const videos = res.filter((e) => e.mimeType === "video/mp4")
.filter((e) => e.size < 50 * 1024 * 1024);
let images = res.filter((e) => e.mimeType === "image/jpeg");
const audios = res.filter((e) => e.mimeType === "audio/mp3")
.filter((e) => e.size < 50 * 1024 * 1024);
const folders = res.filter((e) => e.mimeType === "application/vnd.google-apps.folder");
bot.sendMessage(msg.chat.id, `共有${images.length}个图片结果，${videos.length}个视频，${audios.length}个音乐，${folders.length}个目录，搜索结果：`);
bot.sendChatAction(msg.chat.id, "upload_photo");
images = goindex.group(images, 10);
images.forEach((e, i) => {
setTimeout(() => {
bot.sendMediaGroup(msg.chat.id, e.map((el) => ({
type: "photo",
media: el.thumbnailLink.replace("=s220", "=s0"),
caption: el.name,
})));
}, i * 2000);
bot.sendMessage(msg.chat.id, `共有${images.length}个图片结果，${videos.length}个视频，${audios.length}个音乐，${folders.length}个目录，搜索结果：`);
bot.sendChatAction(msg.chat.id, "upload_photo");
images = goindex.group(images, 10);
images.forEach((e, i) => {
setTimeout(() => {
bot.sendMediaGroup(msg.chat.id, e.map((el) => ({
type: "photo",
media: el.thumbnailLink.replace("=s220", "=s0"),
caption: el.name,
})));
}, i * 2000);
});
bot.sendChatAction(msg.chat.id, "upload_video");
videos.forEach((e, i) => {
setTimeout(() => {
goindex.id2path(e.id)
.then((path) => {
console.log(path);
bot.sendVideo(msg.chat.id, encodeURI(path), {
caption: `${e.name}`,
reply_markup: {
inline_keyboard: [
[{
text: "带我去看片",
url: encodeURI(path)
}],
],
},
});
});
}, i * 2000);
});
bot.sendChatAction(msg.chat.id, "upload_voice");
audios.forEach((e, i) => {
setTimeout(() => {
goindex.id2path(e.id)
.then((path) => {
console.log(path);
bot.sendAudio(msg.chat.id, path, { caption: `${e.name}` });
});
}, i * 2000);
});
});
bot.sendChatAction(msg.chat.id, "upload_video");
videos.forEach((e, i) => {
setTimeout(() => {
goindex.id2path(e.id).then((path) => {
console.log(path);
bot.sendVideo(msg.chat.id, encodeURI(path), {
caption: `${e.name}`,
reply_markup: {
inline_keyboard: [
[{ text: "带我去看片", url: encodeURI(path) }],
],
},
});
});
}, i * 2000);
});
bot.sendChatAction(msg.chat.id, "upload_voice");
audios.forEach((e, i) => {
setTimeout(() => {
goindex.id2path(e.id).then((path) => {
console.log(path);
bot.sendAudio(msg.chat.id, path, { caption: `${e.name}` });
});
}, i * 2000);
});
});
});
bot.onText(/\/senddice/, (msg) => {