overleaf/services/web/scripts/count_encrypted_access_tokens.js
Jakob Ackermann 79f7d259e9 Merge pull request #11833 from overleaf/jpa-v2-fallback
[access-token-encryptor] scripts: fallback version to v2

GitOrigin-RevId: 232e6b1f6b8667a58c5fd0f5caf604209f100809
2023-02-16 09:04:39 +00:00

74 lines
1.8 KiB
JavaScript

const TEN_MINUTES = 1000 * 60 * 10
process.env.MONGO_SOCKET_TIMEOUT =
process.env.MONGO_SOCKET_TIMEOUT || TEN_MINUTES.toString()
const { ReadPreference } = require('mongodb')
const { db, waitForDb } = require('../app/src/infrastructure/mongodb')
const _ = require('lodash')
const {
formatTokenUsageStats,
} = require('@overleaf/access-token-encryptor/scripts/helpers/format-usage-stats')
const CASES = {
users: {
dropbox: 'dropbox.access_token_oauth2.encrypted',
zotero: 'refProviders.zotero.encrypted',
mendeley: 'refProviders.mendeley.encrypted',
},
githubSyncUserCredentials: {
github: 'auth_token_encrypted',
},
}
async function count(collectionName, paths) {
const collection = db[collectionName]
const stats = {}
const projection = { _id: 0 }
for (const path of Object.values(paths)) {
projection[path] = 1
}
const cursor = collection.find(
{},
{
readPreference: ReadPreference.SECONDARY,
projection,
}
)
for await (const doc of cursor) {
for (const [name, path] of Object.entries(paths)) {
const blob = _.get(doc, path)
if (!blob) continue
// Schema: LABEL-VERSION:SALT:CIPHERTEXT:IV
const [label] = blob.split(':')
let [, version] = label.split('-')
version = version || 'v2'
const key = [name, version, collectionName, path, label].join(':')
stats[key] = (stats[key] || 0) + 1
}
}
return stats
}
async function main() {
await waitForDb()
const STATS = {}
for (const [collectionName, paths] of Object.entries(CASES)) {
const stats = await count(collectionName, paths)
Object.assign(STATS, stats)
}
formatTokenUsageStats(STATS)
}
main()
.then(() => {
process.exit(0)
})
.catch(err => {
console.error(err)
process.exit(1)
})