ft_lib.py

from __future__ import print_function

import base64
import datetime
import os
import subprocess

VALID_EXTENSIONS = ('txt', 'text', 'img', 'image', 'vid', 'video')
THIS_DIR = os.path.dirname(os.path.realpath(__file__))


def run_cmd(cmd):
    if '<' in cmd:
        out = subprocess.Popen(cmd.split(),
                               shell=True,
                               stdout=subprocess.PIPE,
                               stderr=subprocess.STDOUT)
    else:
        out = subprocess.Popen(cmd.split(),
                               stdout=subprocess.PIPE,
                               stderr=subprocess.STDOUT)

    stdout_data, stderr_data = out.communicate()
    stdout_data = stdout_data or b''
    stderr_data = stderr_data or b''
    return stdout_data, stderr_data


def _abs_path(path):
    return os.path.join(THIS_DIR, path)


def _write_to_file(contents, ext, perms='wb'):
    filename = '{:%Y-%m-%d-%H-%M--%s}.{}'.format(datetime.datetime.today(), ext)
    f = open(_abs_path(filename), perms)
    f.write(contents)
    f.close()
    return filename


def send(type_str, request):
    json_d = None
    if 'application/json' in request.headers.get('Content-Type'):
        json_d = request.get_json()

    if type_str in ('txt', 'text',):
        d = json_d or request.form
        data = d.get('txt') or d.get('text')
        if data:
            return send_text(data)
        raise Exception('Got no text data! See /api for docs.')

    elif type_str in ('img', 'image',):
        d = json_d or request.files
        data = d.get('img') or d.get('image')
        if data:
            if json_d:
                return send_image(base64.decodestring(bytes(data, 'utf-8')))
            else:
                return send_image(data.read())
        raise Exception('Got no image data! See /api for docs.')

    elif type_str in ('vid', 'video',):
        d = json_d or request.files
        data = d.get('vid') or d.get('video')
        if data:
            if json_d:
                return send_video(base64.decodestring(bytes(data, 'utf-8')))
            else:
                return send_video(data.read())
        raise Exception('Got no video data! See /api for docs.')

    raise Exception('type_str not in ' + ', '.join(VALID_EXTENSIONS))


def send_text(txt):
    filename = _write_to_file(txt, 'txt', perms='w')
    return run_cmd('./bin/send-text -h ft.noise:1337 -l 14 -O -f./fonts/6x10.bdf -i ' +
                   _abs_path(filename))


def send_image(img):
    filename = _write_to_file(img, 'img')
    return run_cmd('./bin/send-image -h ft.noise:1337 -l 12 -t 30 -c ' +
                   _abs_path(filename))


def send_video(vid):
    filename = _write_to_file(vid, 'vid')
    return run_cmd('./bin/send-video -h ft.noise:1337 -l 13 ' +
                   _abs_path(filename))