| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980 |
- const archiver = require('archiver')
- const OutputCacheManager = require('./OutputCacheManager')
- const OutputFileFinder = require('./OutputFileFinder')
- const Settings = require('@overleaf/settings')
- const { open } = require('node:fs/promises')
- const path = require('path')
- const { NotFoundError } = require('./Errors')
- // NOTE: Updating this list requires a corresponding change in
- // * services/web/frontend/js/features/pdf-preview/util/file-list.js
- const ignoreFiles = ['output.fls', 'output.fdb_latexmk']
- function getContentDir(projectId, userId) {
- let subDir
- if (userId != null) {
- subDir = `${projectId}-${userId}`
- } else {
- subDir = projectId
- }
- return `${Settings.path.outputDir}/${subDir}/`
- }
- module.exports = {
- async archiveFilesForBuild(projectId, userId, build) {
- const validFiles = await this._getAllOutputFiles(projectId, userId, build)
- const archive = archiver('zip')
- const missingFiles = []
- for (const file of validFiles) {
- try {
- const fileHandle = await open(file, 'r')
- const fileStream = fileHandle.createReadStream()
- archive.append(fileStream, { name: path.basename(file) })
- } catch (error) {
- missingFiles.push(file)
- }
- }
- if (missingFiles.length > 0) {
- archive.append(missingFiles.join('\n'), {
- name: 'missing_files.txt',
- })
- }
- await archive.finalize()
- return archive
- },
- async _getAllOutputFiles(projectId, userId, build) {
- const contentDir = getContentDir(projectId, userId)
- try {
- const { outputFiles } = await OutputFileFinder.promises.findOutputFiles(
- [],
- `${contentDir}${OutputCacheManager.path(build, '.')}`
- )
- return outputFiles
- .filter(
- // Ignore the pdf and also ignore the files ignored by the frontend.
- ({ path }) => path !== 'output.pdf' && !ignoreFiles.includes(path)
- )
- .map(
- ({ path }) => `${contentDir}${OutputCacheManager.path(build, path)}`
- )
- } catch (error) {
- if (
- error.code === 'ENOENT' ||
- error.code === 'ENOTDIR' ||
- error.code === 'EACCES'
- ) {
- throw new NotFoundError('Output files not found')
- }
- throw error
- }
- },
- }
|