Added stats generation

src/compiler/Compiler.ts

@@ -7,6 +7,7 @@ import { Story as ParsedStory } from "./Parser/ParsedHierarchy/Story";
 import { DebugMetadata } from "../engine/DebugMetadata";
 import { StringValue } from "../engine/Value";
 import { asOrNull } from "../engine/TypeAssertion";
+import { GenerateStoryStats, Stats } from "./Stats";
 
 export { CompilerOptions } from "./CompilerOptions";
 export { InkParser } from "./Parser/InkParser";
@@ -113,6 +114,13 @@ export class Compiler {
     }
   };
 
+  public readonly GenerateStats = (): Stats | null => {
+    if (this._parsedStory === null) {
+      return null;
+    }
+    return GenerateStoryStats(this._parsedStory);
+  };
+
   public readonly DebugMetadataForContentAtOffset = (
     offset: number
   ): DebugMetadata | null => {

src/compiler/Stats.ts

@@ -0,0 +1,52 @@
+import { Choice } from "./Parser/ParsedHierarchy/Choice";
+import { Divert } from "./Parser/ParsedHierarchy/Divert/Divert";
+import { Gather } from "./Parser/ParsedHierarchy/Gather/Gather";
+import { Knot } from "./Parser/ParsedHierarchy/Knot";
+import { Stitch } from "./Parser/ParsedHierarchy/Stitch";
+import { Story } from "./Parser/ParsedHierarchy/Story";
+import { Text } from "./Parser/ParsedHierarchy/Text";
+
+export interface Stats {
+  words: number;
+  knots: number;
+  stitches: number;
+  functions: number;
+  choices: number;
+  gathers: number;
+  diverts: number;
+}
+
+export function GenerateStoryStats(story: Story): Stats {
+  let allText = story.FindAll(Text)();
+  let words = 0;
+  for (const text of allText) {
+    let wordsInThisStr = 0;
+    let wasWhiteSpace = true;
+    for (const c of text.text) {
+      if (c == " " || c == "\t" || c == "\n" || c == "\r") {
+        wasWhiteSpace = true;
+      } else if (wasWhiteSpace) {
+        wordsInThisStr++;
+        wasWhiteSpace = false;
+      }
+    }
+
+    words += wordsInThisStr;
+  }
+
+  const knots = story.FindAll(Knot)();
+  const stitches = story.FindAll(Stitch)();
+  const choices = story.FindAll(Choice)();
+  const gathers = story.FindAll(Gather)((g) => g.debugMetadata != null);
+  const diverts = story.FindAll(Divert)();
+
+  return {
+    words,
+    knots: knots.length,
+    functions: knots.filter((k) => k.isFunction).length,
+    stitches: stitches.length,
+    gathers: gathers.length,
+    diverts: diverts.length - 1,
+    choices: choices.length,
+  };
+}