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

import each from 'lodash/each';
import i18n from '../i18n';
import { capByLevel } from '../statHelpers';
import { MAX_LEVEL } from '../constants';
import {
  NotAuthorized,
} from '../libs/errors';
import equip from './equip';
import { removePinnedGearByClass } from './pinnedGearUtils';
import isFreeRebirth from '../libs/isFreeRebirth';
import setDebuffPotionItems from '../libs/setDebuffPotionItems';
import updateUserBalance from './updateUserBalance';

const USERSTATSLIST = ['per', 'int', 'con', 'str', 'points', 'gp', 'exp', 'mp'];

export default async function rebirth (user, tasks = [], req = {}, analytics) {
  const notFree = !isFreeRebirth(user);

  if (user.balance < 1.5 && notFree) {
    throw new NotAuthorized(i18n.t('notEnoughGems', req.language));
  }

  const analyticsData = {
    uuid: user._id,
    user,
    category: 'behavior',
  };

  if (notFree) {
    await updateUserBalance(user, -1.5, 'rebirth');
    analyticsData.currency = 'Gems';
    analyticsData.gemCost = 6;
  } else {
    analyticsData.currency = 'Free';
    analyticsData.gemCost = 0;
  }

  if (analytics) {
    analyticsData.headers = req.headers;
    analytics.track('Rebirth', analyticsData);
  }

  const lvl = capByLevel(user.stats.lvl);

  each(tasks, task => {
    if (!task.challenge || !task.challenge.id || task.challenge.broken) {
      if (task.type !== 'reward' && task.type !== 'todo') {
        task.value = 0;
      }
      if (task.type === 'daily') {
        task.streak = 0;
      }
      if (task.type === 'habit') {
        task.counterDown = 0;
        task.counterUp = 0;
      }
      if (task.type === 'todo' && task.completed !== true) {
        task.value = 0;
      }
    }
  });

  removePinnedGearByClass(user);

  const { stats } = user;
  stats.buffs = {};
  stats.hp = 50;
  stats.lvl = 1;
  stats.class = 'warrior';

  user.preferences.automaticAllocation = false;

  each(USERSTATSLIST, value => {
    stats[value] = 0;
  });

  if (user.items.currentPet) {
    equip(user, {
      params: {
        type: 'pet',
        key: user.items.currentPet,
      },
    });
  }

  if (user.items.currentMount) {
    equip(user, {
      params: {
        type: 'mount',
        key: user.items.currentMount,
      },
    });
  }

  const { flags } = user;
  flags.itemsEnabled = false;
  flags.dropsEnabled = false;
  flags.classSelected = false;
  flags.rebirthEnabled = false;
  flags.levelDrops = {};

  if (!user.achievements.rebirths) {
    user.achievements.rebirths = 1;
    user.achievements.rebirthLevel = lvl;
  } else if (lvl > user.achievements.rebirthLevel || lvl === MAX_LEVEL) {
    user.achievements.rebirths += 1;
    user.achievements.rebirthLevel = lvl;
  }

  if (!notFree) {
    user.flags.lastFreeRebirth = new Date();
  }

  if (user.addNotification) user.addNotification('REBIRTH_ACHIEVEMENT');

  user.stats.buffs = {};
  setDebuffPotionItems(user);

  return [
    { user, tasks },
    i18n.t('rebirthComplete'),
  ];
}