initial commitHEADmain

2 files changed, 106 insertions, 0 deletions

diff --git a/packet/command/nails/nails.go b/packet/command/nails/nails.go
new file mode 100644
index 0000000..a4a3987
--- /dev/null
+++ b/packet/command/nails/nails.go
@@ -0,0 +1,50 @@
+package nails
+
+import (
+	"github.com/osm/quake/common/buffer"
+	"github.com/osm/quake/common/context"
+	"github.com/osm/quake/protocol"
+)
+
+type Command struct {
+	Count   byte
+	Command []Nail
+}
+
+type Nail struct {
+	Bits []byte
+}
+
+func (cmd *Command) Bytes() []byte {
+	buf := buffer.New()
+
+	buf.PutByte(protocol.SVCNails)
+	buf.PutByte(cmd.Count)
+
+	for i := 0; i < len(cmd.Command); i++ {
+		buf.PutBytes(cmd.Command[i].Bits)
+	}
+
+	return buf.Bytes()
+}
+
+func Parse(ctx *context.Context, buf *buffer.Buffer) (*Command, error) {
+	var err error
+	var cmd Command
+
+	if cmd.Count, err = buf.ReadByte(); err != nil {
+		return nil, err
+	}
+
+	for i := 0; i < int(cmd.Count); i++ {
+		var nail Nail
+
+		if nail.Bits, err = buf.GetBytes(6); err != nil {
+			return nil, err
+		}
+
+		cmd.Command = append(cmd.Command, nail)
+	}
+
+	return &cmd, nil
+}
diff --git a/packet/command/nails2/nails2.go b/packet/command/nails2/nails2.go
new file mode 100644
index 0000000..9bf8038
--- /dev/null
+++ b/packet/command/nails2/nails2.go
@@ -0,0 +1,56 @@
+package nails2
+
+import (
+	"github.com/osm/quake/common/buffer"
+	"github.com/osm/quake/common/context"
+	"github.com/osm/quake/protocol"
+)
+
+type Command struct {
+	Count byte
+	Nails []Nail2
+}
+
+type Nail2 struct {
+	Index byte
+	Bits  []byte
+}
+
+func (cmd *Command) Bytes() []byte {
+	buf := buffer.New()
+
+	buf.PutByte(protocol.SVCNails2)
+	buf.PutByte(cmd.Count)
+
+	for i := 0; i < len(cmd.Nails); i++ {
+		buf.PutByte(cmd.Nails[i].Index)
+		buf.PutBytes(cmd.Nails[i].Bits)
+	}
+
+	return buf.Bytes()
+}
+
+func Parse(ctx *context.Context, buf *buffer.Buffer) (*Command, error) {
+	var err error
+	var cmd Command
+
+	if cmd.Count, err = buf.ReadByte(); err != nil {
+		return nil, err
+	}
+
+	for i := 0; i < int(cmd.Count); i++ {
+		var nail Nail2
+
+		if nail.Index, err = buf.ReadByte(); err != nil {
+			return nil, err
+		}
+
+		if nail.Bits, err = buf.GetBytes(6); err != nil {
+			return nil, err
+		}
+
+		cmd.Nails = append(cmd.Nails, nail)
+	}
+
+	return &cmd, nil
+}