PeerTube/server/initializers/migrations/0560-user-feed-token.ts

52 lines
1.2 KiB
TypeScript

import * as Sequelize from 'sequelize'
import { buildUUID } from '@shared/extra-utils'
async function up (utils: {
transaction: Sequelize.Transaction
queryInterface: Sequelize.QueryInterface
sequelize: Sequelize.Sequelize
db: any
}): Promise<void> {
const q = utils.queryInterface
{
// Create uuid column for users
const userFeedTokenUUID = {
type: Sequelize.UUID,
defaultValue: Sequelize.UUIDV4,
allowNull: true
}
await q.addColumn('user', 'feedToken', userFeedTokenUUID)
}
// Set UUID to previous users
{
const query = 'SELECT * FROM "user" WHERE "feedToken" IS NULL'
const options = { type: Sequelize.QueryTypes.SELECT as Sequelize.QueryTypes.SELECT }
const users = await utils.sequelize.query<any>(query, options)
for (const user of users) {
const queryUpdate = `UPDATE "user" SET "feedToken" = '${buildUUID()}' WHERE id = ${user.id}`
await utils.sequelize.query(queryUpdate)
}
}
{
const userFeedTokenUUID = {
type: Sequelize.UUID,
defaultValue: Sequelize.UUIDV4,
allowNull: false
}
await q.changeColumn('user', 'feedToken', userFeedTokenUUID)
}
}
function down (options) {
throw new Error('Not implemented.')
}
export {
up,
down
}