cherry-studio/src/main/utils/zip.ts

import util from 'node:util'
import zlib from 'node:zlib'

import { loggerService } from '@logger'

const logger = loggerService.withContext('Utils:Zip')

// 将 zlib 的 gzip 和 gunzip 方法转换为 Promise 版本
const gzipPromise = util.promisify(zlib.gzip)
const gunzipPromise = util.promisify(zlib.gunzip)

/**
 * 压缩字符串
 * @param {string} str 要压缩的 JSON 字符串
 * @returns {Promise<Buffer>} 压缩后的 Buffer
 */
export async function compress(str: string): Promise<Buffer> {
  try {
    const buffer = Buffer.from(str, 'utf-8')
    return await gzipPromise(buffer)
  } catch (error) {
    logger.error('Compression failed:', error as Error)
    throw error
  }
}

/**
 * 解压缩 Buffer 到 JSON 字符串
 * @param {Buffer} compressedBuffer - 压缩的 Buffer
 * @returns {Promise<string>} 解压缩后的 JSON 字符串
 */
export async function decompress(compressedBuffer: Buffer): Promise<string> {
  try {
    const buffer = await gunzipPromise(compressedBuffer)
    return buffer.toString('utf-8')
  } catch (error) {
    logger.error('Decompression failed:', error as Error)
    throw error
  }
}