#!/usr/bin/python3
# ups.khirnov.net uploader

import os.path
import shutil
import subprocess
import sys

try:
    import magic
    have_magic = True
except ImportError:
    have_magic = False

server    = 'https://ups.khirnov.net/'
cert_path = os.path.expanduser('~/.local/var/up/cert')

if len(sys.argv) > 2:
    sys.stderr.write('Too many arguments\n')
    sys.exit(1)
elif len(sys.argv) == 2 and sys.argv[1] != '-':
    src_name = sys.argv[1]
    dst_name = os.path.basename(src_name)
    stdin    = subprocess.DEVNULL
else:
    src_name = '-'
    dst_name = 'stdin'
    stdin    = subprocess.PIPE

    # try to guess the extension
    if have_magic:
        m = magic.Magic(extension = True)
        e = m.from_buffer(sys.stdin.buffer.peek(2048))
        if e != '???':
            dst_name += '.' + e.split('/')[0]


curl_cmdline = ['curl', '--silent', '--show-error', '--cert', cert_path,
                '--data-binary', '@' + src_name, server + dst_name]

child = subprocess.Popen(curl_cmdline, stdin = stdin)
if stdin != subprocess.DEVNULL:
    # stream our stdin to the child process
    # cannot pass stdin to the child directly because we peeked at it above
    shutil.copyfileobj(sys.stdin.buffer, child.stdin)
    child.stdin.close()

ret = child.wait()
if ret != 0:
    sys.stderr.write('curl returned an error code: %d\n' % ret)
else:
    sys.stdout.write('\n')