Refactoring

1 files changed, 49 insertions, 0 deletions

diff --git a/src/cmd_query.cc b/src/cmd_query.cc
new file mode 100644
index 0000000..7d4e6d6
--- /dev/null
+++ b/src/cmd_query.cc
@@ -0,0 +1,49 @@
+#include "cmd.h"
+#include "utils.h"
+#include "creature.h"
+#include "dice.h"
+#include "weapon.h"
+#include <sstream>
+
+namespace cmd {
+    std::string attacks(std::vector<std::string> args) {
+        std::stringstream text;
+	auto c = utils::instantiate<creature::Creature>(getTruePath(args[0]));
+	for(auto w : creature::getAttacks(*c)) {
+	    text << w->getName() << std::endl;
+	}
+        return text.str();
+    }
+
+    std::string roll(std::vector<std::string> args) {
+        std::stringstream text;
+	auto c = utils::instantiate<creature::Creature>(getTruePath(args[0]));
+	args.erase(args.begin());
+	std::string rollName = utils::join(args, " ");
+	utils::lower(rollName);
+	int rolled = dice::roll(20);
+	auto printResults = [&text](std::string name, std::string type, int rolled, int bonus) {
+	    text << name << " " << type << ": " << rolled << " (d20) + " << bonus << " (" << name << " " << type << " bonus) = " << rolled + bonus << std::endl;
+	};
+	// Search through skills, saves, and attacks to roll
+	rules::Skill skill = rules::tryGetAbilityOrSkill<rules::Skill>(rollName);
+	rules::Ability ability = rules::tryGetAbilityOrSkill<rules::Ability>(rollName);
+	if(skill) {
+	    printResults(skill.getName(), "check", rolled, c->getSkillBonus(skill));
+	} else if(ability) {
+	    printResults(ability.getFull(), "save", rolled, c->getAbilitySaveBonus(ability));
+	} else {
+	    for(auto w : creature::getAttacks(*c)) {
+		if(w->getName() == rollName) {
+		    text << w->getText(*c) << std::endl;
+		    int abilityBonus = c->getBonus(creature::getBestAbility(getAbilityOptions(*w), *c));
+		    int bonus = abilityBonus + c->getProficiency();
+		    printResults(w->getName(), "attack", rolled, bonus);
+		    text << "  on hit: " << entry::formatDmg(*w, *c) << std::endl;
+		    break;
+		}
+	    }
+	}
+        return text.str();
+    }
+}