overleaf/services/web/scripts/count_files_in_projects.js
Jakob Ackermann ad52f7dc01 Merge pull request #3255 from overleaf/jpa-fix-scripts
[misc] add waitForDb wrapper to scripts with indirect mongo access

GitOrigin-RevId: 009331f91a45f075a35b0b6e90c4352e55908c21
2020-10-06 02:04:28 +00:00

49 lines
1.4 KiB
JavaScript

const readline = require('readline')
const { waitForDb } = require('../app/src/infrastructure/mongodb')
const ProjectEntityHandler = require('../app/src/Features/Project/ProjectEntityHandler')
const ProjectGetter = require('../app/src/Features/Project/ProjectGetter')
const Errors = require('../app/src/Features/Errors/Errors')
/* eslint-disable no-console */
async function countFiles() {
const rl = readline.createInterface({
input: process.stdin
})
for await (const projectId of rl) {
try {
const project = await ProjectGetter.promises.getProject(projectId)
if (!project) {
throw new Errors.NotFoundError('project not found')
}
const {
files,
docs
} = await ProjectEntityHandler.promises.getAllEntitiesFromProject(project)
console.error(
projectId,
files.length,
(project.deletedFiles && project.deletedFiles.length) || 0,
docs.length,
(project.deletedDocs && project.deletedDocs.length) || 0
)
} catch (err) {
if (err instanceof Errors.NotFoundError) {
console.error(projectId, 'NOTFOUND')
} else {
console.log(projectId, 'ERROR', err.name, err.message)
}
}
}
}
waitForDb()
.then(countFiles)
.then(() => {
process.exit(0)
})
.catch(err => {
console.log('Aiee, something went wrong!', err)
process.exit(1)
})