1 files changed, 8 insertions, 3 deletions

diff --git a/etc/b64.lua b/etc/b64.lua
index 7a4607d..de83578 100644
--- a/etc/b64.lua
+++ b/etc/b64.lua
@@ -1,6 +1,11 @@
-local base64 = socket.mime.base64.encode()
-local split = socket.mime.split()
-local convert = socket.mime.chain(base64, split)
+local convert
+if arg and arg[1] == '-d' then
+    convert = socket.mime.decode("base64")
+else
+    local base64 = socket.mime.encode("base64")
+    local wrap = socket.mime.wrap()
+    convert = socket.mime.chain(base64, wrap)
+end
 while 1 do
     local chunk = io.read(4096)
     io.write(convert(chunk))