#!/usr/bin/python3
import logging
from gi.repository import Gst

from lib.config import Config
from lib.tcpsingleconnection import TCPSingleConnection

class AVSource(TCPSingleConnection):
	def __init__(self, name, port):
		self.log = logging.getLogger('AVSource['+name+']')
		super().__init__(port)

		self.name = name

	def on_accepted(self, conn, addr):
		pipeline = """
			fdsrc fd={fd} !
			matroskademux name=demux

			demux. ! 
			{acaps} !
			queue !
			tee name=atee

			atee. ! queue ! interaudiosink channel=audio_{name}_mixer
			atee. ! queue ! interaudiosink channel=audio_{name}_mirror
		""".format(
			fd=conn.fileno(),
			name=self.name,
			acaps=Config.get('mix', 'audiocaps')
		)

		if Config.getboolean('previews', 'enabled'):
			pipeline += """
				atee. ! queue ! interaudiosink channel=audio_{name}_preview
			""".format(
				name=self.name
			)

		pipeline += """
			demux. ! 
			{vcaps} !
			queue !
			tee name=vtee

			vtee. ! queue ! intervideosink channel=video_{name}_mixer
			vtee. ! queue ! intervideosink channel=video_{name}_mirror
		""".format(
			fd=conn.fileno(),
			name=self.name,
			vcaps=Config.get('mix', 'videocaps')
		)

		if Config.getboolean('previews', 'enabled'):
			pipeline += """
				vtee. ! queue ! intervideosink channel=video_{name}_preview
			""".format(
				name=self.name
			)

		self.log.debug('Launching Source-Pipeline:\n%s', pipeline)
		self.receiverPipeline = Gst.parse_launch(pipeline)

		self.log.debug('Binding End-of-Stream-Signal on Source-Pipeline')
		self.receiverPipeline.bus.add_signal_watch()
		self.receiverPipeline.bus.connect("message::eos", self.on_eos)
		self.receiverPipeline.bus.connect("message::error", self.on_error)

		self.receiverPipeline.set_state(Gst.State.PLAYING)


	def on_eos(self, bus, message):
		self.log.debug('Received End-of-Stream-Signal on Source-Pipeline')
		if self.currentConnection is not None:
			self.disconnect()

	def on_error(self, bus, message):
		self.log.debug('Received Error-Signal on Source-Pipeline')
		(error, debug) = message.parse_error()
		self.log.debug('Error-Details: #%u: %s', error.code, debug)

		if self.currentConnection is not None:
			self.disconnect()

	def disconnect(self):
		self.receiverPipeline.set_state(Gst.State.NULL)
		self.receiverPipeline = None
		self.close_connection()