diff --git a/encode.go b/encode.go index 60a9422..9c34d98 100644 --- a/encode.go +++ b/encode.go @@ -2,11 +2,13 @@ package coding import ( "bytes" + "github.com/icza/bitio" "github.com/samber/lo" + "github.com/securemesh/coding/heap" ) -func Encode(h *Heap, msg []byte) []byte { +func Encode(h *heap.Heap, msg []byte) []byte { buf := &bytes.Buffer{} w := bitio.NewWriter(buf) diff --git a/heap.go b/heap/heap.go similarity index 98% rename from heap.go rename to heap/heap.go index d996dd0..c07a31e 100644 --- a/heap.go +++ b/heap/heap.go @@ -1,4 +1,4 @@ -package coding +package heap import ( "maps" diff --git a/seeds.go b/seeds.go index 1836241..29cfbef 100644 --- a/seeds.go +++ b/seeds.go @@ -1,5 +1,9 @@ package coding +import ( + "github.com/securemesh/coding/heap" +) + var chatHeap = newHeapFromSeed([][]byte{ []byte(`]\_}`), []byte(`[ê%=Z`), @@ -18,12 +22,12 @@ var chatHeap = newHeapFromSeed([][]byte{ []byte(` `), }) -func ChatHeap() *Heap { +func ChatHeap() *heap.Heap { return chatHeap.Clone() } -func newHeapFromSeed(seed [][]byte) *Heap { - h := NewHeap() +func newHeapFromSeed(seed [][]byte) *heap.Heap { + h := heap.NewHeap() for i := range seed { for _, s := range seed[i:] {