initial commitHEADmain

6 files changed, 166 insertions, 0 deletions

diff --git a/demo/dem/data.go b/demo/dem/data.go
new file mode 100644
index 0000000..87ff90a
--- /dev/null
+++ b/demo/dem/data.go
@@ -0,0 +1,54 @@
+package dem
+
+import (
+	"github.com/osm/quake/common/buffer"
+	"github.com/osm/quake/common/context"
+	"github.com/osm/quake/packet"
+	"github.com/osm/quake/packet/svc"
+)
+
+type Data struct {
+	Size   uint32
+	Angle  [3]float32
+	Packet packet.Packet
+}
+
+func (d *Data) Bytes() []byte {
+	buf := buffer.New()
+
+	buf.PutUint32(d.Size)
+
+	for i := 0; i < 3; i++ {
+		buf.PutFloat32(d.Angle[i])
+	}
+
+	buf.PutBytes(d.Packet.Bytes())
+
+	return buf.Bytes()
+}
+
+func parseData(ctx *context.Context, buf *buffer.Buffer) (*Data, error) {
+	var err error
+	var data Data
+
+	if data.Size, err = buf.GetUint32(); err != nil {
+		return nil, err
+	}
+
+	for i := 0; i < 3; i++ {
+		if data.Angle[i], err = buf.GetFloat32(); err != nil {
+			return nil, err
+		}
+	}
+
+	bytes, err := buf.GetBytes(int(data.Size))
+	if err != nil {
+		return nil, err
+	}
+
+	if data.Packet, err = svc.Parse(ctx, bytes); err != nil {
+		return nil, err
+	}
+
+	return &data, nil
+}
diff --git a/demo/dem/parse.go b/demo/dem/parse.go
new file mode 100644
index 0000000..203a711
--- /dev/null
+++ b/demo/dem/parse.go
@@ -0,0 +1,57 @@
+package dem
+
+import (
+	"errors"
+
+	"github.com/osm/quake/common/buffer"
+	"github.com/osm/quake/common/context"
+)
+
+var ErrUnknownType = errors.New("unknown type")
+
+type Demo struct {
+	CDTrack []byte
+	Data    []*Data
+}
+
+func (dem *Demo) Bytes() []byte {
+	buf := buffer.New()
+
+	buf.PutBytes(dem.CDTrack)
+
+	for _, d := range dem.Data {
+		buf.PutBytes(d.Bytes())
+	}
+
+	return buf.Bytes()
+}
+
+func Parse(ctx *context.Context, data []byte) (*Demo, error) {
+	var demo Demo
+
+	buf := buffer.New(buffer.WithData(data))
+	ctx.SetIsDem(true)
+
+	for buf.Off() < buf.Len() {
+		b, err := buf.ReadByte()
+		if err != nil {
+			return nil, err
+		}
+
+		demo.CDTrack = append(demo.CDTrack, b)
+		if b == '\n' {
+			break
+		}
+	}
+
+	for buf.Off() < buf.Len() {
+		d, err := parseData(ctx, buf)
+		if err != nil {
+			return nil, err
+		}
+
+		demo.Data = append(demo.Data, d)
+	}
+
+	return &demo, nil
+}
diff --git a/demo/dem/parse_test.go b/demo/dem/parse_test.go
new file mode 100644
index 0000000..d3d33cf
--- /dev/null
+++ b/demo/dem/parse_test.go
@@ -0,0 +1,55 @@
+package dem
+
+import (
+	"crypto/sha256"
+	"fmt"
+	"io/ioutil"
+	"testing"
+
+	"github.com/osm/quake/common/context"
+)
+
+type demTest struct {
+	filePath string
+	checksum string
+}
+
+var demTests = []demTest{
+	{
+		filePath: "testdata/demo1.dem",
+		checksum: "893e5279e3a84fed0416a1101dfc2b6bca1ebd28787323daa8e22cefe54883c1",
+	},
+	{
+		filePath: "testdata/demo2.dem",
+		checksum: "150e6fc54c24a70a44012ba71473a5cbbec081e19204a9eae98249621fef00d7",
+	},
+	{
+		filePath: "testdata/demo3.dem",
+		checksum: "7bee6edc47fe563cbf8f1e873d524d1e4cef7d919b523606e200c0d7c269bd83",
+	},
+}
+
+func TestParse(t *testing.T) {
+	for _, dt := range demTests {
+		t.Run(dt.filePath, func(t *testing.T) {
+			data, err := ioutil.ReadFile(dt.filePath)
+			if err != nil {
+				t.Errorf("unable to open demo file, %v", err)
+			}
+
+			demo, err := Parse(context.New(), data)
+			if err != nil {
+				t.Errorf("unable to parse demo, %v", err)
+			}
+
+			h := sha256.New()
+			h.Write(demo.Bytes())
+			checksum := fmt.Sprintf("%x", h.Sum(nil))
+			if checksum != dt.checksum {
+				t.Errorf("sha256 checksums didn't match")
+				t.Logf("output: %#v", checksum)
+				t.Logf("expected: %#v", dt.checksum)
+			}
+		})
+	}
+}
diff --git a/demo/dem/testdata/demo1.dem b/demo/dem/testdata/demo1.dem
new file mode 100644
index 0000000..2b821c3
--- /dev/null
+++ b/demo/dem/testdata/demo1.dem
diff --git a/demo/dem/testdata/demo2.dem b/demo/dem/testdata/demo2.dem
new file mode 100644
index 0000000..9dd006e
--- /dev/null
+++ b/demo/dem/testdata/demo2.dem
diff --git a/demo/dem/testdata/demo3.dem b/demo/dem/testdata/demo3.dem
new file mode 100644
index 0000000..d25c402
--- /dev/null
+++ b/demo/dem/testdata/demo3.dem