update-hosts-2021-04-30.js

#!/usr/bin/env node
import '../server/env';

import { ArgumentParser } from 'argparse';

import status from '../server/constants/order_status';
import { PLANS_COLLECTIVE_SLUG } from '../server/constants/plans';
import models, { Op } from '../server/models';

const previousPlansSlugs = ['single-host-plan', 'small-host-plan', 'medium-host-plan', 'large-host-plan'];

async function run({ dryRun } = {}) {
  const hosts = await models.Collective.findAll({
    where: {
      isHostAccount: { [Op.is]: true },
      [Op.or]: [{ plan: null }, { plan: { [Op.not]: ['start-plan-2021', 'grow-plan-2021', 'custom', 'owned'] } }],
    },
  });

  for (const host of hosts) {
    console.log(`Checking ${host.slug}`);
    if (host.plan) {
      console.log(`- Current plan: ${host.plan}`);
    }
    if (!host.isActive) {
      console.log(`- Activating Budget`);
      if (!dryRun) {
        await host.activateBudget();
      }
    }

    if (host.hostFeePercent !== 0) {
      console.log(`- Updating Plan to grow-plan-2021`);
      if (!dryRun) {
        await host.update({ plan: 'grow-plan-2021' });
      }
    } else {
      console.log(`- Updating Plan to start-plan-2021`);
      if (!dryRun) {
        await host.update({ plan: 'start-plan-2021' });
      }
    }

    if (host.platformFeePercent !== 0) {
      console.log(`- Activating Platform Tips (updating platformFeePercent to 0)`);
      if (!dryRun) {
        // This will NOT cascade to all Collectives
        await host.update({ platformFeePercent: 0 });
        // TODO: maybe run the following manually after
        // UPDATE "Collectives"
        // SET "platformFeePercent" = 0
        // FROM "Collectives" as "Hosts"
        // WHERE "Hosts"."id" = "Collectives"."HostCollectiveId"
        // AND "Hosts"."platformFeePercent" = 0
        // AND ("Collectives"."platformFeePercent" IS NULL OR "Collectives"."platformFeePercent" != 0);
      }
    }

    const order = await models.Order.findOne({
      where: { status: status.ACTIVE },
      include: [
        { model: models.Collective, as: 'collective', where: { slug: PLANS_COLLECTIVE_SLUG } },
        { model: models.Collective, as: 'fromCollective', where: { id: host.id } },
        { model: models.Subscription, as: 'Subscription' },
        { model: models.Tier, as: 'Tier', where: { slug: { [Op.in]: previousPlansSlugs } } },
      ],
      order: [['updatedAt', 'DESC']],
    });
    if (order) {
      console.log(`- Cancelling previous Plan`);
    }
    if (!dryRun) {
      if (order) {
        await order.update({ status: status.CANCELLED });
        await order.Subscription.deactivate();
      }
    }
  }
}

/* eslint-disable camelcase */
function parseCommandLineArguments() {
  const parser = new ArgumentParser({
    add_help: true,
    description: 'Check and update Hosts in the new model',
  });
  parser.add_argument('--dryrun', {
    help: "Don't perform any change, just log.",
    default: false,
    action: 'store_const',
    const: true,
  });
  const args = parser.parse_args();
  return {
    dryRun: args.dryrun,
  };
}

/* eslint-enable camelcase */

if (require.main === module) {
  run(parseCommandLineArguments())
    .then(() => {
      process.exit(0);
    })
    .catch(e => {
      console.error(e);
      process.exit(1);
    });
}