Fixing some typos and adding a new instance of approach to visitor

Author: stimms

Chapter 5/State.js

@@ -26,6 +26,7 @@ var Westeros;
 
         var BankAccountManager = (function () {
             function BankAccountManager() {
+                this.currentState = new GoodStandingState(this);
             }
             BankAccountManager.prototype.Deposit = function (amount) {
                 this.currentState.Deposit(amount);
@@ -34,8 +35,8 @@ var Westeros;
             BankAccountManager.prototype.Withdraw = function (amount) {
                 this.currentState.Withdraw(amount);
             };
-            BankAccountManager.prototype.addToBalance = function (newBalance) {
-                this.balance = newBalance;
+            BankAccountManager.prototype.addToBalance = function (amount) {
+                this.balance += amount;
             };
             BankAccountManager.prototype.getBalance = function () {
                 return this.balance;

Chapter 5/State.ts

@@ -32,6 +32,10 @@ module Westeros.Banking{
 	export class BankAccountManager{
 		private balance: number;
 		private currentState: IState;
+		constructor() {
+		    this.currentState = new GoodStandingState(this);
+		}
+
 		public Deposit(amount: number)
 		{
 			this.currentState.Deposit(amount);
@@ -41,9 +45,9 @@ module Westeros.Banking{
 		{
 			this.currentState.Withdraw(amount);
 		}
-		public addToBalance(newBalance: number)
+		public addToBalance(amount: number)
 		{
-			this.balance = newBalance;
+			this.balance += amount;
 		}
 		public getBalance():number{
 			return this.balance;

Chapter 5/Visitor.js

@@ -54,6 +54,27 @@ var Westeros;
         })();
         Army.Lord = Lord;
 
+        var InstanceOfExample = (function () {
+            function InstanceOfExample() {
+            }
+            InstanceOfExample.prototype.Execute = function () {
+                var collection = [];
+                collection.push(Object.create(Knight));
+                collection.push(Object.create(FootSoldier));
+                collection.push(new Lord());
+                collection.push(new Archer());
+
+                for (var i = 0; i < collection.length; i++) {
+                    if (collection[i] instanceof Westeros.Army.Knight)
+                        collection[i].printName();
+                    else
+                        console.log("No match");
+                }
+            };
+            return InstanceOfExample;
+        })();
+        Army.InstanceOfExample = InstanceOfExample;
+
         var IfExample = (function () {
             function IfExample() {
             }
@@ -97,7 +118,7 @@ var Westeros;
             function SelectiveNamePrinterVisitor() {
             }
             SelectiveNamePrinterVisitor.prototype.visit = function (memberOfArmy) {
-                if (memberOfArmy._type == "Westeros.Army.Knight") {
+                if (memberOfArmy instanceof Westeros.Army.Knight) {
                     this.VisitKnight(memberOfArmy);
                 } else {
                     console.log("Not a knight");
@@ -113,5 +134,12 @@ var Westeros;
     var Army = Westeros.Army;
 })(Westeros || (Westeros = {}));
 
-var b = new Westeros.Army.VisitorExample();
+console.log("Instance of");
+var a = new Westeros.Army.InstanceOfExample();
+a.Execute();
+console.log("Type of");
+var b = new Westeros.Army.IfExample();
 b.Execute();
+console.log("Internal type");
+var c = new Westeros.Army.VisitorExample();
+c.Execute();

Chapter 5/Visitor.ts

@@ -51,6 +51,25 @@ module Westeros.Army{
 		}
 	}
 
+	export class InstanceOfExample{
+		Execute()
+		{
+			var collection = [];
+			collection.push(new Knight());
+			collection.push(new FootSoldier());
+			collection.push(new Lord());
+			collection.push(new Archer());
+
+			for(var i = 0; i< collection.length; i++)
+			{
+				if(collection[i] instanceof Westeros.Army.Knight)
+					collection[i].printName();
+				else 
+					console.log("No match");
+			}
+		}
+	}
+
 	export class IfExample{
 		Execute()
 		{
@@ -92,7 +111,7 @@ module Westeros.Army{
 	class SelectiveNamePrinterVisitor implements IVisitor{
 		public visit(memberOfArmy: IMemberOfArmy)
 		{
-			if(memberOfArmy._type == "Westeros.Army.Knight")
+			if(memberOfArmy instanceof Westeros.Army.Knight)
 			{
 				this.VisitKnight(memberOfArmy);
 			}
@@ -108,5 +127,12 @@ module Westeros.Army{
 	}
 }
 
-var b = new Westeros.Army.VisitorExample();
+console.log("Instance of");
+var a = new Westeros.Army.InstanceOfExample();
+a.Execute();
+console.log("Type of");
+var b = new Westeros.Army.IfExample();
 b.Execute();
+console.log("Vistor example");
+var c = new Westeros.Army.VisitorExample();
+c.Execute();