initial commit

src/machine.ts

@@ -0,0 +1,84 @@
+import {
+	buildReflectors,
+	buildRotors,
+	ReflectorValues,
+	RotorValues,
+} from "./loadComponentData";
+import { Plugboard } from "./plugboard";
+import { Rotor } from "./rotors";
+import { checkCharacter } from "./validation";
+
+const rotors = buildRotors("rotors.csv");
+const reflectors = buildReflectors("reflectors.csv");
+
+export class Machine {
+	plugboard: Plugboard;
+	reflector: ReflectorValues;
+	rotors: Rotor[];
+
+	constructor(reflector: string, first: string, second: string, third: string) {
+		this.plugboard = new Plugboard();
+		this.reflector = getComponent(reflectors, reflector);
+
+		this.rotors = [];
+		this.rotors.push(getRotor(getComponent(rotors, first)));
+		this.rotors.push(getRotor(getComponent(rotors, second)));
+		this.rotors.push(getRotor(getComponent(rotors, third)));
+	}
+
+	rings(): string[] {
+		return this.rotors.map((r) => r.position());
+	}
+
+	stecker(one: string, two: string) {
+		this.plugboard.stecker(one, two);
+	}
+
+	input(message: string): string {
+		let output = "";
+		for (let char of message) {
+			checkCharacter(char, "character");
+			let turnover = true;
+
+			// plugboard, first pass
+			let scrambled = this.plugboard.in(char);
+
+			// rotors, first pass (in)
+			for (const rotor of this.rotors) {
+				if (turnover) {
+					turnover = rotor.rotate();
+				}
+
+				scrambled = rotor.in(scrambled);
+			}
+
+			// reflector
+			scrambled = this.reflector.reflector.get(scrambled)!;
+
+			// rotors, second pass (out)
+			for (let i = this.rotors.length - 1; i >= 0; i--) {
+				const rotor = this.rotors[i];
+				scrambled = rotor.out(scrambled);
+			}
+
+			// plugboard, second pass
+			output += this.plugboard.in(scrambled);
+		}
+
+		return output;
+	}
+}
+
+function getRotor({ name, values, turnovers }: RotorValues) {
+	return new Rotor(name, values, turnovers, 0);
+}
+
+function getComponent<T extends { name: string }>(
+	components: T[],
+	componentName: string,
+): T {
+	const component = components.find(({ name }) => name === componentName);
+	if (!component) throw new Error(`unknown reflector: ${component}`);
+
+	return component;
+}