balance(ai): Moveset generation rebalancing (#6962)

* [WIP] movegen rebalance part 1

* Add script to generate random movesets

* finalize movegen improvements

* Cleanup licenses in scripts

* Add tm denylist

* Add level based denylist

* Misc cleanup

* Allow specifying ability for movegen

* Allow specifying form index and egg moves

* fix improper conditions for singles moves and level based denylist

* Add final filtering for useless moves

Also update forced signature moves

* Add explicit undefined to allowable payload types

* Remove `src/test.ts`

* Fix license snippet in gen-moveset-driver

* Enable rival-based signature moves

* Adjust forbidden-moves

Disallowed in singles: +Heal Pulse

Disallowed TMs: +Take Down

Disallowed Level-Based: +Teleport, +Gear Up, +Imprison, +Spark, +Dream Eater, +Nightmare

* Remove Rest from its own superceded list

* Sort `forbidden-moves.ts` arrays and add missing TSDoc type import

* Fix function name, fix a couple of typos

* Format `moveset-generation.ts`

* Format `forbidden-moves.ts` and `signature-moves.ts`

* Format `superceded-moves.ts`

* Fix some docs/comments

* Remove duplicate comment

* Fix variable name

* Fix inconsistencies in superceded moves

* Clarify some variable names/comments wrt weather/terrain setting moves

* Balance Team adjustments to Sig List

* Remove duplicate entry in `signature-moves.ts`

* Add new files to `biome.jsonc` so they can be formatted

* misc: address comments from code review

* fix: pnpx changed to pnpm dlx for cross compat with windows

* Update signature-moves.ts

* Eradicate duplicate Togedemaru entry

* Update forbidden-moves.ts

* Fix duplicate comment

* fix: gen-moveset script respects custom reporter

* feat: add take down to level-based denylist

* misc: remove params variable from gen-moveset.ts

* docs: add description of forRival param to generateAndPopulateMoveset

* fix: actually pass forRival when generating movesets for rivals

* feat: Make parental bond impact moveset generation

* feat: Make power ceiling cap configurable

* fix: removeSelfStatBoost works properly now

* Update Forbidden Move and Signature Move List, change Rival Pidgeot ability

* Update signature-moves.ts

* Add Steel Beam, Psywave, SonicBoom, and Synchronoise to denylist

* docs: add docs for `useRivalSignature` to `forceRivalBirdAbility`

* Address Kev's comments from code review

Co-authored-by: NightKev <34855794+DayKev@users.noreply.github.com>

* Address Bertie's comment

---------

Co-authored-by: NightKev <34855794+DayKev@users.noreply.github.com>
Co-authored-by: Blitzy <118096277+Blitz425@users.noreply.github.com>

Author: 3 people

biome.jsonc

@@ -16,7 +16,15 @@
     "enabled": true,
     "useEditorconfig": true,
     "indentStyle": "space",
-    "includes": ["**", "!src/data/balance/**", "src/data/balance/biomes/**", "src/data/balance/timed-events.ts"], // TODO: enable formatting of balance folder
+    "includes": [
+      "**",
+      "!src/data/balance/**",
+      "src/data/balance/moves/forbidden-moves.ts",
+      "src/data/balance/moves/signature-moves.ts",
+      "src/data/balance/moves/superceded-moves.ts",
+      "src/data/balance/biomes/**",
+      "src/data/balance/timed-events.ts"
+    ], // TODO: enable formatting of balance folder
     "lineWidth": 120
   },
   "files": {
@@ -315,6 +323,22 @@
     }
   },
   "overrides": [
+    {
+      "includes": ["**/src/ai/**/*.ts"],
+      "linter": {
+        "rules": {
+          "complexity": {
+            "noExcessiveCognitiveComplexity": {
+              "level": "info",
+              "options": {
+                // Code for AI behavior can have a slightly higher complexity due to the nature of decision trees and such
+                "maxAllowedComplexity": 20
+              }
+            }
+          }
+        }
+      }
+    },
     {
       "includes": ["**/scripts/**/*.js"],
       "linter": {

package.json

@@ -39,7 +39,8 @@
     "update-assets": "pnpm update-submodules assets",
     "update-assets:remote": "pnpm update-submodules:remote assets",
     "update-submodules": "git submodule update --progress --init --recursive --force --depth 1",
-    "update-submodules:remote": "pnpm update-submodules --remote"
+    "update-submodules:remote": "pnpm update-submodules --remote",
+    "sample-movesets": "pnpm dlx vite-node scripts/gen-moveset/gen-moveset-driver.ts"
   },
   "dependencies": {
     "@material/material-color-utilities": "^0.3.0",

pnpm-workspace.yaml

@@ -5,6 +5,7 @@ onlyBuiltDependencies:
   - esbuild
   - lefthook
   - msw
+  - vite-node
 
 overrides:
   ajv@>=7.0.0-alpha.0 <8.18.0: '>=8.18.0'

scripts/gen-moveset/gen-moveset-driver.ts

@@ -0,0 +1,174 @@
+/*
+ * SPDX-FileCopyrightText: 2026 Pagefault Games
+ * SPDX-FileContributor: SirzBenjie
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+import { EGG_MOVE_LEVEL_REQUIREMENT } from "#balance/moves/moveset-generation";
+import { SpeciesId } from "#enums/species-id";
+import { spawn } from "node:child_process";
+import net from "node:net";
+import { confirm, input, number as promptNumber } from "@inquirer/prompts";
+import chalk from "chalk";
+import type { SamplerPayload } from "./types";
+
+/**
+ * Starts a TCP server and returns a Promise that resolves with the dataPromise and port.
+ */
+function spawnServer(payload: SamplerPayload): Promise<{ dataPromise: Promise<string>; port: number }> {
+  const { promise: dataPromise, resolve: dataResolver } = Promise.withResolvers<string>();
+
+  const server = net.createServer(socket => {
+    socket.write(JSON.stringify(payload));
+    let data = "";
+    socket.on("data", chunk => {
+      data += chunk.toString();
+    });
+    socket.on("end", () => {
+      server.close();
+      dataResolver(data);
+    });
+  });
+
+  const { promise, resolve } = Promise.withResolvers<{ dataPromise: Promise<string>; port: number }>();
+  server.listen(0, () => {
+    const port = (server.address() as net.AddressInfo).port;
+    resolve({ dataPromise, port });
+  });
+
+  return promise;
+}
+
+async function getDataFromChildProcess(payload: SamplerPayload): Promise<void> {
+  const { dataPromise, port } = await spawnServer(payload);
+
+  const child = spawn("pnpm", ["vitest", "-c", "scripts/gen-moveset/gen-moveset.config.ts"], {
+    env: {
+      ...process.env,
+      COMMUNICATION_PORT: port.toString(),
+    },
+    stdio: ["inherit", "ignore", "ignore"],
+  });
+
+  child.on("exit", async () => {
+    const data = await dataPromise;
+    if (!data) {
+      console.error("No data received from child process.");
+      process.exit(1);
+    }
+
+    process.stdout.write(data + "\n");
+  });
+}
+
+async function promptInputs(): Promise<SamplerPayload> {
+  const speciesInput = await input({
+    message: "Enter the species ID or name of the Pokémon to generate movesets for:",
+    validate: value => {
+      return SpeciesId[value.toUpperCase()] != null;
+    },
+  });
+
+  let speciesName: string;
+  let speciesId = Number(speciesInput) as SpeciesId;
+  let wasNum = false;
+  if (Number.isNaN(speciesId)) {
+    speciesName = speciesInput;
+    speciesId = SpeciesId[speciesInput.toUpperCase()] as SpeciesId;
+  } else {
+    speciesName = SpeciesId[speciesId];
+    wasNum = true;
+  }
+  speciesName = speciesName[0].toUpperCase() + speciesName.slice(1).toLowerCase();
+  if (wasNum) {
+    console.log(chalk.bold(`  Selected species: ${speciesName}`));
+  }
+
+  const boss = await confirm({
+    message: "Generate as boss (default yes)?",
+    default: true,
+  });
+
+  const forTrainer = await confirm({
+    message: "Generate as for a trainer (default yes)?",
+    default: true,
+  });
+  let forRival = false;
+  if (forTrainer) {
+    forRival = await confirm({
+      message: "Generate as a Rival's Pokémon (default no)?",
+      default: false,
+    });
+  }
+
+  const level = await promptNumber({
+    message: "Enter the level of the Pokémon (default 100):",
+    default: 100,
+    max: 200,
+    min: 3,
+    required: true,
+  });
+
+  let allowEggMoves: boolean | undefined;
+  if (forTrainer && level >= EGG_MOVE_LEVEL_REQUIREMENT) {
+    allowEggMoves = await confirm({
+      message: "Allow egg moves? (default no)?",
+      default: false,
+    });
+  }
+  const formIndex = await promptNumber({
+    message: "Enter the form index to generate (if empty or invalid, will default to the base form):",
+    default: 0,
+    min: 0,
+    required: false,
+  });
+
+  const abilityIndex = await promptNumber({
+    message: "Enter an ability index to force (leave blank to not force any):",
+    min: 0,
+    max: 2,
+    required: false,
+  });
+
+  const trials = await promptNumber({
+    message: "Enter the number of movesets to generate (default 100):",
+    default: 100,
+    min: 1,
+    required: true,
+  });
+
+  const printWeights = await confirm({
+    message: "Print move weight details (default no)?",
+    default: false,
+  });
+
+  return {
+    speciesId,
+    boss,
+    level,
+    trials,
+    printWeights,
+    forTrainer,
+    abilityIndex,
+    allowEggMoves,
+    formIndex,
+    forRival,
+  };
+}
+
+async function main() {
+  let payload: SamplerPayload;
+  try {
+    payload = await promptInputs();
+  } catch (err: any) {
+    // Suppress annoying stack trace on SIGINT
+    if (err?.message.includes("User force closed the prompt with SIGINT")) {
+      process.exit(130);
+    }
+    throw err;
+  }
+
+  await getDataFromChildProcess(payload);
+}
+
+await main();

scripts/gen-moveset/gen-moveset.config.ts

@@ -0,0 +1,39 @@
+/*
+ * SPDX-FileCopyrightText: 2024-2025 Pagefault Games
+ * SPDX-FileContributor: SirzBenjie
+ *
+ * SPDX-License-Identifier: AGPL-3.0-only
+ */
+
+import { CustomDefaultReporter } from "#test/reporters/custom-default-reporter";
+import type { UserConfig } from "vite";
+import { defineConfig } from "vitest/config";
+import { sharedConfig } from "../../vite.config";
+
+// biome-ignore lint/style/noDefaultExport: required for vitest
+export default defineConfig(async config => {
+  const viteConfig = await sharedConfig(config);
+  const opts: UserConfig = {
+    ...viteConfig,
+    test: {
+      passWithNoTests: false,
+      env: {
+        TZ: "UTC",
+      },
+      reporters: [new CustomDefaultReporter()],
+      setupFiles: ["./test/setup/font-face.setup.ts", "./test/setup/vitest.setup.ts"],
+      includeTaskLocation: true,
+      environment: "jsdom",
+      environmentOptions: {
+        jsdom: {
+          resources: "usable",
+        },
+      },
+      restoreMocks: true,
+      watch: false,
+      name: "gen-moveset",
+      include: ["./scripts/gen-moveset/gen-moveset.ts"],
+    },
+  };
+  return opts;
+});