mirror of
https://github.com/lyz05/danmaku.git
synced 2026-02-02 17:59:53 +08:00
124 lines
3.2 KiB
JavaScript
124 lines
3.2 KiB
JavaScript
const express = require("express");
|
||
const axios = require("axios");
|
||
const router = express.Router();
|
||
const URL = require("url");
|
||
const {
|
||
bilibili,
|
||
mgtv,
|
||
tencentvideo,
|
||
youku,
|
||
iqiyi,
|
||
gamer,
|
||
} = require("./api/base");
|
||
const list = [bilibili, mgtv, tencentvideo, youku, iqiyi, gamer];
|
||
const memory = require("../utils/memory");
|
||
const db = require("../utils/db");
|
||
|
||
// 返回对象{msg: "ok", title: "标题", content: []}
|
||
async function build_response(url, req) {
|
||
// 循环找最终url
|
||
for (let q = new URLSearchParams(URL.parse(url).query); q.has("url");) {
|
||
console.log("Redirecting to", url);
|
||
url = q.get("url");
|
||
q = new URLSearchParams(URL.parse(url).query);
|
||
}
|
||
console.log("Real url:", url);
|
||
// 测试url是否能下载
|
||
try {
|
||
await axios.get(url, {
|
||
headers: { "Accept-Encoding": "gzip,deflate,compress" }
|
||
});
|
||
} catch (e) {
|
||
console.log(e);
|
||
// 如果是 403 错误，不报错，继续执行
|
||
if (e.response && e.response.status === 403) {
|
||
console.log("访问视频页面 403 错误，有可能被防火墙拦了");
|
||
} else {
|
||
return { msg: "传入的链接非法！请检查链接是否能在浏览器正常打开" };
|
||
}
|
||
}
|
||
// 循环找到对应的解析器
|
||
let fc = undefined;
|
||
for (let item of list) {
|
||
if (url.indexOf(item.domain) !== -1) {
|
||
fc = item;
|
||
}
|
||
}
|
||
// 找不到对应的解析器
|
||
if (fc === undefined) {
|
||
return { "msg": "不支持的视频网址" };
|
||
}
|
||
// 捕获所有错误并添加日志
|
||
let ret;
|
||
try {
|
||
ret = await fc.work(url);
|
||
} catch (e) {
|
||
console.log(e);
|
||
let err = JSON.stringify(e, Object.getOwnPropertyNames(e));
|
||
db.errorInsert({
|
||
ip: req.ip,
|
||
url,
|
||
err
|
||
});
|
||
return { msg: "弹幕解析过程中程序报错退出，请等待管理员修复！或者换条链接试试！" };
|
||
}
|
||
return ret;
|
||
}
|
||
|
||
async function resolve(req, res) {
|
||
const url = req.query.url;
|
||
const download = (req.query.download === "on");
|
||
const ret = await build_response(url, req);
|
||
memory(); //显示内存使用量
|
||
if (ret.msg !== "ok") {
|
||
res.status(403).send(ret.msg);
|
||
return;
|
||
} else if (download) {
|
||
res.attachment(ret.title + ".xml");
|
||
} else {
|
||
res.type("application/xml");
|
||
}
|
||
// 记录视频信息
|
||
db.videoInfoInsert({url,title:ret.title})
|
||
//B站视频，直接重定向
|
||
if (ret.url)
|
||
res.redirect(ret.url);
|
||
else {
|
||
res.set('Cache-Control', 'public, max-age=86400'); // one year
|
||
res.render("danmaku-xml", { contents: ret.content });
|
||
}
|
||
}
|
||
|
||
async function index(req, res) {
|
||
const urls = list.map(item => item.example_urls[0]);
|
||
const names = list.map(item => item.name);
|
||
const path = req.protocol + "://" + req.headers.host + req.originalUrl;
|
||
const resolve_info = await db.accessCountQuery()
|
||
const hotlist = await db.hotlistQuery()
|
||
res.render("danmaku", {
|
||
path,
|
||
urls,
|
||
names,
|
||
resolve_info,
|
||
hotlist
|
||
});
|
||
}
|
||
|
||
/* GET home page. */
|
||
router.get("/", async function (req, res) {
|
||
db.accessInsert({
|
||
ip: req.ip,
|
||
url: req.query.url,
|
||
UA: req.headers["user-agent"]
|
||
});
|
||
//检查是否包含URL参数
|
||
if (!req.query.url) index(req, res); else resolve(req, res);
|
||
});
|
||
|
||
router.get("/delete", async function (req, res) {
|
||
const rows = db.deleteAccess();
|
||
res.send(`成功请求删除三个月以前的记录，删除情况请查看日志`);
|
||
});
|
||
|
||
module.exports = router;
|