mirror of
https://github.com/qaiu/netdisk-fast-download.git
synced 2026-08-30 05:22:01 +00:00
q
c71d4cfa87
🚀 核心功能 - 新增完整的客户端下载链接生成器系统 - 支持ARIA2、Motrix、比特彗星、迅雷、wget、cURL、IDM、FDM、PowerShell等8种客户端 - 自动处理防盗链参数（User-Agent、Referer、Cookie等） - 提供可扩展的生成器架构，支持自定义客户端 🔧 技术实现 - ClientLinkGeneratorFactory: 工厂模式管理生成器 - DownloadLinkMeta: 元数据存储下载信息 - ClientLinkUtils: 便捷工具类 - 线程安全的ConcurrentHashMap设计 🌐 前端集成 - 新增ClientLinks.vue界面，支持客户端链接展示 - Element Plus图标系统，混合图标显示 - 客户端检测逻辑优化，避免自动打开外部应用 - 移动端和PC端环境判断 📚 文档完善 - 完整的CLIENT_LINK_GENERATOR_GUIDE.md使用指南 - API文档和测试用例 - 输出示例和最佳实践 从单纯的网盘解析工具升级为完整的下载解决方案生态
351 lines
13 KiB
Java
351 lines
13 KiB
Java
package cn.qaiu.parser.customjs;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import cn.qaiu.parser.custom.CustomParserConfig;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.nio.charset.StandardCharsets;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.util.ArrayList;
|
|
import java.util.Enumeration;
|
|
import java.util.List;
|
|
import java.util.jar.JarEntry;
|
|
import java.util.jar.JarFile;
|
|
import java.util.stream.Stream;
|
|
|
|
/**
|
|
* JavaScript脚本加载器
|
|
* 自动加载资源目录和外部目录的JavaScript脚本文件
|
|
*
|
|
* @author <a href="https://qaiu.top">QAIU</a>
|
|
* Create at 2025/10/17
|
|
*/
|
|
public class JsScriptLoader {
|
|
|
|
private static final Logger log = LoggerFactory.getLogger(JsScriptLoader.class);
|
|
|
|
private static final String RESOURCE_PATH = "custom-parsers";
|
|
private static final String EXTERNAL_PATH = "./custom-parsers";
|
|
|
|
// 系统属性配置的外部目录路径
|
|
private static final String EXTERNAL_PATH_PROPERTY = "parser.custom-parsers.path";
|
|
|
|
/**
|
|
* 加载所有JavaScript脚本
|
|
* @return 解析器配置列表
|
|
*/
|
|
public static List<CustomParserConfig> loadAllScripts() {
|
|
List<CustomParserConfig> configs = new ArrayList<>();
|
|
|
|
// 1. 加载资源目录下的JS文件
|
|
try {
|
|
List<CustomParserConfig> resourceConfigs = loadFromResources();
|
|
configs.addAll(resourceConfigs);
|
|
log.info("从资源目录加载了 {} 个JavaScript解析器", resourceConfigs.size());
|
|
} catch (Exception e) {
|
|
log.warn("从资源目录加载JavaScript脚本失败", e);
|
|
}
|
|
|
|
// 2. 加载外部目录下的JS文件
|
|
try {
|
|
List<CustomParserConfig> externalConfigs = loadFromExternal();
|
|
configs.addAll(externalConfigs);
|
|
log.info("从外部目录加载了 {} 个JavaScript解析器", externalConfigs.size());
|
|
} catch (Exception e) {
|
|
log.warn("从外部目录加载JavaScript脚本失败", e);
|
|
}
|
|
|
|
log.info("总共加载了 {} 个JavaScript解析器", configs.size());
|
|
return configs;
|
|
}
|
|
|
|
/**
|
|
* 从资源目录加载JavaScript脚本
|
|
*/
|
|
private static List<CustomParserConfig> loadFromResources() {
|
|
List<CustomParserConfig> configs = new ArrayList<>();
|
|
|
|
try {
|
|
// 尝试使用反射方式获取JAR包内的资源文件列表
|
|
List<String> resourceFiles = getResourceFileList();
|
|
|
|
// 按文件名排序，确保加载顺序一致
|
|
resourceFiles.sort(String::compareTo);
|
|
|
|
for (String resourceFile : resourceFiles) {
|
|
try {
|
|
InputStream inputStream = JsScriptLoader.class.getClassLoader()
|
|
.getResourceAsStream(resourceFile);
|
|
|
|
if (inputStream != null) {
|
|
String jsCode = new String(inputStream.readAllBytes(), StandardCharsets.UTF_8);
|
|
CustomParserConfig config = JsScriptMetadataParser.parseScript(jsCode);
|
|
configs.add(config);
|
|
|
|
String fileName = resourceFile.substring(resourceFile.lastIndexOf('/') + 1);
|
|
log.debug("从资源目录加载脚本: {}", fileName);
|
|
}
|
|
} catch (Exception e) {
|
|
log.warn("加载资源脚本失败: {}", resourceFile, e);
|
|
}
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
log.error("从资源目录加载脚本时发生异常", e);
|
|
}
|
|
|
|
return configs;
|
|
}
|
|
|
|
/**
|
|
* 尝试使用反射方式获取JAR包内的资源文件列表
|
|
*/
|
|
private static List<String> getResourceFileList() {
|
|
List<String> resourceFiles = new ArrayList<>();
|
|
|
|
try {
|
|
// 尝试获取资源目录的URL
|
|
java.net.URL resourceUrl = JsScriptLoader.class.getClassLoader()
|
|
.getResource(RESOURCE_PATH);
|
|
|
|
if (resourceUrl != null) {
|
|
String protocol = resourceUrl.getProtocol();
|
|
|
|
if ("jar".equals(protocol)) {
|
|
// JAR包内的资源
|
|
resourceFiles = getJarResourceFiles(resourceUrl);
|
|
} else if ("file".equals(protocol)) {
|
|
// 文件系统中的资源（开发环境）
|
|
resourceFiles = getFileSystemResourceFiles(resourceUrl);
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
log.debug("使用反射方式获取资源文件列表失败，将使用预定义列表", e);
|
|
}
|
|
|
|
return resourceFiles;
|
|
}
|
|
|
|
/**
|
|
* 获取JAR包内的资源文件列表
|
|
*/
|
|
private static List<String> getJarResourceFiles(java.net.URL jarUrl) {
|
|
List<String> resourceFiles = new ArrayList<>();
|
|
|
|
try {
|
|
String jarPath = jarUrl.getPath().substring(5, jarUrl.getPath().indexOf("!"));
|
|
JarFile jarFile = new JarFile(jarPath);
|
|
|
|
Enumeration<JarEntry> entries = jarFile.entries();
|
|
while (entries.hasMoreElements()) {
|
|
JarEntry entry = entries.nextElement();
|
|
String entryName = entry.getName();
|
|
|
|
if (entryName.startsWith(RESOURCE_PATH + "/") &&
|
|
entryName.endsWith(".js") &&
|
|
!isExcludedFile(entryName.substring(entryName.lastIndexOf('/') + 1))) {
|
|
resourceFiles.add(entryName);
|
|
}
|
|
}
|
|
|
|
jarFile.close();
|
|
} catch (Exception e) {
|
|
log.debug("解析JAR包资源文件失败", e);
|
|
}
|
|
|
|
return resourceFiles;
|
|
}
|
|
|
|
/**
|
|
* 获取文件系统中的资源文件列表
|
|
*/
|
|
private static List<String> getFileSystemResourceFiles(java.net.URL fileUrl) {
|
|
List<String> resourceFiles = new ArrayList<>();
|
|
|
|
try {
|
|
java.io.File resourceDir = new java.io.File(fileUrl.getPath());
|
|
if (resourceDir.exists() && resourceDir.isDirectory()) {
|
|
java.io.File[] files = resourceDir.listFiles();
|
|
if (files != null) {
|
|
for (java.io.File file : files) {
|
|
if (file.isFile() && file.getName().endsWith(".js") &&
|
|
!isExcludedFile(file.getName())) {
|
|
resourceFiles.add(RESOURCE_PATH + "/" + file.getName());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
log.debug("解析文件系统资源文件失败", e);
|
|
}
|
|
|
|
return resourceFiles;
|
|
}
|
|
|
|
|
|
/**
|
|
* 从外部目录加载JavaScript脚本
|
|
*/
|
|
private static List<CustomParserConfig> loadFromExternal() {
|
|
List<CustomParserConfig> configs = new ArrayList<>();
|
|
|
|
try {
|
|
// 获取外部目录路径，支持系统属性配置
|
|
String externalPath = getExternalPath();
|
|
Path externalDir = Paths.get(externalPath);
|
|
|
|
if (!Files.exists(externalDir) || !Files.isDirectory(externalDir)) {
|
|
log.debug("外部目录 {} 不存在或不是目录", externalPath);
|
|
return configs;
|
|
}
|
|
|
|
try (Stream<Path> paths = Files.walk(externalDir)) {
|
|
paths.filter(Files::isRegularFile)
|
|
.filter(path -> path.toString().endsWith(".js"))
|
|
.filter(path -> !isExcludedFile(path.getFileName().toString()))
|
|
.forEach(path -> {
|
|
try {
|
|
String jsCode = Files.readString(path, StandardCharsets.UTF_8);
|
|
CustomParserConfig config = JsScriptMetadataParser.parseScript(jsCode);
|
|
configs.add(config);
|
|
log.debug("从外部目录加载脚本: {}", path.getFileName());
|
|
} catch (Exception e) {
|
|
log.warn("加载外部脚本失败: {}", path.getFileName(), e);
|
|
}
|
|
});
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
log.error("从外部目录加载脚本时发生异常", e);
|
|
}
|
|
|
|
return configs;
|
|
}
|
|
|
|
/**
|
|
* 获取外部目录路径
|
|
* 优先级：系统属性 > 环境变量 > 默认路径
|
|
*/
|
|
private static String getExternalPath() {
|
|
// 1. 检查系统属性
|
|
String systemProperty = System.getProperty(EXTERNAL_PATH_PROPERTY);
|
|
if (systemProperty != null && !systemProperty.trim().isEmpty()) {
|
|
log.debug("使用系统属性配置的外部目录: {}", systemProperty);
|
|
return systemProperty;
|
|
}
|
|
|
|
// 2. 检查环境变量
|
|
String envVariable = System.getenv("PARSER_CUSTOM_PARSERS_PATH");
|
|
if (envVariable != null && !envVariable.trim().isEmpty()) {
|
|
log.debug("使用环境变量配置的外部目录: {}", envVariable);
|
|
return envVariable;
|
|
}
|
|
|
|
// 3. 使用默认路径
|
|
log.debug("使用默认外部目录: {}", EXTERNAL_PATH);
|
|
return EXTERNAL_PATH;
|
|
}
|
|
|
|
/**
|
|
* 从指定文件加载JavaScript脚本
|
|
* @param filePath 文件路径
|
|
* @return 解析器配置
|
|
*/
|
|
public static CustomParserConfig loadFromFile(String filePath) {
|
|
try {
|
|
Path path = Paths.get(filePath);
|
|
if (!Files.exists(path)) {
|
|
throw new IllegalArgumentException("文件不存在: " + filePath);
|
|
}
|
|
|
|
String jsCode = Files.readString(path, StandardCharsets.UTF_8);
|
|
return JsScriptMetadataParser.parseScript(jsCode);
|
|
|
|
} catch (IOException e) {
|
|
throw new RuntimeException("读取文件失败: " + filePath, e);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 从指定文件加载JavaScript脚本（资源路径）
|
|
* @param resourcePath 资源路径
|
|
* @return 解析器配置
|
|
*/
|
|
public static CustomParserConfig loadFromResource(String resourcePath) {
|
|
try {
|
|
InputStream inputStream = JsScriptLoader.class.getClassLoader()
|
|
.getResourceAsStream(resourcePath);
|
|
|
|
if (inputStream == null) {
|
|
throw new IllegalArgumentException("资源文件不存在: " + resourcePath);
|
|
}
|
|
|
|
String jsCode = new String(inputStream.readAllBytes(), StandardCharsets.UTF_8);
|
|
return JsScriptMetadataParser.parseScript(jsCode);
|
|
|
|
} catch (IOException e) {
|
|
throw new RuntimeException("读取资源文件失败: " + resourcePath, e);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 检查外部目录是否存在
|
|
* @return true表示存在，false表示不存在
|
|
*/
|
|
public static boolean isExternalDirectoryExists() {
|
|
Path externalDir = Paths.get(EXTERNAL_PATH);
|
|
return Files.exists(externalDir) && Files.isDirectory(externalDir);
|
|
}
|
|
|
|
/**
|
|
* 创建外部目录
|
|
* @return true表示创建成功，false表示创建失败
|
|
*/
|
|
public static boolean createExternalDirectory() {
|
|
try {
|
|
Path externalDir = Paths.get(EXTERNAL_PATH);
|
|
Files.createDirectories(externalDir);
|
|
log.info("创建外部目录成功: {}", EXTERNAL_PATH);
|
|
return true;
|
|
} catch (IOException e) {
|
|
log.error("创建外部目录失败: {}", EXTERNAL_PATH, e);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取外部目录路径
|
|
* @return 外部目录路径
|
|
*/
|
|
public static String getExternalDirectoryPath() {
|
|
return EXTERNAL_PATH;
|
|
}
|
|
|
|
/**
|
|
* 获取资源目录路径
|
|
* @return 资源目录路径
|
|
*/
|
|
public static String getResourceDirectoryPath() {
|
|
return RESOURCE_PATH;
|
|
}
|
|
|
|
/**
|
|
* 检查文件是否应该被排除
|
|
* @param fileName 文件名
|
|
* @return true表示应该排除，false表示应该加载
|
|
*/
|
|
private static boolean isExcludedFile(String fileName) {
|
|
// 排除类型定义文件和其他非解析器文件
|
|
return fileName.equals("types.js") ||
|
|
fileName.equals("jsconfig.json") ||
|
|
fileName.equals("README.md") ||
|
|
fileName.contains(".test.") ||
|
|
fileName.contains(".spec.");
|
|
}
|
|
}
|