findme-miniprogram-frontend/utils/cos-manager-test.js

/**
 * 腾讯云COS管理器 - 完整版
 * 支持MD5去重、文件记录保存、头像上传和动态图片上传两种场景
 * 参考文档: https://cloud.tencent.com/document/product/436/31953
 */

const COS = require('../dist/cos-wx-sdk-v5.min.js');
const config = require('../config/config.js');
const apiClient = require('./api-client.js');
const errorHandler = require('./error-handler.js');
const SparkMD5 = require('../dist/spark-md5.min.js');

class COSManager {
  constructor() {
    this.cosInstance = null;
    this.credentials = null; // 缓存的临时密钥
    this.credentialsExpireTime = 0; // 密钥过期时间
    this.isInitialized = false;
  }

  /**
   * 初始化COS实例
   */
  init() {
    if (this.isInitialized && this.cosInstance) {
      return;
    }

    try {
      this.cosInstance = new COS({
        // 强烈推荐: 高级上传内部对小文件使用putObject
        SimpleUploadMethod: 'putObject',
        
        getAuthorization: (options, callback) => {
          // 从后端获取临时密钥
          this.getCredentials(options.Scope?.[0]?.action)
            .then(credentials => {
              callback({
                TmpSecretId: credentials.tmpSecretId,
                TmpSecretKey: credentials.tmpSecretKey,
                SecurityToken: credentials.sessionToken,
                StartTime: credentials.startTime,
                ExpiredTime: credentials.expiredTime
              });
            })
            .catch(error => {
              console.error('[COS] 获取临时密钥失败:', error);
              callback({
                TmpSecretId: '',
                TmpSecretKey: '',
                SecurityToken: '',
                StartTime: 0,
                ExpiredTime: 0
              });
            });
        }
      });

      this.isInitialized = true;
      console.log('[COS] 初始化成功');
    } catch (error) {
      console.error('[COS] 初始化失败:', error);
      errorHandler.handleError(error, {
        action: 'cos_init',
        showToast: true
      });
      throw error;
    }
  }

  /**
   * 获取临时密钥
   * @param {string} fileType - 文件类型 (image/video/audio/file/avatar)
   * @returns {Promise<Object>} 临时密钥信息
   */
  async getCredentials(fileType = 'image') {
    try {
      // 检查缓存的密钥是否还有效(提前5分钟过期)
      const now = Math.floor(Date.now() / 1000);
      if (this.credentials && this.credentialsExpireTime > now + 300) {
        return this.credentials;
      }

      // 从后端获取新的临时密钥
      const response = await apiClient.get(config.cos.stsUrl, { fileType });

      if (response.code === 200 && response.data) {
        this.credentials = response.data;
        this.credentialsExpireTime = response.data.expiredTime;
        return this.credentials;
      }

      throw new Error(response.message || '获取临时密钥失败');
    } catch (error) {
      console.error('[COS] 获取临时密钥失败:', error);
      throw error;
    }
  }

  /**
   * 计算文件MD5
   * @param {string} filePath - 微信临时文件路径
   * @returns {Promise<string>} MD5值
   */
  async calculateMD5(filePath) {
    return new Promise((resolve, reject) => {
      try {
        const fileSystemManager = wx.getFileSystemManager();

        // 读取文件内容
        fileSystemManager.readFile({
          filePath,
          success: (res) => {
            try {
              // 使用SparkMD5计算MD5
              const spark = new SparkMD5.ArrayBuffer();
              spark.append(res.data);
              const md5 = spark.end();
              resolve(md5);
            } catch (error) {
              console.error('[COS] MD5计算失败:', error);
              reject(error);
            }
          },
          fail: (error) => {
            console.error('[COS] 读取文件失败:', error);
            reject(error);
          }
        });
      } catch (error) {
        console.error('[COS] 计算MD5异常:', error);
        reject(error);
      }
    });
  }

  /**
   * 检查文件是否已存在(用于去重)
   * @param {string} md5Hash - 文件MD5值
   * @returns {Promise<Object>} 检查结果 {exists, fileUrl, fileId, fileSize}
   */
  async checkFileExists(md5Hash) {
    try {
      const response = await apiClient.post(config.cos.fileCheckUrl, { md5Hash });

      if (response.code === 200 && response.data) {
        return response.data;
      }

      return { exists: false };
    } catch (error) {
      console.error('[COS] 检查文件存在性失败:', error);
      // 检查失败不影响上传流程,返回不存在
      return { exists: false };
    }
  }

  /**
   * 保存文件记录到后端
   * @param {Object} fileInfo - 文件信息
   * @returns {Promise<Object>} 保存结果
   */
  async saveFileRecord(fileInfo) {
    try {
      const response = await apiClient.post(config.cos.fileRecordUrl, fileInfo);

      if (response.code === 200) {
        return response.data;
      }

      throw new Error(response.message || '保存文件记录失败');
    } catch (error) {
      console.error('[COS] 保存文件记录失败:', error);
      throw error;
    }
  }

  /**
   * 生成文件路径
   * @param {string} fileName - 原始文件名
   * @param {string} fileType - 文件类型
   * @param {number} userId - 用户ID
   * @returns {string} 完整文件路径
   */
  generateFilePath(fileName, fileType, userId) {
    const timestamp = Math.floor(Date.now() / 1000);
    // 路径格式: user/{userId}/{fileType}/{timestamp}_{fileName}
    return `user/${userId}/${fileType}/${timestamp}_${fileName}`;
  }

  /**
   * 生成文件URL
   * @param {string} objectPath - 对象路径
   * @param {Object|string} thumbParams - 缩略图参数，支持数组或字符串
   * @returns {string} 完整URL
   */
  generateFileUrl(objectPath, thumbParams) {
    const { bucket, region } = config.cos;
    let baseUrl = `https://${bucket}.cos.${region}.myqcloud.com/${objectPath}`;
    
    // 处理缩略图参数
    if (thumbParams) {
      let paramStr = '';
      // 处理旧方法：字符串参数
      if (typeof thumbParams === 'string' && thumbParams) {
        paramStr = thumbParams;
      } 
      // 处理新方法：数组参数
      else if (Array.isArray(thumbParams) && thumbParams.length) {
        paramStr = thumbParams.join(',');
      }
      // 处理对象参数
      else if (typeof thumbParams === 'object' && thumbParams !== null) {
        const allowedParams = ['w', 'h', 'l', 's', 'q', 'r', 'g', 'bg', 'x', 'y', 'a', 'b', 'p', 'th'];
        const paramItems = [];
        for (const key in thumbParams) {
          if (allowedParams.includes(key) && thumbParams[key] !== undefined) {
            paramItems.push(`${key}_${thumbParams[key]}`);
          }
        }
        paramStr = paramItems.join(',');
      }
      
      if (paramStr) {
        const separator = baseUrl.includes('?') ? '&' : '?';
        baseUrl += `${separator}imageMogr2/auto-orient/thumbnail/${paramStr}`;
      }
    }
    
    return baseUrl;
  }

  /**
   * 核心上传方法 - 统一接口
   * @param {string|string[]} filePath - 文件路径(单个或数组)
   * @param {Object} options - 上传配置
   * @param {string} options.fileType - 文件类型: 'avatar'|'image'|'video'|'audio' (默认'image')
   * @param {boolean} options.enableDedup - 是否去重 (默认false)
   * @param {Function} options.onProgress - 进度回调
   * @param {Object|string|Array} options.thumbParams - 缩略图参数，参考腾讯COS API
   * @returns {Promise<Object|Array>} 上传结果(单个对象或数组)
   */
  async upload(filePath, { fileType = 'image', enableDedup = false, onProgress, thumbParams } = {}) {
    // 批量上传
    if (Array.isArray(filePath)) {
      return this._uploadMultiple(filePath, { fileType, enableDedup, onProgress, thumbParams });
    }

    // 单文件上传
    return this._uploadSingle(filePath, { fileType, enableDedup, onProgress, thumbParams });
  }

  /**
   * 单文件上传(内部方法)
   */
  async _uploadSingle(filePath, { fileType, enableDedup, onProgress, thumbParams }) {
    try {
      if (!this.isInitialized) this.init();

      const userId = this._getUserId();
      const fileName = filePath.split('/').pop();
      const fileInfo = await this.getFileInfo(filePath);

      // 去重检查
      if (enableDedup) {
        const dedupResult = await this._handleDedup(filePath);
        if (dedupResult) {
          // 为去重文件添加缩略图参数
          if (thumbParams && dedupResult.fileUrl) {
            const baseUrl = dedupResult.fileUrl.split('?')[0];
            dedupResult.fileUrl = this.generateFileUrl(baseUrl.replace(/^https?:\/\/.*?\//, ''), thumbParams);
          }
          return dedupResult;
        }
      }

      // 上传流程
      const { objectPath, fileUrl, md5Hash } = await this._executeUpload(
        filePath, fileName, fileType, userId, onProgress, thumbParams
      );

      // 确定usageType(用于后端自动处理)
      const usageType = fileType === 'avatar' ? 'avatar' : 'feed';

      // 保存记录 - 同时保存原始URL和缩略图URL
      const recordData = {
        fileName, fileInfo, fileType, objectPath, fileUrl: fileUrl.split('?')[0], 
        md5Hash, usageType, relatedId: 0
      };
      if (thumbParams) {
        recordData.thumbUrl = fileUrl;
      }
      const recordResult = await this._saveRecord(recordData);

      return {
        success: true,
        fileUrl,
        originalUrl: fileUrl.split('?')[0],
        fileId: recordResult.fileId,
        fileSize: fileInfo.size,
        md5Hash,
        isDeduped: false,
        message: '文件上传成功'
      };
    } catch (error) {
      console.error('[COS] 上传失败:', error);
      errorHandler.handleError(error, {
        action: 'cos_upload',
        showToast: true,
        toastMessage: '文件上传失败,请重试'
      });
      throw error;
    }
  }

  /**
   * 批量上传(内部方法)
   */
  async _uploadMultiple(filePaths, { fileType, enableDedup, onProgress, thumbParams }) {
    const results = [];
    const total = filePaths.length;

    for (let i = 0; i < total; i++) {
      try {
        const result = await this._uploadSingle(filePaths[i], {
          fileType,
          enableDedup,
          thumbParams,
          onProgress: (percent) => {
            if (onProgress) {
              onProgress({
                current: i + 1,
                total,
                currentPercent: percent,
                overallPercent: Math.round(((i + percent / 100) / total) * 100)
              });
            }
          }
        });
        results.push(result);
      } catch (error) {
        console.error(`[COS] 文件 ${i + 1} 上传失败:`, error);
        results.push({
          success: false,
          error: error.message || '上传失败'
        });
      }
    }

    return results;
  }

  // ==================== 内部辅助方法 ====================

  /**
   * 获取用户ID
   */
  _getUserId() {
    const userInfo = wx.getStorageSync('userInfo');
    const customId = userInfo.user?.customId;
    if (!userInfo || !customId) {
      throw new Error('用户未登录');
    }
    return customId;
  }

  /**
   * 处理去重逻辑
   */
  async _handleDedup(filePath) {
    const md5Hash = await this.calculateMD5(filePath);
    const checkResult = await this.checkFileExists(md5Hash);

    if (checkResult.exists) {
      return {
        success: true,
        fileUrl: checkResult.fileUrl,
        fileId: checkResult.fileId,
        fileSize: checkResult.fileSize,
        md5Hash,
        isDeduped: true,
        message: '文件已存在,直接使用'
      };
    }
    return null;
  }

  /**
   * 执行上传到COS
   */
  async _executeUpload(filePath, fileName, fileType, userId, onProgress, thumbParams) {
    await this.getCredentials(fileType);

    const objectPath = this.generateFilePath(fileName, fileType, userId);
    const fileUrl = this.generateFileUrl(objectPath, thumbParams);

    await this.uploadToCOS({ filePath, key: objectPath, onProgress });
    const md5Hash = await this.calculateMD5(filePath);

    return { objectPath, fileUrl, md5Hash };
  }

  /**
   * 保存文件记录
   */
  async _saveRecord({ fileName, fileInfo, fileType, objectPath, fileUrl, md5Hash, usageType, relatedId, thumbUrl }) {
    return await this.saveFileRecord({
      fileName,
      fileSize: fileInfo.size,
      contentType: fileInfo.type || this.getContentType(fileName),
      fileType,
      objectPath,
      fileUrl,
      thumbUrl, // 新增缩略图URL记录
      md5Hash,
      usageType,
      relatedId
    });
  }

  /**
   * 直接上传到COS(不经过去重和记录保存)
   * 使用 uploadFile 高级接口(官方强烈推荐)
   * 自动判断小文件用putObject,大文件用分块上传
   * @param {Object} options - 上传选项
   * @returns {Promise<void>}
   */
  uploadToCOS({ filePath, key, onProgress }) {
    return new Promise((resolve, reject) => {
      // 使用 uploadFile 高级接口(推荐)
      // 小文件自动使用putObject,大文件自动使用分块上传
      this.cosInstance.uploadFile({
        Bucket: config.cos.bucket,
        Region: config.cos.region,
        Key: key,
        FilePath: filePath,
        SliceSize: 1024 * 1024 * 5, // 大于5MB使用分块上传
        onProgress: (progressData) => {
          if (onProgress) {
            const percent = Math.round(progressData.percent * 100);
            onProgress(percent);
          }
        }
      }, (err, data) => {
        if (err) {
          console.error('[COS] 上传失败:', err.code, err.message);
          // 403错误特别提示
          if (err.statusCode === 403 || err.code === 'AccessDenied') {
            console.error('[COS] 权限错误,请检查临时密钥配置');
          }
          reject(err);
        } else {
          resolve(data);
        }
      });
    });
  }

  /**
   * 批量上传文件
   * @deprecated 请使用 upload(filePaths, options) 代替
   */
  async uploadFiles(files, options = {}) {
    console.warn('[COS] uploadFiles已废弃,请使用upload()方法');
    const filePaths = files.map(f => f.filePath || f);
    return this.upload(filePaths, options);
  }

  /**
   * 完整上传方法(保留向后兼容)
   * @deprecated 请使用 upload(filePath, options) 代替
   */
  async uploadFile(options) {
    console.warn('[COS] uploadFile已废弃,请使用upload()方法');
    const { filePath, fileType = 'image', enableDedup = false, onProgress, thumbParams } = options;
    return this.upload(filePath, { fileType, enableDedup, onProgress, thumbParams });
  }

  /**
   * 获取文件信息
   * @param {string} filePath - 文件路径
   * @returns {Promise<Object>} 文件信息
   */
  getFileInfo(filePath) {
    return new Promise((resolve, reject) => {
      wx.getFileInfo({
        filePath,
        success: (res) => {
          // 推断文件类型
          const ext = filePath.split('.').pop().toLowerCase();
          const type = this.getContentType('file.' + ext);

          resolve({
            size: res.size,
            type
          });
        },
        fail: reject
      });
    });
  }

  /**
   * 根据文件名获取Content-Type
   * @param {string} fileName - 文件名
   * @returns {string} Content-Type
   */
  getContentType(fileName) {
    const ext = fileName.split('.').pop().toLowerCase();
    const mimeTypes = {
      // 图片
      jpg: 'image/jpeg',
      jpeg: 'image/jpeg',
      png: 'image/png',
      gif: 'image/gif',
      webp: 'image/webp',
      bmp: 'image/bmp',
      // 视频
      mp4: 'video/mp4',
      avi: 'video/x-msvideo',
      mov: 'video/quicktime',
      wmv: 'video/x-ms-wmv',
      '3gp': 'video/3gpp',
      // 音频
      mp3: 'audio/mpeg',
      wav: 'audio/wav',
      ogg: 'audio/ogg',
      aac: 'audio/aac',
      m4a: 'audio/mp4',
      // 其他
      pdf: 'application/pdf',
      doc: 'application/msword',
      docx: 'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
      xls: 'application/vnd.ms-excel',
      xlsx: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
      txt: 'text/plain',
      json: 'application/json'
    };

    return mimeTypes[ext] || 'application/octet-stream';
  }

  /**
   * 删除文件
   * @param {string} key - 文件key
   * @returns {Promise<void>}
   */
  deleteObject(key) {
    return new Promise((resolve, reject) => {
      if (!this.isInitialized) {
        this.init();
      }

      this.cosInstance.deleteObject({
        Bucket: config.cos.bucket,
        Region: config.cos.region,
        Key: key
      }, (err, data) => {
        if (err) {
          console.error('[COS] 删除文件失败:', err);
          reject(err);
        } else {
          resolve(data);
        }
      });
    });
  }

  /**
   * 获取文件访问URL
   * @param {string} key - 文件key
   * @param {Object} options - 选项
   * @param {Object|string|Array} options.thumbParams - 缩略图参数
   * @returns {string} 访问URL
   */
  getObjectUrl(key, options = {}) {
    if (!this.isInitialized) {
      this.init();
    }

    const baseUrl = this.cosInstance.getObjectUrl({
      Bucket: config.cos.bucket,
      Region: config.cos.region,
      Key: key,
      Sign: options.sign !== false, // 默认需要签名
      Expires: options.expires || 3600 // 默认1小时
    });
    
    // 处理缩略图参数
    if (options.thumbParams) {
      return this.generateFileUrl(key, options.thumbParams);
    }
    
    return baseUrl;
  }
}

// 导出单例
const cosManager = new COSManager();
module.exports = cosManager;