"use strict";

/** Schwangerschaft (Admin / Spiel): erwarteter Geburtstermin + optionaler Vater-Charakter */
module.exports = {
  async up(queryInterface) {
    await queryInterface.sequelize.query(`
      DO $$
      BEGIN
        IF NOT EXISTS (
          SELECT 1 FROM information_schema.columns
          WHERE table_schema = 'falukant_data' AND table_name = 'character' AND column_name = 'pregnancy_due_at'
        ) THEN
          ALTER TABLE falukant_data."character"
          ADD COLUMN pregnancy_due_at TIMESTAMPTZ NULL;
        END IF;
        IF NOT EXISTS (
          SELECT 1 FROM information_schema.columns
          WHERE table_schema = 'falukant_data' AND table_name = 'character' AND column_name = 'pregnancy_father_character_id'
        ) THEN
          ALTER TABLE falukant_data."character"
          ADD COLUMN pregnancy_father_character_id INTEGER NULL
            REFERENCES falukant_data."character"(id) ON DELETE SET NULL;
        END IF;
      END$$;
    `);
  },

  async down(queryInterface) {
    await queryInterface.sequelize.query(`
      ALTER TABLE falukant_data."character" DROP COLUMN IF EXISTS pregnancy_father_character_id;
    `);
    await queryInterface.sequelize.query(`
      ALTER TABLE falukant_data."character" DROP COLUMN IF EXISTS pregnancy_due_at;
    `);
  },
};