'use strict';

module.exports = {
  async up(queryInterface, Sequelize) {
    const table = await queryInterface.describeTable('assessment_sessions');
    if (!table.draft_answers)
      await queryInterface.addColumn('assessment_sessions', 'draft_answers', {
        type: Sequelize.JSONB, allowNull: true, defaultValue: null,
      });
    if (!table.last_heartbeat_at)
      await queryInterface.addColumn('assessment_sessions', 'last_heartbeat_at', {
        type: Sequelize.DATE, allowNull: true, defaultValue: null,
      });
  },

  async down(queryInterface) {
    await queryInterface.removeColumn('assessment_sessions', 'last_heartbeat_at');
    await queryInterface.removeColumn('assessment_sessions', 'draft_answers');
  },
};