package module

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

type hexModule struct{}

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

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

	return nil
}

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

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

func newHexModule(opts.Opts, env.Env) (module, error) {
	return hexModule{}, nil
}

func init() {
	register("hex", newHexModule)
}