habitica/website/common/script/ops/hatch.js

import findIndex from 'lodash/findIndex';
import forEach from 'lodash/forEach';
import get from 'lodash/get';
import keys from 'lodash/keys';
import upperFirst from 'lodash/upperFirst';
import moment from 'moment';
import i18n from '../i18n';
import content from '../content/index';
import {
  BadRequest,
  NotAuthorized,
  NotFound,
} from '../libs/errors';
import { errorMessage } from '../libs/errorMessage';
import { checkOnboardingStatus } from '../libs/onboarding';

export default function hatch (user, req = {}, analytics) {
  const egg = get(req, 'params.egg');
  const hatchingPotion = get(req, 'params.hatchingPotion');

  if (!(egg && hatchingPotion)) {
    throw new BadRequest(errorMessage('missingEggHatchingPotion'));
  }

  if (!(user.items.eggs[egg] > 0 && user.items.hatchingPotions[hatchingPotion] > 0)) {
    throw new NotFound(i18n.t('messageMissingEggPotion', req.language));
  }

  if (
    (
      content.hatchingPotions[hatchingPotion].premium
      || content.hatchingPotions[hatchingPotion].wacky
    )
    && !content.dropEggs[egg]
  ) {
    throw new BadRequest(i18n.t('messageInvalidEggPotionCombo', req.language));
  }

  const pet = `${egg}-${hatchingPotion}`;

  if (user.items.pets[pet] && user.items.pets[pet] > 0) {
    throw new NotAuthorized(i18n.t('messageAlreadyPet', req.language));
  }

  user.items.pets = {
    ...user.items.pets,
    [pet]: 5,
  };
  user.items.eggs[egg] -= 1;
  user.items.hatchingPotions[hatchingPotion] -= 1;
  if (user.markModified) {
    user.markModified('items.pets');
    user.markModified('items.eggs');
    user.markModified('items.hatchingPotions');
  }

  if (!user.achievements.hatchedPet && user.addAchievement) {
    user.addAchievement('hatchedPet');
    checkOnboardingStatus(user, req, analytics);
  }

  if (content.dropEggs[egg]) {
    forEach(content.animalColorAchievements, achievement => {
      if (hatchingPotion !== achievement.color) return;
      if (!user.achievements[achievement.petAchievement]) {
        const petIndex = findIndex(
          keys(content.dropEggs),
          animal => !user.items.pets[`${animal}-${achievement.color}`] || user.items.pets[`${animal}-${achievement.color}`] <= 0,
        );
        if (petIndex === -1) {
          user.achievements[achievement.petAchievement] = true;
          if (user.addNotification) {
            const achievementString = `achievement${upperFirst(achievement.petAchievement)}`;
            user.addNotification(achievement.petNotificationType, {
              label: `${'achievement'}: ${achievementString}`,
              achievement: achievement.petAchievement,
              message: `${i18n.t('modalAchievement')} ${i18n.t(achievementString)}`,
              modalText: i18n.t(`${achievementString}ModalText`),
            });
          }
        }
      }
    });
  }

  if (content.dropHatchingPotions[hatchingPotion]) {
    forEach(content.animalSetAchievements, achievement => {
      if (!user.achievements[achievement.achievementKey]) {
        if (achievement.type === 'pet' || achievement.type === 'petMount') {
          let achieved = true;
          forEach(achievement.species, species => {
            if (!achieved) return;
            const petIndex = findIndex(
              keys(content.dropHatchingPotions),
              color => !user.items.pets[`${species}-${color}`] || user.items.pets[`${species}-${color}`] === -1,
            );
            if (petIndex !== -1) achieved = false;
          });
          if (achievement.type === 'petMount') {
            forEach(achievement.species, species => {
              if (!achieved) return;
              const mountIndex = findIndex(
                keys(content.dropHatchingPotions),
                color => !user.items.mounts[`${species}-${color}`],
              );
              if (mountIndex !== -1) achieved = false;
            });
          }
          if (achieved) {
            user.achievements[achievement.achievementKey] = true;
            if (user.addNotification) {
              const achievementString = `achievement${upperFirst(achievement.achievementKey)}`;
              user.addNotification(achievement.notificationType, {
                label: `${'achievement'}: ${achievementString}`,
                achievement: achievement.achievementKey,
                message: `${i18n.t('modalAchievement')} ${i18n.t(achievementString)}`,
                modalText: i18n.t(`${achievementString}ModalText`),
              });
            }
          }
        }
      }
    });
  }

  if (content.dropEggs[egg] || content.questEggs[egg]) {
    forEach(content.petSetCompleteAchievs, achievement => {
      if (hatchingPotion !== achievement.color) return;
      if (!user.achievements[achievement.petAchievement]) {
        const dropPetIndex = findIndex(
          keys(content.dropEggs),
          animal => !user.items.pets[`${animal}-${achievement.color}`] || user.items.pets[`${animal}-${achievement.color}`] <= 0,
        );
        const questPetIndex = findIndex(
          keys(content.questEggs),
          animal => !user.items.pets[`${animal}-${achievement.color}`] || user.items.pets[`${animal}-${achievement.color}`] <= 0,
        );
        if (dropPetIndex === -1 && questPetIndex === -1) {
          user.achievements[achievement.petAchievement] = true;
          if (user.addNotification) {
            const achievementString = `achievement${upperFirst(achievement.petAchievement)}`;
            user.addNotification(achievement.petNotificationType, {
              label: `${'achievement'}: ${achievementString}`,
              achievement: achievement.petAchievement,
              message: `${i18n.t('modalAchievement')} ${i18n.t(achievementString)}`,
              modalText: i18n.t(`${achievementString}ModalText`),
            });
          }
        }
      }
    });
  }

  if (analytics && moment().diff(user.auth.timestamps.created, 'days') < 7) {
    analytics.track('pet hatch', {
      user,
      uuid: user._id,
      petKey: pet,
      category: 'behavior',
      headers: req.headers,
    });
  }

  return [
    user.items,
    i18n.t('messageHatched', req.language),
  ];
}