Merge pull request #4315 from overleaf/jpa-import-ce-scripts

[scripts] import scripts from server-ce and add tests

GitOrigin-RevId: 07e7e15aa86030ccd3eca40ed40e5492622dc2d1

services/web/modules/server-ce-scripts/scripts/check-mongodb.js

@@ -0,0 +1,12 @@
+const { waitForDb } = require('../../../app/src/infrastructure/mongodb')
+
+waitForDb()
+  .then(() => {
+    console.error('Mongodb is up.')
+    process.exit(0)
+  })
+  .catch(err => {
+    console.error('Cannot connect to mongodb.')
+    console.error(err)
+    process.exit(1)
+  })

services/web/modules/server-ce-scripts/scripts/check-redis.js

@@ -0,0 +1,18 @@
+const RedisWrapper = require('../../../app/src/infrastructure/RedisWrapper')
+const rclient = RedisWrapper.client('health_check')
+rclient.on('error', err => {
+  console.error('Cannot connect to redis.')
+  console.error(err)
+  process.exit(1)
+})
+
+rclient.healthCheck(err => {
+  if (err) {
+    console.error('Cannot connect to redis.')
+    console.error(err)
+    process.exit(1)
+  } else {
+    console.error('Redis is up.')
+    process.exit(0)
+  }
+})

services/web/modules/server-ce-scripts/scripts/create-admin.js

@@ -0,0 +1,72 @@
+const Settings = require('@overleaf/settings')
+const { db, waitForDb } = require('../../../app/src/infrastructure/mongodb')
+const UserRegistrationHandler = require('../../../app/src/Features/User/UserRegistrationHandler')
+const OneTimeTokenHandler = require('../../../app/src/Features/Security/OneTimeTokenHandler')
+
+async function main() {
+  await waitForDb()
+
+  const email = (process.argv.slice(2).pop() || '').replace(/^--email=/, '')
+  if (!email) {
+    console.error(`Usage: node ${__filename} --email=joe@example.com`)
+    process.exit(1)
+  }
+
+  await new Promise((resolve, reject) => {
+    UserRegistrationHandler.registerNewUser(
+      {
+        email,
+        password: require('crypto').randomBytes(32).toString('hex'),
+      },
+      (error, user) => {
+        if (error && error.message !== 'EmailAlreadyRegistered') {
+          return reject(error)
+        }
+        db.users.updateOne(
+          { _id: user._id },
+          { $set: { isAdmin: true } },
+          error => {
+            if (error) {
+              return reject(error)
+            }
+            const ONE_WEEK = 7 * 24 * 60 * 60 // seconds
+            OneTimeTokenHandler.getNewToken(
+              'password',
+              {
+                expiresIn: ONE_WEEK,
+                email: user.email,
+                user_id: user._id.toString(),
+              },
+              (err, token) => {
+                if (err) {
+                  return reject(err)
+                }
+
+                console.log('')
+                console.log(`\
+Successfully created ${email} as an admin user.
+
+Please visit the following URL to set a password for ${email} and log in:
+
+${Settings.siteUrl}/user/password/set?passwordResetToken=${token}
+\
+`)
+                resolve()
+              }
+            )
+          }
+        )
+      }
+    )
+  })
+}
+
+main()
+  .then(() => {
+    console.error('Done.')
+    process.exit(0)
+  })
+  .catch(err => {
+    console.error(err)
+    process.exit(1)
+  })

services/web/modules/server-ce-scripts/scripts/delete-user.js

@@ -0,0 +1,43 @@
+const { waitForDb } = require('../../../app/src/infrastructure/mongodb')
+const UserGetter = require('../../../app/src/Features/User/UserGetter')
+const UserDeleter = require('../../../app/src/Features/User/UserDeleter')
+
+async function main() {
+  await waitForDb()
+
+  const email = (process.argv.slice(2).pop() || '').replace(/^--email=/, '')
+  if (!email) {
+    console.error(`Usage: node ${__filename} --email=joe@example.com`)
+    process.exit(1)
+  }
+
+  await new Promise((resolve, reject) => {
+    UserGetter.getUser({ email }, function (error, user) {
+      if (error) {
+        return reject(error)
+      }
+      if (!user) {
+        console.log(
+          `user ${email} not in database, potentially already deleted`
+        )
+        return resolve()
+      }
+      UserDeleter.deleteUser(user._id, function (err) {
+        if (err) {
+          return reject(err)
+        }
+        resolve()
+      })
+    })
+  })
+}
+
+main()
+  .then(() => {
+    console.error('Done.')
+    process.exit(0)
+  })
+  .catch(err => {
+    console.error(err)
+    process.exit(1)
+  })