initial commitHEADmain

1 files changed, 66 insertions, 0 deletions

diff --git a/demo/qwd/parse.go b/demo/qwd/parse.go
new file mode 100644
index 0000000..458f714
--- /dev/null
+++ b/demo/qwd/parse.go
@@ -0,0 +1,66 @@
+package qwd
+
+import (
+	"errors"
+
+	"github.com/osm/quake/common/buffer"
+	"github.com/osm/quake/common/context"
+	"github.com/osm/quake/protocol"
+)
+
+var ErrUnknownType = errors.New("unknown type")
+
+type Demo struct {
+	Data []*Data
+}
+
+func (dem *Demo) Bytes() []byte {
+	buf := buffer.New()
+
+	for _, d := range dem.Data {
+		buf.PutBytes(d.Bytes())
+	}
+
+	return buf.Bytes()
+}
+
+func Parse(ctx *context.Context, data []byte) (*Demo, error) {
+	var err error
+	var cmd Demo
+
+	buf := buffer.New(buffer.WithData(data))
+	ctx.SetIsQWD(true)
+
+	for buf.Off() < buf.Len() {
+		var data Data
+
+		if data.Timestamp, err = buf.GetFloat32(); err != nil {
+			return nil, err
+		}
+
+		if data.Command, err = buf.ReadByte(); err != nil {
+			return nil, err
+		}
+
+		switch data.Command {
+		case protocol.DemoCmd:
+			if data.Cmd, err = parseCmd(ctx, buf); err != nil {
+				return nil, err
+			}
+		case protocol.DemoRead:
+			if data.Read, err = parseRead(ctx, buf); err != nil {
+				return nil, err
+			}
+		case protocol.DemoSet:
+			if data.Set, err = parseSet(ctx, buf); err != nil {
+				return nil, err
+			}
+		default:
+			return nil, ErrUnknownType
+		}
+
+		cmd.Data = append(cmd.Data, &data)
+	}
+
+	return &cmd, nil
+}