add messaging

src/message/message.go

@@ -0,0 +1,40 @@
+package message
+
+import (
+	"encoding/json"
+
+	"github.com/schollz/croc/v6/src/compress"
+	"github.com/schollz/croc/v6/src/crypt"
+)
+
+type Message struct {
+	Type    string `json:"t,omitempty"`
+	Message string `json:"m,omitempty"`
+	Bytes   []byte `json:"b,omitempty"`
+	Num     int    `json:"n,omitempty"`
+}
+
+func Encode(m Message, e ...crypt.Encryption) (b []byte, err error) {
+	b, err = json.Marshal(m)
+	if err != nil {
+		return
+	}
+
+	b = compress.Compress(b)
+	if len(e) > 0 {
+		b, err = e[0].Encrypt(b)
+	}
+	return
+}
+
+func Decode(b []byte, e ...crypt.Encryption) (m Message, err error) {
+	if len(e) > 0 {
+		b, err = e[0].Decrypt(b)
+		if err != nil {
+			return
+		}
+	}
+	b = compress.Decompress(b)
+	err = json.Unmarshal(b, &m)
+	return
+}

src/message/message_test.go

@@ -0,0 +1,23 @@
+package message
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/schollz/croc/v6/src/crypt"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMessage(t *testing.T) {
+	m := Message{Type: "message", Message: "hello, world"}
+	e, err := crypt.New([]byte("passphrase"), nil)
+	assert.Nil(t, err)
+	fmt.Println(e.Salt())
+	b, err := Encode(m, e)
+	assert.Nil(t, err)
+	fmt.Printf("%x\n", b)
+
+	m2, err := Decode(b, e)
+	assert.Nil(t, err)
+	assert.Equal(t, m, m2)
+}