habitica/website/server/libs/spells.js

import { model as User } from '../models/user';
import * as Tasks from '../models/task';
import {
  NotFound,
  BadRequest,
} from './errors';

// @TODO: After refactoring individual spells, move quantity to the calculations

async function castTaskSpell (res, req, targetId, user, spell, quantity = 1) {
  if (!targetId) throw new BadRequest(res.t('targetIdUUID'));

  const task = await Tasks.Task.findOne({
    _id: targetId,
    userId: user._id,
  }).exec();
  if (!task) throw new NotFound(res.t('taskNotFound'));
  if (task.challenge.id) throw new BadRequest(res.t('challengeTasksNoCast'));
  if (task.group.id) throw new BadRequest(res.t('groupTasksNoCast'));

  for (let i = 0; i < quantity; i += 1) {
    spell.cast(user, task, req);
  }

  const results = await Promise.all([
    user.save(),
    task.save(),
  ]);

  return results;
}

async function castMultiTaskSpell (req, user, spell, quantity = 1) {
  const tasks = await Tasks.Task.find({
    userId: user._id,
    ...Tasks.taskIsGroupOrChallengeQuery,
  }).exec();

  for (let i = 0; i < quantity; i += 1) {
    spell.cast(user, tasks, req);
  }

  const toSave = tasks
    .filter(t => t.isModified())
    .map(t => t.save());
  toSave.unshift(user.save());
  const saved = await Promise.all(toSave);

  const response = {
    tasks: saved,
    user,
  };

  return response;
}

async function castSelfSpell (req, user, spell, quantity = 1) {
  for (let i = 0; i < quantity; i += 1) {
    spell.cast(user, null, req);
  }
  await user.save();
}

async function castPartySpell (req, party, partyMembers, user, spell, quantity = 1) {
  if (!party) {
    partyMembers = [user]; // Act as solo party
  } else {
    partyMembers = await User
      .find({
        'party._id': party._id,
        _id: { $ne: user._id }, // add separately
      })
      // .select(partyMembersFields) Selecting the entire user because otherwise when saving it'll save
      // default values for non-selected fields and pre('save') will mess up thinking some values are missing
      .exec();

    partyMembers.unshift(user);
  }

  for (let i = 0; i < quantity; i += 1) {
    spell.cast(user, partyMembers, req);
  }
  await Promise.all(partyMembers.map(m => m.save()));

  return partyMembers;
}

async function castUserSpell (res, req, party, partyMembers, targetId, user, spell, quantity = 1) {
  if (!party && (!targetId || user._id === targetId)) {
    partyMembers = user;
  } else {
    if (!targetId) throw new BadRequest(res.t('targetIdUUID'));
    if (!party) throw new NotFound(res.t('partyNotFound'));
    partyMembers = await User
      .findOne({_id: targetId, 'party._id': party._id})
      // .select(partyMembersFields) Selecting the entire user because otherwise when saving it'll save
      // default values for non-selected fields and pre('save') will mess up thinking some values are missing
      .exec();
  }

  if (!partyMembers) throw new NotFound(res.t('userWithIDNotFound', {userId: targetId}));

  for (let i = 0; i < quantity; i += 1) {
    spell.cast(user, partyMembers, req);
  }

  if (partyMembers !== user) {
    await Promise.all([
      user.save(),
      partyMembers.save(),
    ]);
  } else {
    await partyMembers.save(); // partyMembers is user
  }

  return partyMembers;
}

export {castTaskSpell, castMultiTaskSpell, castSelfSpell, castPartySpell, castUserSpell};