summaryrefslogtreecommitdiff
path: root/pkg/server/hook/alpha.go
blob: f25d4ded6ffcf2841ccd6547bcb4e4bcf2bc73c7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package hook

import (
	"bufio"
	"io"
	"tunnel/pkg/server/queue"
	"unicode"
)

func alpha(cb func(rune) rune) Func {
	return func(rq, wq queue.Q) error {
		r := bufio.NewReader(rq.Reader())

		for {
			c, _, err := r.ReadRune()
			if err != nil {
				if err == io.EOF {
					break
				}
				return err
			}
			wq <- []byte(string(cb(c)))
		}

		return nil
	}
}

func init() {
	registerFunc("lower", "lowercase filter out/", alpha(unicode.ToLower))
	registerFunc("upper", "uppercase filter out/", alpha(unicode.ToUpper))
}