2013-10-01 21:26:21 +00:00
|
|
|
package transform
|
|
|
|
|
|
|
|
import (
|
2013-11-05 22:28:06 +00:00
|
|
|
"bytes"
|
2013-10-02 02:59:35 +00:00
|
|
|
"io"
|
2013-10-01 21:26:21 +00:00
|
|
|
)
|
|
|
|
|
2013-11-05 22:28:06 +00:00
|
|
|
type trans func([]byte) []byte
|
2013-10-01 21:26:21 +00:00
|
|
|
|
2013-11-05 22:28:06 +00:00
|
|
|
type link trans
|
|
|
|
|
|
|
|
type chain []link
|
|
|
|
|
|
|
|
func NewChain(trs ...link) chain {
|
2013-11-01 05:14:11 +00:00
|
|
|
return trs
|
2013-10-01 21:26:21 +00:00
|
|
|
}
|
|
|
|
|
2014-01-03 23:36:53 +00:00
|
|
|
func NewEmptyTransforms() []link {
|
|
|
|
return make([]link, 0, 20)
|
|
|
|
}
|
|
|
|
|
2013-10-01 21:42:08 +00:00
|
|
|
func (c *chain) Apply(w io.Writer, r io.Reader) (err error) {
|
2013-11-01 05:14:11 +00:00
|
|
|
|
2013-11-05 22:28:06 +00:00
|
|
|
buffer := new(bytes.Buffer)
|
|
|
|
buffer.ReadFrom(r)
|
|
|
|
b := buffer.Bytes()
|
|
|
|
for _, tr := range *c {
|
|
|
|
b = tr(b)
|
2013-10-01 21:26:21 +00:00
|
|
|
}
|
2013-11-05 22:28:06 +00:00
|
|
|
buffer.Reset()
|
|
|
|
buffer.Write(b)
|
|
|
|
buffer.WriteTo(w)
|
|
|
|
return
|
2013-10-01 21:26:21 +00:00
|
|
|
}
|