habitica/website/common/script/ops/buy/purchase.js

import get from 'lodash/get';
import pick from 'lodash/pick';
import forEach from 'lodash/forEach';
import i18n from '../../i18n';
import content from '../../content/index';
import splitWhitespace from '../../libs/splitWhitespace';
import {
  NotFound,
  NotAuthorized,
  BadRequest,
} from '../../libs/errors';

import { removeItemByPath } from '../pinnedGearUtils';
import getItemInfo from '../../libs/getItemInfo';
import updateUserBalance from '../updateUserBalance';
import { getScheduleMatchingGroup } from '../../content/constants/schedule';

function getItemAndPrice (user, type, key, req) {
  let item;
  let price;

  if (type === 'gear') {
    item = content.gear.flat[key];

    if (!item) {
      throw new NotFound(i18n.t('contentKeyNotFound', { type }, req.language));
    }

    if (user.items.gear.owned[key]) {
      throw new NotAuthorized(i18n.t('alreadyHave', req.language));
    }

    price = (item.twoHanded || item.gearSet === 'animal' ? 2 : 1) / 4;
  } else {
    item = content[type][key];

    if (!item) {
      throw new NotFound(i18n.t('contentKeyNotFound', { type }, req.language));
    }

    price = item.value / 4;
  }

  return { item, price };
}

async function purchaseItem (user, item, price, type, key) {
  await updateUserBalance(user, -price, 'spend', item.key, `${item.text()} ${type}`);

  if (type === 'gear') {
    user.items.gear.owned = {
      ...user.items.gear.owned,
      [key]: true,
    };
    if (user.markModified) user.markModified('items.gear.owned');
  } else if (type === 'bundles') {
    const subType = item.type;
    forEach(item.bundleKeys, bundledKey => {
      if (!user.items[subType][bundledKey] || user.items[subType][bundledKey] < 0) {
        user.items[subType][bundledKey] = 0;
      }
      user.items[subType][bundledKey] += 1;
    });
    if (user.markModified) user.markModified(`items.${subType}`);
  } else {
    if (!user.items[type][key] || user.items[type][key] < 0) {
      user.items[type][key] = 0;
    }
    user.items[type] = {
      ...user.items[type],
      [key]: user.items[type][key] + 1,
    };
    if (user.markModified) user.markModified(`items.${type}`);
  }
}

const acceptedTypes = ['eggs', 'hatchingPotions', 'food', 'gear', 'bundles'];
const singlePurchaseTypes = ['gear'];
export default async function purchase (user, req = {}, analytics) {
  const type = get(req.params, 'type');
  const key = get(req.params, 'key');

  const quantity = req.quantity ? Number(req.quantity) : 1;
  if (quantity < 1 || !Number.isInteger(quantity)) throw new BadRequest(i18n.t('invalidQuantity', req.language));

  if (!type) {
    throw new BadRequest(i18n.t('typeRequired', req.language));
  }

  if (!key) {
    throw new BadRequest(i18n.t('missingKeyParam', req.language));
  }

  if (!acceptedTypes.includes(type)) {
    throw new NotFound(i18n.t('notAccteptedType', req.language));
  }

  const { price, item } = getItemAndPrice(user, type, key, req);

  if (type === 'hatchingPotions' && (item.premium === true || item.wacky === true) && item.questPotion !== true) {
    const matchers = getScheduleMatchingGroup('premiumHatchingPotions');
    if (!matchers.match(item.key)) {
      throw new NotAuthorized(i18n.t('messageNotAvailable', req.language));
    }
  } else if (item.klass === 'special' && item.gearSet !== 'animal') {
    const matchers = getScheduleMatchingGroup('seasonalGear');
    if (!matchers.match(item.set)) {
      throw new NotAuthorized(i18n.t('messageNotAvailable', req.language));
    }
  } else if (type === 'bundles') {
    const matchers = getScheduleMatchingGroup('bundles');
    if (!matchers.match(item.key)) {
      throw new NotAuthorized(i18n.t('notAvailable', { key: item.key }));
    }
  } else if (!item.canBuy || !item.canBuy(user)) {
    throw new NotAuthorized(i18n.t('messageNotAvailable', req.language));
  }

  if (!user.balance || user.balance < price * quantity) {
    throw new NotAuthorized(i18n.t('notEnoughGems', req.language));
  }

  if (singlePurchaseTypes.includes(type)) {
    const itemInfo = getItemInfo(user, type, item);
    removeItemByPath(user, itemInfo.path);
  }

  /* eslint-disable no-await-in-loop */
  for (let i = 0; i < quantity; i += 1) {
    await purchaseItem(user, item, price, type, key);
  }
  /* eslint-enable no-await-in-loop */
  if (analytics) {
    analytics.track('buy', {
      user,
      uuid: user._id,
      itemKey: key,
      itemType: type,
      currency: 'Gems',
      gemCost: price * 4,
      quantityPurchased: quantity,
      category: 'behavior',
      headers: req.headers,
    });
  }

  return [
    pick(user, splitWhitespace('items balance')),
  ];
}