package module

import (
	"encoding/hex"
	"tunnel/pkg/server/env"
	"tunnel/pkg/server/opts"
	"tunnel/pkg/server/queue"
)

type hexModule struct{}

func (h hexModule) Send(rq, wq queue.Q) error {
	enc := hex.NewEncoder(wq.Writer())

	for b := range rq {
		enc.Write(b)
	}

	return nil
}

func (h hexModule) Recv(rq, wq queue.Q) error {
	r := hex.NewDecoder(rq.Reader())
	w := wq.Writer()
	return queue.IoCopy(r, w)
}

func (m hexModule) Open(env env.Env) (interface{}, error) {
	return m, nil
}

func init() {
	register("hex", func(opts.Opts, env.Env) (module, error) {
		return hexModule{}, nil
	})
}