124 lines
3.3 KiB
TypeScript
124 lines
3.3 KiB
TypeScript
/**
|
|
* download 命令模块
|
|
* 用于从 Kevisual 平台下载项目文件
|
|
*/
|
|
import path from 'path';
|
|
import fs from 'fs';
|
|
import { queryLogin } from '@/module/query.ts';
|
|
import { program, Command } from '@/program.ts';
|
|
import { fetchLink } from '@/module/download/install.ts';
|
|
import { chalk } from '@/module/chalk.ts';
|
|
|
|
/**
|
|
* 文件项类型定义
|
|
* 代表从 Kevisual 平台获取的文件元信息
|
|
*/
|
|
export type FileItem = {
|
|
/** 文件名称，包含完整路径 */
|
|
name: string;
|
|
/** 文件大小（字节） */
|
|
size: number;
|
|
/** 最后修改时间 */
|
|
lastModified: string;
|
|
/** 文件 ETag 值，用于缓存验证 */
|
|
etag: string;
|
|
/** 相对路径 */
|
|
path: string;
|
|
/** 完整路径名 */
|
|
pathname: string;
|
|
/** 文件下载 URL */
|
|
url: string;
|
|
};
|
|
|
|
const downloadCmd = new Command('download')
|
|
.description('下载项目')
|
|
.option('-l, --link <link>', '下载链接')
|
|
.option('-d, --directory <directory>', '下载目录', process.cwd())
|
|
.action(async (opts) => {
|
|
let link = opts.link || '';
|
|
if (!link) {
|
|
console.log('请提供下载链接');
|
|
return;
|
|
}
|
|
let url = new URL(link);
|
|
if (!url.pathname.endsWith('/')) {
|
|
url.pathname += '/';
|
|
}
|
|
url.searchParams.set('recursive', 'true');
|
|
const directory = opts.directory || process.cwd();
|
|
const token = await queryLogin.getToken();
|
|
|
|
const res = await queryLogin.query.fetchText({
|
|
url: url.toString(),
|
|
method: 'GET',
|
|
headers: {
|
|
'Authorization': `Bearer ${token}`
|
|
}
|
|
});
|
|
|
|
if (res.code === 200 && res.data) {
|
|
const files = res.data as FileItem[];
|
|
console.log(`获取到 ${files.length} 个文件`);
|
|
await downloadFiles(files, { directory });
|
|
} else {
|
|
console.log(chalk.red('获取文件列表失败:'), res.message || '未知错误');
|
|
}
|
|
});
|
|
|
|
program.addCommand(downloadCmd);
|
|
|
|
/**
|
|
* 下载文件列表
|
|
* @param files 文件列表
|
|
* @param opts 下载选项
|
|
* @param opts.directory 下载目录，默认为当前目录
|
|
*/
|
|
export const downloadFiles = async (files: FileItem[], opts?: { directory?: string }) => {
|
|
const directory = opts?.directory || process.cwd();
|
|
let successCount = 0;
|
|
let failCount = 0;
|
|
|
|
for (const file of files) {
|
|
try {
|
|
const downloadPath = path.join(directory, file.path);
|
|
const dir = path.dirname(downloadPath);
|
|
|
|
// 创建目录
|
|
if (!fs.existsSync(dir)) {
|
|
fs.mkdirSync(dir, { recursive: true });
|
|
}
|
|
|
|
// 下载文件
|
|
console.log(`下载中: ${file.name}`);
|
|
const { blob, type } = await fetchLink(file.url);
|
|
|
|
// 检查是否为错误响应
|
|
if (type.includes('text/html')) {
|
|
const text = await blob.text();
|
|
if (text === 'fetchRes is error') {
|
|
console.log(chalk.red('下载失败:'), file.name);
|
|
failCount++;
|
|
continue;
|
|
}
|
|
}
|
|
|
|
// 写入文件
|
|
fs.writeFileSync(downloadPath, Buffer.from(await blob.arrayBuffer()));
|
|
successCount++;
|
|
console.log(chalk.green('下载成功:'), file.name);
|
|
} catch (error) {
|
|
failCount++;
|
|
console.log(chalk.red('下载失败:'), file.name, error);
|
|
}
|
|
}
|
|
|
|
console.log(chalk.blue('下载完成'));
|
|
console.log(chalk.green(`成功: ${successCount}`));
|
|
console.log(chalk.red(`失败: ${failCount}`));
|
|
|
|
return {
|
|
successCount,
|
|
failCount,
|
|
};
|
|
};
|