speckle-server/packages/server/scripts/duplicateUserMigration.js

const appRoot = require( 'app-root-path' )
const knex = require( `${appRoot}/db/knex` )
const bcrypt = require( 'bcrypt' )
const crs = require( 'crypto-random-string' )
const roles = require( `${appRoot}/modules/core/roles.js` )

const Users = ( ) => knex( 'users' )
const Acl = ( ) => knex( 'server_acl' )

// tableName, columnName that need migration
const migrationTargets = [
  [ 'api_tokens' , 'owner' ],
  [ 'authorization_codes' , 'userId' ],
  [ 'branches' , 'authorId' ],
  [ 'commits' , 'author' ],
  [ 'file_uploads' , 'userId' ],
  [ 'personal_api_tokens' , 'userId' ],
  [ 'refresh_tokens' , 'userId' ],
  // [ 'server_acl' , 'userId' ], //userId is a PrimaryKey in this table, act accordingly
  [ 'server_apps' , 'authorId' ],
  [ 'server_invites' , 'inviterId' ],
  // [ 'stream_acl' , 'userId' ],//userId, with resourceId is a PrimaryKey in this table, act accordingly
  [ 'stream_activity' , 'userId' ],
]

const migrateColumnValue = async( tableName, columnName, oldUser, newUser ) => {
  try {
    const query = knex( tableName ).where( { [columnName]: oldUser.id } ).update( { [columnName]: newUser.id } ) 
    console.log( `${query}` )
    await query
  } catch ( err ) {
    console.log( err )
  }
}

const serverAclMigration = async ( { lowerUser, upperUser } ) => {
  const oldAcl = await knex( 'server_acl' ).where( { userId: upperUser.id } ).first()
  // if the old user was admin, make the target admin too
  if ( oldAcl.role === 'server:admin' ) await knex( 'server_acl' ).where( { userId: lowerUser.id } ).update( { role: 'server:admin' } )
}

const _migrateSingleStreamAccess = async ( { lowerUser, upperUser, upperStreamAcl } ) => {
  const upperRole = roles.filter( r => r.name === upperStreamAcl.role )[0]
  const lowerAcl = await knex( 'stream_acl' ).where( { userId: lowerUser.id, resourceId: upperStreamAcl.resourceId } ).first()
  // see if the lowerUser has access to the stream
  if ( lowerAcl ) {
    // if the upper user had more access, migrate the lower user up
    const lowerRole = roles.filter( r => r.name === lowerAcl.role )[0]
    if ( lowerRole.weight < upperRole.weight ) 
      await knex( 'stream_acl' )
        .where( { userId: lowerUser.id, resourceId: upperStreamAcl.resourceId } )
        .update( { role: upperRole.name } )
  } else {
    // if it didn't have access, just add it
    let lowerStreamAcl = { ...upperStreamAcl }
    lowerStreamAcl.userId = lowerUser.id
    await knex( 'stream_acl' ).insert( lowerStreamAcl )
  }
}

const streamAclMigration =  async ( { lowerUser, upperUser } ) => {
  const upperAcl = await knex( 'stream_acl' ).where( { userId: upperUser.id } )

  await Promise.all( upperAcl.map( async upperStreamAcl => await _migrateSingleStreamAccess( { lowerUser, upperUser, upperStreamAcl } ) ) )
}


const createMigrations = ( { lowerUser, upperUser } ) => migrationTargets.map( ( [ tableName, columnName ] ) => {
  migrateColumnValue( tableName, columnName, upperUser,lowerUser ) } ) 


const userByEmailQuery = email => Users( ).where( { email } )
const createUser = async user => {
  user.id = crs( { length: 10 } )

  if ( user.password ) {
    let pwdigest =await bcrypt.hash( user.password, 10 )
    user.passwordDigest = pwdigest
  }
  delete user.password

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

  let res = await Users( ).returning( 'id' ).insert( user )
    
  let userRole = 'server:user' 

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

  return res[ 0 ]
}

const createData = async ( ) => {
  const users = [
    { email: 'asdf@asdf.asdf', password: '12345678', name: 'Asdf Asdf' } ,
    { email: 'AsDf@asdf.asdf', password: '12345678', name: 'Asdf Asdf' } ,
    { email: 'fdsa@asdf.asdf', password: '12345678', name: 'Asdf Asdf' } ,
    { email: 'Fdsa@asdf.asdf', password: '12345678', name: 'Asdf Asdf' } 
  ]
  await Promise.all( users.map( async user => {
    try {
      let userId = await createUser( user )
      console.log( userId )
    } catch ( err ) {
      console.log( err )
    }  
  } ) )
}

const getDuplicateUsers = async ( ) => {
  let duplicates = await knex.raw( 'select lower(email) as lowered, count(id) as reg_count from users group by lowered having count(id) > 1' )
  return await Promise.all( duplicates.rows.map( async dup => {
    let lowerEmail = dup.lowered

    let lowerUser = await userByEmailQuery( lowerEmail ).first( )
    // if no user found migrate to a random one?
    // TODO: decide 👆
    // my idea, take the first one and run with it
    if ( !lowerUser ) lowerUser = await Users( ).whereRaw( 'lower(email) = lower(?)',[ lowerEmail ] ).first() 
    let upperUser = await Users( ).whereRaw( 'lower(email) = lower(?)',[ lowerEmail ] ).whereNot( { id: lowerUser.id } ).first( )
    return { lowerUser,upperUser }
  } ) )
}

const runMigrations = async ( ) => {
  const duplicateUsers = await getDuplicateUsers( )
  await Promise.all( duplicateUsers.map( async userDouble => {
    const migrations = createMigrations( userDouble )
    await Promise.all( migrations.map( async migrationStep => await migrationStep ) )
    await serverAclMigration( userDouble )
    await streamAclMigration( userDouble )

    // remove the now defunct user
    await Users( ).where( { email: userDouble.upperUser.email } ).delete( )
  } ) )
}

( async function () {
  try {
    await createData()
    await runMigrations()
  } catch ( err ) {
    console.log( err )
  } finally { process.exit() } 
}() )