#!/usr/bin/env python3
import socket
import sys
import json
import shlex
import subprocess
import logging
from configparser import SafeConfigParser

logging.basicConfig(level=logging.DEBUG)
log = logging.getLogger('record-all-audio-streams')

host = 'localhost'
port = 9999

log.info('Connecting to %s:%u', host, port)
conn = socket.create_connection((host, port))
fd = conn.makefile('rw')

log.info('Fetching Config from Server')
fd.write("get_config\n")
fd.flush()
for line in fd:
    if line.startswith('server_config'):
        [cmd, arg] = line.split(' ', 1)
        server_config_json = arg
        log.info('Received Config from Server')
        break

log.info('Parsing Server-Config')
server_config = json.loads(server_config_json)


def getlist(self, section, option):
    return [x.strip() for x in self.get(section, option).split(',')]

SafeConfigParser.getlist = getlist

config = SafeConfigParser()
config.read_dict(server_config)

sources = config.getlist('mix', 'sources')

inputs = []
maps = []
for idx, source in enumerate(sources):
    inputs.append('-i tcp://localhost:{:d}'.format(13000 + idx))
    maps.append('-map {0:d}:a -metadata:s:a:{0:d} language=und'.format(idx))

try:
    output = sys.argv[1]
except:
    output = 'output.ts'

cmd = """
ffmpeg \
    -hide_banner
    -y -nostdin
    {}
    -ac 2 -channel_layout stereo
    {}
    -c:a mp2 -b:a 192k -ac:a 2 -ar:a 48000
    -flags +global_header -flags +ilme+ildct
    -f mpegts
    {}
""".format(' '.join(inputs), ' '.join(maps), output)
log.info('running command:\n%s', cmd)
args = shlex.split(cmd)
p = subprocess.run(args)
sys.exit(p.returncode)