huanld/speckle-server
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

chore: re-formatted everything with prettier

Browse Source
This commit is contained in:
Fabians
2022-03-29 16:53:49 +03:00
parent 4e97360e48
commit abe6d5e086
307 changed files with 12690 additions and 8378 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/server/scripts/duplicateUserMigration.js
+94 -67
View File
@@ -1,106 +1,133 @@
const appRoot = require( 'app-root-path' )
const knex = require( `${appRoot}/db/knex` )
const roles = require( `${appRoot}/modules/core/roles.js` )
const appRoot = require('app-root-path')
const knex = require(`${appRoot}/db/knex`)
const roles = require(`${appRoot}/modules/core/roles.js`)
const Users = ( ) => knex( 'users' )
const Users = () => knex('users')
// 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' ],
['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' ],
['server_apps', 'authorId'],
['server_invites', 'inviterId'],
// [ 'stream_acl' , 'userId' ],//userId, with resourceId is a PrimaryKey in this table, act accordingly
[ 'stream_activity' , 'userId' ],
['stream_activity', 'userId']
]
const migrateColumnValue = async( tableName, columnName, oldUser, newUser ) => {
const migrateColumnValue = async (tableName, columnName, oldUser, newUser) => {
try {
const query = knex( tableName ).where( { [columnName]: oldUser.id } ).update( { [columnName]: newUser.id } )
console.log( `${query}` )
const query = knex(tableName)
.where({ [columnName]: oldUser.id })
.update({ [columnName]: newUser.id })
console.log(`${query}`)
await query
} catch ( err ) {
console.log( err )
} catch (err) {
console.log(err)
}
}
const serverAclMigration = async ( { lowerUser, upperUser } ) => {
const oldAcl = await knex( 'server_acl' ).where( { userId: upperUser.id } ).first()
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' } )
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()
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 (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 } )
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 )
await knex('stream_acl').insert(lowerStreamAcl)
}
}
const streamAclMigration = async ( { lowerUser, upperUser } ) => {
const upperAcl = await knex( 'stream_acl' ).where( { userId: upperUser.id } )
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 } ) ) )
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 createMigrations = ({ lowerUser, upperUser }) =>
migrationTargets.map(([tableName, columnName]) => {
migrateColumnValue(tableName, columnName, upperUser, lowerUser)
})
const userByEmailQuery = (email) => Users().where({ email })
const userByEmailQuery = email => Users( ).where( { email } )
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
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 }
} ) )
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( )
console.log( duplicateUsers )
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 )
const runMigrations = async () => {
const duplicateUsers = await getDuplicateUsers()
console.log(duplicateUsers)
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 userByEmailQuery( userDouble.upperUser.email ).delete( )
} ) )
// remove the now defunct user
await userByEmailQuery(userDouble.upperUser.email).delete()
})
)
}
( async function () {
;(async function () {
try {
// await createData()
await runMigrations()
} catch ( err ) {
console.log( err )
} finally { process.exit() }
}() )
} catch (err) {
console.log(err)
} finally {
process.exit()
}
})()