Chocobozzz/PeerTube

import { createWriteStream } from 'fs'
import { ensureDir } from 'fs-extra/esm'
import { dirname, join } from 'path'
import { pipeline } from 'stream'
import * as yauzl from 'yauzl'
import { logger, loggerTagsFactory } from './logger.js'

const lTags = loggerTagsFactory('unzip')

export async function unzip (source: string, destination: string) {
  await ensureDir(destination)

  logger.info(`Unzip ${source} to ${destination}`, lTags())

  return new Promise<void>((res, rej) => {
    yauzl.open(source, { lazyEntries: true }, (err, zipFile) => {
      if (err) return rej(err)

      zipFile.readEntry()

      zipFile.on('entry', async entry => {
        const entryPath = join(destination, entry.fileName)

        try {
          if (/\/$/.test(entry.fileName)) {
            await ensureDir(entryPath)
            logger.debug(`Creating directory from zip ${entryPath}`, lTags())

            zipFile.readEntry()
            return
          }

          await ensureDir(dirname(entryPath))
        } catch (err) {
          return rej(err)
        }

        zipFile.openReadStream(entry, (readErr, readStream) => {
          if (readErr) return rej(readErr)

          logger.debug(`Creating file from zip ${entryPath}`, lTags())

          const writeStream = createWriteStream(entryPath)
          writeStream.on('close', () => zipFile.readEntry())

          pipeline(readStream, writeStream, pipelineErr => {
            if (pipelineErr) return rej(pipelineErr)
          })
        })
      })

      zipFile.on('end', () => res())
    })
  })
}