speckle-server/packages/server/modules/core/services/users.js

'use strict'
const bcrypt = require('bcrypt')
const crs = require('crypto-random-string')
const knex = require('@/db/knex')
const {
  ServerAcl: ServerAclSchema,
  Users: UsersSchema
} = require('@/modules/core/dbSchema')

const { saveActivity } = require('@/modules/activitystream/services')

const Users = () => UsersSchema.knex()
const Acl = () => ServerAclSchema.knex()

const debug = require('debug')
const { deleteStream } = require('./streams')
const { LIMITED_USER_FIELDS } = require('@/modules/core/helpers/userHelper')
const { deleteAllUserInvites } = require('@/modules/serverinvites/repositories')

const changeUserRole = async ({ userId, role }) =>
  await Acl().where({ userId }).update({ role })

const countAdminUsers = async () => {
  const [{ count }] = await Acl().where({ role: 'server:admin' }).count()
  return parseInt(count)
}
const _ensureAtleastOneAdminRemains = async (userId) => {
  if ((await countAdminUsers()) === 1) {
    const currentAdmin = await Acl().where({ role: 'server:admin' }).first()
    if (currentAdmin.userId === userId) {
      throw new Error('Cannot remove the last admin role from the server')
    }
  }
}

const userByEmailQuery = (email) => Users().whereRaw('lower(email) = lower(?)', [email])

const getUsersBaseQuery = (searchQuery = null) => {
  const query = Users()
  if (searchQuery) {
    query.where((queryBuilder) => {
      queryBuilder
        .where('email', 'ILIKE', `%${searchQuery}%`)
        .orWhere('name', 'ILIKE', `%${searchQuery}%`)
        .orWhere('company', 'ILIKE', `%${searchQuery}%`)
    })
  }
  return query
}

module.exports = {
  /*

        Users

  */

  async createUser(user) {
    user.id = crs({ length: 10 })
    user.email = user.email.toLowerCase()

    if (user.password) {
      if (user.password.length < 8)
        throw new Error('Password to short; needs to be 8 characters or longer.')
      user.passwordDigest = await bcrypt.hash(user.password, 10)
    }
    delete user.password

    const usr = await userByEmailQuery(user.email).select('id').first()
    if (usr) throw new Error('Email taken. Try logging in?')

    const res = await Users().returning('id').insert(user)

    const userRole = (await countAdminUsers()) === 0 ? 'server:admin' : 'server:user'

    await Acl().insert({ userId: res[0].id, role: userRole })

    const loggedUser = { ...user }
    delete loggedUser.passwordDigest
    await saveActivity({
      streamId: null,
      resourceType: 'user',
      resourceId: user.id,
      actionType: 'user_create',
      userId: user.id,
      info: { user: loggedUser },
      message: 'User created'
    })

    return res[0].id
  },

  async findOrCreateUser({ user }) {
    const existingUser = await userByEmailQuery(user.email).select('id').first()

    if (existingUser) {
      if (user.suuid) {
        await module.exports.updateUser(existingUser.id, { suuid: user.suuid })
      }

      existingUser.suuid = user.suuid
      return existingUser
    }

    user.password = crs({ length: 20 })
    user.verified = true // because we trust the external identity provider, no?
    return {
      id: await module.exports.createUser(user),
      email: user.email,
      isNewUser: true
    }
  },

  /**
   * @param {{userId: string}} param0
   * @returns {import('@/modules/core/helpers/userHelper').UserRecord | null}
   */
  async getUserById({ userId }) {
    const user = await Users().where({ id: userId }).select('*').first()
    if (user) delete user.passwordDigest
    return user
  },

  // TODO: deprecate
  async getUser(id) {
    const user = await Users().where({ id }).select('*').first()
    if (user) delete user.passwordDigest
    return user
  },

  async getUserByEmail({ email }) {
    const user = await userByEmailQuery(email).select('*').first()
    if (!user) return null
    delete user.passwordDigest
    return user
  },

  async getUserRole(id) {
    const { role } = (await Acl().where({ userId: id }).select('role').first()) || {
      role: null
    }
    return role
  },

  async updateUser(id, user) {
    delete user.id
    delete user.passwordDigest
    delete user.password
    delete user.email
    await Users().where({ id }).update(user)
  },

  async updateUserPassword({ id, newPassword }) {
    if (newPassword.length < 8)
      throw new Error('Password to short; needs to be 8 characters or longer.')
    const passwordDigest = await bcrypt.hash(newPassword, 10)
    await Users().where({ id }).update({ passwordDigest })
  },

  /**
   * User search available for normal server users. It's more limited because of the lower access level.
   */
  async searchUsers(searchQuery, limit, cursor, archived = false) {
    const query = Users()
      .join('server_acl', 'users.id', 'server_acl.userId')
      .select(...LIMITED_USER_FIELDS)
      .where((queryBuilder) => {
        queryBuilder.where({ email: searchQuery }) //match full email or partial name
        queryBuilder.orWhere('name', 'ILIKE', `%${searchQuery}%`)
        if (!archived) queryBuilder.andWhere('role', '!=', 'server:archived-user')
      })

    if (cursor) query.andWhere('users.createdAt', '<', cursor)

    const defaultLimit = 25
    query.orderBy('users.createdAt', 'desc').limit(limit || defaultLimit)

    const rows = await query
    return {
      users: rows,
      cursor: rows.length > 0 ? rows[rows.length - 1].createdAt.toISOString() : null
    }
  },

  async validatePasssword({ email, password }) {
    const { passwordDigest } = await userByEmailQuery(email)
      .select('passwordDigest')
      .first()
    return bcrypt.compare(password, passwordDigest)
  },

  async deleteUser(id) {
    //TODO: check for the last admin user to survive
    debug('speckle:db')('Deleting user ' + id)
    await _ensureAtleastOneAdminRemains(id)
    const streams = await knex.raw(
      `
      -- Get the stream ids with only this user as owner
      SELECT "resourceId" as id
      FROM (
        -- Compute (streamId, ownerCount) table for streams on which the user is owner
        SELECT acl."resourceId", count(*) as cnt
        FROM stream_acl acl
        INNER JOIN 
          (
          -- Get streams ids on which the user is owner
          SELECT "resourceId" FROM stream_acl
          WHERE role = 'stream:owner' AND "userId" = ?
          ) AS us ON acl."resourceId" = us."resourceId"
        WHERE acl.role = 'stream:owner'
        GROUP BY (acl."resourceId")
      ) AS soc
      WHERE cnt = 1
      `,
      [id]
    )
    for (const i in streams.rows) {
      await deleteStream({ streamId: streams.rows[i].id })
    }

    // Delete all invites (they don't have a FK, so we need to do this manually)
    await deleteAllUserInvites(id)

    return await Users().where({ id }).del()
  },

  /**
   * Get all users or filter them with the specified searchQuery. This is meant for
   * server admins, because it exposes the User object (& thus the email).
   * @returns {Promise<import('@/modules/core/helpers/userHelper').UserRecord[]>}
   */
  async getUsers(limit = 10, offset = 0, searchQuery = null) {
    // sanitize limit
    const maxLimit = 200
    if (limit > maxLimit) limit = maxLimit

    const query = getUsersBaseQuery(searchQuery)
    query.limit(limit).offset(offset)

    const users = await query
    users.map((user) => delete user.passwordDigest)
    return users
  },

  async makeUserAdmin({ userId }) {
    await changeUserRole({ userId, role: 'server:admin' })
  },

  async unmakeUserAdmin({ userId }) {
    // dont delete last admin role
    await _ensureAtleastOneAdminRemains(userId)
    await changeUserRole({ userId, role: 'server:user' })
  },

  async archiveUser({ userId }) {
    // dont change last admin to archived
    await _ensureAtleastOneAdminRemains(userId)
    await changeUserRole({ userId, role: 'server:archived-user' })
  },

  async countUsers(searchQuery = null) {
    const query = getUsersBaseQuery(searchQuery)
    const [userCount] = await query.count()
    return parseInt(userCount.count)
  }
}