use b64 encoding

[mtxbot.git] / src / mtxbot-post.py

diff --git a/src/mtxbot-post.py b/src/mtxbot-post.py
index 1ba56c4f2de103fc1b28852ac655dcb6b7583950..3e97ef2043f68675abe6929881002f13f0afab5f 100755 (executable)
--- a/src/mtxbot-post.py
+++ b/src/mtxbot-post.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 
+import base64
 import errno
 import os
 import posix
@@ -11,23 +12,28 @@ assert sys.version_info >= (3, 5)
 
 if len(sys.argv) != 2:
     print("Usage: %s <channel>" % sys.argv[0])
+    sys.exit(1)
 
 fifo_dir = os.getenv('MTXBOT_FIFO_DIR', '/run/mtxbot')
 fifo_path = "%s/%s" % (fifo_dir, sys.argv[1])
 
 if not os.path.isdir(fifo_dir):
     print("Fifo directory %s does not exist!" % fifo_dir, file=sys.stderr)
-    sys.exit(1)
+    sys.exit(2)
 
-if not stat.S_ISFIFO( os.stat(fifo_path).st_mode ):
+if not ( os.path.exists(fifo_path) and
+         stat.S_ISFIFO(os.stat(fifo_path).st_mode) ):
     print("Channel %s does not exist!" % sys.argv[1], file=sys.stderr)
-    sys.exit(2)
+    sys.exit(3)
+
+inBytes = sys.stdin.read().encode("UTF-8")
+inB64 = base64.b64encode(inBytes)
 
 for i in range(0, 10):
     fifo = -1
     try:
         fifo = posix.open(fifo_path, posix.O_WRONLY | posix.O_NONBLOCK)
-        posix.write(fifo, sys.stdin.read().encode())
+        posix.write(fifo, inB64)
         sys.exit(0)
     except OSError as e:
         if e.errno == errno.ENXIO:
@@ -37,4 +43,4 @@ for i in range(0, 10):
             posix.close(fifo)
 
 print("Error posting to channel %s!" % sys.argv[1], file=sys.stderr)
-sys.exit(3)
+sys.exit(4)