Unify all input/output plugins

examples/middleware/echo_modifier.sh

@@ -22,7 +22,6 @@ while read line; do
     case ${header:0:1} in
     "1")
         log "Request type: Request"
-        echo "$encoded"
         ;;
     "2")
         log "Request type: Original Response"
@@ -33,6 +32,8 @@ while read line; do
     *)
         log "Unknown request type $header"
     esac
+    echo "$encoded"
+
     log "==================================="
 
     log "Original data: $line"

examples/middleware/token_modifier.go

@@ -80,21 +80,13 @@ func process(buf []byte) {
 					payload = proto.SetPathParam(payload, []byte("token"), alias)
 
 					// Copy modified payload to our buffer
-					copy(buf[headerSize:], payload)
+					buf = append(buf[:headerSize], payload...)
 				}
 			}
 		}
 
-		// Re-compute length in case if payload was modified
-		bufLen := headerSize + len(payload)
-		// Encoding request and sending it back
-		dst := make([]byte, bufLen*2+1)
-		hex.Encode(dst, buf[:bufLen])
-		dst[len(dst)-1] = '\n'
-
-		os.Stdout.Write(dst)
-
-		return
+		// Emitting data back
+		os.Stdout.Write(encode(buf))
 	case '2': // Original response
 		if _, ok := originalTokens[reqID]; ok {
 			// Token is inside response body
@@ -113,6 +105,14 @@ func process(buf []byte) {
 	}
 }
 
+func encode(buf []byte) []byte {
+	dst := make([]byte, len(buf)*2+1)
+	hex.Encode(dst, buf)
+	dst[len(dst)-1] = '\n'
+
+	return dst
+}
+
 func Debug(args ...interface{}) {
 	fmt.Fprint(os.Stderr, "[DEBUG][TOKEN-MOD] ")
 	fmt.Fprintln(os.Stderr, args...)