rename module to hook

1 files changed, 32 insertions, 0 deletions

diff --git a/pkg/server/hook/alpha.go b/pkg/server/hook/alpha.go
new file mode 100644
index 0000000..d1fefcc
--- /dev/null
+++ b/pkg/server/hook/alpha.go
@@ -0,0 +1,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", alpha(unicode.ToLower))
+	registerFunc("upper", alpha(unicode.ToUpper))
+}