Clean up example in README

Author: markus-wa

README.md

@@ -46,19 +46,20 @@ func main() {
 	p := dem.NewParser(f, dem.WarnToStdErr)
 
 	// Parse header
-	p.ParseHeader()
-
-	// Get T / CT team state references (contain scores)
-	tState := p.TState()
-	ctState := p.CTState()
+	h, err := p.ParseHeader()
+	if err != nil {
+		t.Fatal(err)
+	}
+	fmt.Println("Map:", h.MapName)
 
 	// Register handler on round end to figure out who won
 	p.RegisterEventHandler(func(e events.RoundEndedEvent) {
-		if e.Winner == common.TeamTerrorists {
-			fmt.Println("T-side won the round - score:", tState.Score()+1) // Score + 1 because it hasn't actually been updated yet
-		} else if e.Winner == common.TeamCounterTerrorists {
-			fmt.Println("CT-side won the round - score:", ctState.Score()+1)
-		} else {
+		switch e.Winner {
+		case common.TeamTerrorists:
+			fmt.Println("T-side won the round - score:", p.TState().Score()+1) // Score + 1 because it hasn't actually been updated yet
+		case common.TeamCounterTerrorists:
+			fmt.Println("CT-side won the round - score:", p.CTState().Score()+1)
+		default:
 			fmt.Println("Apparently neither the Ts nor CTs won the round, interesting")
 		}
 	})

example_test.go

@@ -21,19 +21,20 @@ func TestExample(t *testing.T) {
 	p := dem.NewParser(f, dem.WarnToStdErr)
 
 	// Parse header
-	p.ParseHeader()
-
-	// Get T / CT team state references (contain scores)
-	tState := p.TState()
-	ctState := p.CTState()
+	h, err := p.ParseHeader()
+	if err != nil {
+		t.Fatal(err)
+	}
+	fmt.Println("Map:", h.MapName)
 
 	// Register handler on round end to figure out who won
 	p.RegisterEventHandler(func(e events.RoundEndedEvent) {
-		if e.Winner == common.TeamTerrorists {
-			fmt.Println("T-side won the round - score:", tState.Score()+1) // Score + 1 because it hasn't actually been updated yet
-		} else if e.Winner == common.TeamCounterTerrorists {
-			fmt.Println("CT-side won the round - score:", ctState.Score()+1)
-		} else {
+		switch e.Winner {
+		case common.TeamTerrorists:
+			fmt.Println("T-side won the round - score:", p.TState().Score()+1) // Score + 1 because it hasn't actually been updated yet
+		case common.TeamCounterTerrorists:
+			fmt.Println("CT-side won the round - score:", p.CTState().Score()+1)
+		default:
 			fmt.Println("Apparently neither the Ts nor CTs won the round, interesting")
 		}
 	})