Add string and byte slice (#5)

encode.go

@@ -40,6 +40,20 @@ func (e *Encoder) marshal(v any) error {
 		}
 		e.buf.Write(raw)
 
+	case []byte:
+		var b [4]byte
+		binary.LittleEndian.PutUint32(b[:], uint32(len(v)+1))
+		e.buf.Write(b[:])
+		e.buf.WriteByte(0x80) // TODO(cristaloleg): better binary type?
+		e.buf.Write(v)
+
+	case string:
+		var b [4]byte
+		binary.LittleEndian.PutUint32(b[:], uint32(len(v)+1))
+		e.buf.Write(b[:])
+		e.buf.Write([]byte(v))
+		e.buf.WriteByte(0)
+
 	case int32:
 		var b [4]byte
 		binary.LittleEndian.PutUint32(b[:], uint32(v))

encode_test.go

@@ -24,3 +24,35 @@ func TestEncode(t *testing.T) {
 	wantBytes(t, buf.Bytes(), "01")
 	buf.Reset()
 }
+
+func TestEncodeBytes(t *testing.T) {
+	var err error
+	var buf bytes.Buffer
+	enc := NewEncoder(&buf)
+
+	err = enc.Encode([]byte("foo"))
+	mustOk(t, err)
+	wantBytes(t, buf.Bytes(), "0400000080666f6f")
+	buf.Reset()
+
+	err = enc.Encode([]byte{0x00})
+	mustOk(t, err)
+	wantBytes(t, buf.Bytes(), "020000008000")
+	buf.Reset()
+}
+
+func TestEncodeString(t *testing.T) {
+	var err error
+	var buf bytes.Buffer
+	enc := NewEncoder(&buf)
+
+	err = enc.Encode("foo")
+	mustOk(t, err)
+	wantBytes(t, buf.Bytes(), "04000000666f6f00")
+	buf.Reset()
+
+	err = enc.Encode("")
+	mustOk(t, err)
+	wantBytes(t, buf.Bytes(), "0100000000")
+	buf.Reset()
+}