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


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

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
}