1 files changed, 10 insertions, 9 deletions

diff --git a/pkg/classes/customData.go b/pkg/classes/customData.go
index 1d6f30a..4c1a79e 100644
--- a/pkg/classes/customData.go
+++ b/pkg/classes/customData.go
@@ -2,29 +2,30 @@ package classes
 
 import (
         "github.com/pektezol/bitreader"
-        "github.com/pektezol/sdp.go/pkg/writer"
+        "github.com/pektezol/sdp.go/pkg/types"
 )
 
 type CustomData struct {
-        Type int32
-        Size int32
-        Data string
+        Type int32 `json:"type"`
+        Size int32 `json:"size"`
+        Data any   `json:"data"`
 }
 
-func (customData *CustomData) ParseCustomData(reader *bitreader.Reader, tickNumber int32, packetType uint8) {
+func (customData *CustomData) ParseCustomData(reader *bitreader.Reader, tickNumber int32, packetType uint8, demo *types.Demo) {
         customData.Type = reader.TryReadSInt32()
         customData.Size = reader.TryReadSInt32()
         if customData.Type != 0 || customData.Size == 8 {
                 // Not SAR data
-                writer.AppendLine("[%d] %s (%d):", tickNumber, "CUSTOMDATA", packetType)
+                demo.Writer.AppendLine("[%d] %s (%d):", tickNumber, "CUSTOMDATA", packetType)
                 customData.Data = string(reader.TryReadBytesToSlice(uint64(customData.Size)))
-                writer.AppendLine("\t%s", customData.Data)
+                demo.Writer.AppendLine("\t%s", customData.Data)
                 return
         }
         // SAR data
-        writer.AppendLine("[%d] %s (%d):", tickNumber, "SARDATA", packetType)
+        demo.Writer.AppendLine("[%d] %s (%d):", tickNumber, "SARDATA", packetType)
         sarData := SarData{}
         data := reader.TryReadBytesToSlice(uint64(customData.Size))
         sarReader := bitreader.NewReaderFromBytes(data, true)
-        sarData.ParseSarData(sarReader)
+        sarData.ParseSarData(sarReader, demo)
+        customData.Data = sarData
 }