habitica/common/script/fns/autoAllocate.js

import _ from 'lodash';
import splitWhitespace from '../libs/splitWhitespace';

/*
  Updates user stats with new stats. Handles death, leveling up, etc
  {stats} new stats
  {update} if aggregated changes, pass in userObj as update. otherwise commits will be made immediately
 */

function getStatToAllocate (user) {
  let suggested;

  let statsObj = user.stats.toObject ? user.stats.toObject() : user.stats;

  switch (user.preferences.allocationMode) {
    case 'flat': {
      let stats = _.pick(statsObj, splitWhitespace('con str per int'));
      return _.invert(stats)[_.min(stats)];
    }
    case 'classbased': {
      let lvlDiv7 = statsObj.lvl / 7;
      let ideal = [lvlDiv7 * 3, lvlDiv7 * 2, lvlDiv7, lvlDiv7];

      let preference;
      switch (statsObj.class) {
        case 'wizard': {
          preference = ['int', 'per', 'con', 'str'];
          break;
        }
        case 'rogue': {
          preference = ['per', 'str', 'int', 'con'];
          break;
        }
        case 'healer': {
          preference = ['con', 'int', 'str', 'per'];
          break;
        }
        default: {
          preference = ['str', 'con', 'per', 'int'];
        }
      }

      let diff = [
        statsObj[preference[0]] - ideal[0],
        statsObj[preference[1]] - ideal[1],
        statsObj[preference[2]] - ideal[2],
        statsObj[preference[3]] - ideal[3],
      ];

      suggested = _.findIndex(diff, (val) => {
        if (val === _.min(diff)) return true;
      });

      return suggested !== -1 ? preference[suggested] : 'str';
    }
    case 'taskbased': {
      suggested = _.invert(statsObj.training)[_.max(statsObj.training)];

      let training = statsObj.training;
      training.str = 0;
      training.int = 0;
      training.con = 0;
      training.per = 0;

      return suggested || 'str';
    }
    default: {
      return 'str';
    }
  }
}

module.exports = function autoAllocate (user) {
  return user.stats[getStatToAllocate(user)]++;
};