299 lines
13 KiB
TypeScript
299 lines
13 KiB
TypeScript
/*
|
|
METADATA
|
|
{
|
|
"name": "douyin_download",
|
|
|
|
"display_name": {
|
|
"zh": "抖音下载工具",
|
|
"en": "Douyin Download Tool"
|
|
},
|
|
"description": { "zh": "抖音工具包，提供从分享链接或分享口令中提取并下载无水印视频的功能。", "en": "Douyin toolkit for extracting and downloading watermark-free videos from share links or share codes." },
|
|
"enabledByDefault": true,
|
|
"category": "Media",
|
|
"tools": [
|
|
{
|
|
"name": "get_douyin_download_link",
|
|
"description": { "zh": "解析抖音分享链接或口令，下载无水印视频到本地", "en": "Parse a Douyin share link/code and download a watermark-free video to local storage." },
|
|
"parameters": [
|
|
{
|
|
"name": "input",
|
|
"description": { "zh": "抖音分享链接或包含链接的分享口令文本", "en": "Douyin share link, or share-code text that contains a link." },
|
|
"type": "string",
|
|
"required": true
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "get_douyin_video_info",
|
|
"description": { "zh": "解析抖音分享链接或口令，仅获取视频信息和无水印下载链接，不下载视频", "en": "Parse a Douyin share link/code and return video info + watermark-free download URL (without downloading)." },
|
|
"parameters": [
|
|
{
|
|
"name": "input",
|
|
"description": { "zh": "抖音分享链接或包含链接的分享口令文本", "en": "Douyin share link, or share-code text that contains a link." },
|
|
"type": "string",
|
|
"required": true
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}*/
|
|
const douyin = (function () {
|
|
/**
|
|
* 包装函数调用，提供标准化的成功/错误处理
|
|
*/
|
|
async function douyin_wrap<T, U>(
|
|
func: (params: T) => Promise<U>,
|
|
params: T,
|
|
successMessage: string,
|
|
failMessage: string
|
|
): Promise<void> {
|
|
try {
|
|
console.log(`开始执行函数: ${func.name || '匿名函数'}`);
|
|
const result = await func(params);
|
|
complete({ success: true, message: successMessage, data: result });
|
|
} catch (error: unknown) {
|
|
console.error(`函数 ${func.name || '匿名函数'} 执行失败! 错误: ${error instanceof Error ? error.message : String(error)}`);
|
|
complete({ success: false, message: `${failMessage}: ${error instanceof Error ? error.message : String(error)}`, error_stack: error instanceof Error ? error.stack : undefined });
|
|
}
|
|
}
|
|
|
|
const client = OkHttp.newClient();
|
|
|
|
// 抖音相关URL模式
|
|
const DOUYIN_URL_PATTERNS = [
|
|
/https?:\/\/v\.douyin\.com\/[^\s]+/,
|
|
/https?:\/\/www\.douyin\.com\/video\/[0-9]+/,
|
|
/https?:\/\/www\.douyin\.com\/share\/video\/[0-9]+/
|
|
];
|
|
|
|
/**
|
|
* 从文本中提取抖音链接
|
|
*/
|
|
function extractDouyinUrl(text: string): string | null {
|
|
for (const pattern of DOUYIN_URL_PATTERNS) {
|
|
const match = text.match(pattern);
|
|
if (match) {
|
|
return match[0];
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
/**
|
|
* 解析抖音分享链接，获取视频信息
|
|
*/
|
|
async function resolveDouyinUrl(shareUrl: string): Promise<{ videoId: string, downloadUrl: string, videoTitle: string }> {
|
|
try {
|
|
console.log(`正在解析抖音链接: ${shareUrl}`);
|
|
|
|
const request = client.newRequest()
|
|
.url(shareUrl)
|
|
.method('GET')
|
|
.headers({
|
|
"User-Agent": "Mozilla/5.0 (iPhone; CPU iPhone OS 17_2 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) EdgiOS/121.0.2277.107 Version/17.0 Mobile/15E148 Safari/604.1",
|
|
"Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8",
|
|
"Accept-Language": "zh-CN,zh;q=0.9,en;q=0.8",
|
|
"Connection": "keep-alive"
|
|
});
|
|
|
|
const response = await request.build().execute();
|
|
|
|
if (!response.isSuccessful()) {
|
|
throw new Error(`链接解析失败 (${response.statusCode})`);
|
|
}
|
|
|
|
const html = response.content;
|
|
|
|
// 尝试从ROUTER_DATA中提取视频信息
|
|
const routerDataMatch = html.match(/window\._ROUTER_DATA\s*=\s*({[\s\S]*?})<\/script>/);
|
|
if (routerDataMatch && routerDataMatch[1]) {
|
|
try {
|
|
const routerData = JSON.parse(routerDataMatch[1]);
|
|
// The actual key might vary, let's find it dynamically
|
|
const pageDataKey = Object.keys(routerData.loaderData).find(k => k.includes('/page'));
|
|
if (!pageDataKey) {
|
|
throw new Error("无法在 _ROUTER_DATA 中找到页面数据");
|
|
}
|
|
const videoData = routerData.loaderData[pageDataKey]?.videoInfoRes?.item_list?.[0];
|
|
|
|
if (videoData) {
|
|
const videoId = videoData.aweme_id;
|
|
const videoTitle = videoData.desc;
|
|
const watermarkedUrl = videoData.video?.play_addr?.url_list?.[0];
|
|
|
|
if (videoId && videoTitle && watermarkedUrl) {
|
|
console.log(`通过 _ROUTER_DATA 找到视频信息: ID=${videoId}`);
|
|
const downloadUrl = watermarkedUrl.replace("playwm", "play");
|
|
return { videoId, downloadUrl, videoTitle };
|
|
}
|
|
}
|
|
} catch (e: unknown) {
|
|
console.error(`解析 _ROUTER_DATA JSON 失败: ${e instanceof Error ? e.message : String(e)}`);
|
|
}
|
|
}
|
|
|
|
throw new Error("无法从HTML中解析出 _ROUTER_DATA");
|
|
|
|
} catch (error) {
|
|
console.error(`解析抖音链接时出错: ${error instanceof Error ? error.message : String(error)}`);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取抖音视频的无水印下载链接并下载视频
|
|
*/
|
|
async function get_douyin_download_link(params: { input: string }): Promise<string> {
|
|
const { input } = params;
|
|
|
|
if (!input) {
|
|
throw new Error("输入内容不能为空");
|
|
}
|
|
|
|
console.log("开始获取并下载抖音无水印视频...");
|
|
|
|
try {
|
|
// 1. 提取和解析URL
|
|
const douyinUrl = extractDouyinUrl(input);
|
|
if (!douyinUrl) {
|
|
throw new Error("未找到有效的抖音链接");
|
|
}
|
|
console.log(`找到抖音链接: ${douyinUrl}`);
|
|
|
|
// 2. 解析链接并获取所有需要的视频信息
|
|
console.log("正在解析分享链接...");
|
|
const { videoId, downloadUrl, videoTitle: rawVideoTitle } = await resolveDouyinUrl(douyinUrl);
|
|
console.log(`解析到视频ID: ${videoId}`);
|
|
console.log(`获取到无水印下载链接: ${downloadUrl}`);
|
|
|
|
// 获取视频标题（用于文件命名）
|
|
let videoTitle = (rawVideoTitle || `douyin_${videoId}`).trim();
|
|
// 替换文件名中的非法字符
|
|
videoTitle = videoTitle.replace(/[\\/:*?"<>|]/g, '_');
|
|
|
|
// 逐级创建目录并下载视频
|
|
const baseDir = getPluginConfigDir("douyin_download");
|
|
const destinationDir = `${baseDir}/downloads`;
|
|
const destinationPath = `${destinationDir}/${videoTitle}_${videoId}.mp4`;
|
|
|
|
console.log(`确保目录存在: ${destinationDir}`);
|
|
// 逐级创建目录以确保路径存在
|
|
const dirsToCreate = [baseDir, destinationDir];
|
|
for (const dir of dirsToCreate) {
|
|
const mkdirResult = await Tools.Files.mkdir(dir);
|
|
if (!mkdirResult.successful) {
|
|
// 忽略"目录已存在"的错误，但记录其他可能的错误
|
|
console.warn(`创建目录 '${dir}' 失败 (可能已存在): ${mkdirResult.details}`);
|
|
}
|
|
}
|
|
|
|
console.log(`开始下载视频到: ${destinationPath}`);
|
|
const downloadResult = await Tools.Files.download(downloadUrl, destinationPath);
|
|
|
|
if (!downloadResult.successful) {
|
|
throw new Error(`视频下载失败: ${downloadResult.details}`);
|
|
}
|
|
|
|
const successMessage = `视频"${videoTitle}"成功下载到: ${destinationPath}`;
|
|
console.log(successMessage);
|
|
return successMessage;
|
|
|
|
} catch (error) {
|
|
const message = error instanceof Error ? error.message : String(error);
|
|
console.error(`处理失败: ${message}`);
|
|
// 重新抛出错误，以便包装器可以捕获它
|
|
throw new Error(message);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取抖音视频信息和无水印下载链接（不下载视频）
|
|
*/
|
|
async function get_douyin_video_info(params: { input: string }): Promise<{ videoId: string; videoTitle: string; downloadUrl: string; originalUrl: string }> {
|
|
const { input } = params;
|
|
|
|
if (!input) {
|
|
throw new Error("输入内容不能为空");
|
|
}
|
|
|
|
console.log("开始解析抖音视频信息...");
|
|
|
|
try {
|
|
// 1. 提取URL
|
|
const douyinUrl = extractDouyinUrl(input);
|
|
if (!douyinUrl) {
|
|
throw new Error("未找到有效的抖音链接");
|
|
}
|
|
console.log(`找到抖音链接: ${douyinUrl}`);
|
|
|
|
// 2. 解析链接并获取视频信息
|
|
console.log("正在解析分享链接...");
|
|
const { videoId, downloadUrl, videoTitle } = await resolveDouyinUrl(douyinUrl);
|
|
console.log(`解析到视频ID: ${videoId}`);
|
|
console.log(`获取到无水印下载链接: ${downloadUrl}`);
|
|
console.log(`视频标题: ${videoTitle}`);
|
|
|
|
const result = {
|
|
videoId,
|
|
videoTitle,
|
|
downloadUrl,
|
|
originalUrl: douyinUrl
|
|
};
|
|
|
|
console.log("视频信息解析成功");
|
|
return result;
|
|
|
|
} catch (error) {
|
|
const message = error instanceof Error ? error.message : String(error);
|
|
console.error(`解析失败: ${message}`);
|
|
throw new Error(message);
|
|
}
|
|
}
|
|
|
|
//region 测试函数
|
|
/**
|
|
* 抖音工具功能测试主函数
|
|
*/
|
|
async function main() {
|
|
console.log("🚀 开始执行抖音工具功能测试...");
|
|
await test_download_link();
|
|
const summary = `抖音下载工具测试完成！\n` +
|
|
`✅ 下载功能已测试`;
|
|
console.log(`\n${summary}`);
|
|
return summary;
|
|
}
|
|
|
|
/**
|
|
* 测试下载功能
|
|
*/
|
|
async function test_download_link() {
|
|
const testUrl = "3.00 05/24 Z@M.JI TLW:/ 这里有几首melodic dubstep，看你认识几个 # 电子音乐 # 音乐分享 # 顶级旋律 # 热门音乐🔥百听不厌 # 戴上耳机 https://v.douyin.com/AT8AfEbuP_k/ 复制此链接，打开Dou音搜索，直接观看视频！"; // 使用真实分享文本进行测试
|
|
console.log(`1. 测试视频下载功能 (使用真实分享文本)`);
|
|
// 我们预期这个调用会成功提取URL并尝试下载
|
|
try {
|
|
const result = await get_douyin_download_link({ input: testUrl });
|
|
console.log(`✅ 下载功能测试成功, 结果: ${result}`);
|
|
return result;
|
|
} catch (e) {
|
|
const errorMessage = e instanceof Error ? e.message : String(e);
|
|
console.error(`❌ 下载功能测试失败: ${errorMessage}`);
|
|
// 在测试中，即使是预期的失败，也应该被视为一个需要注意的问题
|
|
throw e;
|
|
}
|
|
}
|
|
//endregion
|
|
|
|
return {
|
|
get_douyin_download_link: (params: { input: string; }) => douyin_wrap(get_douyin_download_link, params, '抖音视频下载完成', '抖音视频下载失败'),
|
|
get_douyin_video_info: (params: { input: string; }) => douyin_wrap(get_douyin_video_info, params, '抖音视频信息获取成功', '抖音视频信息获取失败'),
|
|
main: (params: Record<string, never>) => douyin_wrap(main, params, '抖音工具测试完成', '抖音工具测试失败'),
|
|
test_download_link: (params: Record<string, never>) => douyin_wrap(test_download_link, params, '下载链接获取测试成功', '下载链接获取测试失败'),
|
|
};
|
|
})();
|
|
|
|
// 导出所有功能
|
|
exports.get_douyin_download_link = douyin.get_douyin_download_link;
|
|
exports.get_douyin_video_info = douyin.get_douyin_video_info;
|
|
exports.main = douyin.main;
|
|
exports.test_download_link = douyin.test_download_link;
|