referendum result calculator

src/tools/referendum-result-calculator.ts

@@ -0,0 +1,47 @@
+import yargs from "yargs";
+import chalk from "chalk";
+import { BN, bnSqrt } from "@polkadot/util";
+import { calcPassing } from "@polkadot/api-derive/democracy/util";
+import { TypeRegistry } from "@polkadot/types/create";
+
+const registry = new TypeRegistry();
+
+const choices = ["isSuperMajorityApprove", "isSuperMajorityAgainst", "isSimpleMajority"];
+
+const args = yargs.options({
+  yes: { type: "string", demandOption: true, alias: "y" },
+  no: { type: "string", demandOption: true, alias: "n" },
+  turnout: { type: "string", demandOption: true, alias: "t" },
+  electorate: { type: "string", demandOption: true, alias: "e" },
+  voteThreshold: {
+    choices: choices,
+    demandOption: true,
+    alias: "vt",
+  },
+}).argv;
+
+async function main() {
+  const yes = new BN(args["yes"]);
+  const no = new BN(args["no"]);
+  const voters = new BN(args["turnout"]);
+  const sqrtElectorate = bnSqrt(new BN(args["electorate"]));
+
+  const voteThreshold = registry.createType(
+    "VoteThreshold",
+    choices.findIndex((x) => x == args["voteThreshold"])
+  );
+  let result = calcPassing(voteThreshold as any, sqrtElectorate, {
+    votedAye: yes,
+    votedNay: no,
+    votedTotal: voters,
+  });
+
+  console.log(`          Vote: ${chalk.green(yes.toString().padStart(26))}/${chalk.red(no.toString().padEnd(26))}`);
+  console.log(`Total issuance: ${new BN(args["electorate"]).toString().padStart(26)}`);
+  console.log(`        Voters: ${voters.toString().padStart(26)}`);
+  console.log("     Result is:", result);
+}
+
+main()
+  .catch(console.error)
+  .finally(() => process.exit());