import { ZodError } from "zod";
import { resolveMove } from "@/tools/calculateDamage/handlers/helpers/resolveMove";
import { calculateDamageWithContext } from "@/utils/calculateDamageWithContext";
import { calculateHp } from "@/utils/calculateHp";
import { calculateStat } from "@/utils/calculateStat";
import { NATURE_MODIFIER_MAP } from "@/utils/natureModifier";
import {
type CalculateDamageMatrixVaryingDefenseInput,
calculateDamageMatrixVaryingDefenseInputSchema,
} from "./schemas/damageMatrixVaryingDefenseSchema";
interface DamageMatrixEntry {
ev: number;
defenseStat: number;
hpStat: number;
damages: number[];
}
/**
* 防御側の努力値を総当たりしてダメージ計算を行う
*/
export const calculateDamageMatrixVaryingDefenseHandler = async (
args: unknown,
): Promise<{
content: Array<{ type: "text"; text: string }>;
}> => {
try {
const input = calculateDamageMatrixVaryingDefenseInputSchema.parse(args);
// 技の解決処理を追加
const resolvedMove = resolveMove(input.move);
const inputWithResolvedMove = {
...input,
move: resolvedMove,
};
const results = calculateDamageMatrix(inputWithResolvedMove);
return {
content: [
{
type: "text",
text: JSON.stringify(results, null, 2),
},
],
};
} catch (error) {
if (error instanceof ZodError) {
return {
content: [
{
type: "text",
text: JSON.stringify(
{
error: "入力値が不正です",
details: error.errors.map((e) => ({
path: e.path.join("."),
message: e.message,
})),
},
null,
2,
),
},
],
};
}
return {
content: [
{
type: "text",
text: JSON.stringify(
{
error:
error instanceof Error
? error.message
: "不明なエラーが発生しました",
},
null,
2,
),
},
],
};
}
};
/**
* ダメージマトリックスを計算
*/
const calculateDamageMatrix = (
input: CalculateDamageMatrixVaryingDefenseInput & {
move: ReturnType<typeof resolveMove>;
},
): { damageMatrix: DamageMatrixEntry[] } => {
const { move, attacker, defender, options } = input;
// 攻撃側のステータスを計算
const attackStat = (() => {
if ("value" in attacker.stat) {
return attacker.stat.value;
}
const baseStat = move.isPhysical
? (attacker.pokemon?.baseStats.atk ?? 100)
: (attacker.pokemon?.baseStats.spa ?? 100);
return calculateStat({
baseStat,
level: attacker.level,
iv: attacker.stat.iv,
ev: attacker.stat.ev,
natureModifier:
NATURE_MODIFIER_MAP[attacker.stat.natureModifier || "neutral"],
});
})();
// 防御側の基礎情報
const defenderBaseStat = defender.isPhysicalDefense
? defender.pokemon.baseStats.def
: defender.pokemon.baseStats.spd;
const defenderHpBaseStat = defender.pokemon.baseStats.hp;
// 努力値の範囲（0, 4, 12, 20, ..., 252）
const evValues: number[] = [0];
for (let ev = 4; ev <= 252; ev += 8) {
evValues.push(ev);
}
// 各努力値でダメージを計算
const damageMatrix: DamageMatrixEntry[] = evValues.map((ev) => {
// 防御/特防の実数値を計算
const defenseStat = calculateStat({
baseStat: defenderBaseStat,
level: defender.level,
iv: defender.stat.iv,
ev,
natureModifier:
NATURE_MODIFIER_MAP[defender.stat.natureModifier || "neutral"],
});
// HPの実数値を計算（参考値として）
const hpStat = calculateHp({
baseStat: defenderHpBaseStat,
level: defender.level,
iv: 31, // HP個体値は31で固定（一般的な仮定）
ev: 0, // HP努力値は0で固定（防御/特防に振る想定）
});
// ダメージ計算
const damages = calculateDamageWithContext({
move: {
name: move.name,
type: move.type,
power: move.power,
isPhysical: move.isPhysical,
},
attackStat,
defenseStat,
attacker: {
level: attacker.level,
statModifier: attacker.statModifier,
pokemon: attacker.pokemon,
ability: attacker.ability,
abilityActive: attacker.abilityActive,
item: attacker.item,
},
defender: {
statModifier: defender.statModifier,
pokemon: defender.pokemon,
ability: defender.ability,
abilityActive: defender.abilityActive,
item: defender.item,
},
options,
});
return {
ev,
defenseStat,
hpStat,
damages,
};
});
return { damageMatrix };
};
