add strings builder, customize ALL outputs (#6)

1 files changed, 14 insertions, 1 deletions

diff --git a/pkg/classes/stringTable.go b/pkg/classes/stringTable.go
index 3983c1f..44d8a73 100644
--- a/pkg/classes/stringTable.go
+++ b/pkg/classes/stringTable.go
@@ -2,6 +2,7 @@ package classes
 
 import (
         "github.com/pektezol/bitreader"
+        "github.com/pektezol/demoparser/pkg/writer"
 )
 
 type StringTable struct {
@@ -38,6 +39,7 @@ func ParseStringTables(reader *bitreader.Reader) []StringTable {
 func (stringTable *StringTable) ParseStream(reader *bitreader.Reader) {
         stringTable.Name = reader.TryReadString()
         entryCount := reader.TryReadBits(16)
+        writer.AppendLine("\tTable Name: %s", stringTable.Name)
         stringTable.TableEntries = make([]StringTableEntry, entryCount)
 
         for i := 0; i < int(entryCount); i++ {
@@ -45,7 +47,12 @@ func (stringTable *StringTable) ParseStream(reader *bitreader.Reader) {
                 entry.Parse(reader)
                 stringTable.TableEntries[i] = entry
         }
-
+        if entryCount != 0 {
+                writer.AppendLine("\t\t%d Table Entries:", entryCount)
+                writer.AppendOutputFromTemp()
+        } else {
+                writer.AppendLine("\t\tNo Table Entries")
+        }
         if reader.TryReadBool() {
                 classCount := reader.TryReadBits(16)
                 stringTable.Classes = make([]StringTableClass, classCount)
@@ -55,6 +62,10 @@ func (stringTable *StringTable) ParseStream(reader *bitreader.Reader) {
                         class.Parse(reader)
                         stringTable.Classes[i] = class
                 }
+                writer.AppendLine("\t\t%d Classes:", classCount)
+                writer.AppendOutputFromTemp()
+        } else {
+                writer.AppendLine("\t\tNo Class Entries")
         }
 }
 
@@ -73,8 +84,10 @@ func (stringTableEntry *StringTableEntry) Parse(reader *bitreader.Reader) {
 
 func (stringTableClass *StringTableClass) Parse(reader *bitreader.Reader) {
         stringTableClass.Name = reader.TryReadString()
+        writer.TempAppendLine("\t\t\tName: %s", stringTableClass.Name)
         if reader.TryReadBool() {
                 dataLen := reader.TryReadBits(16)
                 stringTableClass.Data = reader.TryReadStringLength(dataLen)
+                writer.TempAppendLine("\t\t\tData: %s", stringTableClass.Data)
         }
 }