mirror of https://github.com/Chocobozzz/PeerTube
56 lines
1.5 KiB
TypeScript
56 lines
1.5 KiB
TypeScript
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())
|
|
})
|
|
})
|
|
}
|