修复注册扫描识别失败问题

2025-10-09 09:01:29 +08:00
parent f32feafd17
commit 1483f93b99
4 changed files with 245 additions and 26 deletions
@@ -0,0 +1,178 @@
+const ConfigManager = require('../config/config-manager');
+const { defaultLogger } = require('./logger');
+
+const logger = defaultLogger.child('ArtworkUtils');
+
+class ArtworkUtils {
+  constructor() {
+    this.configManager = new ConfigManager();
+    this._cachedConfig = null;
+    this._configCacheTime = 0;
+    this.CACHE_DURATION = 30000; // 30秒缓存
+  }
+
+  /**
+   * 获取配置（带缓存）
+   * @returns {Promise<Object>} 配置对象
+   */
+  async getConfig() {
+    const now = Date.now();
+    if (this._cachedConfig && (now - this._configCacheTime) < this.CACHE_DURATION) {
+      return this._cachedConfig;
+    }
+
+    this._cachedConfig = await this.configManager.readConfig();
+    this._configCacheTime = now;
+    return this._cachedConfig;
+  }
+
+  /**
+   * 从作品目录名中提取作品ID
+   * @param {string} artworkDir - 作品目录名
+   * @returns {Promise<number|null>} 作品ID，如果无法提取则返回null
+   */
+  async extractArtworkIdFromDir(artworkDir) {
+    try {
+      // 获取配置中的命名模式
+      const config = await this.getConfig();
+      const namingPattern = config.namingPattern || "{artist_name}/{artwork_id}_{title}";
+      
+      // 从命名模式中提取作品目录部分（去掉 {artist_name}/ 前缀）
+      let artworkPattern = namingPattern;
+      if (artworkPattern.includes('/')) {
+        artworkPattern = artworkPattern.split('/').pop(); // 取最后一部分
+      }
+      
+      // 将命名模式转换为正则表达式
+      // 替换占位符为对应的正则表达式组
+      let regexPattern = artworkPattern
+        .replace(/\{artwork_id\}/g, '(\\d+)')  // artwork_id 匹配数字
+        .replace(/\{title\}/g, '(.+)')         // title 匹配任意字符
+        .replace(/\{artist_name\}/g, '(.+)')   // artist_name 匹配任意字符（虽然在这里不应该出现）
+        .replace(/\{[^}]+\}/g, '(.+)');        // 其他占位符匹配任意字符
+      
+      // 转义特殊字符
+      regexPattern = regexPattern.replace(/[.*+?^${}()|[\]\\]/g, (match) => {
+        // 不转义我们添加的正则表达式组
+        if (match === '(' || match === ')' || match === '\\') {
+          return match;
+        }
+        return '\\' + match;
+      });
+      
+      // 创建正则表达式
+      const regex = new RegExp(`^${regexPattern}$`);
+      const match = artworkDir.match(regex);
+      
+      if (match) {
+        // 找到 artwork_id 在模式中的位置
+        const placeholders = artworkPattern.match(/\{[^}]+\}/g) || [];
+        const artworkIdIndex = placeholders.findIndex(placeholder => placeholder === '{artwork_id}');
+        
+        if (artworkIdIndex !== -1 && match[artworkIdIndex + 1]) {
+          return parseInt(match[artworkIdIndex + 1]);
+        }
+      }
+      
+      // 如果动态解析失败，回退到默认格式 {artwork_id}_{title}
+      const fallbackMatch = artworkDir.match(/^(\d+)_(.+)$/);
+      if (fallbackMatch) {
+        logger.debug(`使用回退模式解析作品目录: ${artworkDir}`);
+        return parseInt(fallbackMatch[1]);
+      }
+      
+      return null;
+    } catch (error) {
+      logger.warn(`解析作品目录名失败: ${artworkDir}`, error.message);
+      
+      // 发生错误时回退到默认格式
+      const fallbackMatch = artworkDir.match(/^(\d+)_(.+)$/);
+      if (fallbackMatch) {
+        return parseInt(fallbackMatch[1]);
+      }
+      
+      return null;
+    }
+  }
+
+  /**
+   * 检查目录名是否匹配作品目录格式
+   * @param {string} dirName - 目录名
+   * @returns {Promise<boolean>} 是否匹配作品目录格式
+   */
+  async isArtworkDirectory(dirName) {
+    const artworkId = await this.extractArtworkIdFromDir(dirName);
+    return artworkId !== null;
+  }
+
+  /**
+   * 从作品目录名中提取标题
+   * @param {string} artworkDir - 作品目录名
+   * @returns {Promise<string|null>} 作品标题，如果无法提取则返回null
+   */
+  async extractTitleFromDir(artworkDir) {
+    try {
+      // 获取配置中的命名模式
+      const config = await this.getConfig();
+      const namingPattern = config.namingPattern || "{artist_name}/{artwork_id}_{title}";
+      
+      // 从命名模式中提取作品目录部分（去掉 {artist_name}/ 前缀）
+      let artworkPattern = namingPattern;
+      if (artworkPattern.includes('/')) {
+        artworkPattern = artworkPattern.split('/').pop(); // 取最后一部分
+      }
+      
+      // 将命名模式转换为正则表达式
+      let regexPattern = artworkPattern
+        .replace(/\{artwork_id\}/g, '(\\d+)')  // artwork_id 匹配数字
+        .replace(/\{title\}/g, '(.+)')         // title 匹配任意字符
+        .replace(/\{artist_name\}/g, '(.+)')   // artist_name 匹配任意字符
+        .replace(/\{[^}]+\}/g, '(.+)');        // 其他占位符匹配任意字符
+      
+      // 转义特殊字符
+      regexPattern = regexPattern.replace(/[.*+?^${}()|[\]\\]/g, (match) => {
+        if (match === '(' || match === ')' || match === '\\') {
+          return match;
+        }
+        return '\\' + match;
+      });
+      
+      // 创建正则表达式
+      const regex = new RegExp(`^${regexPattern}$`);
+      const match = artworkDir.match(regex);
+      
+      if (match) {
+        // 找到 title 在模式中的位置
+        const placeholders = artworkPattern.match(/\{[^}]+\}/g) || [];
+        const titleIndex = placeholders.findIndex(placeholder => placeholder === '{title}');
+        
+        if (titleIndex !== -1 && match[titleIndex + 1]) {
+          return match[titleIndex + 1];
+        }
+      }
+      
+      // 如果动态解析失败，回退到默认格式 {artwork_id}_{title}
+      const fallbackMatch = artworkDir.match(/^(\d+)_(.+)$/);
+      if (fallbackMatch) {
+        return fallbackMatch[2];
+      }
+      
+      return null;
+    } catch (error) {
+      logger.warn(`解析作品标题失败: ${artworkDir}`, error.message);
+      
+      // 发生错误时回退到默认格式
+      const fallbackMatch = artworkDir.match(/^(\d+)_(.+)$/);
+      if (fallbackMatch) {
+        return fallbackMatch[2];
+      }
+      
+      return null;
+    }
+  }
+}
+
+// 创建单例实例
+const artworkUtils = new ArtworkUtils();
+
+module.exports = artworkUtils;