github/overleaf
1
0
Fork 0
mirror of https://github.com/overleaf/overleaf.git synced 2024-11-21 20:47:08 -05:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #21997 from overleaf/jpa-object-id-from-timestamp

Browse source 
[misc] add bin/object-id-from-timestamp helper

GitOrigin-RevId: e0eb5f90084dd6dd1d020f0b20fbd5fe9e9ede0f
This commit is contained in:
Jakob Ackermann 2024-11-20 10:36:38 +01:00 committed by Copybot
parent 0253130c36
commit 64fffb9d4d
1 changed files with 0 additions and 144 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

144
services/web/scripts/export_institution_chat.mjs
View file

@ -1,144 +0,0 @@
import {
db,
READ_PREFERENCE_SECONDARY,
ObjectId,
} from '../app/src/infrastructure/mongodb.js'
import minimist from 'minimist'
import InstitutionHubsController from '../modules/metrics/app/src/InstitutionHubsController.mjs'
function usage() {
console.log(
`Usage: node export_institution_chat.js --institution <id> --from <date> --to <date> [--pretty] [--help]
--institution=ID
The V1 institution ID
--from=DATE
The start of the report period. Specified as an ISO 8601 date string
(e.g. 2024-08-01T00:00:00.000Z)
--to=DATE
The end of the report period. Specified as an ISO 8601 date string
(e.g. 2024-09-01T00:00:00.000Z)
--help
Prints this help page\n`
)
}
function parseArgs() {
const argv = minimist(process.argv.slice(2), {
string: ['institution', 'from', 'to'],
bool: ['help'],
default: {
help: false,
},
})
if (argv.help) {
usage()
process.exit(0)
}
if (!argv.institution || !argv.from || !argv.to) {
usage()
process.exit(1)
}
const institutionId = parseInt(argv.institution, 10)
const from = new Date(argv.from).getTime()
const to = new Date(argv.to).getTime()
if (to < from) {
console.error('The end date must be after the start date.')
process.exit(1)
}
return { institutionId, from, to }
}
async function fetchInstitutionAndAffiliations(institutionId) {
const { json: affiliations } =
await InstitutionHubsController.promises.v1InstitutionsApi(
institutionId,
'csv_affiliations'
)
return affiliations.filter(({ license }) => license === 'pro_plus')
}
function getUserMappings(affiliations) {
const entries = affiliations.map(({ user_id: userId, email }) => [
userId,
email,
])
return new Map(entries)
}
async function main() {
const args = parseArgs()
const affiliations = await fetchInstitutionAndAffiliations(args.institutionId)
const userMappings = getUserMappings(affiliations)
const projectRecords = []
for (const [userId, email] of userMappings.entries()) {
projectRecords.push(...(await processUser(userId, email, args)))
}
console.log(JSON.stringify(projectRecords, null, 2))
}
async function processUser(userId, email, args) {
const projectsOwnedByUser = db.projects.find(
{ owner_ref: new ObjectId(userId) },
{ projection: { name: 1 }, readPreference: READ_PREFERENCE_SECONDARY }
)
const projectRecords = []
for await (const project of projectsOwnedByUser) {
const hasMessages = await processProject(project, args)
if (hasMessages) {
projectRecords.push({
projectId: project._id,
owner: email,
})
}
}
return projectRecords
}
async function processProject(project, args) {
const { _id: projectId } = project
const globalRoom = await db.rooms.findOne(
{
project_id: new ObjectId(projectId),
thread_id: { $exists: false },
},
{ readPreference: READ_PREFERENCE_SECONDARY }
)
if (!globalRoom) {
return null
}
const messages = await db.messages
.find(
{
room_id: globalRoom._id,
timestamp: { $gte: args.from, $lte: args.to },
},
{
projection: {
user_id: 1,
timestamp: 1,
},
readPreference: READ_PREFERENCE_SECONDARY,
}
)
.sort({ timestamp: 1 })
.toArray()
return messages.length > 0
}
try {
await main()
process.exit(0)
} catch (err) {
console.error(err)
process.exit(1)
}